azalea/MTfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix only one torrent being detected when multiple exist

Browse Source
This commit is contained in:
Azalea Gui
2026-03-10 21:10:59 -04:00
parent 995d9ff4b5
commit 44fb80ad21
1 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflow.py
+15 -13
View File
@@ -105,16 +105,17 @@ def check_local_filesystem(dl_dir: str, imdb_id: str):
return item return item
return None return None
def check_qbittorrent(qb, imdb_id: str): def check_qbittorrent(qb, imdb_id: str) -> list[str]:
""" """
Checks qBittorrent for any existing torrent that has the IMDb ID in its name. Checks qBittorrent for any existing torrent that has the IMDb ID in its name.
Returns the torrent hash if found, or None. Returns a list of torrent hashes found.
""" """
existing_torrents = qb.torrents_info() existing_torrents = qb.torrents_info()
hashes = []
for t in existing_torrents: for t in existing_torrents:
if f"[{imdb_id}]" in t.name: if f"[{imdb_id}]" in t.name:
return t.hash hashes.append(t.hash)
return None return hashes
def search_and_download_mteam(qb, imdb_id: str, new_name: str, dl_dir: str) -> list: def search_and_download_mteam(qb, imdb_id: str, new_name: str, dl_dir: str) -> list:
""" """
@@ -278,18 +279,19 @@ def process_imdb_workflow(imdb_id: str, dl_dir: str = DEFAULT_DL_DIR, jellyfin_b
print(f"\n=== [0.2] Checking if torrent already exists in qBittorrent ===") print(f"\n=== [0.2] Checking if torrent already exists in qBittorrent ===")
qb = get_qb_client() qb = get_qb_client()
existing_t_hash = check_qbittorrent(qb, imdb_id) existing_t_hashes = check_qbittorrent(qb, imdb_id)
hashes_to_process = [] hashes_to_process = []
if existing_t_hash: if existing_t_hashes:
print(f"Found existing torrent with hash {existing_t_hash}, skipping local check, search, and download.") print(f"Found {len(existing_t_hashes)} existing torrent(s), skipping local check, search, and download.")
rename_torrent_and_folder(qb, existing_t_hash, new_name) for existing_t_hash in existing_t_hashes:
rename_torrent_and_folder(qb, existing_t_hash, new_name)
print(f"\n=== [0.3] Waiting for existing download to finish ===")
wait_for_download(qb, existing_t_hash) print(f"\n=== [0.3] Waiting for existing download to finish ===")
wait_for_download(qb, existing_t_hash)
hashes_to_process.append((existing_t_hash, "existing"))
hashes_to_process.append((existing_t_hash, "existing"))
else: else:
print(f"\n=== [0.5] Checking if already exists in file system ===") print(f"\n=== [0.5] Checking if already exists in file system ===")
fs_match_dir = check_local_filesystem(dl_dir, imdb_id) fs_match_dir = check_local_filesystem(dl_dir, imdb_id)