154 lines
4.2 KiB
Python
154 lines
4.2 KiB
Python
# 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}addnote <word><reply to a message>`
|
|
add note in the used chat with replied message and choosen word.
|
|
|
|
• `{i}remnote <word>`
|
|
Remove the note from used chat.
|
|
|
|
• `{i}listnote`
|
|
list all notes.
|
|
|
|
• Use :
|
|
set notes in group so all can use it.
|
|
type `#(Keyword of note)` to get it
|
|
"""
|
|
import os
|
|
|
|
from . import upload_file as uf
|
|
from telethon.utils import pack_bot_file_id
|
|
|
|
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
|
|
|
|
# Functions moved from notes_db.py
|
|
def get_stuff():
|
|
return udB.get_key("NOTE") or {}
|
|
|
|
|
|
def add_note(chat, word, msg, media, button):
|
|
ok = get_stuff()
|
|
if ok.get(int(chat)):
|
|
ok[int(chat)].update({word: {"msg": msg, "media": media, "button": button}})
|
|
else:
|
|
ok.update({int(chat): {word: {"msg": msg, "media": media, "button": button}}})
|
|
udB.set_key("NOTE", ok)
|
|
|
|
|
|
def rem_note(chat, word):
|
|
ok = get_stuff()
|
|
if ok.get(int(chat)) and ok[int(chat)].get(word):
|
|
ok[int(chat)].pop(word)
|
|
return udB.set_key("NOTE", ok)
|
|
|
|
|
|
def rem_all_note(chat):
|
|
ok = get_stuff()
|
|
if ok.get(int(chat)):
|
|
ok.pop(int(chat))
|
|
return udB.set_key("NOTE", ok)
|
|
|
|
|
|
def get_notes(chat, word):
|
|
ok = get_stuff()
|
|
if ok.get(int(chat)) and ok[int(chat)].get(word):
|
|
return ok[int(chat)][word]
|
|
|
|
|
|
def list_note(chat):
|
|
ok = get_stuff()
|
|
if ok.get(int(chat)):
|
|
return "".join(f"👉 #{z}\n" for z in ok[chat])
|
|
|
|
|
|
@ultroid_cmd(pattern="addnote( (.*)|$)", admins_only=True)
|
|
async def an(e):
|
|
wrd = (e.pattern_match.group(1)).strip()
|
|
wt = await e.get_reply_message()
|
|
chat = e.chat_id
|
|
if not (wt and wrd):
|
|
return await e.eor(get_string("notes_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"))
|
|
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_note(chat, wrd, txt, m, btn)
|
|
else:
|
|
add_note(chat, wrd, None, m, btn)
|
|
else:
|
|
txt = wt.text
|
|
if not btn:
|
|
txt, btn = get_msg_button(wt.text)
|
|
add_note(chat, wrd, txt, None, btn)
|
|
await e.eor(get_string("notes_2").format(wrd))
|
|
ultroid_bot.add_handler(notes, events.NewMessage())
|
|
|
|
|
|
@ultroid_cmd(pattern="remnote( (.*)|$)", admins_only=True)
|
|
async def rn(e):
|
|
wrd = (e.pattern_match.group(1)).strip()
|
|
chat = e.chat_id
|
|
if not wrd:
|
|
return await e.eor(get_string("notes_3"))
|
|
if wrd.startswith("#"):
|
|
wrd = wrd.replace("#", "")
|
|
rem_note(chat, wrd)
|
|
await e.eor(get_string("notes_4").format(wrd))
|
|
|
|
|
|
@ultroid_cmd(pattern="listnote$", admins_only=True)
|
|
async def lsnote(e):
|
|
if x := list_note(e.chat_id):
|
|
sd = get_string("notes_5")
|
|
return await e.eor(sd + x)
|
|
await e.eor(get_string("notes_6"))
|
|
|
|
|
|
async def notes(e):
|
|
if not e.text:
|
|
return
|
|
xx = e.text.lower().strip()
|
|
if not xx.startswith("#"):
|
|
return
|
|
xx = xx.replace("#", "", 1)
|
|
chat = e.chat_id
|
|
if not get_notes(chat, xx):
|
|
return
|
|
info = get_notes(chat, xx)
|
|
msg = info["msg"]
|
|
media = info["media"]
|
|
if info.get("button"):
|
|
btn = create_tl_btn(info["button"])
|
|
return await something(e, msg, media, btn)
|
|
await e.reply(msg, file=media)
|
|
|
|
|
|
if get_stuff():
|
|
ultroid_bot.add_handler(notes, events.NewMessage())
|