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

FileStream/bot/plugins/FileHandlers/callback.py

@@ -113,6 +113,29 @@ async def cb_data(bot: Client, update: CallbackQuery):
         disable_web_page_preview=True,
         reply_markup=reply_markup,
     )
+  elif usr_cmd[0] == "usercheck":
+    print("Checking User :", usr_cmd)
+    user_id = str(usr_cmd[1])
+    message_id = int(usr_cmd[2])
+    if str(user_id)== str(Telegram.OWNER_ID):
+        await update.message.edit_text(
+          text="You are OWNER , Please Upload The File to controller",
+          parse_mode=ParseMode.HTML,
+          disable_web_page_preview=True,
+          reply_markup= InlineKeyboardMarkup([
+              [InlineKeyboardButton("PUBLIC Upload",callback_data=f"pubup_{file_info['user_id']}_{file_info['message_id']}")],
+              [InlineKeyboardButton("ᴄʟᴏsᴇ", callback_data="close")]
+            ]),
+      )
+    else:
+        await update.message.edit_text(
+          text="You are Not OWNER , Permission Denied",
+          parse_mode=ParseMode.HTML,
+          disable_web_page_preview=True,
+          reply_markup= InlineKeyboardMarkup([
+              [InlineKeyboardButton("ᴄʟᴏsᴇ", callback_data="close")]
+            ]),
+      )
 
   elif usr_cmd[0] == "pubup":
     print("Public Upload :", usr_cmd)
@@ -120,76 +143,69 @@ async def cb_data(bot: Client, update: CallbackQuery):
     message_id = int(usr_cmd[2])
     message = await FileStream.get_messages(user_id, message_id)
     #response = await FileStream.get_messages(user_id, response_id)     
-    name = getattr(message, "caption",  f"{get_name(message)}" ) or "None/Unknown"
-    print(f"User:{user_id}","FileName :",name) 
+    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
-    
-    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:
+    try:
+        print(f"User:{user_id}","FileName :", name)
+        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="You are Not An Admin ,Please Get Admin Access",
+          text=stream_text,
           parse_mode=ParseMode.HTML,
-          disable_web_page_preview=True
-    )
+          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: