Spaces:
Sleeping
Sleeping
File size: 2,979 Bytes
cddafb3 007891e cddafb3 007891e cddafb3 007891e af1662b ca4eb6d c3ccb7b ca4eb6d |
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 |
from Powers import LOGGER
from traceback import format_exc
from Powers.bot_class import Gojo
from pyrogram.types import Message
from Powers.utils.custom_filters import command
from pyrogram.errors import RPCError, PeerIdInvalid
from Powers.database.group_blacklist import GroupBlacklist
# initialise database
db = GroupBlacklist()
@Gojo.on_message(command("blchat", dev_cmd=True))
async def blacklist_chat(c: Gojo, m: Message):
if len(m.text.split()) >= 2:
chat_ids = m.text.split()[1:]
replymsg = await m.reply_text(f"Adding {len(chat_ids)} chats to blacklist")
LOGGER.info(f"{m.from_user.id} blacklisted {chat_ids} groups for bot")
for chat in chat_ids:
try:
get_chat = await c.get_chat(chat)
chat_id = get_chat.id
db.add_chat(chat_id)
except PeerIdInvalid:
await replymsg.edit_text(
"Haven't seen this group in this session, maybe try again later?",
)
except RPCError as ef:
LOGGER.error(ef)
LOGGER.error(format_exc())
await replymsg.edit_text(
f"Added the following chats to Blacklist.\n<code>{', '.join(chat_ids)}</code>.",
)
return
@Gojo.on_message(
command(["rmblchat", "unblchat"], dev_cmd=True),
)
async def unblacklist_chat(c: Gojo, m: Message):
if len(m.text.split()) >= 2:
chat_ids = m.text.split()[1:]
replymsg = await m.reply_text(f"Removing {len(chat_ids)} chats from blacklist")
LOGGER.info(
f"{m.from_user.id} removed blacklisted {chat_ids} groups for bot")
bl_chats = db.list_all_chats()
for chat in chat_ids:
try:
get_chat = await c.get_chat(chat)
chat_id = get_chat.id
if chat_id not in bl_chats:
# If chat is not blaklisted, continue loop
continue
db.remove_chat(chat_id)
except PeerIdInvalid:
await replymsg.edit_text(
"Haven't seen this group in this session, maybe try again later?",
)
except RPCError as ef:
LOGGER.error(ef)
LOGGER.error(format_exc())
await replymsg.edit_text(
f"Removed the following chats to Blacklist.\n<code>{', '.join(chat_ids)}</code>.",
)
return
@Gojo.on_message(
command(["blchatlist", "blchats"], dev_cmd=True),
)
async def list_blacklist_chats(_, m: Message):
bl_chats = db.list_all_chats()
LOGGER.info(f"{m.from_user.id} checking group blacklists in {m.chat.id}")
if bl_chats:
txt = (
(
"These Chats are Blacklisted:\n"
+ "\n".join(f"<code>{i}</code>" for i in bl_chats)
),
)
else:
txt = "No chats are currently blacklisted!"
await m.reply_text(txt)
return
|