g0ldyy/comet
1
0
Fork 0
mirror of https://github.com/g0ldyy/comet.git synced 2026-01-12 01:16:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed info hash retriever

Browse Source
This commit is contained in:
Goldy
2024-06-24 18:55:01 +02:00
committed by GitHub
parent 6a81f7aebf
commit 949f17ca9b
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comet/main.py
+6 -1
View File
@@ -144,6 +144,11 @@ async def getJackett(session: aiohttp.ClientSession, indexers: list, query: str)
logger.warning(f"Exception while getting Jackett results for {query} with {indexers}: {e}")
async def getTorrentHash(session: aiohttp.ClientSession, url: str):
if url["InfoHash"] != None:
return url["InfoHash"]
url = url["Link"]
try:
timeout = aiohttp.ClientTimeout(total=int(os.getenv("GET_TORRENT_TIMEOUT", 5)))
response = await session.get(url, allow_redirects=False, timeout=timeout)
@@ -273,7 +278,7 @@ async def stream(request: Request, b64config: str, type: str, id: str):
continue
tasks.append(getTorrentHash(session, torrent["Link"]))
tasks.append(getTorrentHash(session, torrent))
torrentHashes = await asyncio.gather(*tasks)
torrentHashes = list(set([hash for hash in torrentHashes if hash]))