diff --git a/misskaty/plugins/web_scraper.py b/misskaty/plugins/web_scraper.py
index ce2f5436..82a0fa3a 100644
--- a/misskaty/plugins/web_scraper.py
+++ b/misskaty/plugins/web_scraper.py
@@ -854,32 +854,32 @@ async def zonafilmpage_callback(client, callback_query):
### Scrape DDL Link From Web ###
# Savefilm21 DDL
-@app.on_message(filters.command(["savefilm21_scrap"], COMMAND_HANDLER))
-async def savefilm21_scrap(_, message):
+@app.on_callback_query(filters.create(lambda _, __, query: 'sf21extract#' in query.data))
+async def savefilm21_scrap(_, 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-2].get("link")
+ except KeyError:
+ return await callback_query.answer("Invalid callback data, please send CMD again..")
+
+ keyboard = InlineKeyboard()
+ keyboard.row(
+ InlineButton("Back", f"page_savefilm#{CurrentPage}#{message_id}#{callback_query.from_user.id}"),
+ InlineButton("❌ Close", f"close#{callback_query.from_user.id}")
+ )
try:
- link = message.text.split(" ", maxsplit=1)[1]
html = await http.get(link, headers=headers)
soup = BeautifulSoup(html.text, "lxml")
res = soup.find_all(class_="button button-shadow")
res = "".join(f"{i.text}\n{i['href']}\n\n" for i in res)
- await message.reply(
- f"Hasil Scrap dari {link}:\n\n{res}",
- disable_web_page_preview=True,
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- text="❌ Close",
- callback_data=f"close#{message.from_user.id}",
- )
- ]
- ]
- ),
- )
- except IndexError:
- return await message.reply(f"Gunakan command /{message.command[0]} [link] untuk scrap link download")
- except Exception as e:
- await message.reply(f"ERROR: {str(e)}")
+ except Exception as err:
+ await editPesan(callback_query.message, f"ERROR: {err}", reply_markup=keyboard)
+ return
+ await editPesan(callback_query.message, f"Hasil Scrap dari {link}:\n\n{res}", reply_markup=keyboard)
# Scrape DDL Link Nodrakor
@app.on_message(filters.command(["nodrakor_scrap"], COMMAND_HANDLER))