notnull/banned
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Reuse same code for is_admin/is_owner

Browse Source
This commit is contained in:
nyuszika7h 2015-07-19 21:22:04 +02:00
parent c13e3087d6
commit 66a8826eb1
1 changed files with 27 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/settings.py
View File

@ -281,43 +281,40 @@ PING_IF_NUMS_ACCS = {}
is_role = lambda plyr, rol: rol in ROLES and plyr in ROLES[rol]
def is_admin(nick, cloak=None, acc=None):
if nick in USERS.keys():
if not cloak:
cloak = USERS[nick]["cloak"]
if not acc:
acc = USERS[nick]["account"]
def check_priv(priv):
assert priv in ("owner", "admin")
if acc and acc != "*":
for pattern in set(botconfig.OWNERS_ACCOUNTS + botconfig.ADMINS_ACCOUNTS):
if fnmatch.fnmatch(acc.lower(), pattern.lower()):
return True
# Owners can do everything
hosts = set(botconfig.OWNERS)
accounts = set(botconfig.OWNERS_ACCOUNTS)
if cloak:
for pattern in set(botconfig.OWNERS + botconfig.ADMINS):
if fnmatch.fnmatch(cloak.lower(), pattern.lower()):
return True
if priv == "admin":
hosts.update(botconfig.ADMINS)
accounts.update(botconfig.ADMINS_ACCOUNTS)
return False
def do_check(nick, cloak=None, acc=None):
if nick in USERS.keys():
if not cloak:
cloak = USERS[nick]["cloak"]
if not acc:
acc = USERS[nick]["account"]
def is_owner(nick, cloak=None, acc=None):
if nick in USERS.keys():
if not cloak:
cloak = USERS[nick]["cloak"]
if not acc:
acc = USERS[nick]["account"]
if acc and acc != "*":
for pattern in set(accounts):
if fnmatch.fnmatch(acc.lower(), pattern.lower()):
return True
if acc and acc != "*":
for pattern in botconfig.OWNERS_ACCOUNTS:
if fnmatch.fnmatch(acc.lower(), pattern.lower()):
return True
if cloak:
for pattern in set(hosts):
if fnmatch.fnmatch(cloak.lower(), pattern.lower()):
return True
if cloak:
for pattern in botconfig.OWNERS:
if fnmatch.fnmatch(cloak.lower(), pattern.lower()):
return True
return False
return False
return do_check
is_admin = check_priv("admin")
is_owner = check_priv("owner")
def plural(role):
bits = role.split()