Dr.Caduceus commited on
Commit
e5a0382
·
unverified ·
1 Parent(s): 25061b6

Bump to v1.5

Browse files
Files changed (1) hide show
  1. bot/plugins/callback.py +16 -21
bot/plugins/callback.py CHANGED
@@ -1,27 +1,22 @@
1
- from pyrogram.types import CallbackQuery
2
  from bot import TelegramBot
3
- from bot.modules.static import *
4
  from bot.modules.decorators import verify_user
 
5
  from bot.modules.telegram import get_message
6
 
7
- @TelegramBot.on_callback_query()
8
- @verify_user
9
- async def manage_callback(bot, q: CallbackQuery):
10
- query = q.data
11
- if query.startswith('rm_'):
12
- sq = query.split('_')
 
 
 
 
 
 
13
 
14
- if len(sq) != 3:
15
- return await q.answer(InvalidQueryText, show_alert=True)
16
-
17
- message = await get_message(int(sq[1]))
18
-
19
- if not message:
20
- return await q.answer(MessageNotExist, show_alert=True)
21
- if sq[2] != message.caption:
22
- return await q.answer(InvalidQueryText, show_alert=True)
23
 
24
- await message.delete()
25
- await q.answer(LinkRevokedText, show_alert=True)
26
- else:
27
- await q.answer(InvalidQueryText, show_alert=True)
 
1
+ from telethon.events import CallbackQuery
2
  from bot import TelegramBot
 
3
  from bot.modules.decorators import verify_user
4
+ from bot.modules.static import *
5
  from bot.modules.telegram import get_message
6
 
7
+ @TelegramBot.on(CallbackQuery(pattern=r'^rm_'))
8
+ @verify_user(private=True)
9
+ async def delete_file(event: CallbackQuery.Event):
10
+ query_data = event.query.data.decode().split('_')
11
+
12
+ if len(query_data) != 3:
13
+ return await event.answer(InvalidQueryText, alert=True)
14
+
15
+ message = await get_message(int(query_data[1]))
16
+
17
+ if not message:
18
+ return await event.answer(MessageNotExist, alert=True)
19
 
20
+ await message.delete()
 
 
 
 
 
 
 
 
21
 
22
+ return await event.answer(LinkRevokedText, alert=True)