Add subdl beta

misskaty/plugins/subscene_dl.py

@@ -38,13 +38,17 @@ async def getTitleSub(msg, kueri, CurrentPage, user):
     try:
         index = int(CurrentPage - 1)
         PageLen = len(SUB_TITLE_DICT[msg.id][0])
-        extractbtn = []
+        extractbtn1 = []
+        extractbtn2 = []
         subResult = f"<b>#Subscene Results For:</b> <code>{kueri}</code>\n\n"
         for c, i in enumerate(SUB_TITLE_DICT[msg.id][0][index], start=1):
             subResult += f"<b>{c}. <a href='{i['link']}'>{i['title']}</a></b>\n"
-            extractbtn.append(InlineButton(c, f"sublist#{CurrentPage}#{c}#{msg.id}#{user}"))
+            if c < 6:
+                extractbtn1.append(InlineButton(c, f"sublist#{CurrentPage}#{c}#{msg.id}#{user}"))
+            else:
+                extractbtn2.append(InlineButton(c, f"sublist#{CurrentPage}#{c}#{msg.id}#{user}"))
         subResult = "".join(i for i in subResult if i not in "[]")
-        return subResult, PageLen, extractbtn
+        return subResult, PageLen, extractbtn1, extractbtn2
     except (IndexError, KeyError):
         await editPesan(msg, "Sorry could not find any matching results!")
         return None, 0, None
@@ -71,13 +75,17 @@ async def getListSub(msg, link, CurrentPage, user):
     try:
         index = int(CurrentPage - 1)
         PageLen = len(SUB_DL_DICT[msg.id][0])
-        extractbtn = []
+        extractbtn1 = []
+        extractbtn2 = []
         subResult = f"<b>#Subscene Results For:</b> <code>{link}</code>\n\n"
         for c, i in enumerate(SUB_DL_DICT[msg.id][0][index], start=1):
             subResult += f"<b>{c}. {i['title']}</b> [{i['rate']}]\n{i['lang']}\n"
-            extractbtn.append(InlineButton(c, f"extractsubs#{CurrentPage}#{c}#{msg.id}#{user}"))
+            if c < 6:
+                extractbtn1.append(InlineButton(c, f"extractsubs#{CurrentPage}#{c}#{msg.id}#{user}"))
+            else:
+                extractbtn2.append(InlineButton(c, f"extractsubs#{CurrentPage}#{c}#{msg.id}#{user}"))
         subResult = "".join(i for i in subResult if i not in "[]")
-        return subResult, PageLen, extractbtn
+        return subResult, PageLen, extractbtn1, extractbtn2
     except (IndexError, KeyError):
         await editPesan(msg, "Sorry could not find any matching results!")
         return None, 0, None
@@ -117,14 +125,16 @@ async def subpage_callback(client, callback_query):
         return await callback_query.message.delete()
 
     try:
-        subres, PageLen, btn = await getTitleSub(callback_query.message, kueri, CurrentPage, callback_query.from_user.id)
+        subres, PageLen, btn1, btn2 = await getTitleSub(callback_query.message, kueri, CurrentPage, callback_query.from_user.id)
     except TypeError:
         return
 
     keyboard = InlineKeyboard()
     keyboard.paginate(PageLen, CurrentPage, "subscenepage#{number}" + f"#{message_id}#{callback_query.from_user.id}")
     keyboard.row(InlineButton("👇 Get Subtitle List", "Hmmm"))
-    keyboard.row(*btn)
+    keyboard.row(*btn1)
+    if btn2:
+        keyboard.row(*btn2)
     keyboard.row(InlineButton("❌ Close", f"close#{callback_query.from_user.id}"))
     await editPesan(callback_query.message, subres, disable_web_page_preview=True, reply_markup=keyboard)
 
@@ -145,14 +155,16 @@ async def subdlpage_callback(client, callback_query):
         return await callback_query.message.delete()
 
     try:
-        subres, PageLen, btn = await getListSub(callback_query.message, link, CurrentPage, callback_query.from_user.id)
+        subres, PageLen, btn1, btn2 = await getListSub(callback_query.message, link, CurrentPage, callback_query.from_user.id)
     except TypeError:
         return
 
     keyboard = InlineKeyboard()
     keyboard.paginate(PageLen, CurrentPage, "sublist#{number}" + f"#{idlink}#{message_id}#{callback_query.from_user.id}")
     keyboard.row(InlineButton("👇 Download Subtitle", "Hmmm"))
-    keyboard.row(*btn)
+    keyboard.row(*btn1)
+    if btn2:
+        keyboard.row(*btn2)
     keyboard.row(InlineButton("❌ Close", f"close#{callback_query.from_user.id}"))
     await editPesan(callback_query.message, subres, disable_web_page_preview=True, reply_markup=keyboard)