fix: postgre stupid

g0ldyy · Feb 20, 2025 · 5c4fa67 · 5c4fa67
1 parent 5148997
commit 5c4fa67
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/comet/api/stream.py b/comet/api/stream.py
@@ -324,14 +324,17 @@ async def playback(
     config = config_check(b64config)
     if not config:
         return FileResponse("comet/assets/invalidconfig.mp4")
+
+    season = int(season) if season != "n" else None
+    episode = int(episode) if episode != "n" else None
 
     async with aiohttp.ClientSession() as session:
         cached_link = await database.fetch_one(
             f"SELECT download_url FROM download_links_cache WHERE debrid_key = '{config['debridApiKey']}' AND info_hash = '{hash}' AND ((cast(:season as INTEGER) IS NULL AND season IS NULL) OR season = cast(:season as INTEGER)) AND ((cast(:episode as INTEGER) IS NULL AND episode IS NULL) OR episode = cast(:episode as INTEGER)) AND timestamp + 3600 >= :current_time",
             {
                 "current_time": time.time(),
-                "season": season if season != "n" else None,
-                "episode": episode if episode != "n" else None,
+                "season": season,
+                "episode": season,
             },
         )
 
@@ -367,8 +370,8 @@ async def playback(
                     "debrid_key": config["debridApiKey"],
                     "info_hash": hash,
                     "name": name,
-                    "season": season if season != "n" else None,
-                    "episode": episode if episode != "n" else None,
+                    "season": season,
+                    "episode": episode,
                     "download_url": download_url,
                     "timestamp": time.time(),
                 },

diff --git a/comet/debrid/stremthru.py b/comet/debrid/stremthru.py
@@ -166,9 +166,6 @@ async def generate_download_link(
             name_parsed = parse(name)
             target_file = None
 
-            season = int(season) if season != "n" else None
-            episode = int(episode) if episode != "n" else None
-
             for file in magnet["data"]["files"]:
                 filename = file["name"]
                 file_parsed = parse(filename)