Changes in Database

FileStream/bot/plugins/FileHandlers/callback.py

@@ -22,6 +22,7 @@ from FileStream.Database import Database
 from FileStream.config import Telegram, Server
 from FileStream.bot import FileStream
 from FileStream.bot import MULTI_CLIENTS
+from FileStream.Tools.cleanup import Get_Title_Year
 from FileStream.Exceptions import FileNotFound  # Corrected the import statement
 from FileStream.utils.FileProcessors.translation import LANG, BUTTON
 from FileStream.utils.FileProcessors.human_readable import humanbytes
@@ -196,7 +197,6 @@ async def cb_data(bot: Client, update: CallbackQuery):
       name = getattr(message, "caption",  f"{get_name(message)}" ) or "None/Unknown"
       #file_name = get_name(message)
       #name = file_caption if (file_caption) != "None/Unknown" else file_name
-      from FileStream.Tools.cleanup import Get_Title_Year
       title, year = Get_Title_Year(name)
       print(f"** ANY SEARCH :{name} \nFinal :{title} YEAR :{year}")
       get_imdb_id = json.loads(imdb.search(title, year=int(year)))
@@ -259,10 +259,33 @@ async def cb_data(bot: Client, update: CallbackQuery):
       user_id = str(usr_cmd[1])
       message_id = int(usr_cmd[2])
       message = await FileStream.get_messages(user_id, message_id)
+      name = getattr(message, "caption",  f"{get_name(message)}" ) or "None/Unknown"
+      #file_name = get_name(message)
+      #name = file_caption if (file_caption) != "None/Unknown" else file_name
+      from FileStream.Tools.cleanup import Get_Title_Year
+      title, year = Get_Title_Year(name)
+      print(f"** ANY SEARCH :{name} \nFinal :{title} YEAR :{year}")
+      get_imdb_id = json.loads(imdb.search(title, year=int(year)))
+      #print(f"ANY SEARCH :{title} YEAR :{year} IMDB Response :{res}")
+      #print("TMDB",search_tmdb_any(title, year))
+      #print(f"IMDB ID :{get_imdb_id['results'][0]['id']}")
+      print(f"\n** GET_ID {get_imdb_id}")
+      #pprint(get_imdb_id)
+      #res=json.loads(imdb.get_by_id(get_imdb_id['results'][0]['id']))
+      res=json.loads(imdb.get_by_id(get_imdb_id['results'][0]['id']))
+      print(f"\n** IMDB Response :{res}")
       instruction = {
         "privacy_type": "PRIVATE",
         "user_id": user_id,
-        "user_type": "TELEGRAM"
+        "user_type": "TELEGRAM",
+        "IMDB_id": get_imdb_id['results'][0]['id'],
+        "poster": res.get("poster", "None"),   
+        "title": res['name'],
+        "release_date": res["datePublished"],
+        "type": res['type'],
+        "description":res.get("description", "None"),
+        "keywords":res["keywords"].split(",") if res.get("keywords") else [],
+        "genre": res.get("genre", []) ,
       }
       file_info = get_file_info(message, instruction)
       # Here we are Adding the File Into the database First