112 lines
4.4 KiB
Python
112 lines
4.4 KiB
Python
import re
|
|
import argparse
|
|
from pathlib import Path
|
|
from utils import DEFAULT_DL_DIR, DEFAULT_JELLYFIN_DIR
|
|
from utils_qb import get_qb_client
|
|
|
|
def get_tt_ids_in_jellyfin(jellyfin_dir):
|
|
"""
|
|
Scans the Jellyfin directory (and its TV/Movie subdirectories) for any folders
|
|
that contain an IMDb ID [tt...] in their name and returns a set of those IDs.
|
|
"""
|
|
jellyfin_path = Path(jellyfin_dir)
|
|
found_ids = set()
|
|
|
|
# Check immediate children of Jellyfin root, and children of TV/Movie folders
|
|
dirs_to_check = []
|
|
if jellyfin_path.exists():
|
|
dirs_to_check.extend(jellyfin_path.iterdir())
|
|
for subdir in ["Movie", "TV"]:
|
|
subpath = jellyfin_path / subdir
|
|
if subpath.exists():
|
|
dirs_to_check.extend(subpath.iterdir())
|
|
|
|
for path in dirs_to_check:
|
|
if path.is_dir():
|
|
match = re.search(r'\[(tt\d+)\]', path.name)
|
|
if match:
|
|
found_ids.add(match.group(1))
|
|
|
|
return found_ids
|
|
|
|
def detect_anomalies(expected_tt_ids=None):
|
|
print("Gathering basic info...")
|
|
jellyfin_tt_ids = get_tt_ids_in_jellyfin(DEFAULT_JELLYFIN_DIR)
|
|
|
|
print(f"Found {len(jellyfin_tt_ids)} linked titles in Jellyfin directories.")
|
|
|
|
qb = get_qb_client()
|
|
torrents = qb.torrents_info()
|
|
|
|
print(f"\n=== Anomaly 1: Torrents with missing Jellyfin links ===")
|
|
torrents_with_missing_links = set()
|
|
|
|
for t in torrents:
|
|
match = re.search(r'\[(tt\d+)\]', t.name)
|
|
if match:
|
|
tt_id = match.group(1)
|
|
if tt_id not in jellyfin_tt_ids:
|
|
torrents_with_missing_links.add((t.name, tt_id))
|
|
|
|
if torrents_with_missing_links:
|
|
for name, tt_id in torrents_with_missing_links:
|
|
print(f"Warning: Torrent '{name}' has ID {tt_id} but no corresponding Jellyfin folder!")
|
|
else:
|
|
print("No torrent anomalies found.")
|
|
|
|
print(f"\n=== Anomaly 2: Local DL folders with missing Jellyfin links ===")
|
|
local_folders_with_missing_links = set()
|
|
|
|
dl_path = Path(DEFAULT_DL_DIR)
|
|
if dl_path.exists():
|
|
for path in dl_path.iterdir():
|
|
match = re.search(r'\[(tt\d+)\]', path.name)
|
|
if match:
|
|
tt_id = match.group(1)
|
|
if tt_id not in jellyfin_tt_ids:
|
|
local_folders_with_missing_links.add((path.name, tt_id))
|
|
|
|
if local_folders_with_missing_links:
|
|
for name, tt_id in local_folders_with_missing_links:
|
|
print(f"Warning: Local folder '{name}' has ID {tt_id} but no corresponding Jellyfin folder!")
|
|
else:
|
|
print("No local folder anomalies found.")
|
|
else:
|
|
print(f"Download directory {DEFAULT_DL_DIR} does not exist.")
|
|
|
|
print(f"\n=== Anomaly 3: TV series with < 6 episodes (Possible Movies) ===")
|
|
tv_path = Path(DEFAULT_JELLYFIN_DIR) / "TV"
|
|
short_series = []
|
|
video_exts = {".mkv", ".mp4", ".avi", ".ts", ".m2ts", ".webm"}
|
|
|
|
if tv_path.exists():
|
|
for series_dir in tv_path.iterdir():
|
|
if series_dir.is_dir():
|
|
video_count = sum(1 for p in series_dir.rglob('*') if p.is_file() and p.suffix.lower() in video_exts)
|
|
if 0 < video_count < 6:
|
|
short_series.append((series_dir.name, video_count))
|
|
|
|
if short_series:
|
|
for name, count in short_series:
|
|
print(f"Warning: TV series '{name}' has only {count} video file(s).")
|
|
else:
|
|
print("No short TV series anomalies found.")
|
|
else:
|
|
print(f"TV directory {tv_path} does not exist.")
|
|
|
|
if expected_tt_ids:
|
|
print(f"\n=== Anomaly 4: Provided IMDb IDs missing from Jellyfin ===")
|
|
missing_ids = [tt_id for tt_id in expected_tt_ids if tt_id not in jellyfin_tt_ids]
|
|
if missing_ids:
|
|
for tt_id in missing_ids:
|
|
print(f"Warning: Expected ID '{tt_id}' is not linked in Jellyfin!")
|
|
else:
|
|
print("All provided IMDb IDs are present in Jellyfin.")
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Find anomalies between torrents, local downloads, and Jellyfin folders.")
|
|
parser.add_argument("tt_ids", nargs="*", help="Optional space-separated list of IMDb IDs (e.g., tt1234567 tt7654321) to verify their presence in Jellyfin.")
|
|
args = parser.parse_args()
|
|
|
|
detect_anomalies(expected_tt_ids=args.tt_ids)
|