torrents: add --missing and --start

Author: chapmanjacobd

library/createdb/av.py

@@ -157,7 +157,7 @@ def munge_av_tags(args, media) -> dict:
 
     if not media.get("size"):
         if size:
-            media["size"] = size
+            media["size"] = int(size)
         elif bitrate_bps and duration:
             total_bits = duration * float(bitrate_bps)
             total_bytes = math.ceil(total_bits / 8)

library/playback/torrents_info.py

@@ -21,9 +21,9 @@ def parse_args():
         help="Sort by priority, ratio, download, upload, download+upload, size, avg_size, remaining",
     )
 
-    parser.add_argument(
-        "--force-start", "--start", action=argparse.BooleanOptionalAction, help="Force start matching torrents"
-    )
+    parser.add_argument("--start", action=argparse.BooleanOptionalAction, help="Start matching torrents")
+    parser.add_argument("--force-start", action=argparse.BooleanOptionalAction, help="Force start matching torrents")
+    parser.add_argument("--check", "--recheck", action="store_true", help="Check matching torrents")
     parser.add_argument("--stop", action="store_true", help="Stop matching torrents")
     parser.add_argument("--move", help="Directory to move folders/files")
     parser.add_argument(
@@ -43,6 +43,9 @@ def parse_args():
 
     arggroups.qBittorrent_torrents_post(args)
 
+    if args.move and not (args.stop or args.delete_rows or args.delete_files):
+        processes.exit_error("--move requires --stop or --delete-rows")
+
     return args
 
 
@@ -83,7 +86,7 @@ def is_matching(args, t):
         if t.num_complete == 0 and t.seen_complete > 0
         else (
             t.added_on
-            if t.state in ['downloading', 'forceDL', 'stalledDL', 'uploading', 'forcedUP', 'stalledUP']
+            if t.state in ["downloading", "forceDL", "stalledDL", "uploading", "forcedUP", "stalledUP"]
             and t.num_complete == 0
             else 0
         )
@@ -114,6 +117,10 @@ def is_matching(args, t):
 def filter_torrents_by_activity(args, torrents):
     if args.stopped:
         torrents = [t for t in torrents if t.state_enum.is_stopped]
+    if args.errored:
+        torrents = [t for t in torrents if t.state == "error"]
+    if args.missing:
+        torrents = [t for t in torrents if t.state == "missingFiles"]
     if args.complete:
         torrents = [t for t in torrents if t.state_enum.is_complete]
     if args.incomplete:
@@ -351,14 +358,22 @@ def gen_row(t):
 
     torrent_hashes = [t.hash for t in torrents]
 
-    if args.force_start is not None:
-        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.check:
+        print("Checking", len(torrents))
+        qbt_client.torrents_recheck(torrent_hashes=torrent_hashes)
+
+    if args.start is not None:
+        print("Starting", len(torrents))
+        qbt_client.torrents_start(torrent_hashes=torrent_hashes)
+
+    if args.force_start is not None:
+        print("Force-starting", len(torrents))
+        qbt_client.torrents_set_force_start(args.force_start, torrent_hashes=torrent_hashes)
+
     if args.delete_incomplete:
         for t in torrents:
             if not os.path.exists(t.content_path):
@@ -375,7 +390,7 @@ def gen_row(t):
                     if file.progress < args.delete_incomplete:
                         path.unlink(missing_ok=True)
 
-    if (args.stop or args.mark_deleted or args.delete_rows or args.delete_files) and args.move:
+    if (args.stop or args.delete_rows or args.delete_files) and args.move:
         for t in torrents:
             if os.path.exists(t.content_path):
                 new_path = Path(args.move)

library/playback/torrents_status.py

@@ -98,13 +98,15 @@ def shorten(s, width):
 
     if len(torrents_by_state) > 1:
         remaining = sum(t.amount_left for t in torrents)
-        categories.append({
-            "state": 'total',
-            "count": len(torrents),
-            "size": strings.file_size(sum(t.total_size for t in torrents)),
-            "remaining": strings.file_size(remaining) if remaining else None,
-            "files": (sum(len(t.files) for t in torrents) if args.file_counts else None),
-        })
+        categories.append(
+            {
+                "state": "total",
+                "count": len(torrents),
+                "size": strings.file_size(sum(t.total_size for t in torrents)),
+                "remaining": strings.file_size(remaining) if remaining else None,
+                "files": (sum(len(t.files) for t in torrents) if args.file_counts else None),
+            }
+        )
     printing.table(iterables.list_dict_filter_bool(categories))
     print()
 
@@ -123,9 +125,7 @@ def shorten(s, width):
                     "count": len(tracker_torrents),
                     "size": sum(t.total_size for t in tracker_torrents),
                     "remaining": remaining,
-                    "files": (
-                        sum(len(t.files) for t in tracker_torrents) if args.file_counts else None
-                    ),  # a bit slow
+                    "files": (sum(len(t.files) for t in tracker_torrents) if args.file_counts else None),  # a bit slow
                 }
             )
         if trackers:

library/utils/arggroups.py

@@ -2001,6 +2001,8 @@ def qBittorrent_torrents(parent_parser):
     )
     parser.add_argument("--inactive", "--dead", action="store_true", help="Include inactive torrents")
     parser.add_argument("--stopped", "--paused", action="store_true", help="Include stopped torrents")
+    parser.add_argument("--missing", action="store_true", help="Include missing torrents")
+    parser.add_argument("--errored", action="store_true", help="Include errored torrents")
     parser.add_argument("--complete", "--completed", action="store_true", help="Include completed torrents")
     parser.add_argument("--incomplete", action="store_true", help="Include incomplete torrents")