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

FileStream/bot/plugins/FileHandlers/callback.py

@@ -114,31 +114,31 @@ async def cb_data(bot: Client, update: CallbackQuery):
     )
 
   elif usr_cmd[0] == "pubup":
-
-    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}")
-      #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 = {
+    if int(usr_cmd[1]) in Telegram.OWNER_ID :
+     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}")
+       #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",
@@ -150,39 +150,44 @@ async def cb_data(bot: Client, update: CallbackQuery):
         "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(
+       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(
+     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)}`",
+          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(exc_type, fname, exc_tb.tb_lineno)
-      print(f"An error occurred: a{str(e)}")
-      await FileStream.send_message(
+     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
+      )    
 
   elif usr_cmd[0] == "tempup":
     try: