def func(*args, **kwargs): try: if g.user.admin: guild = Guild.get(Guild.guild_id == kwargs.pop('gid')) guild.role = 'admin' else: guild = Guild.select( Guild, Guild.config['web'][str(g.user.user_id)].alias('role') ).where((Guild.guild_id == kwargs.pop('gid')) & (~(Guild.config['web'][str(g.user.user_id)] >> None)) ).get() return f(guild, *args, **kwargs) except Guild.DoesNotExist: return 'Invalid Guild', 404
def rmv_whitelist(self, event, guild, flag): flag = Guild.WhitelistFlags.get(flag) if not flag: raise CommandFail('invalid flag') try: guild = Guild.get(guild_id=guild) except Guild.DoesNotExist: raise CommandFail('no guild exists with that id') if not guild.is_whitelisted(flag): raise CommandFail('this guild doesn\'t have this flag') guild.whitelist.remove(int(flag)) guild.save() guild.emit('GUILD_UPDATE') event.msg.reply('Ok, removed flag `{}` from guild {}'.format(str(flag), guild.guild_id))
def add_whitelist(self, event, guild, flag): flag = Guild.WhitelistFlags.get(flag) if not flag: raise CommandFail('invalid flag') try: guild = Guild.get(guild_id=guild) except Guild.DoesNotExist: raise CommandFail('no guild exists with that id') if guild.is_whitelisted(flag): raise CommandFail('this guild already has this flag') guild.whitelist.append(int(flag)) guild.save() guild.emit('GUILD_UPDATE') event.msg.reply('Ok, added flag `{}` to guild {}'.format(str(flag), guild.guild_id))
def rmv_whitelist(guild_id, flag): from rowboat.models.guild import Guild init_db(ENV) flag = Guild.WhitelistFlags.get(flag) if not flag: print 'Invalid flag' return try: guild = Guild.get(guild_id=guild_id) except Guild.DoesNotExist: print 'No guild exists with that id' return guild.whitelist.remove(int(flag)) guild.save() guild.emit_update() print 'removed flag'
def add_whitelist(guild_id, flag): from rowboat.models.guild import Guild init_db(ENV) flag = Guild.WhitelistFlags.get(flag) if not flag: print('Flag not found') return try: guild = Guild.get(guild_id=guild_id) except Guild.DoesNotExist: print('No guild exists with that id') return guild.whitelist.append(int(flag)) guild.save() guild.emit_update() print('Added flag')
def rmv_whitelist(guild_id, flag): from rowboat.models.guild import Guild init_db(ENV) flag = Guild.WhitelistFlags.get(flag) if not flag: print 'Invalid flag' return try: guild = Guild.get(guild_id=guild_id) except Guild.DoesNotExist: print 'No guild exists with that id' return if not guild.is_whitelisted(flag): print 'This guild doesn\'t have this flag' return guild.whitelist.remove(int(flag)) guild.save() guild.emit('GUILD_UPDATE') print 'Removed flag'
def add_whitelist(guild_id, flag): from rowboat.models.guild import Guild init_db(ENV) flag = Guild.WhitelistFlags.get(flag) if not flag: print 'Invalid flag' return try: guild = Guild.get(guild_id=guild_id) except Guild.DoesNotExist: print 'No guild exists with that id' return if guild.is_whitelisted(flag): print 'This guild already has this flag' return guild.whitelist.append(int(flag)) guild.save() guild.emit('GUILD_UPDATE') print 'Added flag'