diff --git a/misskaty/plugins/subscene_dl.py b/misskaty/plugins/subscene_dl.py
index 47ddd37c..cff22bd1 100644
--- a/misskaty/plugins/subscene_dl.py
+++ b/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"#Subscene Results For: {kueri}\n\n"
for c, i in enumerate(SUB_TITLE_DICT[msg.id][0][index], start=1):
subResult += f"{c}. {i['title']}\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"#Subscene Results For: {link}\n\n"
for c, i in enumerate(SUB_DL_DICT[msg.id][0][index], start=1):
subResult += f"{c}. {i['title']} [{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)