From 52e6cb9130ae03a6d308a6c0976d12710cd6707f Mon Sep 17 00:00:00 2001 From: Feramance Date: Thu, 14 Nov 2024 09:17:46 +0100 Subject: [PATCH] Fix headless mode category path checks --- qBitrr/arss.py | 35 +++++++++++++++++++---------------- 1 file changed, 19 insertions(+), 16 deletions(-) diff --git a/qBitrr/arss.py b/qBitrr/arss.py index ab91b227..b411978d 100755 --- 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)