Callback Update

FileStream/bot/plugins/FileHandlers/callback.py

@@ -39,18 +39,15 @@ db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
 async def cb_data(bot: Client, update: CallbackQuery):
   usr_cmd = update.data.split("_")
   if usr_cmd[0] == "home":
-    await update.message.edit_text(text=LANG.START_TEXT.format(
-        update.from_user.mention, FileStream.username),
+    await update.message.edit_text(text=LANG.START_TEXT.format(update.from_user.mention, FileStream.username),
                                    disable_web_page_preview=True,
                                    reply_markup=BUTTON.START_BUTTONS)
   elif usr_cmd[0] == "help":
-    await update.message.edit_text(text=LANG.HELP_TEXT.format(
-        Telegram.OWNER_ID),
+    await update.message.edit_text(text=LANG.HELP_TEXT.format(Telegram.OWNER_ID),
                                    disable_web_page_preview=True,
                                    reply_markup=BUTTON.HELP_BUTTONS)
   elif usr_cmd[0] == "about":
-    await update.message.edit_text(text=LANG.ABOUT_TEXT.format(
-        FileStream.fname, __version__),
+    await update.message.edit_text(text=LANG.ABOUT_TEXT.format(FileStream.fname, __version__),
                                    disable_web_page_preview=True,
                                    reply_markup=BUTTON.ABOUT_BUTTONS)
 
@@ -133,16 +130,16 @@ async def cb_data(bot: Client, update: CallbackQuery):
       res=json.loads(imdb.get_by_id(get_imdb_id['results'][0]['id']))
       print(f"\n** IMDB Response :{res}")
       instruction = {
-          "privacy_type": "PUBLIC",
-          "user_id": user_id,
-          "user_type": "TELEGRAM",
-          "IMDB_id": get_imdb_id['results'][0]['id'],
-          "poster": res['poster'],   
-          "title": res['name'],
-          "release_date":res["datePublished"],
-          "type": res['type'],
-          "description":res["description"],
-          "genre":res["genre"]+[res["keywords"]],
+        "privacy_type": "PUBLIC",
+        "user_id": user_id,
+        "user_type": "TELEGRAM",
+        "IMDB_id": get_imdb_id['results'][0]['id'],
+        "poster": res['poster'],   
+        "title": res['name'],
+        "release_date": res["datePublished"],
+        "type": res['type'],
+        "description":res["description"],
+        "genre": res.get("genre", []) + (res["keywords"].split(",") if res.get("keywords") else []),
       }
       
       file_info = get_file_info(message, instruction)
@@ -320,10 +317,7 @@ async def gen_file_menu(_id, file_list_no, update: CallbackQuery):
                 "ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",
                 callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
         ],
-        [
-            InlineKeyboardButton(
-                "ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
-        ]
+        [InlineKeyboardButton("ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))]
     ])
   else:
     MYFILES_BUTTONS = InlineKeyboardMarkup([
@@ -335,12 +329,8 @@ async def gen_file_menu(_id, file_list_no, update: CallbackQuery):
                 "ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",
                 callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
         ],
-        [
-            InlineKeyboardButton(
-                "ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
-        ]
+        [InlineKeyboardButton("ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))]
     ])
-
   TiMe = myfile_info['time']
   if type(TiMe) == float:
     date = datetime.datetime.fromtimestamp(TiMe)
@@ -420,30 +410,22 @@ async def gen_privfile_menu(_id, file_list_no, update: CallbackQuery):
             InlineKeyboardButton("ᴅᴏᴡɴʟᴏᴀᴅ", url=stream_link)
         ],
         [
-            InlineKeyboardButton(
-                "ɢᴇᴛ ғɪʟᴇ", callback_data=f"sendfile_{myfile_info['_id']}"),
-            InlineKeyboardButton(
-                "ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",
-                callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
+          InlineKeyboardButton("ɢᴇᴛ ғɪʟᴇ", callback_data=f"sendfile_{myfile_info['_id']}"),
+          InlineKeyboardButton("ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
         ],
         [
-            InlineKeyboardButton(
-                "ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
+          InlineKeyboardButton("ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
         ]
     ])
   else:
     MYFILES_BUTTONS = InlineKeyboardMarkup([
         [InlineKeyboardButton("ᴅᴏᴡɴʟᴏᴀᴅ", url=stream_link)],
         [
-            InlineKeyboardButton(
-                "ɢᴇᴛ ғɪʟᴇ", callback_data=f"sendfile_{myfile_info['_id']}"),
-            InlineKeyboardButton(
-                "ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",
-                callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
+          InlineKeyboardButton("ɢᴇᴛ ғɪʟᴇ", callback_data=f"sendfile_{myfile_info['_id']}"),
+          InlineKeyboardButton("ʀᴇᴠᴏᴋᴇ ғɪʟᴇ",callback_data=f"msgdelete_{myfile_info['_id']}_{file_list_no}")
         ],
         [
-            InlineKeyboardButton(
-                "ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
+          InlineKeyboardButton("ʙᴀᴄᴋ", callback_data="userfiles_{}".format(file_list_no))
         ]
     ])