Callback Update

FileStream/APIs/TMDB/Endpoint.py

@@ -1,4 +1,4 @@
-
+import logging
 import urllib3
 from .__init__ import tmdb,search,movie,tv
 from FileStream.Tools.cleanup import Get_Title_Year
@@ -50,8 +50,62 @@ def search_tmdb_tv(name):
         print("* Error at Endpoint", e, result)
         return None
     return None    
-    
-def search_tmdb_any(name):
+
+import logging
+from tmdbv3api import Search  # Assuming you're using tmdbv3api
+from FileStream.TMDB.Endpoint import get_tvshows  # Assuming this is your function
+
+# Create an instance of the Search class
+search = Search()
+
+def search_tmdb_any(title: str, year: int):
+    """
+    Search for a movie or TV show on TMDb based on title and release year.
+    :param title: The name of the movie or TV show.
+    :param year: The release year.
+    :return: Matched TMDb result or None if no match is found.
+    """
+    try:
+        # Validate inputs
+        if not isinstance(title, str) or not isinstance(year, int):
+            logging.warning("Invalid input: Title must be a string, and Year must be an integer.")
+            return None
+
+        # Perform a search query
+        search_results = search.multi(title)
+        if not search_results:
+            logging.info(f"No results found for '{title}' ({year})")
+            return None
+
+        # Iterate through the results and find a match by media type and year
+        for result in search_results:
+            try:
+                media_type = getattr(result, "media_type", None)
+                release_date = (
+                    getattr(result, "release_date", None) or 
+                    getattr(result, "first_air_date", None)
+                )
+
+                if release_date and release_date.split('-')[0].isdigit():
+                    result_year = int(release_date.split('-')[0])
+
+                    if result_year == year:
+                        if media_type == "tv":
+                            return get_tvshows(getattr(result, "id", None))
+                        elif media_type == "movie":
+                            return result
+
+            except Exception as inner_error:
+                logging.error(f"Error processing result {result}: {inner_error}")
+
+    except Exception as e:
+        logging.error(f"Error in search_tmdb_any: {e}")
+
+    return None
+
+
+
+def search_tmdb_any_old(name):
     # Search for the title in TMDb
     title, year = Get_Title_Year(name)
     #print("*", title, year,"\n Name :", name)