torrent-info: fix categories sorting

chapmanjacobd · Dec 11, 2024 · 8497c29 · 8497c29
1 parent 400d8fc
commit 8497c29
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/xklb/playback/torrents_info.py b/xklb/playback/torrents_info.py
@@ -185,6 +185,14 @@ def torrents_info():
                 "file_count": sum(len(t.files) for t in torrents) if args.verbose >= 1 else None,  # a bit slow
             }
         )
+
+    categories = sorted(
+        categories,
+        key=lambda d: (
+            d["state"].endswith(("downloading", "DL")),
+            iterables.safe_index(interesting_states, d["state"]),
+        ),
+    )
     printing.table(iterables.list_dict_filter_bool(categories))
     print()
 

diff --git a/xklb/utils/iterables.py b/xklb/utils/iterables.py
@@ -55,6 +55,15 @@ def safe_len(list_) -> Any | None:
         return len(str(list_))
 
 
+def safe_index(list_, value) -> int:
+    if not list_:
+        return 0
+    try:
+        return list_.index(value)
+    except ValueError:
+        return -1
+
+
 def get_all_lists(nested_dict):
     list_ = []