[patch] Merge pull request #131 from Feramance/127-attributeerror-in-…

…headless-mode Fix headless mode category path checks
Feramance · Nov 14, 2024 · 4506eb9 · 4506eb9
2 parents fd06a19 + 52e6cb9
commit 4506eb9
Showing 1 changed file with 19 additions and 16 deletions.
diff --git a/qBitrr/arss.py b/qBitrr/arss.py
@@ -106,23 +106,26 @@ def __init__(
             fh = logging.FileHandler(logfile)
             self.logger.addHandler(fh)
         run_logs(self.logger)
-        categories = self.manager.qbit_manager.client.torrent_categories.categories
-        try:
-            categ = categories[self.category]
-            path = categ["savePath"]
-            if path:
-                self.logger.trace("Category exists with save path [%s]", path)
-                self.completed_folder = pathlib.Path(path)
-            else:
-                self.logger.trace("Category exists without save path")
-                self.completed_folder = pathlib.Path(COMPLETED_DOWNLOAD_FOLDER).joinpath(
-                    self.category
+
+        if not QBIT_DISABLED:
+            categories = self.manager.qbit_manager.client.torrent_categories.categories
+            try:
+                categ = categories[self.category]
+                path = categ["savePath"]
+                if path:
+                    self.logger.trace("Category exists with save path [%s]", path)
+                    self.completed_folder = pathlib.Path(path)
+                else:
+                    self.logger.trace("Category exists without save path")
+                    self.completed_folder = pathlib.Path(COMPLETED_DOWNLOAD_FOLDER).joinpath(
+                        self.category
+                    )
+            except KeyError:
+                self.completed_folder = pathlib.Path(COMPLETED_DOWNLOAD_FOLDER).joinpath(self.category)
+                self.manager.qbit_manager.client.torrent_categories.create_category(
+                    self.category, save_path=self.completed_folder
                 )
-        except KeyError:
-            self.completed_folder = pathlib.Path(COMPLETED_DOWNLOAD_FOLDER).joinpath(self.category)
-            self.manager.qbit_manager.client.torrent_categories.create_category(
-                self.category, save_path=self.completed_folder
-            )
+
         if not self.completed_folder.exists() and not SEARCH_ONLY:
             try:
                 self.completed_folder.mkdir(parents=True, exist_ok=True)