[patch] Added exception handling

Feramance · Oct 16, 2024 · 3a3f4f1 · 3a3f4f1
1 parent b132e74
commit 3a3f4f1
Showing 1 changed file with 78 additions and 54 deletions.
diff --git a/qBitrr/arss.py b/qBitrr/arss.py
@@ -2245,25 +2245,34 @@ def db_update_single_series(
                         ).execute()
 
                     if self.use_temp_for_missing:
-                        if (
-                            searched
-                            and db_entry["qualityProfileId"] == self.temp_quality_profile_id
-                        ):
-                            data: JsonObject = {"qualityProfileId": self.main_quality_profile_id}
-                            self.logger.debug(
-                                "Updating quality profile for %s to %s",
-                                db_entry["title"],
-                                self.temp_quality_profile,
-                            )
-                        elif (
-                            not searched
-                            and db_entry["qualityProfileId"] == self.main_quality_profile_id
-                        ):
-                            data: JsonObject = {"qualityProfileId": self.temp_quality_profile_id}
-                            self.logger.debug(
-                                "Updating quality profile for %s to %s",
-                                db_entry["title"],
-                                self.temp_quality_profile,
+                        try:
+                            if (
+                                searched
+                                and db_entry["qualityProfileId"] == self.temp_quality_profile_id
+                            ):
+                                data: JsonObject = {
+                                    "qualityProfileId": self.main_quality_profile_id
+                                }
+                                self.logger.debug(
+                                    "Updating quality profile for %s to %s",
+                                    db_entry["title"],
+                                    self.temp_quality_profile,
+                                )
+                            elif (
+                                not searched
+                                and db_entry["qualityProfileId"] == self.main_quality_profile_id
+                            ):
+                                data: JsonObject = {
+                                    "qualityProfileId": self.temp_quality_profile_id
+                                }
+                                self.logger.debug(
+                                    "Updating quality profile for %s to %s",
+                                    db_entry["title"],
+                                    self.temp_quality_profile,
+                                )
+                        except KeyError:
+                            self.logger.warning(
+                                "Check quality profile settings for %s", db_entry["title"]
                             )
                         while True:
                             try:
@@ -2439,25 +2448,32 @@ def db_update_single_series(
                         else:
                             searched = (episodeCount + monitoredEpisodeCount) == episodeFileCount
                         if self.use_temp_for_missing:
-                            if (
-                                searched
-                                and db_entry["qualityProfileId"] == self.temp_quality_profile_id
-                            ):
-                                db_entry["qualityProfileId"] = self.main_quality_profile_id
-                                self.logger.debug(
-                                    "Updating quality profile for %s to %s",
-                                    db_entry["title"],
-                                    self.temp_quality_profile,
-                                )
-                            elif (
-                                not searched
-                                and db_entry["qualityProfileId"] == self.main_quality_profile_id
-                            ):
-                                db_entry["qualityProfileId"] = self.temp_quality_profile_id
-                                self.logger.debug(
-                                    "Updating quality profile for %s to %s",
-                                    db_entry["title"],
-                                    self.temp_quality_profile,
+                            try:
+                                if (
+                                    searched
+                                    and db_entry["qualityProfileId"]
+                                    == self.temp_quality_profile_id
+                                ):
+                                    db_entry["qualityProfileId"] = self.main_quality_profile_id
+                                    self.logger.debug(
+                                        "Updating quality profile for %s to %s",
+                                        db_entry["title"],
+                                        self.temp_quality_profile,
+                                    )
+                                elif (
+                                    not searched
+                                    and db_entry["qualityProfileId"]
+                                    == self.main_quality_profile_id
+                                ):
+                                    db_entry["qualityProfileId"] = self.temp_quality_profile_id
+                                    self.logger.debug(
+                                        "Updating quality profile for %s to %s",
+                                        db_entry["title"],
+                                        self.temp_quality_profile,
+                                    )
+                            except KeyError:
+                                self.logger.warning(
+                                    "Check quality profile settings for %s", db_entry["title"]
                                 )
                             while True:
                                 try:
@@ -2574,22 +2590,30 @@ def db_update_single_series(
                     ).execute()
 
                 if self.use_temp_for_missing:
-                    if searched and db_entry["qualityProfileId"] == self.temp_quality_profile_id:
-                        db_entry["qualityProfileId"] = self.main_quality_profile_id
-                        self.logger.debug(
-                            "Updating quality profile for %s to %s",
-                            db_entry["title"],
-                            self.temp_quality_profile,
-                        )
-                    elif (
-                        not searched
-                        and db_entry["qualityProfileId"] == self.main_quality_profile_id
-                    ):
-                        db_entry["qualityProfileId"] = self.temp_quality_profile_id
-                        self.logger.debug(
-                            "Updating quality profile for %s to %s",
-                            db_entry["title"],
-                            self.temp_quality_profile,
+                    try:
+                        if (
+                            searched
+                            and db_entry["qualityProfileId"] == self.temp_quality_profile_id
+                        ):
+                            db_entry["qualityProfileId"] = self.main_quality_profile_id
+                            self.logger.debug(
+                                "Updating quality profile for %s to %s",
+                                db_entry["title"],
+                                self.temp_quality_profile,
+                            )
+                        elif (
+                            not searched
+                            and db_entry["qualityProfileId"] == self.main_quality_profile_id
+                        ):
+                            db_entry["qualityProfileId"] = self.temp_quality_profile_id
+                            self.logger.debug(
+                                "Updating quality profile for %s to %s",
+                                db_entry["title"],
+                                self.temp_quality_profile,
+                            )
+                    except KeyError:
+                        self.logger.warning(
+                            "Check quality profile settings for %s", db_entry["title"]
                         )
                     while True:
                         try: