Require youtube PO token

.env.example

@@ -1,5 +1,8 @@
 TOKEN=""
+PO_TOKEN=""
+COOKIES_FILE=""
 GUILD_ID=""
+QUEUE_PATH=""
 MAX_MIN="8"
 MAX_QUEUE="5"
 ALLOW_SKIP="TRUE"

.gitignore

@@ -31,7 +31,10 @@ share/python-wheels/
 *.egg
 *.mp4
 *.vtt
+cookies*
 testing/
+*.db
+next*
 MANIFEST
 
 # PyInstaller
@@ -129,7 +132,7 @@ celerybeat.pid
 
 # Environments
 .env
-.venv
+.*venv
 env/
 venv/
 ENV/

newbot.py

@@ -19,7 +19,8 @@ import sqlite3
 
 if len(sys.argv) > 1:
     if sys.argv[1] == "--clear-queue":
-        os.remove("queue.db")
+        if os.path.isfile(os.getenv("QUEUE_PATH","queue.db")):
+            os.remove(os.getenv("QUEUE_PATH","queue.db"))
         print("Queue cleared!")
 
 load_dotenv()
@@ -34,7 +35,7 @@ vid_dir = Path(temp_dir.name)
 download_dir = TemporaryDirectory()
 full_dl_dir = Path(download_dir.name)
 vid_details = {"title": "", "channel": ""}
-con = sqlite3.connect("queue.db", check_same_thread=False)
+con = sqlite3.connect(os.getenv("QUEUE_PATH","queue.db"), check_same_thread=False)
 cur = con.cursor()
 
 def countuser(usrid):
@@ -80,6 +81,9 @@ ydl_opts = {
     'match_filter': video_check,
     'hls_prefer_native': True,
     'extract_flat': 'discard_in_playlist',
+    'cookiefile': f'{os.getenv("COOKIES_FILE")}',
+     'extractor_args': {'youtube': {'player_client': ['web', 'default'],
+                                     'po_token': [f'web+{os.getenv("PO_TOKEN")}']}},
     'format': 'bestvideo[height<=1080][vcodec!*=av01][ext=mp4]+bestaudio[abr<=256][ext=m4a]/best[ext=mp4]/best',
     'fragment_retries': 10,
     'noplaylist': True,
@@ -153,7 +157,7 @@ def download_video(index):
         sleep(2) #allow ytdlp to fully cleanup
     except Exception as e:
         print("handling youtube exception")
-        if "Sign in to confirm" in str(e):
+        if "Sign in to confirm" in str(e) or "forbidden" in str(e).lower():
             print("youtube ban detected")
             #This is the worst possible case and therefore all activity must halt
             obs.set_current_program_scene("banned")
@@ -264,7 +268,7 @@ async def stats(inter: disnake.AppCmdInter):
     message = message + f"Link: <{queue[0]}>\n"
     playing = obs.get_media_input_status("player")
     message = message + f"Video Duration: {str(datetime.timedelta(seconds=(round(playing.media_cursor/1000))))}/{str(datetime.timedelta(seconds=(round(playing.media_duration/1000))))}\n"
-    if inter.permissions.moderate_members and not (os.getenv("PERMANENT_MAX_QUEUE","FALSE") == "TRUE"):
+    if inter.permissions.moderate_members:
         res = cur.execute(f"SELECT user FROM queue WHERE hasplayed = true AND link = '{queue[0]}'")
         res = res.fetchall() #We can't gaurentee the result so just show likely possibilites
         message = message + f"Users who have queued this video: "
@@ -601,4 +605,5 @@ bot.run(os.getenv("TOKEN"))
 print("cleaning up tempdir")
 temp_dir.cleanup()
 download_dir.cleanup()
+con.close()
 obs.set_current_program_scene("nosignal")

shell.nix

@@ -10,8 +10,8 @@ pkgs.mkShell {
     pkgs.python311Packages.python-dotenv
     pkgs.python311Packages.venvShellHook
     pkgs.ffmpeg
-    pkgs.icewm
     pkgs.discord
+    pkgs.mpv
     ];
   venvDir = "./.venv";
 }