git

Seedbox/30 KontrolaCoJsmeStahliZUltraCC.py

@@ -0,0 +1,214 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import os
+import pymysql
+import bencodepy
+from tqdm import tqdm
+
+# =====================================
+# CONFIG
+# =====================================
+
+ULTRACC_ROOT = r"\\tower\torrents\ultracc"
+DRY_MODE = False
+
+DB_CONFIG = dict(
+    host="192.168.1.50",
+    user="root",
+    password="Vlado9674+",
+    database="torrents",
+    charset="utf8mb4"
+)
+
+# =====================================
+# TORRENT PARSER
+# =====================================
+
+def parse_torrent(blob):
+
+    data = bencodepy.decode(blob)
+    info = data[b'info']
+
+    files = []
+
+    # multi file
+    if b'files' in info:
+        for f in info[b'files']:
+            rel_path = "/".join(p.decode(errors="ignore") for p in f[b'path'])
+            size = f[b'length']
+            files.append((rel_path, size))
+        multi = True
+    else:
+        name = info[b'name'].decode(errors="ignore")
+        size = info[b'length']
+        files.append((name, size))
+        multi = False
+
+    return files, multi
+
+# =====================================
+# BUILD FILESYSTEM INDEX
+# =====================================
+
+import pickle
+import os
+
+INDEX_FILE = r"U:\PycharmProjects\Torrents\fs_index_ultracc.pkl"
+FORCE_REBUILD = False
+
+
+def build_fs_index():
+
+    # ============================
+    # LOAD EXISTING INDEX
+    # ============================
+
+    if os.path.exists(INDEX_FILE) and not FORCE_REBUILD:
+        print("Načítám uložený filesystem index...")
+
+        with open(INDEX_FILE, "rb") as f:
+            index = pickle.load(f)
+
+        print(f"Načten index ({len(index)} klíčů)")
+        return index
+
+    # ============================
+    # BUILD NEW INDEX
+    # ============================
+
+    print("Indexuji filesystem...")
+
+    index = {}
+    pocet = 0
+
+    for root, _, files in os.walk(ULTRACC_ROOT):
+        for f in files:
+            full = os.path.join(root, f)
+
+            try:
+                size = os.path.getsize(full)
+            except OSError:
+                continue
+
+            key = (f.lower(), size)
+
+            pocet += 1
+            if pocet % 100 == 0:
+                print(pocet)
+
+            index.setdefault(key, []).append(full)
+
+    print(f"Index obsahuje {len(index)} unikátních souborů")
+
+    # ============================
+    # SAVE INDEX
+    # ============================
+
+    print("Ukládám index na disk...")
+
+    with open(INDEX_FILE, "wb") as f:
+        pickle.dump(index, f, protocol=pickle.HIGHEST_PROTOCOL)
+
+    print("Index uložen")
+
+    return index
+
+
+# =====================================
+# VALIDACE ROOTU
+# =====================================
+
+def validate_root(root, torrent_files):
+
+    for rel_path, size in torrent_files:
+
+        check = os.path.join(root, rel_path.replace("/", os.sep))
+
+        if not os.path.exists(check):
+            return False
+
+    return True
+
+# =====================================
+# MAIN
+# =====================================
+
+def main():
+
+    fs_index = build_fs_index()
+
+    conn = pymysql.connect(**DB_CONFIG)
+    cur = conn.cursor()
+
+    cur.execute("""
+        SELECT id, torrent_content
+        FROM torrents
+        WHERE torrent_content IS NOT NULL and physical_exists=FALSE
+    """)
+
+    rows = cur.fetchall()
+
+    print(f"Torrentů ke kontrole: {len(rows)}")
+
+    success = 0
+
+    for tid, blob in tqdm(rows):
+
+        try:
+            torrent_files, multi = parse_torrent(blob)
+
+            # vezmeme největší soubor pro lookup
+            rel_path, size = max(torrent_files, key=lambda x: x[1])
+            fname = os.path.basename(rel_path).lower()
+
+            key = (fname, size)
+
+            if key not in fs_index:
+                continue
+
+            found = False
+
+            for full_path in fs_index[key]:
+
+                if multi:
+                    # ROOT = odečtení relativní cesty
+                    root = full_path[:-len(rel_path)]
+                    root = root.rstrip("\\/")
+                else:
+                    # single file
+                    root = os.path.dirname(full_path)
+
+                if validate_root(root, torrent_files):
+
+                    found = True
+                    success += 1
+
+                    print(f"[FOUND] Torrent {tid} → {root}")
+
+                    if not DRY_MODE:
+                        cur.execute("""
+                            UPDATE torrents
+                            SET physical_exists = 1
+                            WHERE id = %s
+                        """, (tid,))
+
+                    break
+
+            if not found:
+                pass
+
+        except Exception as e:
+            print(f"ERROR torrent {tid}: {e}")
+
+    if not DRY_MODE:
+        conn.commit()
+
+    print(f"Celkem nalezeno: {success}")
+
+    cur.close()
+    conn.close()
+
+
+if __name__ == "__main__":
+    main()