90ad5646ff
Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-authored-by: Aditya <xditya@ultroid.tech> Co-authored-by: Kaif <88398455+kaif-00z@users.noreply.github.com> Co-authored-by: Aditya <me@xditya.me> Co-authored-by: 1Danish-00 <danish@ultroid.tech> Co-authored-by: smartman_ru <bblkovo@gmail.com>
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2021-2022 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/>.
|
|
"""
|
|
✘ Commands Available -
|
|
|
|
•`{i}addcmd <new cmd> <reply>`
|
|
It will set new cmd for your assistant bot with that reply message.
|
|
|
|
•`{i}remcmd <cmd name>`
|
|
It will remove your cmd.
|
|
|
|
•`{i}listcmd`
|
|
To Get list of all your custom cmd.
|
|
"""
|
|
import os
|
|
|
|
from pyUltroid.dB.asstcmd_db import add_cmd, cmd_reply, list_cmds, rem_cmd
|
|
from pyUltroid.functions.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
|
|
|
|
|
|
@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)
|
|
variable = uf(dl)
|
|
os.remove(dl)
|
|
m = "https://telegra.ph" + 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 = "https://telegra.ph" + 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))
|
|
|
|
|
|
@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 += "/" + 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())
|
|
),
|
|
)
|