| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| | from . import get_help |
| |
|
| | __doc__ = get_help("help_autoban") |
| |
|
| | from telethon import events |
| |
|
| | from pyUltroid.dB.base import KeyManager |
| |
|
| | from . import LOGS, asst, ultroid_bot, ultroid_cmd |
| |
|
| | Keym = KeyManager("DND_CHATS", cast=list) |
| |
|
| |
|
| | def join_func(e): |
| | return e.user_joined and Keym.contains(e.chat_id) |
| |
|
| |
|
| | async def dnd_func(event): |
| | for user in event.users: |
| | try: |
| | await (await event.client.kick_participant(event.chat_id, user)).delete() |
| | except Exception as ex: |
| | LOGS.error("Error in DND:") |
| | LOGS.exception(ex) |
| | await event.delete() |
| |
|
| |
|
| | @ultroid_cmd( |
| | pattern="autokick (on|off)$", |
| | admins_only=True, |
| | manager=True, |
| | require="ban_users", |
| | fullsudo=True, |
| | ) |
| | async def _(event): |
| | match = event.pattern_match.group(1) |
| | if match == "on": |
| | if Keym.contains(event.chat_id): |
| | return await event.eor("`Chat already in do not disturb mode.`", time=3) |
| | Keym.add(event.chat_id) |
| | event.client.add_handler(dnd_func, events.ChatAction(func=join_func)) |
| | await event.eor("`Do not disturb mode activated for this chat.`", time=3) |
| | elif match == "off": |
| | if not Keym.contains(event.chat_id): |
| | return await event.eor("`Chat is not in do not disturb mode.`", time=3) |
| | Keym.remove(event.chat_id) |
| | await event.eor("`Do not disturb mode deactivated for this chat.`", time=3) |
| |
|
| |
|
| | if Keym.get(): |
| | ultroid_bot.add_handler(dnd_func, events.ChatAction(func=join_func)) |
| | asst.add_handler(dnd_func, events.ChatAction(func=join_func)) |
| |
|
