From 819cec7e605d88695769cdfcdd68898237f9a153 Mon Sep 17 00:00:00 2001 From: Tiemen Date: Mon, 4 Sep 2017 04:12:03 +0200 Subject: [PATCH] Add mban command --- rowboat/plugins/admin.py | 41 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) diff --git a/rowboat/plugins/admin.py b/rowboat/plugins/admin.py index 205f8045d..1032e4abb 100644 --- a/rowboat/plugins/admin.py +++ b/rowboat/plugins/admin.py @@ -765,6 +765,47 @@ def ban(self, event, user, reason=None): u=member.user if member else user, )) + @Plugin.command('mban', parser=True, level=CommandLevels.MOD) + @Plugin.parser.add_argument('users', type=long, nargs='+') + @Plugin.parser.add_argument('-r', '--reason', default='', help='reason for modlog') + def mban(self, event, args): + members = [] + for user_id in args.users: + member = event.guild.get_member(user_id) + if not member: + raise CommandFail('failed to ban {}, user not found'.format(user_id)) + + if not self.can_act_on(event, member.id, throw=False): + raise CommandFail('failed to ban {}, invalid permissions'.format(user_id)) + + members.append(member) + + msg = event.msg.reply('Ok, ban {} users for `{}`?'.format(len(members), args.reason or 'no reason')) + msg.chain(False).\ + add_reaction(GREEN_TICK_EMOJI).\ + add_reaction(RED_TICK_EMOJI) + + try: + mra_event = self.wait_for_event( + 'MessageReactionAdd', + message_id=msg.id, + conditional=lambda e: ( + e.emoji.id in (GREEN_TICK_EMOJI_ID, RED_TICK_EMOJI_ID) and + e.user_id == event.author.id + )).get(timeout=10) + except gevent.Timeout: + return + finally: + msg.delete() + + if mra_event.emoji.id != GREEN_TICK_EMOJI_ID: + return + + for member in members: + Infraction.ban(self, event, member, args.reason, guild=event.guild) + + raise CommandSuccess('banned {} users'.format(len(members))) + @Plugin.command('softban', ' [reason:str...]', level=CommandLevels.MOD) def softban(self, event, user, reason=None): """