._.
This commit is contained in:
parent
28dd4cf5f7
commit
e6d669be2b
2 changed files with 41 additions and 80 deletions
|
@ -1,6 +1,7 @@
|
|||
import re
|
||||
|
||||
from util import hook
|
||||
from util import hook, text
|
||||
from fnmatch import fnmatch
|
||||
|
||||
|
||||
@hook.sieve
|
||||
|
@ -29,11 +30,46 @@ def sieve_suite(bot, input, func, kind, args):
|
|||
if input.chan.lower() in denied_channels:
|
||||
return None
|
||||
|
||||
# shim so plugins using the old "adminonly" permissions format still work
|
||||
if args.get('adminonly', False):
|
||||
admins = bot.config.get('admins', [])
|
||||
args['perms'] = ["adminonly"]
|
||||
|
||||
if input.nick not in admins and input.mask not in admins:
|
||||
input.notice("Sorry, you are not allowed to use this command.")
|
||||
return None
|
||||
|
||||
if args.get('permissions', False):
|
||||
groups = bot.config.get("permission_groups", [])
|
||||
group_users = bot.config.get("permission_users", [])
|
||||
|
||||
allowed_permissions = args.get('permissions', [])
|
||||
|
||||
|
||||
allowed_groups = []
|
||||
|
||||
# loop over every group
|
||||
for name, permissions in groups.iteritems():
|
||||
# loop over every permission the command allows
|
||||
for permission in allowed_permissions:
|
||||
# see if the group has that permission
|
||||
if permission in permissions:
|
||||
# if so, add the group name to the allowed_groups list
|
||||
allowed_groups.append(name)
|
||||
|
||||
|
||||
if not allowed_groups:
|
||||
print "Something is wrong. A hook requires {} but" \
|
||||
" there are no groups with that permission!".format(str(allowed_permissions))
|
||||
|
||||
mask = input.mask.lower()
|
||||
|
||||
# make all masks lowercase
|
||||
for group, masks in group_users.iteritems():
|
||||
group_users[group] = [_mask.lower() for _mask in masks]
|
||||
|
||||
for group in allowed_groups:
|
||||
for pattern in group_users[group]:
|
||||
if fnmatch(mask, pattern):
|
||||
return input
|
||||
|
||||
input.notice("Sorry, you are not allowed to use this command.")
|
||||
return None
|
||||
|
||||
return input
|
||||
|
|
|
@ -1,75 +0,0 @@
|
|||
import re
|
||||
|
||||
from util import hook, text
|
||||
from fnmatch import fnmatch
|
||||
|
||||
|
||||
@hook.sieve
|
||||
def sieve_suite(bot, input, func, kind, args):
|
||||
if input.command == 'PRIVMSG' and\
|
||||
input.nick.endswith('bot') and args.get('ignorebots', True):
|
||||
return None
|
||||
|
||||
if kind == "command":
|
||||
if input.trigger in bot.config.get('disabled_commands', []):
|
||||
return None
|
||||
|
||||
fn = re.match(r'^plugins.(.+).py$', func._filename)
|
||||
disabled = bot.config.get('disabled_plugins', [])
|
||||
if fn and fn.group(1).lower() in disabled:
|
||||
return None
|
||||
|
||||
acl = bot.config.get('acls', {}).get(func.__name__)
|
||||
if acl:
|
||||
if 'deny-except' in acl:
|
||||
allowed_channels = map(unicode.lower, acl['deny-except'])
|
||||
if input.chan.lower() not in allowed_channels:
|
||||
return None
|
||||
if 'allow-except' in acl:
|
||||
denied_channels = map(unicode.lower, acl['allow-except'])
|
||||
if input.chan.lower() in denied_channels:
|
||||
return None
|
||||
|
||||
# shim so plugins using the old "adminonly" permissions format still work
|
||||
if args.get('adminonly', False):
|
||||
args['perms'] = ["adminonly"]
|
||||
|
||||
|
||||
if args.get('permissions', False):
|
||||
groups = bot.config.get("permission_groups", [])
|
||||
group_users = bot.config.get("permission_users", [])
|
||||
|
||||
allowed_permissions = args.get('permissions', [])
|
||||
|
||||
|
||||
allowed_groups = []
|
||||
|
||||
# loop over every group
|
||||
for name, permissions in groups.iteritems():
|
||||
# loop over every permission the command allows
|
||||
for permission in allowed_permissions:
|
||||
# see if the group has that permission
|
||||
if permission in permissions:
|
||||
# if so, add the group name to the allowed_groups list
|
||||
allowed_groups.append(name)
|
||||
|
||||
|
||||
if not allowed_groups:
|
||||
print "Something is wrong. A hook requires {} but" \
|
||||
" there are no groups with that permission!".format(str(allowed_permissions))
|
||||
|
||||
mask = input.mask.lower()
|
||||
|
||||
# make all masks lowercase
|
||||
for group, masks in group_users.iteritems():
|
||||
group_users[group] = [_mask.lower() for _mask in masks]
|
||||
|
||||
for group in allowed_groups:
|
||||
for pattern in group_users[group]:
|
||||
if fnmatch(mask, pattern):
|
||||
return input
|
||||
|
||||
input.notice("Sorry, you are not allowed to use this command.")
|
||||
return None
|
||||
|
||||
return input
|
Reference in a new issue