Callback Update

FileStream/bot/plugins/FileHandlers/callback.py

@@ -159,10 +159,11 @@ async def cb_data(bot: Client, update: CallbackQuery):
       #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 \n {title}\n {year} Resp :",search_tmdb_any(title, year))
+      print(f"ANY SEARCH :{title} YEAR :{year} Resp : {search_tmdb_any(title, year)}")
+      res = imdb.search(title, year=year)[0]
+      print(f"IMDB Response :{imdb.get_by_id(res['id'])}")
       if msg_type == 'MOVIE':
         resp=search_tmdb_movies(name)
-        res = imdb.search(title, year=year)
         print("TMDB Movie Response:",resp,"\n IMDB",res)
         result={
           'title': resp['title'],
@@ -185,13 +186,13 @@ async def cb_data(bot: Client, update: CallbackQuery):
           "privacy_type": "PUBLIC",
           "user_id": user_id,
           "user_type": "TELEGRAM",
+          "IMDB_id": res['id'],
           "TMDB_id": result['id'],    
           "title": result['title'],
           "type": result['media_type'],
           "subtype":"Unknown",
           "quality":"Unknown",
       }
-      
       file_info = get_file_info(message, instruction)
       # Here we are Adding the File Into the database First
       inserted_id = await db.add_file(file_info)
@@ -227,20 +228,17 @@ async def cb_data(bot: Client, update: CallbackQuery):
       message_id = int(usr_cmd[2])
       message = await FileStream.get_messages(user_id, message_id)
       instruction = {
-
-          "privacy_type": "PRIVATE",
-          "user_id": user_id,
-          "user_type": "TELEGRAM"
+        "privacy_type": "PRIVATE",
+        "user_id": user_id,
+        "user_type": "TELEGRAM"
       }
       file_info = get_file_info(message, instruction)
       # Here we are Adding the File Into the database First
       db_id = await db.add_file(file_info)
       await get_file_ids(False, db_id, message)
-
       if True:
         file_info = await db.get_file(db_id)
         reply_markup, stream_text = await priv_func(file_info['file']['file_name'], file_info['file']['file_size'])
-
         await update.message.edit_text(
             text=stream_text,
             parse_mode=ParseMode.HTML,

FileStream/bot/plugins/FileHandlers/stream.py

@@ -61,6 +61,7 @@ async def private_receive_handler(bot: Client, message: Message):
             "privacy_type":"PUBLIC",
             "user_id":message.from_user.id if (message.chat.type == ChatType.PRIVATE) else message.chat.id,
             "user_type": "TELEGRAM",
+            "IMDB_id": "Unknown",
             "TMDB_id": "Unknown",    
             "title":"Unknown",
             "type": "Unknown",

FileStream/utils/FileProcessors/file_properties.py

@@ -191,6 +191,7 @@ def get_file_info(message, instruction):
       "user_id": instruction['user_id'],
       "user_type": instruction['user_type'],
       "message_id": message.id,
+      "IMDB_id": instruction['IMDB_id'],
       "TMDB_id": instruction['TMDB_id'],    
       "title": instruction['title'],
       "type": instruction['type'],