feat: ability to enable/disable title match check (TITLE_MATCH_CHECK …

…env var)

.env-sample

@@ -18,3 +18,4 @@ SCRAPE_TORRENTIO=False # scrape Torrentio
 CUSTOM_HEADER_HTML=None # only set it if you know what it is
 PROXY_DEBRID_STREAM=False # Proxy Debrid Streams (very useful to use your debrid service on multiple IPs at same time)
 PROXY_DEBRID_STREAM_PASSWORD=CHANGE_ME # Secret password to enter on configuration page to prevent people from abusing your debrid stream proxy
+TITLE_MATCH_CHECK=True # disable if you only use Jackett / Prowlarr / Torrentio and are sure you're only scraping good titles, for example (not at all recommended with Zilean enabled)

comet/api/stream.py

@@ -229,30 +229,31 @@ async def stream(request: Request, b64config: str, type: str, id: str):
         if len(torrents) == 0:
             return {"streams": []}
 
-        indexed_torrents = [(i, torrents[i]["Title"]) for i in range(len(torrents))]
-        chunk_size = 50
-        chunks = [
-            indexed_torrents[i : i + chunk_size]
-            for i in range(0, len(indexed_torrents), chunk_size)
-        ]
+        if settings.TITLE_MATCH_CHECK:
+            indexed_torrents = [(i, torrents[i]["Title"]) for i in range(len(torrents))]
+            chunk_size = 50
+            chunks = [
+                indexed_torrents[i : i + chunk_size]
+                for i in range(0, len(indexed_torrents), chunk_size)
+            ]
 
-        tasks = []
-        for chunk in chunks:
-            tasks.append(filter(chunk, name))
+            tasks = []
+            for chunk in chunks:
+                tasks.append(filter(chunk, name))
 
-        filtered_torrents = await asyncio.gather(*tasks)
-        index_less = 0
-        for result in filtered_torrents:
-            for filtered in result:
-                if not filtered[1]:
-                    del torrents[filtered[0] - index_less]
-                    index_less += 1
-                    continue
+            filtered_torrents = await asyncio.gather(*tasks)
+            index_less = 0
+            for result in filtered_torrents:
+                for filtered in result:
+                    if not filtered[1]:
+                        del torrents[filtered[0] - index_less]
+                        index_less += 1
+                        continue
 
-        logger.info(f"{len(torrents)} torrents passed title match check for {log_name}")
+            logger.info(f"{len(torrents)} torrents passed title match check for {log_name}")
 
-        if len(torrents) == 0:
-            return {"streams": []}
+            if len(torrents) == 0:
+                return {"streams": []}
 
         tasks = []
         for i in range(len(torrents)):

comet/utils/models.py

@@ -30,6 +30,7 @@ class AppSettings(BaseSettings):
     CUSTOM_HEADER_HTML: Optional[str] = None
     PROXY_DEBRID_STREAM: Optional[bool] = False
     PROXY_DEBRID_STREAM_PASSWORD: Optional[str] = "CHANGE_ME"
+    TITLE_MATCH_CHECK: Optional[bool] = True
 
 
 settings = AppSettings()