From 218037200c003334c631d797977721b1be585e31 Mon Sep 17 00:00:00 2001
From: Luigi311 <git@luigi311.com>
Date: Sat, 25 Feb 2023 18:27:01 -0700
Subject: [PATCH] Jellyfin: Fix tv show searching for watched

---
 src/jellyfin.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/jellyfin.py b/src/jellyfin.py
index c300714..4034f7f 100644
--- a/src/jellyfin.py
+++ b/src/jellyfin.py
@@ -1,4 +1,4 @@
-import asyncio, aiohttp
+import asyncio, aiohttp, traceback
 from src.functions import (
     logger,
     search_mapping,
@@ -46,9 +46,14 @@ class Jellyfin:
                 ) as response:
                     results = await response.json()
 
+            if type(results) is str:
+                logger(f"Jellyfin: Query {query_type} {query} {results}", 2)
+                raise Exception(results)
+
             # append identifiers to results
             if identifiers:
                 results["Identifiers"] = identifiers
+
             return results
 
         except Exception as e:
@@ -148,7 +153,7 @@ class Jellyfin:
                             )
 
                 # TV Shows
-                if library_type == "Series":
+                if library_type in ["Series", "Episode"]:
                     # Initialize an empty dictionary for the given user and library
                     user_watched[user_name][library_title] = {}
 
@@ -184,6 +189,7 @@ class Jellyfin:
                             "show_guids": show_guids,
                             "show_id": show["Id"],
                         }
+
                         season_task = asyncio.ensure_future(
                             self.query(
                                 f"/Shows/{show['Id']}/Seasons"
@@ -309,6 +315,8 @@ class Jellyfin:
                 f"Jellyfin: Failed to get watched for {user_name} in library {library_title}, Error: {e}",
                 2,
             )
+
+            logger(traceback.format_exc(), 2)
             return {}
 
     async def get_users_watched(
@@ -362,7 +370,7 @@ class Jellyfin:
                         [
                             x["Type"]
                             for x in watched["Items"]
-                            if x["Type"] in ["Movie", "Series"]
+                            if x["Type"] in ["Movie", "Series", "Episode"]
                         ]
                     )
 
@@ -385,8 +393,14 @@ class Jellyfin:
 
                     # If there are multiple types in library raise error
                     if types is None or len(types) < 1:
+                        all_types = set(
+                            [
+                                x["Type"]
+                                for x in watched["Items"]
+                            ]
+                        )
                         logger(
-                            f"Jellyfin: Skipping Library {library_title} not a single type: {types}",
+                            f"Jellyfin: Skipping Library {library_title} found types: {types}, all types: {all_types}",
                             1,
                         )
                         continue