Spaces:
Paused
Paused
File size: 3,354 Bytes
2f67506 | 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 | #
# Copyright (C) 2021-2022 by TeamYukki@Github, < https://github.com/TeamYukki >.
#
# This file is part of < https://github.com/TeamYukki/YukkiMusicBot > project,
# and is released under the "GNU v3.0 License Agreement".
# Please see < https://github.com/TeamYukki/YukkiMusicBot/blob/master/LICENSE >
#
# All rights reserved.
from pyrogram import filters
from pyrogram.types import Message
import config
from strings import get_command
from YukkiMusic import app
from YukkiMusic.misc import SUDOERS
from YukkiMusic.utils.database import (add_private_chat,
get_private_served_chats,
is_served_private_chat,
remove_private_chat)
from YukkiMusic.utils.decorators.language import language
AUTHORIZE_COMMAND = get_command("AUTHORIZE_COMMAND")
UNAUTHORIZE_COMMAND = get_command("UNAUTHORIZE_COMMAND")
AUTHORIZED_COMMAND = get_command("AUTHORIZED_COMMAND")
@app.on_message(filters.command(AUTHORIZE_COMMAND) & SUDOERS)
@language
async def authorize(client, message: Message, _):
if config.PRIVATE_BOT_MODE != str(True):
return await message.reply_text(_["pbot_12"])
if len(message.command) != 2:
return await message.reply_text(_["pbot_1"])
try:
chat_id = int(message.text.strip().split()[1])
except:
return await message.reply_text(_["pbot_7"])
if not await is_served_private_chat(chat_id):
await add_private_chat(chat_id)
await message.reply_text(_["pbot_3"])
else:
await message.reply_text(_["pbot_5"])
@app.on_message(filters.command(UNAUTHORIZE_COMMAND) & SUDOERS)
@language
async def unauthorize(client, message: Message, _):
if config.PRIVATE_BOT_MODE != str(True):
return await message.reply_text(_["pbot_12"])
if len(message.command) != 2:
return await message.reply_text(_["pbot_2"])
try:
chat_id = int(message.text.strip().split()[1])
except:
return await message.reply_text(_["pbot_7"])
if not await is_served_private_chat(chat_id):
return await message.reply_text(_["pbot_6"])
else:
await remove_private_chat(chat_id)
return await message.reply_text(_["pbot_4"])
@app.on_message(filters.command(AUTHORIZED_COMMAND) & SUDOERS)
@language
async def authorized(client, message: Message, _):
if config.PRIVATE_BOT_MODE != str(True):
return await message.reply_text(_["pbot_12"])
m = await message.reply_text(_["pbot_8"])
served_chats = []
text = _["pbot_9"]
chats = await get_private_served_chats()
for chat in chats:
served_chats.append(int(chat["chat_id"]))
count = 0
co = 0
msg = _["pbot_13"]
for served_chat in served_chats:
try:
title = (await app.get_chat(served_chat)).title
count += 1
text += f"{count}:- {title[:15]} [{served_chat}]\n"
except Exception:
title = _["pbot_10"]
co += 1
msg += f"{co}:- {title} [{served_chat}]\n"
if co == 0:
if count == 0:
return await m.edit(_["pbot_11"])
else:
return await m.edit(text)
else:
if count == 0:
await m.edit(msg)
else:
text = f"{text} {msg}"
return await m.edit(text)
|