Better handle exceptions

Add validate to tests

Author: Buried-In-Code

mediux_posters/__main__.py

@@ -160,11 +160,19 @@ def process_image(
                 set_data.username,
             )
             should_log = False
-        mediux.download_image(file_id=file.id, output=image_file)
+        try:
+            mediux.download_image(file_id=file.id, output=image_file)
+        except ServiceError as err:
+            LOGGER.error("[Mediux] %s", err)
+            return
 
-        success = service.upload_image(
-            object_id=obj.id, image_file=image_file, kometa_integration=kometa_integration
-        )
+        try:
+            success = service.upload_image(
+                object_id=obj.id, image_file=image_file, kometa_integration=kometa_integration
+            )
+        except ServiceError as err:
+            LOGGER.error("[%s] %s", type(service).__name__, err)
+            success = False
         setattr(obj, uploaded_attr, success)
         if success:
             service.cache.insert(
@@ -190,7 +198,12 @@ def process_image(
         filename="backdrop.jpg",
     )
     if isinstance(entry, Show) and isinstance(set_data, ShowSet):
-        for season in entry.seasons or service.list_seasons(show_id=entry.id):
+        try:
+            seasons = service.list_seasons(show_id=entry.id)
+        except ServiceError as err:
+            LOGGER.error("[%s] %s", type(service).__name__, err)
+            seasons = []
+        for season in seasons:
             entry.seasons.append(season)
             mediux_season = next(
                 (x for x in set_data.show.seasons if x.number == season.number), None
@@ -204,9 +217,12 @@ def process_image(
                 parent=slugify(value=entry.display_name),
                 filename=f"s{season.number:02}.jpg",
             )
-            for episode in season.episodes or service.list_episodes(
-                show_id=entry.id, season_id=season.id
-            ):
+            try:
+                episodes = service.list_episodes(show_id=entry.id, season_id=season.id)
+            except ServiceError as err:
+                LOGGER.error("[%s] %s", type(service).__name__, err)
+                episodes = []
+            for episode in episodes:
                 season.episodes.append(episode)
                 mediux_episode = next(
                     (x for x in mediux_season.episodes if x.number == episode.number), None
@@ -221,7 +237,12 @@ def process_image(
                     filename=f"s{season.number:02}e{episode.number:02}.jpg",
                 )
     elif isinstance(entry, Collection) and isinstance(set_data, CollectionSet):
-        for movie in service.list_collection_movies(collection_id=entry.id):
+        try:
+            movies = service.list_collection_movies(collection_id=entry.id)
+        except ServiceError as err:
+            LOGGER.error("[%s] %s", type(service).__name__, err)
+            movies = []
+        for movie in movies:
             entry.movies.append(movie)
             mediux_movie = next(
                 (x for x in set_data.collection.movies if x.id == movie.tmdb_id), None

mediux_posters/mediux/service.py

@@ -102,7 +102,7 @@ def __init__(self, base_url: str, token: str):
             headers={
                 "Accept": "application/json",
                 "Authorization": f"Bearer {token}",
-                "User-Agent": f"{__project__}/{__version__}/{system()}: {release()}",
+                "User-Agent": f"{__project__.title()}/{__version__}/{system()}: {release()}",
             },
         )
 
@@ -130,8 +130,8 @@ def _perform_graphql_request(self, query: str) -> dict[str, Any]:
 
     def validate(self) -> bool:
         try:
-            # TODO: Do single set call to validate credentials
-            return True
+            results = self.list_movie_sets(tmdb_id=324857)
+            return results != []
         except ServiceError as err:
             LOGGER.error("[Mediux] %s", err)
         return False

mediux_posters/services/jellyfin/service.py

@@ -36,7 +36,7 @@ def __init__(self, base_url: str, token: str):
             headers={
                 "Accept": "application/json",
                 "X-Emby-Token": token,
-                "User-Agent": f"{__project__}/{__version__}/{system()}: {release()}",
+                "User-Agent": f"{__project__.title()}/{__version__}/{system()}: {release()}",
             },
         )
 

mediux_posters/services/plex/service.py

@@ -35,7 +35,7 @@ def __init__(self, base_url: str, token: str):
             headers={
                 "Accept": "application/json",
                 "X-Plex-Token": token,
-                "User-Agent": f"{__project__}/{__version__}/{system()}: {release()}",
+                "User-Agent": f"{__project__.title()}/{__version__}/{system()}: {release()}",
             },
         )
 

tests/conftest.py

@@ -19,7 +19,9 @@ def mediux_token() -> str:
 
 @pytest.fixture(scope="session")
 def mediux_session(mediux_base_url: str, mediux_token: str) -> Mediux:
-    return Mediux(base_url=mediux_base_url, token=mediux_token)
+    mediux = Mediux(base_url=mediux_base_url, token=mediux_token)
+    assert mediux.validate() is True
+    return mediux
 
 
 @pytest.fixture(scope="session")
@@ -35,7 +37,9 @@ def jellyfin_token() -> str | None:
 @pytest.fixture(scope="session")
 def jellyfin_session(jellyfin_base_url: str | None, jellyfin_token: str | None) -> Jellyfin | None:
     if jellyfin_base_url and jellyfin_token:
-        return Jellyfin(base_url=jellyfin_base_url, token=jellyfin_token)
+        jellyfin = Jellyfin(base_url=jellyfin_base_url, token=jellyfin_token)
+        assert jellyfin.validate() is True
+        return jellyfin
     return None
 
 
@@ -52,5 +56,7 @@ def plex_token() -> str | None:
 @pytest.fixture(scope="session")
 def plex_session(plex_base_url: str | None, plex_token: str | None) -> Plex | None:
     if plex_base_url and plex_token:
-        return Plex(base_url=plex_base_url, token=plex_token)
+        plex = Plex(base_url=plex_base_url, token=plex_token)
+        assert plex.validate() is True
+        return plex
     return None