catch Arr NotFounds

Kometa-Team · Dec 29, 2021 · 4e54942 · 4e54942
1 parent 8f6960e
commit 4e54942
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/modules/radarr.py b/modules/radarr.py
@@ -2,7 +2,7 @@
 from modules import util
 from modules.util import Failed
 from arrapi import RadarrAPI
-from arrapi.exceptions import ArrException, Invalid
+from arrapi.exceptions import ArrException, Invalid, NotFound
 
 logger = logging.getLogger("Plex Meta Manager")
 
@@ -107,7 +107,7 @@ def add_tmdb(self, tmdb_ids, **options):
                     exists.extend(_e)
                     invalid.extend(_i)
                     movies = []
-                except Invalid as e:
+                except (Invalid, NotFound) as e:
                     raise Failed(f"Radarr Error: {e}")
 
         if len(added) > 0:

diff --git a/modules/sonarr.py b/modules/sonarr.py
@@ -2,7 +2,7 @@
 from modules import util
 from modules.util import Failed
 from arrapi import SonarrAPI
-from arrapi.exceptions import ArrException, Invalid
+from arrapi.exceptions import ArrException, Invalid, NotFound
 
 logger = logging.getLogger("Plex Meta Manager")
 
@@ -133,7 +133,7 @@ def add_tvdb(self, tvdb_ids, **options):
                     exists.extend(_e)
                     invalid.extend(_i)
                     shows = []
-                except Invalid as e:
+                except (Invalid, NotFound) as e:
                     raise Failed(f"Sonarr Error: {e}")
 
         if len(added) > 0: