From c2fe066efde1981901631c5cb68e1a78c2015c95 Mon Sep 17 00:00:00 2001 From: yasir Date: Sun, 15 Jan 2023 19:52:52 +0700 Subject: [PATCH] Tes --- misskaty/plugins/web_scraper.py | 54 ++++++++++++++++++++++++++++----- 1 file changed, 47 insertions(+), 7 deletions(-) diff --git a/misskaty/plugins/web_scraper.py b/misskaty/plugins/web_scraper.py index 619059b4..c1e668cd 100644 --- a/misskaty/plugins/web_scraper.py +++ b/misskaty/plugins/web_scraper.py @@ -272,7 +272,7 @@ async def getDataSavefilm21(msg, kueri, CurrentPage, user): return None, 0, None # Lendrive GetData -async def getDataLendrive(msg, kueri, CurrentPage): +async def getDataLendrive(msg, kueri, CurrentPage, user): if not SCRAP_DICT.get(msg.id): data = await http.get(f'https://lendrive.web.id/?s={kueri}', headers=headers) soup = BeautifulSoup(data.text, "lxml") @@ -285,21 +285,25 @@ async def getDataLendrive(msg, kueri, CurrentPage): lenddata.append({"judul": title, "link": link, "quality": kualitas, "status": status}) if not lenddata: await editPesan(msg, "Sorry could not find any results!") - return None, None + return None, 0, None SCRAP_DICT[msg.id] = [split_arr(lenddata, 6), kueri] try: index = int(CurrentPage - 1) PageLen = len(SCRAP_DICT[msg.id][0]) + extractbtn = [] lenddataResult = f"#LenDrive Latest:\nšŸŒ€ Use /lendrive [title] to start search with title.\n\n" if kueri == "" else f"#LenDrive Results For: {kueri}\n\n" for c, i in enumerate(SCRAP_DICT[msg.id][0][index], start=1): lenddataResult += f"{c}. {i['judul']}\nQuality: {i['quality']}\nStatus: {i['status']}\n\n" + extractbtn.append( + InlineButton(c, f"lendriveextract#{CurrentPage}#{c}#{user}#{msg.id}") + ) IGNORE_CHAR = "[]" lenddataResult = ''.join(i for i in lenddataResult if not i in IGNORE_CHAR) - return lenddataResult, PageLen + return lenddataResult, PageLen, extractbtn except (IndexError, KeyError): await editPesan(msg, "Sorry could not find any matching results!") - return None, None + return None, 0, None # MelongMovie GetData async def getDataMelong(msg, kueri, CurrentPage, user): @@ -413,7 +417,7 @@ async def getDataGomov(msg, kueri, CurrentPage, user): ) IGNORE_CHAR = "[]" gomovResult = ''.join(i for i in gomovResult if not i in IGNORE_CHAR) - return gomovResult, PageLen + return gomovResult, PageLen, extractbtn except (IndexError, KeyError): await editPesan(msg, "Sorry could not find any matching results!") return None, None @@ -593,10 +597,12 @@ async def lendrive_s(client, message): kueri = "" pesan = await kirimPesan(message, "ā³ Please wait, scraping data from Lendrive..", quote=True) CurrentPage = 1 - lendres, PageLen = await getDataLendrive(pesan, kueri, CurrentPage) + lendres, PageLen, btn = await getDataLendrive(pesan, kueri, CurrentPage) if not lendres: return keyboard = InlineKeyboard() keyboard.paginate(PageLen, CurrentPage, 'page_lendrive#{number}' + f'#{pesan.id}#{message.from_user.id}') + keyboard.row(InlineButton("šŸ‘‡ Extract Data ", "Hmmm")) + keyboard.row(*btn) keyboard.row( InlineButton("āŒ Close", f"close#{message.from_user.id}") ) @@ -710,12 +716,14 @@ async def moviekupage_callback(client, callback_query): return await callback_query.answer("Invalid callback data, please send CMD again..") try: - lendres, PageLen = await getDataLendrive(callback_query.message, kueri, CurrentPage) + lendres, PageLen, btn = await getDataLendrive(callback_query.message, kueri, CurrentPage, callback_query.from_user.id) except TypeError: return keyboard = InlineKeyboard() keyboard.paginate(PageLen, CurrentPage, 'page_lendrive#{number}' + f'#{message_id}#{callback_query.from_user.id}') + keyboard.row(InlineButton("šŸ‘‡ Extract Data ", "Hmmm")) + keyboard.row(*btn) keyboard.row( InlineButton("āŒ Close", f"close#{callback_query.from_user.id}") ) @@ -1091,3 +1099,35 @@ async def zonafilm_dl(_, callback_query): await editPesan(callback_query.message, f"ERROR: {err}", reply_markup=keyboard) return await editPesan(callback_query.message, f"Scrape result from {link}:\n\n{hasil}", reply_markup=keyboard) + +@app.on_callback_query(filters.create(lambda _, __, query: 'lendriveextract#' in query.data)) +async def zonafilm_dl(_, callback_query): + if callback_query.from_user.id != int(callback_query.data.split('#')[3]): + return await callback_query.answer("Not yours..", True) + idlink = int(callback_query.data.split("#")[2]) + message_id = int(callback_query.data.split('#')[4]) + CurrentPage = int(callback_query.data.split('#')[1]) + try: + link = SCRAP_DICT[message_id][0][CurrentPage-1][idlink-1].get("link") + except KeyError: + return await callback_query.answer("Invalid callback data, please send CMD again..") + + keyboard = InlineKeyboard() + keyboard.row( + InlineButton("ā†©ļø Back", f"page_lendrive#{CurrentPage}#{message_id}#{callback_query.from_user.id}"), + InlineButton("āŒ Close", f"close#{callback_query.from_user.id}") + ) + try: + html = await http.get(link, headers=headers) + soup = BeautifulSoup(html.text, "lxml") + j = soup.findAll("div", class_="soraurlx") + kl = "#Lendrive Results Download URL:\n\n" + for i in j: + if not i.find("a"): + continue + kl += f"{i.find('strong')}:\n" + kl += "".join(f"[ {a.text} ]\n" for a in i.findAll("a")) + except Exception as err: + await editPesan(callback_query.message, f"ERROR: {err}", reply_markup=keyboard) + return + await editPesan(callback_query.message, f"Scrape result from {link}:\n\n{kl}", reply_markup=keyboard)