/label command improved
This commit is contained in:
parent
9c611b436c
commit
32ebad29ca
@ -1,20 +1,33 @@
|
|||||||
from app import app
|
from app import app, isAnAdmin
|
||||||
from pyrogram import filters
|
from pyrogram import filters
|
||||||
from modules.utils import should_quote, find_user
|
from pyrogram.types import ChatPrivileges
|
||||||
|
from modules.utils import should_quote, find_user, configGet
|
||||||
|
from classes.holo_user import HoloUser
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.private & filters.command(["label"], prefixes=["/"]))
|
@app.on_message(~ filters.scheduled & filters.private & filters.command(["label"], prefixes=["/"]))
|
||||||
async def cmd_label(app, msg):
|
async def cmd_label(app, msg):
|
||||||
|
|
||||||
if len(msg.command) < 3:
|
if msg.chat.id == configGet("admin_group") or await isAnAdmin(msg.from_user.id):
|
||||||
await msg.reply_text("Invalid syntax:\n`/label USER NICKNAME`")
|
|
||||||
return
|
|
||||||
|
|
||||||
target = await find_user(app, msg.command[1])
|
if len(msg.command) < 3:
|
||||||
|
await msg.reply_text("Invalid syntax:\n`/label USER NICKNAME`")
|
||||||
|
return
|
||||||
|
|
||||||
if target is not None:
|
target = await find_user(app, msg.command[1])
|
||||||
|
|
||||||
nickname = " ".join(msg.command[2:])
|
if target is not None:
|
||||||
await msg.reply_text(f"Setting **{target.id}**'s label to **{nickname}**...", quote=should_quote(msg))
|
|
||||||
|
|
||||||
else:
|
target = HoloUser(target)
|
||||||
await msg.reply_text(f"User not found")
|
|
||||||
|
nickname = " ".join(msg.command[2:])
|
||||||
|
|
||||||
|
if nickname.lower() == "reset":
|
||||||
|
await target.reset_label(app, msg.chat)
|
||||||
|
await msg.reply_text(f"Resetting **{target.id}**'s label...", quote=should_quote(msg))
|
||||||
|
|
||||||
|
else:
|
||||||
|
await target.set_label(app, msg.chat, nickname)
|
||||||
|
await msg.reply_text(f"Setting **{target.id}**'s label to **{nickname}**...", quote=should_quote(msg))
|
||||||
|
|
||||||
|
else:
|
||||||
|
await msg.reply_text(f"User not found")
|
Reference in New Issue
Block a user