torrents: add --status

Author: chapmanjacobd

library/playback/links_open.py

@@ -1,4 +1,4 @@
-import argparse, shlex
+import argparse
 
 from library import usage
 from library.mediadb import db_history, db_media
@@ -38,9 +38,6 @@ def parse_args() -> argparse.Namespace:
         args.title_prefix = ["https://www.google.com/search?q=%"]
     args.title_prefix = [s if "%" in s else s + "%" for s in args.title_prefix]
 
-    if args.browser:
-        args.browser = shlex.split(args.browser)
-
     return args
 
 

library/playback/torrents_info.py

@@ -161,24 +161,8 @@ def shorten(s, width):
     torrents = qbt_client.torrents_info()
 
     error_torrents = [t for t in torrents if t.state_enum.is_errored]
-    error_torrents = sorted(
-        error_torrents, key=lambda t: (t.amount_left == t.total_size, t.eta, t.amount_left), reverse=True
-    )
     if error_torrents:
-        print("Error Torrents")
-        tbl = [
-            {
-                "state": t.state,
-                "name": shorten(t.name, width=40),
-                "progress": strings.safe_percent(t.progress),
-                "eta": strings.duration_short(t.eta) if t.eta < 8640000 else None,
-                "remaining": strings.file_size(t.amount_left) if t.amount_left > 0 else None,
-                "files": len(t.files) if args.file_counts else None,
-            }
-            for t in error_torrents
-        ]
-        printing.table(tbl)
-        print()
+        args.status = True
 
     torrents = filter_torrents_by_activity(args, torrents)
     torrents = [t for t in torrents if is_matching(args, t)]
@@ -274,6 +258,8 @@ def gen_row(t):
                 d |= {"priority": str(t.priority) + (" [F]" if t.force_start else "")}
             if args.trackers:
                 d |= {"tracker": qbt_get_tracker(qbt_client, t)}
+            if args.status:
+                d |= {"state": t.state}
 
             if args.verbose >= consts.LOG_INFO:
                 d |= {
@@ -338,10 +324,11 @@ def gen_row(t):
                 d |= {"priority": str(t.priority) + (" [F]" if t.force_start else "")}
             if args.trackers:
                 d |= {"tracker": qbt_get_tracker(qbt_client, t)}
+            if args.status:
+                d |= {"state": t.state}
 
             if args.verbose >= consts.LOG_INFO:
                 d |= {
-                    "state": t.state,
                     "added_on": strings.relative_datetime(t.added_on),
                     "size": strings.file_size(t.total_size),
                     "remaining": strings.file_size(t.amount_left),

library/playback/torrents_status.py

@@ -96,12 +96,7 @@ def shorten(s, width):
             }
         )
 
-    categories = sorted(
-        categories,
-        key=lambda d: (
-            iterables.safe_index(interesting_states, d["state"])
-        )
-    )
+    categories = sorted(categories, key=lambda d: (iterables.safe_index(interesting_states, d["state"])))
 
     if len(torrents_by_state) > 1:
         remaining = sum(t.amount_left for t in torrents)

library/utils/arggroups.py

@@ -2058,6 +2058,7 @@ def qBittorrent_torrents(parent_parser):
         "--file-counts", "--files", "--counts", action="store_true", help="Include file counts column (a bit slow)"
     )
     parser.add_argument("--trackers", action="store_true", help="Include tracker column")
+    parser.add_argument("--status", "--state", action="store_true", help="Include state column")
 
     parser.add_argument("--file-search", "-s", nargs="+", help="The file path substring to search for")
     parser.add_argument("torrent_search", nargs="*", help="The info_hash, name, or save_path substring to search for")

library/utils/date_utils.py

@@ -5,16 +5,19 @@
 
 from library.utils import iterables, nums
 
+
 def is_tz_aware(dt: datetime.datetime) -> bool:
     return dt.tzinfo is not None and dt.tzinfo.utcoffset(dt) is not None
 
+
 def maybe_tz_now(maybe_tz_dt: datetime.datetime):
     if is_tz_aware(maybe_tz_dt):
         now = datetime.datetime.now(tz=tz.utc).astimezone()
     else:
         now = datetime.datetime.now()
     return now
 
+
 def super_parser(date_str, fallback=False):
     parsing_strategies = [
         {},  # default