sigh

main.py

@@ -0,0 +1,118 @@
+import os,json,base64,requests,time,subprocess,sys,concurrent.futures
+from mutagen.easyid3 import EasyID3;from mutagen.mp3 import MP3
+from mutagen.id3 import ID3,APIC,error
+
+CONFIG_FILE="config.json";DOWNLOAD_DIR="downloads"
+SPOTIFY_TRACK_API="https://api.spotify.com/v1/tracks/{}"
+SPOTIFY_PLAYLIST_API="https://api.spotify.com/v1/playlists/{}/tracks"
+INVIDIOUS_API=None
+
+def load_config(): # opens the config file
+ with open(CONFIG_FILE) as f:return json.load(f)
+
+def ensure_valid_token():
+ global INVIDIOUS_API
+ c=load_config()
+ 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):
+  print("no token running oauth")
+  subprocess.run(["python3","spotify_oauth.py"],check=True)
+  c=load_config()
+ return c["access_token"]
+
+def get_liked_tracks(tok): # gets liked songs i guess
+ l,r=[],0;h={"Authorization":f"Bearer {tok}"}
+ while 1:
+  u=f"https://api.spotify.com/v1/me/tracks?limit=50&offset={r}"
+  d=requests.get(u,headers=h)
+  if d.status_code==401:print("nah bro token's dead");break
+  j=d.json();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
+ l,r=[],0;h={"Authorization":f"Bearer {tok}"}
+ while 1:
+  u=f"{SPOTIFY_PLAYLIST_API.format(pid)}?limit=100&offset={r}"
+  d=requests.get(u,headers=h)
+  if d.status_code==401:print("nope, bad token");break
+  if d.status_code==404:print("what playlist");break
+  j=d.json();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
+ 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
+ return"".join(c for c in n if c not in r'\/:*?"<>|').strip()
+
+def search_invidious(q): # looks up stuff on invidious, yeah
+ try:
+  r=requests.get(INVIDIOUS_API.format(query=q),timeout=10);r.raise_for_status()
+  for v in r.json():
+   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
+ os.makedirs(DOWNLOAD_DIR,exist_ok=True)
+ p=os.path.join(DOWNLOAD_DIR,f)
+ if os.path.exists(p):print(f"{f}? already here bro");return True
+ c=[];copt=["--cookies","cookies.txt"] if os.path.exists("cookies.txt") else []
+ cmd=["yt-dlp",*copt,"-x","--audio-format","mp3","--no-playlist","--no-live-from-start","-o",p,url]
+ 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
+ try:
+  i=requests.get(img).content
+  a=MP3(path,ID3=ID3)
+  try:a.add_tags()
+  except:error()
+  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
+ try:
+  a=MP3(p,ID3=EasyID3)
+  a["title"]=m.get("name","")
+  a["artist"]=", ".join(z["name"]for z in m.get("artists",[]))
+  a["album"]=m.get("album",{}).get("name","")
+  a["date"]=m.get("album",{}).get("release_date","")
+  a.save();print(f"tagged: {p}")
+  i=m.get("album",{}).get("images",[])
+  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
+ m=fetch_metadata(tid,tok)
+ if not m:print(f"nah nothing for {tid}");return False
+ a=m["artists"][0]["name"];t=m["name"]
+ s=f"{a} - {t} audio";f=sanitize_filename(f"{a} - {t}.mp3")
+ 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__":
+ p=sys.argv[1]if len(sys.argv)>1 else None
+ c=load_config();t=ensure_valid_token()
+ if p:
+  print(f"playlist: {p}");ids=get_playlist_tracks(t,p)
+ else:
+  print("liked songs");ids=get_liked_tracks(t)
+ print(f"{len(ids)} songs found, whew")
+ with concurrent.futures.ThreadPoolExecutor(max_workers=8)as x:
+  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}")