torrents: add torrents export option

Author: chapmanjacobd

library/__main__.py

@@ -129,7 +129,6 @@
         "gallery_update": "Update online gallery media",
         "links_update": "Update a link-scraping database",
         "reddit_update": "Update reddit media",
-        "computers_update": "Update computer stats",
     },
     "Misc subcommands": {
         "export_text": "Export HTML files from SQLite databases",
@@ -162,7 +161,6 @@ def print_help(parser) -> None:
 
 modules = {
     "library.createdb.computers_add.computers_add": ["computer-add", "pc-add", "ssh-add"],
-    "library.createdb.computers_add.computers_update": ["computer-update", "pc-update", "ssh-update"],
     "library.createdb.fs_add.fs_add": ["filesystem-add", "x", "extract"],
     "library.createdb.fs_add.fs_update": ["filesystem-update", "xu"],
     "library.createdb.gallery_add.gallery_add": ["gdl-add", "ga"],

library/createdb/computers_add.py

@@ -7,14 +7,13 @@
 from library.utils.log_utils import log
 
 
-def parse_args(action, usage):
-    parser = argparse_utils.ArgumentParser(usage=usage)
+def parse_args():
+    parser = argparse_utils.ArgumentParser(usage=usage.computers_add)
     arggroups.debug(parser)
 
     arggroups.database(parser)
-    if action == consts.SC.computers_add:
-        parser.add_argument("--ignore-mounts", nargs="+", default=[], help="List of mountpoints to ignore")
-        parser.add_argument("hostnames", nargs="+", help="List of hostnames to connect to")
+    parser.add_argument("--ignore-mounts", nargs="+", default=[], help="List of mountpoints to ignore")
+    parser.add_argument("hostnames", nargs="+", help="List of hostnames to connect to")
     args = parser.parse_args()
     arggroups.args_post(args, parser, create_db=True)
 
@@ -101,12 +100,6 @@ def computer_add(args, hostnames):
 
 
 def computers_add():
-    args = parse_args(consts.SC.computers_add, usage.computers_add)
+    args = parse_args()
 
     computer_add(args, args.hostnames)
-
-
-def computers_update():
-    args = parse_args(consts.SC.computers_update, usage.computers_update)
-
-    computer_add(args, [d["path"] for d in args.db.query("select path from playlists")])

library/playback/media_printer.py

@@ -161,7 +161,7 @@ def media_printer(args, data, units=None, media_len=None) -> None:
             D["duration"] = total_duration
             D["avg_duration"] = nums.safe_mean(m.get("duration") for m in media)
 
-        if hasattr(args, "action") and "history" in tables:
+        if hasattr(args, "action") and "history" in tables and "id" in m_columns:
             if action in (SC.download, SC.download_status) and "time_downloaded" in m_columns:
                 D["download_duration"] = cadence_adjusted_items(args, D["count"], time_column="time_downloaded")
             elif total_duration > 0:

library/playback/torrents_info.py

@@ -4,7 +4,7 @@
 
 from library import usage
 from library.mediafiles import torrents_start
-from library.utils import arggroups, argparse_utils, consts, iterables, printing, processes, strings
+from library.utils import arggroups, argparse_utils, consts, iterables, path_utils, printing, processes, strings
 from library.utils.path_utils import domain_from_url
 
 
@@ -45,10 +45,13 @@ def parse_args():
     parser.add_argument("--all", action="store_true", help="Show active and inactive torrents")
     parser.add_argument("--active", action="store_true", help="Show active torrents")
     parser.add_argument("--inactive", "--dead", action="store_true", help="Show inactive torrents")
+    parser.add_argument("--stopped", "--paused", action="store_true", help="Show stopped torrents")
 
     parser.add_argument(
         "--force-start", "--start", action=argparse.BooleanOptionalAction, help="Force start matching torrents"
     )
+    parser.add_argument("--stop", action="store_true", help="Stop matching torrents")
+    parser.add_argument("--export", action="store_true", help="Export matching torrent files")
     arggroups.capability_soft_delete(parser)
     arggroups.capability_delete(parser)
     arggroups.debug(parser)
@@ -133,6 +136,9 @@ def shorten(s, width):
         printing.table(tbl)
         print()
 
+    if args.stopped:
+        torrents = [t for t in torrents if t.state_enum.is_stopped]
+
     torrents = filter_torrents_by_activity(args, torrents)
 
     if args.torrent_search or args.file_search:
@@ -187,7 +193,7 @@ def shorten(s, width):
         or (t.state_enum.is_complete and t.uploaded_session > 0)
     ]
     if active_torrents:
-        print("Active Torrents")
+        print(f"Active Torrents ({len(active_torrents)})")
 
         def gen_row(t):
             d = {
@@ -248,7 +254,7 @@ def gen_row(t):
         or (t.state_enum.is_complete and t.uploaded_session == 0)
     ]
     if inactive_torrents:
-        print("Inactive Torrents")
+        print(f"Inactive Torrents ({len(inactive_torrents)})")
 
         def gen_row(t):
             d = {
@@ -318,6 +324,20 @@ def gen_row(t):
         print("Force-starting", len(torrents))
         qbt_client.torrents_set_force_start(args.force_start, torrent_hashes=torrent_hashes)
 
+    if args.stop:
+        print("Stopping", len(torrents))
+        qbt_client.torrents_stop(torrent_hashes=torrent_hashes)
+
+    if args.export:
+        p = Path("exported_torrents")
+        p.mkdir(exist_ok=True)
+        for idx, t in enumerate(torrents):
+            printing.print_overwrite("Exporting", idx + 1, "of", len(torrents), "to", p)
+
+            file_name = f"{qbt_get_tracker(qbt_client, t)}_{t.name}_{t.hash}.torrent"
+            file_name = path_utils.clean_path(file_name.encode())
+            (p / file_name).write_bytes(qbt_client.torrents_export(torrent_hash=t.hash))
+
     if args.temp_drive and Path(args.temp_drive).is_absolute():
         temp_prefix = Path(args.temp_drive)
     else: