async def _(event): if event.is_private: return user = event.sender # type: Optional[User] chat = event.chat_id # type: Optional[Chat] # ignore admins and owner if await is_register_admin(chat, user.id) or user.id == OWNER_ID: sql.update_flood(chat, None) return should_ban = sql.update_flood(chat, user.id) if not should_ban: return try: getmode, getvalue = sql.get_flood_setting(chat) if getmode == 1: await tbot(EditBannedRequest(chat, user.id, BANNED_RIGHTS)) execstrings = "Banned" elif getmode == 2: await tbot.kick_participant(chat, user.id) execstrings = "Kicked" elif getmode == 3: await tbot(EditBannedRequest(chat, user.id, MUTE_RIGHTS)) execstrings = "Muted" elif getmode == 4: bantime = await extract_time(event, getvalue) NEW_RIGHTS = ChatBannedRights( until_date=bantime, view_messages=True, send_messages=True, send_media=True, send_stickers=True, send_gifs=True, send_games=True, send_inline=True, embed_links=True, ) await tbot(EditBannedRequest(chat, user.id, NEW_RIGHTS)) execstrings = "Banned for {}".format(getvalue) elif getmode == 5: mutetime = await extract_time(event, getvalue) print(mutetime) NEW_RIGHTS = ChatBannedRights(until_date=mutetime, send_messages=True) await tbot(EditBannedRequest(chat, user.id, NEW_RIGHTS)) execstrings = "Muted for {}".format(getvalue) await event.reply("Spammer Detected !\n{}!".format(execstrings)) except Exception: await event.reply( "I can't restrict people here, give me permissions first! Until then, I'll disable anti-flood." ) sql.set_flood(chat, 0)
def set_flood_mode(update, context): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] msg = update.effective_message # type: Optional[Message] args = context.args if update.effective_message.chat.type == "private": send_message( update.effective_message, "This command is meant to use in group not in PM", ) return "" chat = update.effective_chat chat_id = update.effective_chat.id chat_name = update.effective_message.chat.title if args: if args[0].lower() == "ban": settypeflood = "ban" sql.set_flood_strength(chat_id, 1, "0") elif args[0].lower() == "kick": settypeflood = "kick" sql.set_flood_strength(chat_id, 2, "0") elif args[0].lower() == "mute": settypeflood = "mute" sql.set_flood_strength(chat_id, 3, "0") elif args[0].lower() == "tban": if len(args) == 1: teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tban <timevalue>`. Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" send_message(update.effective_message, teks, parse_mode="markdown") return settypeflood = "tban for {}".format(args[1]) sql.set_flood_strength(chat_id, 4, str(args[1])) elif args[0].lower() == "tmute": if len(args) == 1: teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tmute <timevalue>`. Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" send_message(update.effective_message, teks, parse_mode="markdown") return settypeflood = "tmute for {}".format(args[1]) sql.set_flood_strength(chat_id, 5, str(args[1])) else: send_message(update.effective_message, "I only understand ban/kick/mute/tban/tmute!") return text = msg.reply_text( "Exceeding consecutive flood limit will result in {}!".format( settypeflood)) send_message(update.effective_message, text, parse_mode="markdown") return ("<b>{}:</b>\n" "<b>Admin:</b> {}\n" "Has changed antiflood mode. User will {}.".format( settypeflood, html.escape(chat.title), mention_html(user.id, user.first_name), )) getmode, getvalue = sql.get_flood_setting(chat.id) if getmode == 1: settypeflood = "ban" elif getmode == 2: settypeflood = "kick" elif getmode == 3: settypeflood = "mute" elif getmode == 4: settypeflood = "tban for {}".format(getvalue) elif getmode == 5: settypeflood = "tmute for {}".format(getvalue) text = msg.reply_text( "Sending more message than flood limit will result in {}.".format( settypeflood)) send_message(update.effective_message, text, parse_mode=ParseMode.MARKDOWN) return ""
async def _(event): try: if event.is_private: return if event.is_group: if not await can_change_info(message=event): return chat_id = event.chat_id args = event.pattern_match.group(1) time = args.split() if time[0]: if time[0] == "ban": settypeflood = "ban" sql.set_flood_strength(chat_id, 1, "0") elif time[0] == "kick": settypeflood = "kick" sql.set_flood_strength(chat_id, 2, "0") elif time[0] == "mute": settypeflood = "mute" sql.set_flood_strength(chat_id, 3, "0") elif time[0] == "tban": try: ttime = time[1] except: await event.reply("Please provide the tban time interval.") return if len(ttime) == 1: teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tban <timevalue>`. Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" await event.reply(teks, parse_mode="markdown") return if not any(ttime.endswith(unit) for unit in ("m", "h", "d")): await event.reply( "Invalid time type specified. Expected m,h, or d, got: {}" .format(ttime)) return settypeflood = "tban for {}".format(ttime) sql.set_flood_strength(chat_id, 4, str(ttime)) elif time[0] == "tmute": try: ttime = time[1] except: await event.reply("Please provide the tmute time interval." ) return if len(ttime) == 1: teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tmute <timevalue>`. Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" await event.reply(teks, parse_mode="markdown") return if not any(ttime.endswith(unit) for unit in ("m", "h", "d")): await event.reply( "Invalid time type specified. Expected m,h, or d, got: {}" .format(ttime)) return settypeflood = "tmute for {}".format(ttime) sql.set_flood_strength(chat_id, 5, str(ttime)) else: await event.reply("I only understand ban/kick/mute/tban/tmute!" ) return await event.reply( "Exceeding consecutive flood limit will result in {}!".format( settypeflood)) else: getmode, getvalue = sql.get_flood_setting(chat_id) if getmode == 1: settypeflood = "ban" elif getmode == 2: settypeflood = "kick" elif getmode == 3: settypeflood = "mute" elif getmode == 4: settypeflood = "tban for {}".format(getvalue) elif getmode == 5: settypeflood = "tmute for {}".format(getvalue) await event.reply( "Sending more message than flood limit will result in {}.". format(settypeflood)) except Exception as e: print(e)
def check_flood(update, context) -> str: user = update.effective_user # type: Optional[User] chat = update.effective_chat # type: Optional[Chat] msg = update.effective_message # type: Optional[Message] if not user: # ignore channels return "" # ignore admins if is_user_admin(chat, user.id): sql.update_flood(chat.id, None) return "" should_ban = sql.update_flood(chat.id, user.id) if not should_ban: return "" try: getmode, getvalue = sql.get_flood_setting(chat.id) if getmode == 1: chat.kick_member(user.id) execstrings = "Banned" tag = "BANNED" elif getmode == 2: chat.kick_member(user.id) chat.unban_member(user.id) execstrings = "Kicked" tag = "KICKED" elif getmode == 3: context.bot.restrict_chat_member( chat.id, user.id, permissions=ChatPermissions(can_send_messages=False)) execstrings = "Muted" tag = "MUTED" elif getmode == 4: bantime = extract_time(msg, getvalue) chat.kick_member(user.id, until_date=bantime) execstrings = "Banned for {}".format(getvalue) tag = "TBAN" elif getmode == 5: mutetime = extract_time(msg, getvalue) context.bot.restrict_chat_member( chat.id, user.id, until_date=mutetime, permissions=ChatPermissions(can_send_messages=False), ) execstrings = "Muted for {}".format(getvalue) tag = "TMUTE" send_message( update.effective_message, "Great, I like to leave flooding to natural disasters but you, " "you were just a disappointment. {}!".format(execstrings), ) return ("<b>{}:</b>" "\n#{}" "\n<b>User:</b> {}" "\nFlooded the group.".format( tag, html.escape(chat.title), mention_html(user.id, user.first_name))) except BadRequest: msg.reply_text( "I can't restrict people here, give me permissions first! Until then, I'll disable anti-flood." ) sql.set_flood(chat.id, 0) return ( "<b>{}:</b>" "\n#INFO" "\nDon't have enough permission to restrict users so automatically disabled anti-flood" .format(chat.title))