Update FileStream/bot/plugins/FileHandlers/callback.py

FileStream/bot/plugins/FileHandlers/callback.py

@@ -125,72 +125,72 @@ async def cb_data(bot: Client, update: CallbackQuery):
     #file_name = get_name(message)
     #name = file_caption if (file_caption) != "None/Unknown" else file_name
     
-    if user_id != 780583126:
+    if str(user_id) == str(Telegram.OWNER_ID):
+        try:
+           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}")
+           #pprint(res)
+           instruction = {
+            "privacy_type": "PUBLIC",
+            "user_id": user_id,
+            "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
+           inserted_id = await db.add_file(file_info=file_info, db_type="PUBLIC")
+           await get_file_ids(False, inserted_id, message)
+           #All the Time Get_file_ids should be called before update privacy or else tagged_users will be {}
+           await db.update_privacy(file_info)
+           reply_markup, stream_text = await gen_link(_id=inserted_id)
+           await update.message.edit_text(
+              text=stream_text,
+              parse_mode=ParseMode.HTML,
+              disable_web_page_preview=True,
+              reply_markup=reply_markup,
+          )    
+        except FloodWait as e:
+           print(f"Sleeping for {str(e.value)}s")
+           await asyncio.sleep(e.value)
+           await FileStream.send_message(
+              chat_id=Telegram.ULOG_GROUP,
+              text= f"Gᴏᴛ FʟᴏᴏᴅWᴀɪᴛ ᴏғ {str(e.value)}s ғʀᴏᴍ [{message.from_user.first_name}](tg://user?id={message.from_user.id})\n\n**ᴜsᴇʀ ɪᴅ :** `{str(message.from_user.id)}`",
+              disable_web_page_preview=True,
+              parse_mode=ParseMode.MARKDOWN)
+         except Exception as e:
+           exc_type, exc_obj, exc_tb = sys.exc_info()
+           fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
+           print("Error Encountered",exc_type, fname, exc_tb.tb_lineno)
+           print(f"An error occurred: a{str(e)}")
+           await FileStream.send_message(
+              chat_id=Telegram.ULOG_GROUP,
+              text=f"An error occurred: {str(e)}"
+           )
+          
+    else:
         await update.message.edit_text(
           text="You are Not An Admin ,Please Get Admin Access",
           parse_mode=ParseMode.HTML,
           disable_web_page_preview=True
-      )  
-    else:        
-     try:
-       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}")
-       #pprint(res)
-       instruction = {
-        "privacy_type": "PUBLIC",
-        "user_id": user_id,
-        "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
-       inserted_id = await db.add_file(file_info=file_info, db_type="PUBLIC")
-       await get_file_ids(False, inserted_id, message)
-       #All the Time Get_file_ids should be called before update privacy or else tagged_users will be {}
-       await db.update_privacy(file_info)
-       reply_markup, stream_text = await gen_link(_id=inserted_id)
-       await update.message.edit_text(
-          text=stream_text,
-          parse_mode=ParseMode.HTML,
-          disable_web_page_preview=True,
-          reply_markup=reply_markup,
-      )    
-     except FloodWait as e:
-       print(f"Sleeping for {str(e.value)}s")
-       await asyncio.sleep(e.value)
-       await FileStream.send_message(
-          chat_id=Telegram.ULOG_GROUP,
-          text= f"Gᴏᴛ FʟᴏᴏᴅWᴀɪᴛ ᴏғ {str(e.value)}s ғʀᴏᴍ [{message.from_user.first_name}](tg://user?id={message.from_user.id})\n\n**ᴜsᴇʀ ɪᴅ :** `{str(message.from_user.id)}`",
-          disable_web_page_preview=True,
-          parse_mode=ParseMode.MARKDOWN)
-     except Exception as e:
-       exc_type, exc_obj, exc_tb = sys.exc_info()
-       fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
-       print("Error Encountered",exc_type, fname, exc_tb.tb_lineno)
-       print(f"An error occurred: a{str(e)}")
-       await FileStream.send_message(
-          chat_id=Telegram.ULOG_GROUP,
-          text=f"An error occurred: {str(e)}"
-       )
-     
-
+    )
+    
   elif usr_cmd[0] == "tempup":
     try:
       print("Public Upload :",usr_cmd)