File size: 2,092 Bytes
a8e9b84 |
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 |
from pyrogram import filters
from pyrogram.types import Message
from DragMusic import app
from DragMusic.misc import SUDOERS
from DragMusic.utils.database import blacklist_chat, blacklisted_chats, whitelist_chat
from DragMusic.utils.decorators.language import language
from config import BANNED_USERS
@app.on_message(filters.command(["blchat", "blacklistchat"]) & SUDOERS)
@language
async def blacklist_chat_func(client, message: Message, _):
if len(message.command) != 2:
return await message.reply_text(_["black_1"])
chat_id = int(message.text.strip().split()[1])
if chat_id in await blacklisted_chats():
return await message.reply_text(_["black_2"])
blacklisted = await blacklist_chat(chat_id)
if blacklisted:
await message.reply_text(_["black_3"])
else:
await message.reply_text(_["black_9"])
try:
await app.leave_chat(chat_id)
except:
pass
@app.on_message(
filters.command(["whitelistchat", "unblacklistchat", "unblchat"]) & SUDOERS
)
@language
async def white_funciton(client, message: Message, _):
if len(message.command) != 2:
return await message.reply_text(_["black_4"])
chat_id = int(message.text.strip().split()[1])
if chat_id not in await blacklisted_chats():
return await message.reply_text(_["black_5"])
whitelisted = await whitelist_chat(chat_id)
if whitelisted:
return await message.reply_text(_["black_6"])
await message.reply_text(_["black_9"])
@app.on_message(filters.command(["blchats", "blacklistedchats"]) & ~BANNED_USERS)
@language
async def all_chats(client, message: Message, _):
text = _["black_7"]
j = 0
for count, chat_id in enumerate(await blacklisted_chats(), 1):
try:
title = (await app.get_chat(chat_id)).title
except:
title = "ᴘʀɪᴠᴀᴛᴇ ᴄʜᴀᴛ"
j = 1
text += f"{count}. {title}[<code>{chat_id}</code>]\n"
if j == 0:
await message.reply_text(_["black_8"].format(app.mention))
else:
await message.reply_text(text)
|