first commit

Author: tryfan

.gitignore

@@ -0,0 +1,3 @@
+.idea/
+venv/
+config.yml

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3
+COPY requirements.txt /
+RUN pip install -r /requirements.txt
+WORKDIR /code
+ADD watcher.py /code/
+ADD config.yml /code/
+CMD [ "python", "watcher.py" ]

README.md

@@ -0,0 +1,3 @@
+# Deluge-Watcher
+
+Simple script to watch Deluge and pause any torrents that have completed that are not in your whitelisted seeding trackers. 

config.yml.dist

@@ -0,0 +1,9 @@
+deluge:
+    host: 127.0.0.1
+    port: 58846
+    username: user
+    password: passwd
+    tracker_whitelist:
+        - bigtorrent
+        - minitracker
+    retry_seconds: 60

requirements.txt

@@ -0,0 +1,2 @@
+deluge-client
+pyyaml

watcher.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+
+import time
+import logging
+import yaml
+from deluge_client import DelugeRPCClient
+
+logging.basicConfig(level="INFO")
+
+
+def convert(data):
+    if isinstance(data, bytes):
+        return data.decode()
+    if isinstance(data, (str, int)):
+        return str(data)
+    if isinstance(data, dict):
+        return dict(map(convert, data.items()))
+    if isinstance(data, tuple):
+        return tuple(map(convert, data))
+    if isinstance(data, list):
+        return list(map(convert, data))
+    if isinstance(data, set):
+        return set(map(convert, data))
+    return data
+
+
+with open("config.yml", "r") as yamlfile:
+    cfg = yaml.load(yamlfile, Loader=yaml.BaseLoader)
+
+client = DelugeRPCClient(
+    cfg['deluge']['host'],
+    int(cfg['deluge']['port']),
+    cfg['deluge']['username'],
+    cfg['deluge']['password']
+)
+
+while True:
+    if not client.connected:
+        try:
+            client.connect()
+        except client.Exception:
+            logging.error(client.Exception)
+
+    all_torrent_status_bytes = client.core.get_torrents_status(
+        {},
+        ['name', 'torrent_id', 'paused', 'is_finished', 'state', 'trackers']
+    )
+
+    for torrent_id, torrent in all_torrent_status_bytes.items():
+        tracker_whitelist = False
+        converted_torrent_id = convert(torrent_id)
+        converted_torrent = convert(torrent)
+        for tracker in converted_torrent['trackers']:
+            for whitelisted_tracker in cfg['deluge']['tracker_whitelist']:
+                if whitelisted_tracker.lower() in tracker['url'].lower():
+                    tracker_whitelist = True
+                    continue
+        if tracker_whitelist:
+            continue
+        if converted_torrent['is_finished'] == 'False':
+            continue
+        if converted_torrent['paused'] == 'True':
+            continue
+        print("Found torrent not in whitelist that is active and complete: %s\nid: %s" %
+              (converted_torrent['name'], converted_torrent_id))
+        client.core.pause_torrent([converted_torrent_id])
+    time.sleep(int(cfg['deluge']['retry_seconds']))