Example #1
0
 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
Example #2
0
    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))
Example #3
0
    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))
Example #4
0
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'
Example #5
0
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')
Example #6
0
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'
Example #7
0
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'