forked from SilentDemonSD/WZML-X
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsave_msg.py
26 lines (23 loc) · 1.23 KB
/
save_msg.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/usr/bin/env python3
from pyrogram.types import InlineKeyboardMarkup
from pyrogram.handlers import CallbackQueryHandler
from pyrogram.filters import regex
from asyncio import sleep
from bot import bot, bot_name, user_data
async def save_message(_, query):
usr = query.from_user.id
user_dict = user_data.get(usr, {})
if query.data == "save":
if user_dict.get('save_mode'):
usr = next(iter(user_dict.get('ldump', {}).values()))
try:
await query.message.copy(usr, reply_markup=InlineKeyboardMarkup(BTN) if (BTN := query.message.reply_markup.inline_keyboard[:-1]) else None)
await query.answer("Message/Media Successfully Saved !", show_alert=True)
except Exception:
if user_dict.get('save_mode'):
await query.answer('Make Bot as Admin and give Post Permissions and Try Again', show_alert=True)
else:
await query.answer(url=f"https://t.me/{bot_name}?start=start")
await sleep(1)
await query.message.copy(usr, reply_markup=InlineKeyboardMarkup(BTN) if (BTN := query.message.reply_markup.inline_keyboard[:-1]) else None)
bot.add_handler(CallbackQueryHandler(save_message, filters=regex(r"^save")))