mirror of
https://github.com/THIS-IS-NOT-A-BACKUP/zspotify.git
synced 2024-11-29 19:24:34 +01:00
Added experimental podcast support
This commit is contained in:
parent
064bc563ff
commit
6e36eb40d1
1
.gitignore
vendored
1
.gitignore
vendored
@ -142,3 +142,4 @@ credentials.json
|
||||
|
||||
#Download Folder
|
||||
ZSpotify\ Music/
|
||||
ZSpotify\ Podcasts/
|
52
zspotify.py
Normal file → Executable file
52
zspotify.py
Normal file → Executable file
@ -18,13 +18,14 @@ import music_tag
|
||||
import requests
|
||||
from librespot.audio.decoders import AudioQuality, VorbisOnlyAudioQuality
|
||||
from librespot.core import Session
|
||||
from librespot.metadata import TrackId
|
||||
from librespot.metadata import TrackId, EpisodeId
|
||||
from pydub import AudioSegment
|
||||
|
||||
SESSION: Session = None
|
||||
sanitize = ["\\", "/", ":", "*", "?", "'", "<", ">", '"']
|
||||
|
||||
ROOT_PATH = "ZSpotify Music/"
|
||||
ROOT_PODCAST_PATH = "ZSpotify Podcasts/"
|
||||
SKIP_EXISTING_FILES = True
|
||||
MUSIC_FORMAT = "mp3" # or "ogg"
|
||||
FORCE_PREMIUM = False # set to True if not detecting your premium account automatically
|
||||
@ -107,7 +108,8 @@ def client():
|
||||
elif sys.argv[1] == "-ls" or sys.argv[1] == "--liked-songs":
|
||||
for song in get_saved_tracks(token):
|
||||
if not song['track']['name']:
|
||||
print("### SKIPPING: SONG DOES NOT EXISTS ON SPOTIFY ANYMORE ###")
|
||||
print(
|
||||
"### SKIPPING: SONG DOES NOT EXISTS ON SPOTIFY ANYMORE ###")
|
||||
else:
|
||||
download_track(song['track']['id'], "Liked Songs/")
|
||||
print("\n")
|
||||
@ -133,6 +135,13 @@ def client():
|
||||
sys.argv[1],
|
||||
)
|
||||
|
||||
episode_uri_search = re.search(
|
||||
r"^spotify:episode:(?P<EpisodeID>[0-9a-zA-Z]{22})$", sys.argv[1])
|
||||
episode_url_search = re.search(
|
||||
r"^(https?://)?open\.spotify\.com/episode/(?P<EpisodeID>[0-9a-zA-Z]{22})(\?si=.+?)?$",
|
||||
sys.argv[1],
|
||||
)
|
||||
|
||||
if track_uri_search is not None or track_url_search is not None:
|
||||
track_id_str = (track_uri_search
|
||||
if track_uri_search is not None else
|
||||
@ -156,13 +165,50 @@ def client():
|
||||
download_track(song['track']['id'],
|
||||
sanitize_data(name) + "/")
|
||||
print("\n")
|
||||
elif episode_uri_search is not None or episode_url_search is not None:
|
||||
episode_id_str = (episode_uri_search
|
||||
if episode_uri_search is not None else
|
||||
episode_url_search).group("EpisodeID")
|
||||
|
||||
downloadEpisode(episode_id_str)
|
||||
else:
|
||||
search_text = input("Enter search: ")
|
||||
search(search_text)
|
||||
wait()
|
||||
|
||||
|
||||
def getEpisodeInfo(episode_id_str):
|
||||
token = SESSION.tokens().get("user-read-email")
|
||||
info = json.loads(requests.get("https://api.spotify.com/v1/episodes/" +
|
||||
episode_id_str, headers={"Authorization": "Bearer %s" % token}).text)
|
||||
|
||||
return info["show"]["name"], info["name"]
|
||||
|
||||
|
||||
def downloadEpisode(episode_id_str):
|
||||
global ROOT_PODCAST_PATH
|
||||
|
||||
podcastName, episodeName = getEpisodeInfo(episode_id_str)
|
||||
filename = podcastName + " - " + episodeName + ".wav"
|
||||
|
||||
episode_id = EpisodeId.from_base62(episode_id_str)
|
||||
stream = SESSION.content_feeder().load(
|
||||
episode_id, VorbisOnlyAudioQuality(QUALITY), False, None)
|
||||
print("### DOWNLOADING ENTIRE PODCAST EPISODE - THIS MAY TAKE A WHILE ###")
|
||||
|
||||
if not os.path.isdir(ROOT_PODCAST_PATH):
|
||||
os.makedirs(ROOT_PODCAST_PATH)
|
||||
|
||||
with open(ROOT_PODCAST_PATH + filename, 'wb') as file:
|
||||
while True:
|
||||
byte = stream.input_stream.stream().read(1024 * 1024)
|
||||
if byte == b'':
|
||||
break
|
||||
file.write(byte)
|
||||
|
||||
# related functions that do stuff with the spotify API
|
||||
|
||||
|
||||
def search(search_term):
|
||||
""" Searches Spotify's API for relevant data """
|
||||
token = SESSION.tokens().get("user-read-email")
|
||||
@ -501,7 +547,7 @@ def download_from_user_playlist():
|
||||
def checkRaw():
|
||||
global RAW_AUDIO_AS_IS, MUSIC_FORMAT
|
||||
if RAW_AUDIO_AS_IS:
|
||||
MUSIC_FORMAT = "raw"
|
||||
MUSIC_FORMAT = "wav"
|
||||
|
||||
|
||||
def main():
|
||||
|
Loading…
Reference in New Issue
Block a user