| |
| |
| |
| |
| |
| |
|
|
| from . import get_help |
|
|
| __doc__ = get_help("help_antiflood") |
|
|
|
|
| import re |
|
|
| from telethon.events import NewMessage as NewMsg |
|
|
| from pyUltroid.dB import DEVLIST |
| from pyUltroid.dB.antiflood_db import get_flood, get_flood_limit, rem_flood, set_flood |
| from pyUltroid.fns.admins import admin_check |
|
|
| from . import Button, Redis, asst, callback, eod, get_string, ultroid_bot, ultroid_cmd |
|
|
| _check_flood = {} |
|
|
| if Redis("ANTIFLOOD"): |
|
|
| @ultroid_bot.on( |
| NewMsg( |
| chats=list(get_flood().keys()), |
| ), |
| ) |
| async def flood_checm(event): |
| count = 1 |
| chat = (await event.get_chat()).title |
| if event.chat_id in _check_flood.keys(): |
| if event.sender_id == list(_check_flood[event.chat_id].keys())[0]: |
| count = _check_flood[event.chat_id][event.sender_id] |
| _check_flood[event.chat_id] = {event.sender_id: count + 1} |
| else: |
| _check_flood[event.chat_id] = {event.sender_id: count} |
| else: |
| _check_flood[event.chat_id] = {event.sender_id: count} |
| if await admin_check(event, silent=True) or getattr(event.sender, "bot", None): |
| return |
| if event.sender_id in DEVLIST: |
| return |
| if _check_flood[event.chat_id][event.sender_id] >= int( |
| get_flood_limit(event.chat_id) |
| ): |
| try: |
| name = event.sender.first_name |
| await event.client.edit_permissions( |
| event.chat_id, event.sender_id, send_messages=False |
| ) |
| del _check_flood[event.chat_id] |
| await event.reply(f"#AntiFlood\n\n{get_string('antiflood_3')}") |
| await asst.send_message( |
| int(Redis("LOG_CHANNEL")), |
| f"#Antiflood\n\n`Muted `[{name}](tg://user?id={event.sender_id})` in {chat}`", |
| buttons=Button.inline( |
| "Unmute", data=f"anti_{event.sender_id}_{event.chat_id}" |
| ), |
| ) |
| except BaseException: |
| pass |
|
|
|
|
| @callback( |
| re.compile( |
| "anti_(.*)", |
| ), |
| ) |
| async def unmuting(e): |
| ino = (e.data_match.group(1)).decode("UTF-8").split("_") |
| user = int(ino[0]) |
| chat = int(ino[1]) |
| user_name = (await ultroid_bot.get_entity(user)).first_name |
| chat_title = (await ultroid_bot.get_entity(chat)).title |
| await ultroid_bot.edit_permissions(chat, user, send_messages=True) |
| await e.edit( |
| f"#Antiflood\n\n`Unmuted `[{user_name}](tg://user?id={user})` in {chat_title}`" |
| ) |
|
|
|
|
| @ultroid_cmd( |
| pattern="setflood ?(\\d+)", |
| admins_only=True, |
| ) |
| async def setflood(e): |
| input_ = e.pattern_match.group(1).strip() |
| if not input_: |
| return await e.eor("`What?`", time=5) |
| if not input_.isdigit(): |
| return await e.eor(get_string("com_3"), time=5) |
| if m := set_flood(e.chat_id, input_): |
| return await eod(e, get_string("antiflood_4").format(input_)) |
|
|
|
|
| @ultroid_cmd( |
| pattern="remflood$", |
| admins_only=True, |
| ) |
| async def remove_flood(e): |
| hmm = rem_flood(e.chat_id) |
| try: |
| del _check_flood[e.chat_id] |
| except BaseException: |
| pass |
| if hmm: |
| return await e.eor(get_string("antiflood_1"), time=5) |
| await e.eor(get_string("antiflood_2"), time=5) |
|
|
|
|
| @ultroid_cmd( |
| pattern="getflood$", |
| admins_only=True, |
| ) |
| async def getflood(e): |
| if ok := get_flood_limit(e.chat_id): |
| return await e.eor(get_string("antiflood_5").format(ok), time=5) |
| await e.eor(get_string("antiflood_2"), time=5) |
|
|
