Add Jellyseerr Support #58

Author: ghomasHudson

main.py

@@ -1,5 +1,6 @@
 from typing import cast
 from utils.jellyfin import JellyfinClient
+from utils.jellyseerr import JellyseerrClient
 import pluginlib
 from loguru import logger
 from pyaml_env import parse_config
@@ -23,6 +24,14 @@ def main(config):
         user_id=config['jellyfin']['user_id']
     )
 
+    if "jellyseerr" in config:
+        js_client = JellyseerrClient(
+            server_url=config['jellyseerr']['server_url'],
+            api_key=config['jellyseerr']['api_key']
+        )
+    else:
+        js_client = None
+
     # Load plugins
     loader = pluginlib.PluginLoader(modules=['plugins'])
     plugins = loader.plugins['list_scraper']
@@ -45,15 +54,27 @@ def main(config):
                 list_info = plugins[plugin_name].get_list(list_id, config['plugins'][plugin_name])
 
                 # Find jellyfin collection or create it
-                collection_id = jf_client.find_collection_with_name_or_create(list_info['name'], list_id, list_info.get("description", None), plugin_name)
+                collection_id = jf_client.find_collection_with_name_or_create(
+                    list_info['name'], 
+                    list_id,
+                    list_info.get("description", None),
+                    plugin_name
+                )
 
                 if config["plugins"][plugin_name].get("clear_collection", False):
                     # Optionally clear everything from the collection first
                     jf_client.clear_collection(collection_id)
 
                 # Add items to the collection
                 for item in list_info['items']:
-                    jf_client.add_item_to_collection(collection_id, item, year_filter=config["plugins"][plugin_name].get("year_filter", True))
+                    matched = jf_client.add_item_to_collection(
+                        collection_id,
+                        item,
+                        year_filter=config["plugins"][plugin_name].get("year_filter", True)
+                    )
+                    if not matched and js_client is not None:
+                        js_client.make_request(item)
+
 
 
 if __name__ == "__main__":

utils/jellyseerr.py

@@ -0,0 +1,82 @@
+import requests
+import urllib.parse
+from loguru import logger
+
+class JellyseerrClient:
+    def __init__(self, server_url: str, api_key: str):
+        # Fix common url issues
+        if server_url.endswith("/"):
+            server_url = server_url[:-1]  # Remove trailing slash 
+        if not server_url.endswith("/api/v1"):
+            server_url += "/api/v1"
+        self.server_url = server_url
+        self.api_key = api_key
+
+        # Check if server is reachable
+        try:
+            r = requests.get(self.server_url + "/status")
+            if r.status_code != 200:
+                raise Exception("Jellyseerr Server is not reachable")
+        except requests.exceptions.ConnectionError:
+            raise Exception("Jellyseerr Server is not reachable")
+
+        # Check if api key is valid
+        r = requests.get(f"{self.server_url}/auth/me")
+
+
+    def make_request(self, item):
+        '''Request item from jellyseerr'''
+
+        # Search for item
+        r = requests.get(f"{self.server_url}/search", params={
+            "query": urllib.parse.quote_plus(item["title"])
+        }, headers={
+            "X-Api-Key": self.api_key
+        })
+        
+        # Find matching item
+        mediaId = None
+        for result in r.json()["results"]:
+            # Try IMDB match first
+            if "mediaInfo" in result and "ImdbId" in result["mediaInfo"]:
+                imdb_id = result["mediaInfo"]["ImdbId"]
+                if imdb_id == item["imdb_id"]:
+                    mediaId = result["id"]
+                    logger.debug(f"Found exact IMDB match for {item['title']}")
+                    break
+            elif "releaseDate" in result:
+                # Try year match
+                release_year = result["releaseDate"].split("-")[0]
+                if release_year == str(item["release_year"]):
+                    mediaId = result["id"]
+                    logger.debug(f"Found year match for {item['title']}")
+                    break
+
+        # Request item if not found
+        if mediaId is not None:
+            if "mediaInfo" not in result or result["mediaInfo"]["jellyfinMediaId"] is None:
+                # If it's not already in Jellyfin
+                # Request item
+                r = requests.post(f"{self.server_url}/request", json={
+                    "mediaType": result["mediaType"],
+                    "mediaId": mediaId,
+                }, headers={
+                    "X-Api-Key": self.api_key
+                })
+                logger.info(f"Requested {item['title']} from Jellyseerr")
+
+
+
+if __name__ == "__main__":
+    from pyaml_env import parse_config
+    config = parse_config("/home/thomas/Documents/Jellyfin-Auto-Collections/config.yaml", default_value=None)
+
+    client = JellyseerrClient(
+        server_url=config["jellyseerr"]["server_url"],
+        api_key=config["jellyseerr"]["api_key"]
+    )
+    client.make_request({
+        "title": "The Matrix",
+        "imdb_id": "tt0133093",
+        "release_year": 1999
+    })