|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from . import get_help |
|
|
|
|
|
__doc__ = get_help("help_asstcmd") |
|
|
|
|
|
import os |
|
|
|
|
|
from pyUltroid.dB.asstcmd_db import add_cmd, cmd_reply, list_cmds, rem_cmd |
|
|
from pyUltroid.fns.tools import create_tl_btn, format_btn, get_msg_button |
|
|
from telethon import events, utils |
|
|
|
|
|
from . import asst, get_string, mediainfo, udB, ultroid_cmd, upload_file |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="addcmd( (.*)|$)") |
|
|
async def ac(e): |
|
|
wrd = (e.pattern_match.group(1).strip()).lower() |
|
|
wt = await e.get_reply_message() |
|
|
if not (wt and wrd): |
|
|
return await e.eor(get_string("asstcmd_1"), time=5) |
|
|
if "/" in wrd: |
|
|
wrd = wrd.replace("/", "") |
|
|
btn = format_btn(wt.buttons) if wt.buttons else None |
|
|
if wt and wt.media: |
|
|
wut = mediainfo(wt.media) |
|
|
if wut.startswith(("pic", "gif")): |
|
|
dl = await e.client.download_media(wt.media) |
|
|
m = upload_file(dl) |
|
|
os.remove(dl) |
|
|
elif wut == "video": |
|
|
if wt.media.document.size > 8 * 1000 * 1000: |
|
|
return await e.eor(get_string("com_4"), time=5) |
|
|
dl = await e.client.download_media(wt.media) |
|
|
m = upload_file(dl) |
|
|
os.remove(dl) |
|
|
else: |
|
|
m = utils.pack_bot_file_id(wt.media) |
|
|
if wt.text: |
|
|
txt = wt.text |
|
|
if not btn: |
|
|
txt, btn = get_msg_button(wt.text) |
|
|
add_cmd(wrd, txt, m, btn) |
|
|
else: |
|
|
add_cmd(wrd, None, m, btn) |
|
|
else: |
|
|
txt = wt.text |
|
|
if not btn: |
|
|
txt, btn = get_msg_button(wt.text) |
|
|
add_cmd(wrd, txt, None, btn) |
|
|
asst.add_handler( |
|
|
astcmds, |
|
|
events.NewMessage( |
|
|
func=lambda x: x.text.startswith("/") and x.text[1:] in list(list_cmds()) |
|
|
), |
|
|
) |
|
|
await e.eor(get_string("asstcmd_4").format(wrd)) |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="remcmd( (.*)|$)") |
|
|
async def rc(e): |
|
|
wrd = (e.pattern_match.group(1).strip()).lower() |
|
|
if not wrd: |
|
|
return await e.eor(get_string("asstcmd_2"), time=5) |
|
|
wrd = wrd.replace("/", "") |
|
|
rem_cmd(wrd) |
|
|
await e.eor(get_string("asstcmd_3").format(wrd)) |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="listcmd$") |
|
|
async def lscmd(e): |
|
|
if list_cmds(): |
|
|
ok = get_string("asstcmd_6") |
|
|
for x in list_cmds(): |
|
|
ok += f"/{x}" + "\n" |
|
|
return await e.eor(ok) |
|
|
return await e.eor(get_string("asstcmd_5")) |
|
|
|
|
|
|
|
|
async def astcmds(e): |
|
|
xx = (e.text.replace("/", "")).lower().split()[0] |
|
|
if cmd_reply(xx): |
|
|
msg, media, bt = cmd_reply(xx) |
|
|
if bt: |
|
|
bt = create_tl_btn(bt) |
|
|
await e.reply(msg, file=media, buttons=bt) |
|
|
|
|
|
|
|
|
if udB.get_key("ASST_CMDS"): |
|
|
asst.add_handler( |
|
|
astcmds, |
|
|
events.NewMessage( |
|
|
func=lambda x: x.text.startswith("/") and x.text[1:] in list(list_cmds()) |
|
|
), |
|
|
) |
|
|
|
