Spaces:
Running
Running
# Ultroid - UserBot | |
# Copyright (C) 2021-2023 TeamUltroid | |
# | |
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ > | |
# PLease read the GNU Affero General Public License in | |
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>. | |
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 | |
try: | |
from telegraph import upload_file as uf | |
except ImportError: | |
uf = None | |
from telethon import events, utils | |
from . import asst, get_string, mediainfo, udB, ultroid_cmd | |
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) | |
variable = uf(dl) | |
os.remove(dl) | |
m = f"https://graph.org{variable[0]}" | |
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) | |
variable = uf(dl) | |
os.remove(dl) | |
m = f"https://graph.org{variable[0]}" | |
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)) | |
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)) | |
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()) | |
), | |
) | |