got GPT to clean this up because i genuienly could not care
This commit is contained in:
WhatDidYouExpect
parent
4a42531632
commit
9ad86bc3eb
1 changed files with 154 additions and 104 deletions
258
main.py
258
main.py
|
|
@ -1,118 +1,168 @@
|
||||||
import os,json,base64,requests,time,subprocess,sys,concurrent.futures
|
import os, json, base64, requests, time, subprocess, sys, concurrent.futures
|
||||||
from mutagen.easyid3 import EasyID3;from mutagen.mp3 import MP3
|
from mutagen.easyid3 import EasyID3
|
||||||
from mutagen.id3 import ID3,APIC,error
|
from mutagen.mp3 import MP3
|
||||||
|
from mutagen.id3 import ID3, APIC, error
|
||||||
|
|
||||||
CONFIG_FILE="config.json";DOWNLOAD_DIR="downloads"
|
CONFIG_FILE = "config.json"
|
||||||
SPOTIFY_TRACK_API="https://api.spotify.com/v1/tracks/{}"
|
DOWNLOAD_DIR = "downloads"
|
||||||
SPOTIFY_PLAYLIST_API="https://api.spotify.com/v1/playlists/{}/tracks"
|
SPOTIFY_TRACK_API = "https://api.spotify.com/v1/tracks/{}"
|
||||||
INVIDIOUS_API=None
|
SPOTIFY_PLAYLIST_API = "https://api.spotify.com/v1/playlists/{}/tracks"
|
||||||
|
INVIDIOUS_API = None
|
||||||
|
|
||||||
def load_config(): # opens the config file
|
def load_config(): # opens the config file
|
||||||
with open(CONFIG_FILE) as f:return json.load(f)
|
with open(CONFIG_FILE) as f:
|
||||||
|
return json.load(f)
|
||||||
|
|
||||||
def ensure_valid_token():
|
def ensure_valid_token():
|
||||||
global INVIDIOUS_API
|
global INVIDIOUS_API
|
||||||
c=load_config()
|
c = load_config()
|
||||||
INVIDIOUS_API=c.get("invidious_url","https://invidious.snopyta.org/api/v1/search?type=video&q={query}")
|
INVIDIOUS_API = c.get("invidious_url", "https://invidious.snopyta.org/api/v1/search?type=video&q={query}")
|
||||||
if "access_token"not in c or int(time.time())>=c.get("expires_at",0):
|
if "access_token" not in c or int(time.time()) >= c.get("expires_at", 0):
|
||||||
print("no token running oauth")
|
print("no token running oauth")
|
||||||
subprocess.run(["python3","spotify_oauth.py"],check=True)
|
subprocess.run(["python3", "spotify_oauth.py"], check=True)
|
||||||
c=load_config()
|
c = load_config()
|
||||||
return c["access_token"]
|
return c["access_token"]
|
||||||
|
|
||||||
def get_liked_tracks(tok): # gets liked songs i guess
|
def get_liked_tracks(tok): # gets liked songs i guess
|
||||||
l,r=[],0;h={"Authorization":f"Bearer {tok}"}
|
l, r = [], 0
|
||||||
while 1:
|
h = {"Authorization": f"Bearer {tok}"}
|
||||||
u=f"https://api.spotify.com/v1/me/tracks?limit=50&offset={r}"
|
while 1:
|
||||||
d=requests.get(u,headers=h)
|
u = f"https://api.spotify.com/v1/me/tracks?limit=50&offset={r}"
|
||||||
if d.status_code==401:print("nah bro token's dead");break
|
d = requests.get(u, headers=h)
|
||||||
j=d.json();i=j.get("items",[])
|
if d.status_code == 401:
|
||||||
for it in i:l.append(it["track"]["id"])
|
print("nah bro token's dead")
|
||||||
if len(i)<50:break
|
break
|
||||||
r+=50;time.sleep(0.5)
|
j = d.json()
|
||||||
return l
|
i = j.get("items", [])
|
||||||
|
for it in i:
|
||||||
|
l.append(it["track"]["id"])
|
||||||
|
if len(i) < 50:
|
||||||
|
break
|
||||||
|
r += 50
|
||||||
|
time.sleep(0.5)
|
||||||
|
return l
|
||||||
|
|
||||||
def get_playlist_tracks(tok,pid): # does same thing but like for playlists
|
def get_playlist_tracks(tok, pid): # does same thing but like for playlists
|
||||||
l,r=[],0;h={"Authorization":f"Bearer {tok}"}
|
l, r = [], 0
|
||||||
while 1:
|
h = {"Authorization": f"Bearer {tok}"}
|
||||||
u=f"{SPOTIFY_PLAYLIST_API.format(pid)}?limit=100&offset={r}"
|
while 1:
|
||||||
d=requests.get(u,headers=h)
|
u = f"{SPOTIFY_PLAYLIST_API.format(pid)}?limit=100&offset={r}"
|
||||||
if d.status_code==401:print("nope, bad token");break
|
d = requests.get(u, headers=h)
|
||||||
if d.status_code==404:print("what playlist");break
|
if d.status_code == 401:
|
||||||
j=d.json();i=j.get("items",[])
|
print("nope, bad token")
|
||||||
for t in i:
|
break
|
||||||
tr=t.get("track")
|
if d.status_code == 404:
|
||||||
if tr and tr.get("id"):l.append(tr["id"])
|
print("what playlist")
|
||||||
if len(i)<100:break
|
break
|
||||||
r+=100;time.sleep(0.5)
|
j = d.json()
|
||||||
return l
|
i = j.get("items", [])
|
||||||
|
for t in i:
|
||||||
|
tr = t.get("track")
|
||||||
|
if tr and tr.get("id"):
|
||||||
|
l.append(tr["id"])
|
||||||
|
if len(i) < 100:
|
||||||
|
break
|
||||||
|
r += 100
|
||||||
|
time.sleep(0.5)
|
||||||
|
return l
|
||||||
|
|
||||||
def fetch_metadata(tid,tok): # gets info about a track
|
def fetch_metadata(tid, tok): # gets info about a track
|
||||||
return requests.get(SPOTIFY_TRACK_API.format(tid),headers={"Authorization":f"Bearer {tok}"}).json()
|
return requests.get(SPOTIFY_TRACK_API.format(tid), headers={"Authorization": f"Bearer {tok}"}).json()
|
||||||
|
|
||||||
def sanitize_filename(n): # removes stuff that windows would cry about
|
def sanitize_filename(n): # removes stuff that windows would cry about
|
||||||
return"".join(c for c in n if c not in r'\/:*?"<>|').strip()
|
return "".join(c for c in n if c not in r'\/:*?"<>|').strip()
|
||||||
|
|
||||||
def search_invidious(q): # looks up stuff on invidious, yeah
|
def search_invidious(q): # looks up stuff on invidious
|
||||||
try:
|
try:
|
||||||
r=requests.get(INVIDIOUS_API.format(query=q),timeout=10);r.raise_for_status()
|
r = requests.get(INVIDIOUS_API.format(query=q), timeout=10)
|
||||||
for v in r.json():
|
r.raise_for_status()
|
||||||
if v.get("videoId"):return f"https://www.youtube.com/watch?v={v['videoId']}"
|
for v in r.json():
|
||||||
except:print("couldn't search :(");return None
|
if v.get("videoId"):
|
||||||
|
return f"https://www.youtube.com/watch?v={v['videoId']}"
|
||||||
|
except:
|
||||||
|
print("couldn't search :(")
|
||||||
|
return None
|
||||||
|
|
||||||
def download_from_youtube(url,f): # yoink an mp3
|
def download_from_youtube(url, f): # yoink an mp3
|
||||||
os.makedirs(DOWNLOAD_DIR,exist_ok=True)
|
os.makedirs(DOWNLOAD_DIR, exist_ok=True)
|
||||||
p=os.path.join(DOWNLOAD_DIR,f)
|
p = os.path.join(DOWNLOAD_DIR, f)
|
||||||
if os.path.exists(p):print(f"{f}? already here bro");return True
|
if os.path.exists(p):
|
||||||
c=[];copt=["--cookies","cookies.txt"] if os.path.exists("cookies.txt") else []
|
print(f"{f}? already here bro")
|
||||||
cmd=["yt-dlp",*copt,"-x","--audio-format","mp3","--no-playlist","--no-live-from-start","-o",p,url]
|
return True
|
||||||
try:
|
copt = ["--cookies", "cookies.txt"] if os.path.exists("cookies.txt") else []
|
||||||
subprocess.run(cmd,check=True);print(f"yanked {f}");return True
|
cmd = ["yt-dlp", *copt, "-x", "--audio-format", "mp3", "--no-playlist", "--no-live-from-start", "-o", p, url]
|
||||||
except:print(f"nope, didn't work: {f}");return False
|
try:
|
||||||
|
subprocess.run(cmd, check=True)
|
||||||
|
print(f"yanked {f}")
|
||||||
|
return True
|
||||||
|
except:
|
||||||
|
print(f"nope, didn't work: {f}")
|
||||||
|
return False
|
||||||
|
|
||||||
def embed_album_art(path,img): # slap on some pic
|
def embed_album_art(path, img): # slap on some pic
|
||||||
try:
|
try:
|
||||||
i=requests.get(img).content
|
i = requests.get(img).content
|
||||||
a=MP3(path,ID3=ID3)
|
a = MP3(path, ID3=ID3)
|
||||||
try:a.add_tags()
|
try:
|
||||||
except:error()
|
a.add_tags()
|
||||||
a.tags.add(APIC(encoding=3,mime='image/jpeg',type=3,desc='Cover',data=i))
|
except error:
|
||||||
a.save();print(f"art slapped on {path}")
|
pass
|
||||||
except Exception as e:print(f"eh: {e}")
|
a.tags.add(APIC(encoding=3, mime='image/jpeg', type=3, desc='Cover', data=i))
|
||||||
|
a.save()
|
||||||
|
print(f"art slapped on {path}")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"eh: {e}")
|
||||||
|
|
||||||
def tag_mp3_with_spotify_metadata(p,m): # add title n stuff
|
def tag_mp3_with_spotify_metadata(p, m): # add title n stuff
|
||||||
try:
|
try:
|
||||||
a=MP3(p,ID3=EasyID3)
|
a = MP3(p, ID3=EasyID3)
|
||||||
a["title"]=m.get("name","")
|
a["title"] = m.get("name", "")
|
||||||
a["artist"]=", ".join(z["name"]for z in m.get("artists",[]))
|
a["artist"] = ", ".join(z["name"] for z in m.get("artists", []))
|
||||||
a["album"]=m.get("album",{}).get("name","")
|
a["album"] = m.get("album", {}).get("name", "")
|
||||||
a["date"]=m.get("album",{}).get("release_date","")
|
a["date"] = m.get("album", {}).get("release_date", "")
|
||||||
a.save();print(f"tagged: {p}")
|
a.save()
|
||||||
i=m.get("album",{}).get("images",[])
|
print(f"tagged: {p}")
|
||||||
if i:embed_album_art(p,i[0]["url"])
|
i = m.get("album", {}).get("images", [])
|
||||||
except Exception as e:print(f"tag fail on {p}: {e}")
|
if i:
|
||||||
|
embed_album_art(p, i[0]["url"])
|
||||||
|
except Exception as e:
|
||||||
|
print(f"tag fail on {p}: {e}")
|
||||||
|
|
||||||
def process_track(tid,tok): # the whole shabang
|
def process_track(tid, tok): # the whole shabang
|
||||||
m=fetch_metadata(tid,tok)
|
m = fetch_metadata(tid, tok)
|
||||||
if not m:print(f"nah nothing for {tid}");return False
|
if not m:
|
||||||
a=m["artists"][0]["name"];t=m["name"]
|
print(f"nah nothing for {tid}")
|
||||||
s=f"{a} - {t} audio";f=sanitize_filename(f"{a} - {t}.mp3")
|
return False
|
||||||
u=search_invidious(s)
|
a = m["artists"][0]["name"]
|
||||||
if not u:print(f"nothing on invidious for {s}");return False
|
t = m["name"]
|
||||||
if not download_from_youtube(u,f):print(f"skipping {f}");return False
|
s = f"{a} - {t} audio"
|
||||||
print(f"yoinked {f}");tag_mp3_with_spotify_metadata(os.path.join(DOWNLOAD_DIR,f),m)
|
f = sanitize_filename(f"{a} - {t}.mp3")
|
||||||
return True
|
u = search_invidious(s)
|
||||||
|
if not u:
|
||||||
|
print(f"nothing on invidious for {s}")
|
||||||
|
return False
|
||||||
|
if not download_from_youtube(u, f):
|
||||||
|
print(f"skipping {f}")
|
||||||
|
return False
|
||||||
|
print(f"yoinked {f}")
|
||||||
|
tag_mp3_with_spotify_metadata(os.path.join(DOWNLOAD_DIR, f), m)
|
||||||
|
return True
|
||||||
|
|
||||||
if __name__=="__main__":
|
if __name__ == "__main__":
|
||||||
p=sys.argv[1]if len(sys.argv)>1 else None
|
p = sys.argv[1] if len(sys.argv) > 1 else None
|
||||||
c=load_config();t=ensure_valid_token()
|
c = load_config()
|
||||||
if p:
|
t = ensure_valid_token()
|
||||||
print(f"playlist: {p}");ids=get_playlist_tracks(t,p)
|
if p:
|
||||||
else:
|
print(f"playlist: {p}")
|
||||||
print("liked songs");ids=get_liked_tracks(t)
|
ids = get_playlist_tracks(t, p)
|
||||||
print(f"{len(ids)} songs found, whew")
|
else:
|
||||||
with concurrent.futures.ThreadPoolExecutor(max_workers=8)as x:
|
print("liked songs")
|
||||||
fut={x.submit(process_track,i,t):i for i in ids}
|
ids = get_liked_tracks(t)
|
||||||
for f in concurrent.futures.as_completed(fut):
|
print(f"{len(ids)} songs found, whew")
|
||||||
try:f.result()
|
with concurrent.futures.ThreadPoolExecutor(max_workers=8) as x:
|
||||||
except Exception as e:print(f"welp: {e}")
|
fut = {x.submit(process_track, i, t): i for i in ids}
|
||||||
|
for f in concurrent.futures.as_completed(fut):
|
||||||
|
try:
|
||||||
|
f.result()
|
||||||
|
except Exception as e:
|
||||||
|
print(f"welp: {e}")
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue