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

FileStream/bot/plugins/FileHandlers/callback.py

@@ -33,6 +33,7 @@ from FileStream.utils.FileProcessors.bot_utils import gen_link, priv_func, gen_p
 from FileStream.APIs.TMDB.Endpoint import search_tmdb_any, search_tmdb_tv, search_tmdb_movies
 #from FileStream.APIs.IMDB import imdb
 from FileStream.APIs.IMDB_CORE import imdb
+from FileStream.Tools.cleanup import Get_Title_Year
 #-----------------Starting Point --------------------------#
 
 db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
@@ -114,19 +115,25 @@ async def cb_data(bot: Client, update: CallbackQuery):
     )
 
   elif usr_cmd[0] == "pubup":
-    if int(usr_cmd[1]) in [ Telegram.OWNER_ID ]:
+    print("Public Upload :",usr_cmd)
+    user_id = str(usr_cmd[1])
+    message_id = int(usr_cmd[2])
+    message = await FileStream.get_messages(user_id, message_id)
+    #response = await FileStream.get_messages(user_id, response_id)
+    #print(f"Message:{message}")      
+    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
+    title, year = Get_Title_Year(name)
+             
+    if user_id not in [ Telegram.OWNER_ID ]:
+        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:
-       print("Public Upload :",usr_cmd)
-       user_id = str(usr_cmd[1])
-       message_id = int(usr_cmd[2])
-       message = await FileStream.get_messages(user_id, message_id)
-       #response = await FileStream.get_messages(user_id, response_id)
-       #print(f"Message:{message}")      
-       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}")
@@ -182,12 +189,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
           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
-      )    
+     
 
   elif usr_cmd[0] == "tempup":
     try: