File size: 3,820 Bytes
618430a |
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 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
# Ultroid - UserBot
# Copyright (C) 2021-2025 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
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)
|