Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Add !givepermission and !removepermission to permissions.py

Browse source
This commit is contained in:
jesopo 2018-08-28 16:53:21 +01:00
parent b47bf49830
commit b9d4e11b77
1 changed files with 50 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
modules/permissions.py
View file

@ -14,11 +14,15 @@ class Module(object):
bot.events.on("received").on("command").on("register" bot.events.on("received").on("command").on("register"
).hook(self.register, private_only=True, min_args=1, ).hook(self.register, private_only=True, min_args=1,
usage="<password>", help="Register your nickname") usage="<password>", help="Register your nickname")
bot.events.on("received").on("command").on("logout" bot.events.on("received.command.logout").hook(self.logout,
).hook(self.logout, private_only=True, private_only=True, help="Sign out from the bot")
help="Sign out from the bot")
bot.events.on("received").on("command").on("mypermissions" bot.events.on("received.command.mypermissions").hook(
).hook(self.my_permissions, private_only=True) self.my_permissions, authenticated=True)
bot.events.on("received.command.givepermission").hook(
self.give_permission, min_args=2, permission="givepermission")
bot.events.on("received.command.removepermission").hook(
self.remove_permission, min_args=2, permission="removepermission")
def new_user(self, event): def new_user(self, event):
self._logout(event["user"]) self._logout(event["user"])
@ -104,3 +108,44 @@ class Module(object):
def my_permissions(self, event): def my_permissions(self, event):
permissions = event["user"].get_setting("permissions", []) permissions = event["user"].get_setting("permissions", [])
event["stdout"].write("Your permissions: %s" % ", ".join(permissions)) event["stdout"].write("Your permissions: %s" % ", ".join(permissions))
def _get_user_details(self, server, nickname):
target = server.get_user(nickname)
registered = bool(target.get_setting("authentication", None))
permissions = target.get_setting("permissions", [])
return [target, registered, permissions]
def give_permission(self, event):
permission = event["args_split"][1].lower()
target, registered, permissions = self._get_user_details(
event["server"], event["args_split"][0])
if not registered:
event["stderr"].write("%s isn't registered" % target.nickname)
return
if permission in permissions:
event["stderr"].write("%s already has permission '%s'" % (
target.nickname, permission))
else:
permissions.append(permission)
target.set_setting("permissions", permissions)
event["stdout"].write("Gave permission '%s' to %s" % (
permission, target.nickname))
def remove_permission(self, event):
permission = event["args_split"][1].lower()
target, registered, permissions = self._get_user_details(
event["server"], event["args_split"][0])
if not registered:
event["stderr"].write("%s isn't registered" % target.nickname)
return
if not permission in permissions:
event["stderr"].write("%s already has permission '%s'" % (
target.nickname, permission))
else:
permissions.remove(permission)
target.set_setting("permissions", permissions)
event["stdout"].write("Removed permission '%s' from %s" % (
permission, target.nickname))