File size: 3,623 Bytes
1f26706 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
# Ultroid - UserBot
# Copyright (C) 2021-2025 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}addsnip <word><reply to a message>`
add the used word as snip relating to replied message.
• `{i}remsnip <word>`
Remove the snip word..
• `{i}listsnip`
list all snips.
• Use :
type `$(ur snip word)` get setted reply.
"""
import os
from . import upload_file as uf
from telethon.utils import pack_bot_file_id
from pyUltroid._misc import sudoers
from pyUltroid.dB.snips_db import add_snip, get_snips, list_snip, rem_snip
from pyUltroid.fns.tools import create_tl_btn, format_btn, get_msg_button
from . import events, get_string, mediainfo, udB, ultroid_bot, ultroid_cmd
from ._inline import something
@ultroid_cmd(pattern="addsnip( (.*)|$)")
async def an(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("snip_1"))
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 wt.download_media()
m = uf(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 wt.download_media()
m = uf(dl)
os.remove(dl)
else:
m = pack_bot_file_id(wt.media)
if wt.text:
txt = wt.text
if not btn:
txt, btn = get_msg_button(wt.text)
add_snip(wrd, txt, m, btn)
else:
add_snip(wrd, None, m, btn)
else:
txt = wt.text
if not btn:
txt, btn = get_msg_button(wt.text)
add_snip(wrd, txt, None, btn)
await e.eor(f"Done : snip `${wrd}` Saved.")
ultroid_bot.add_handler(add_snips, events.NewMessage())
@ultroid_cmd(pattern="remsnip( (.*)|$)")
async def rs(e):
wrd = (e.pattern_match.group(1).strip()).lower()
if not wrd:
return await e.eor(get_string("snip_2"))
if wrd.startswith("$"):
wrd = wrd.replace("$", "")
rem_snip(wrd)
await e.eor(f"Done : snip `${wrd}` Removed.")
@ultroid_cmd(pattern="listsnip")
async def lsnote(e):
if x := list_snip():
sd = "SNIPS Found :\n\n"
return await e.eor(sd + x)
await e.eor("No Snips Found Here")
async def add_snips(e):
if not e.out and e.sender_id not in sudoers():
return
xx = [z.replace("$", "") for z in e.text.lower().split() if z.startswith("$")]
for z in xx:
if k := get_snips(z):
msg = k["msg"]
media = k["media"]
rep = await e.get_reply_message()
if rep:
if k.get("button"):
btn = create_tl_btn(k["button"])
return await something(rep, msg, media, btn)
await rep.reply(msg, file=media)
else:
await e.delete()
if k.get("button"):
btn = create_tl_btn(k["button"])
return await something(e, msg, media, btn, reply=None)
await ultroid_bot.send_message(e.chat_id, msg, file=media)
if udB.get_key("SNIP"):
ultroid_bot.add_handler(add_snips, events.NewMessage())
|