Refactor

bin/seed_armbian_torrents

@@ -1,12 +1,27 @@
 #!/usr/bin/env python3
-#
-# Armbian torrents seed script
-#
-# This script will download Armbian torrent files and add them to a qBittorrent
-# instance. It's intended to be run under /etc/cron.weekly or used in a systemd
-# timer.
-#
-# Python implementation of https://docs.armbian.com/Community_Torrent/
+"""seed_armbian_torrents
+
+Description:
+Armbian torrents seed script
+
+This script will download Armbian torrent files and add them to a qBittorrent
+instance. It's intended to be run under /etc/cron.weekly or used in a systemd
+timer.
+
+This is a Python implementation of https://docs.armbian.com/Community_Torrent/
+for qBittorrent.
+
+Usage:
+    seed_armbian_torrents (HOSTNAME) (USERNAME) (PASSWORD)
+    seed_armbian_torrents -h
+
+Examples:
+    seed_armbian_torrents "http://localhost:8080" "admin" "adminadmin"
+    seed_armbian_torrents "https://cat.seedhost.eu/lol/qbittorrent" "lol" "pw"
+
+Options:
+    -h, --help         show this help message and exit.
+"""
 
 import os
 from io import BytesIO
@@ -14,19 +29,25 @@ from tempfile import TemporaryDirectory
 from zipfile import ZipFile
 
 import requests
+from docopt import docopt
 from qbittorrent import Client
 
 if __name__ == "__main__":
+    args = docopt(__doc__)
 
-    qb = Client("http://localhost:8080/")
-    qb.login()
+    # Initialize client and login
+    qb = Client(args["HOSTNAME"])
+    qb.login(username=args["USERNAME"], password=args["PASSWORD"])
 
     with TemporaryDirectory() as tmp_dir:
         req = requests.get(
-            "https://dl.armbian.com/torrent/all-torrents.zip", stream=True
+            "https://dl.armbian.com/torrent/all-torrents.zip",
+            stream=True,
+            timeout=60,
         )
-        zip_file = ZipFile(BytesIO(req.content))
-        zip_file.extractall(tmp_dir)
+
+        with ZipFile(BytesIO(req.content)) as zip_file:
+            zip_file.extractall(tmp_dir)
 
         torrents = qb.torrents()
         for torrent in torrents:
@@ -38,4 +59,6 @@ if __name__ == "__main__":
         for torrent_file in torrent_files:
             with open(os.path.join(tmp_dir, torrent_file), "rb") as tf:
                 qb.download_from_file(tf, category="distro")
-            print(f"Added {os.path.basename(os.path.join(tmp_dir, torrent_file))}")
+            print(
+                f"Added {os.path.basename(os.path.join(tmp_dir, torrent_file))}"
+            )