new functions

pytorrent/services/rtorrent.py

@@ -371,6 +371,66 @@ def browse_path(profile: dict, path: str | None = None) -> dict:
     return {"path": base, "parent": parent, "dirs": dirs[:300], "source": "rtorrent"}
 
 
+POST_CHECK_DOWNLOAD_LABEL = "To download after check"
+
+
+def _label_names(value: str) -> list[str]:
+    names: list[str] = []
+    for part in str(value or "").replace(";", ",").replace("|", ",").split(","):
+        label = part.strip()
+        if label and label not in names:
+            names.append(label)
+    return names
+
+
+def _label_value(labels: list[str]) -> str:
+    return ", ".join([label for label in labels if str(label or "").strip()])
+
+
+def _row_progress_complete(row: dict) -> bool:
+    size = int(row.get("size") or 0)
+    completed = int(row.get("completed_bytes") or 0)
+    return bool(row.get("complete")) or (size > 0 and completed >= size) or float(row.get("progress") or 0) >= 100.0
+
+
+def apply_post_check_policy(profile: dict, rows: list[dict], previous_rows: dict[str, dict] | None = None) -> list[dict]:
+    """Start complete torrents after check; pause and label incomplete ones."""
+    previous_rows = previous_rows or {}
+    c = client_for(profile)
+    changes: list[dict] = []
+    for row in rows:
+        h = str(row.get("hash") or "")
+        prev = previous_rows.get(h) or {}
+        was_checking = str(prev.get("status") or "") == "Checking" or int(prev.get("hashing") or 0) > 0
+        is_checking = str(row.get("status") or "") == "Checking" or int(row.get("hashing") or 0) > 0
+        if not h or not was_checking or is_checking:
+            continue
+        complete = _row_progress_complete(row)
+        try:
+            if complete:
+                # Note: Po zakonczonym checku pelny torrent jest automatycznie startowany, zeby od razu seedowal.
+                c.call("d.start", h)
+                labels = [label for label in _label_names(str(row.get("label") or "")) if label != POST_CHECK_DOWNLOAD_LABEL]
+                if _label_value(labels) != str(row.get("label") or ""):
+                    c.call("d.custom1.set", h, _label_value(labels))
+                    row["label"] = _label_value(labels)
+                row.update({"state": 1, "active": 1, "paused": False, "status": "Seeding"})
+                changes.append({"hash": h, "action": "start", "complete": True})
+            else:
+                # Note: Niepelny torrent po checku trafia do pauzy i dostaje etykiete informujaca, ze wymaga dalszego pobierania.
+                c.call("d.start", h)
+                c.call("d.pause", h)
+                labels = _label_names(str(row.get("label") or ""))
+                if POST_CHECK_DOWNLOAD_LABEL not in labels:
+                    labels.append(POST_CHECK_DOWNLOAD_LABEL)
+                    c.call("d.custom1.set", h, _label_value(labels))
+                row.update({"state": 1, "active": 0, "paused": True, "status": "Paused", "label": _label_value(labels)})
+                changes.append({"hash": h, "action": "pause_and_label", "complete": False, "label": POST_CHECK_DOWNLOAD_LABEL})
+        except Exception as exc:
+            changes.append({"hash": h, "action": "post_check_policy_failed", "error": str(exc)})
+    return changes
+
+
 TORRENT_FIELDS = [
     "d.hash=", "d.name=", "d.state=", "d.complete=", "d.size_bytes=", "d.completed_bytes=",
     "d.ratio=", "d.up.rate=", "d.down.rate=", "d.up.total=", "d.down.total=", "d.peers_connected=",