New option cat in trackers (#400)

* New option cat in trackers * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: bobokun <[email protected]>
StuffAnThings · Oct 7, 2023 · 51a6f12 · 51a6f12
1 parent a02a066
commit 51a6f12
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/modules/core/category.py b/modules/core/category.py
@@ -23,7 +23,7 @@ def category(self):
         logger.separator("Updating Categories", space=False, border=False)
         torrent_list = self.qbt.get_torrents({"category": "", "status_filter": "completed"})
         for torrent in torrent_list:
-            new_cat = self.qbt.get_category(torrent.save_path)
+            new_cat = self.get_tracker_cat(torrent) or self.qbt.get_category(torrent.save_path)
             if new_cat == self.uncategorized_mapping:
                 logger.print_line(f"{torrent.name} remains uncategorized.", self.config.loglevel)
                 continue
@@ -44,6 +44,10 @@ def category(self):
         else:
             logger.print_line("No new torrents to categorize.", self.config.loglevel)
 
+    def get_tracker_cat(self, torrent):
+        tracker = self.qbt.get_tags(torrent.trackers)
+        return tracker["cat"]
+
     def update_cat(self, torrent, new_cat, cat_change):
         """Update category based on the torrent information"""
         tracker = self.qbt.get_tags(torrent.trackers)

diff --git a/modules/qbittorrent.py b/modules/qbittorrent.py
@@ -204,6 +204,7 @@ def get_tags(self, trackers):
         urls = [x.url for x in trackers if x.url.startswith("http")]
         tracker = {}
         tracker["tag"] = None
+        tracker["cat"] = None
         tracker["notifiarr"] = None
         tracker["url"] = None
         tracker_other_tag = self.config.util.check_for_attribute(
@@ -238,6 +239,16 @@ def get_tags(self, trackers):
                         tracker["tag"] = self.config.util.check_for_attribute(
                             self.config.data, "tag", parent="tracker", subparent=tag_url, default=tag_url, var_type="list"
                         )
+                        tracker["cat"] = self.config.util.check_for_attribute(
+                            self.config.data,
+                            "cat",
+                            parent="tracker",
+                            subparent=tag_url,
+                            default_is_none=True,
+                            var_type="str",
+                            save=False,
+                            do_print=False,
+                        )
                         if tracker["tag"] == [tag_url]:
                             self.config.data["tracker"][tag_url]["tag"] = [tag_url]
                         if isinstance(tracker["tag"], str):