Telegram/modules/commands/label.py

from app import app, isAnAdmin
from pyrogram import filters
from modules.utils import locale, should_quote, find_user
from classes.holo_user import HoloUser, LabelTooLongError

@app.on_message(~ filters.scheduled & filters.private & filters.command(["label"], prefixes=["/"]))
async def cmd_label(app, msg):

    if await isAnAdmin(msg.from_user.id) is True:

        if len(msg.command) < 3:
            await msg.reply_text("Invalid syntax:\n`/label USER LABEL`")
            return

        target = await find_user(app, msg.command[1])

        if target is not None:
            
            target = HoloUser(target)

            label = " ".join(msg.command[2:])

            if label.lower() == "reset":
                await target.label_reset(msg.chat)
                await msg.reply_text(f"Resetting **{target.id}**'s label...", quote=should_quote(msg))
                
            else:
                try:
                    await target.label_set(msg.chat, label)
                except LabelTooLongError:
                    await msg.reply_text(locale("label_too_long", "message"))
                    return
                await msg.reply_text(f"Setting **{target.id}**'s label to **{label}**...", quote=should_quote(msg))
                
        else:
            await msg.reply_text(f"User not found")