This commit is contained in:
yasir 2023-01-15 19:02:46 +07:00
parent 9316505eb7
commit 25a164978b

View file

@ -176,25 +176,31 @@ async def getDataKuso(msg, kueri, CurrentPage, user):
kusodata.append({"title": title, "link": link}) kusodata.append({"title": title, "link": link})
if not kusodata: if not kusodata:
await editPesan(msg, "Sorry could not find any results!") await editPesan(msg, "Sorry could not find any results!")
return None, 0, None return None, 0, None, None
SCRAP_DICT[msg.id] = [split_arr(kusodata, 10), kueri] SCRAP_DICT[msg.id] = [split_arr(kusodata, 10), kueri]
try: try:
index = int(CurrentPage - 1) index = int(CurrentPage - 1)
PageLen = len(SCRAP_DICT[msg.id][0]) PageLen = len(SCRAP_DICT[msg.id][0])
extractbtn = [] extractbtn1 = []
extractbtn2 = []
kusoResult = f"<b>#Kusonime Latest Post\n\n" if kueri == "" else f"<b>#Kusonime Results For:</b> <code>{kueri}</code>\n\n" kusoResult = f"<b>#Kusonime Latest Post\n\n" if kueri == "" else f"<b>#Kusonime Results For:</b> <code>{kueri}</code>\n\n"
for c, i in enumerate(SCRAP_DICT[msg.id][0][index], start=1): for c, i in enumerate(SCRAP_DICT[msg.id][0][index], start=1):
kusoResult += f"<b>{c}</b>. {i['title']}\n{i['link']}\n\n" kusoResult += f"<b>{c}</b>. {i['title']}\n{i['link']}\n\n"
extractbtn.append( if c < 6:
extractbtn1.append(
InlineButton(c, f"kusoextract#{CurrentPage}#{c}#{user}#{msg.id}")
)
else:
extractbtn2.append(
InlineButton(c, f"kusoextract#{CurrentPage}#{c}#{user}#{msg.id}") InlineButton(c, f"kusoextract#{CurrentPage}#{c}#{user}#{msg.id}")
) )
IGNORE_CHAR = "[]" IGNORE_CHAR = "[]"
kusoResult = ''.join(i for i in kusoResult if not i in IGNORE_CHAR) kusoResult = ''.join(i for i in kusoResult if not i in IGNORE_CHAR)
return kusoResult, PageLen, extractbtn return kusoResult, PageLen, extractbtn1, extractbtn2
except (IndexError, KeyError): except (IndexError, KeyError):
await editPesan(msg, "Sorry could not find any matching results!") await editPesan(msg, "Sorry could not find any matching results!")
return None, 0, None return None, 0, None, None
# Movieku GetData # Movieku GetData
async def getDataMovieku(msg, kueri, CurrentPage): async def getDataMovieku(msg, kueri, CurrentPage):
@ -548,12 +554,14 @@ async def kusonime_s(client, message):
kueri = "" kueri = ""
pesan = await kirimPesan(message, "⏳ Please wait, scraping data from Kusonime..", quote=True) pesan = await kirimPesan(message, "⏳ Please wait, scraping data from Kusonime..", quote=True)
CurrentPage = 1 CurrentPage = 1
kusores, PageLen, btn = await getDataKuso(pesan, kueri, CurrentPage, message.from_user.id) kusores, PageLen, btn1, btn2 = await getDataKuso(pesan, kueri, CurrentPage, message.from_user.id)
if not kusores: return if not kusores: return
keyboard = InlineKeyboard() keyboard = InlineKeyboard()
keyboard.paginate(PageLen, CurrentPage, 'page_kuso#{number}' + f'#{pesan.id}#{message.from_user.id}') keyboard.paginate(PageLen, CurrentPage, 'page_kuso#{number}' + f'#{pesan.id}#{message.from_user.id}')
keyboard.row(InlineButton("👇 Extract Data ", "Hmmm")) keyboard.row(InlineButton("👇 Extract Data ", "Hmmm"))
keyboard.row(*btn) keyboard.row(*btn1)
if btn2:
keyboard.row(*btn2)
keyboard.row( keyboard.row(
InlineButton("❌ Close", f"close#{message.from_user.id}") InlineButton("❌ Close", f"close#{message.from_user.id}")
) )
@ -632,14 +640,16 @@ async def kusopage_callback(client, callback_query):
return await callback_query.answer("Invalid callback data, please send CMD again..") return await callback_query.answer("Invalid callback data, please send CMD again..")
try: try:
kusores, PageLen, btn = await getDataKuso(callback_query.message, kueri, CurrentPage, callback_query.from_user.id) kusores, PageLen, btn1, btn2 = await getDataKuso(callback_query.message, kueri, CurrentPage, callback_query.from_user.id)
except TypeError: except TypeError:
return return
keyboard = InlineKeyboard() keyboard = InlineKeyboard()
keyboard.paginate(PageLen, CurrentPage, 'page_kuso#{number}' + f'#{message_id}#{callback_query.from_user.id}') keyboard.paginate(PageLen, CurrentPage, 'page_kuso#{number}' + f'#{message_id}#{callback_query.from_user.id}')
keyboard.row(InlineButton("👇 Extract Data ", "Hmmm")) keyboard.row(InlineButton("👇 Extract Data ", "Hmmm"))
keyboard.row(*btn) keyboard.row(*btn1)
if btn2:
keyboard.row(*btn2)
keyboard.row( keyboard.row(
InlineButton("❌ Close", f"close#{callback_query.from_user.id}") InlineButton("❌ Close", f"close#{callback_query.from_user.id}")
) )