add torrents-info, torrents-stop, and torrents-stop-incomplete

Author: chapmanjacobd

tests/createdb/test_computers_add.py

@@ -0,0 +1 @@
+#

tests/createdb/test_torrents_add.py

@@ -0,0 +1 @@
+#

tests/mediafiles/test_torrents_list.py

@@ -0,0 +1 @@
+#

tests/mediafiles/test_torrents_start.py

@@ -0,0 +1 @@
+#

tests/mediafiles/test_torrents_stop.py

@@ -0,0 +1 @@
+#

tests/mediafiles/test_torrents_stop_incomplete.py

@@ -0,0 +1 @@
+#

tests/multidb/test_allocate_torrents.py

@@ -0,0 +1 @@
+#

tests/playback/test_torrents_info.py

@@ -75,7 +75,8 @@
         "images_to_pdf": "Convert folders of images into image PDFs",
         "pdf_edit": "Apply brightness, contrast, saturation, and sharpness adjustments to PDFs",
         "torrents_start": "Start torrents (qBittorrent-nox)",
-        "torrents_stop": "Stop torrents (qBittorrent-nox)",
+        "torrents_stop": "Stop seeding torrents (qBittorrent-nox)",
+        "torrents_stop_incomplete": "Stop downloading torrents (qBittorrent-nox)",
     },
     "Multi-database subcommands": {
         "merge_dbs": "Merge SQLite databases",
@@ -108,6 +109,7 @@
         "tabs_open": "Open your tabs for the day",
         "links_open": "Open links from link dbs",
         "surf": "Auto-load browser tabs in a streaming way (stdin)",
+        "torrents_info": "List torrents (qBittorrent-nox)",
     },
     "Database enrichment subcommands": {
         "dedupe_db": "Dedupe SQLite tables",
@@ -227,6 +229,7 @@ def print_help(parser) -> None:
     "xklb.mediafiles.pdf_edit.pdf_edit": [],
     "xklb.mediafiles.torrents_start.torrents_start": ["torrent-start"],
     "xklb.mediafiles.torrents_stop.torrents_stop": ["torrent-stop"],
+    "xklb.mediafiles.torrents_stop_incomplete.torrents_stop_incomplete": ["torrent-stop-incomplete"],
     "xklb.misc.dedupe_czkawka.czkawka_dedupe": ["dedupe-czkawka"],
     "xklb.misc.export_text.export_text": [],
     "xklb.multidb.copy_play_counts.copy_play_counts": [],
@@ -242,10 +245,11 @@ def print_help(parser) -> None:
     "xklb.playback.playback_control.playback_next": ["next"],
     "xklb.playback.playback_control.playback_now": ["now"],
     "xklb.playback.playback_control.playback_pause": ["pause", "play"],
-    "xklb.playback.playback_control.playback_stop": ["stop"],
     "xklb.playback.playback_control.playback_seek": ["ffwd", "rewind", "seek"],  # TODO: make rewind negative...
+    "xklb.playback.playback_control.playback_stop": ["stop"],
     "xklb.playback.surf.streaming_tab_loader": ["surf"],
     "xklb.playback.tabs_open.tabs_open": ["tb", "tabs", "open_tabs"],
+    "xklb.playback.torrents_info.torrents_info": ["torrent-info", "torrents", "torrent"],
     "xklb.tablefiles.eda.eda": ["preview"],
     "xklb.tablefiles.incremental_diff.incremental_diff": [],
     "xklb.tablefiles.columns.columns": [],
@@ -277,7 +281,20 @@ def create_subcommands_parser() -> argparse.ArgumentParser:
     subparsers = parser.add_subparsers()
 
     # this needs to stay inside the function to prevent side-effects during testing
-    known_subcommands = ["fs", "media", "open", "table", "tables", "tabs", "du", "search", "links", "images"]
+    known_subcommands = [
+        "fs",
+        "media",
+        "open",
+        "table",
+        "tables",
+        "tabs",
+        "du",
+        "search",
+        "links",
+        "images",
+        "torrents",
+        "torrent",
+    ]
 
     def consecutive_prefixes(s):
         prefixes = [s[:j] for j in range(5, len(s)) if s[:j] and s[:j] not in known_subcommands]
@@ -325,7 +342,12 @@ def add_parser(subparsers, func, aliases=None):
 
 def library(args=None) -> None:
     if args:
-        sys.argv = ["lb", *args]
+        original_argv = sys.argv
+        try:
+            sys.argv = ["lb", *args]
+            return library()
+        finally:
+            sys.argv = original_argv
 
     parser.exit_on_error = False  # type: ignore
     try:

xklb/__main__.py

@@ -1,13 +1,13 @@
 import concurrent.futures, os, statistics
 from pathlib import Path
-from urllib.parse import urlparse
 
 from torrentool.api import Torrent
 
 from xklb import usage
 from xklb.mediadb import db_playlists
 from xklb.utils import arg_utils, arggroups, argparse_utils, consts, db_utils, iterables, nums, objects, printing
 from xklb.utils.log_utils import log
+from xklb.utils.path_utils import domain_from_url
 
 
 def parse_args():
@@ -29,8 +29,7 @@ def get_tracker(torrent: Torrent):
     log.debug(torrent.announce_urls)
 
     for tracker in iterables.flatten(torrent.announce_urls):
-        url = urlparse(tracker)
-        domain = ".".join(url.netloc.rsplit(":")[0].rsplit(".", 2)[-2:]).lower()
+        domain = domain_from_url(tracker)
         return domain
 
     return torrent.source