Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add a parameter to knight.py to remove user rights. 

(imported from commit 4527955f688298ba56296e9f24d42e33f7d04948)
This commit is contained in:
Luke Faraone 2013-06-24 12:57:40 -07:00
parent f22abf2b8a
commit 2daf8db3a5
1 changed files with 19 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
zephyr/management/commands/knight.py
View File

@ -8,7 +8,7 @@ from django.core.exceptions import ValidationError
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from django.core import validators from django.core import validators
from guardian.shortcuts import assign_perm from guardian.shortcuts import assign_perm, remove_perm
from zephyr.models import Realm, UserProfile from zephyr.models import Realm, UserProfile
@ -19,6 +19,12 @@ ONLY perform this on customer request from an authorized person.
""" """
option_list = BaseCommand.option_list + ( option_list = BaseCommand.option_list + (
make_option('--revoke',
dest='grant',
action="store_false",
default=True,
help='Remove an administrator\'s rights.'),
)
def handle(self, *args, **options): def handle(self, *args, **options):
try: try:
@ -29,8 +35,16 @@ ONLY perform this on customer request from an authorized person.
profile = UserProfile.objects.get(email=email) profile = UserProfile.objects.get(email=email)
except ValidationError: except ValidationError:
raise CommandError("No such user.") raise CommandError("No such user.")
if profile.has_perm('administer', profile.realm):
raise CommandError("User already has permission for this realm.") if options['grant']:
if profile.has_perm('administer', profile.realm):
raise CommandError("User already has permission for this realm.")
else:
assign_perm('administer', profile, profile.realm)
print "Done!"
else: else:
assign_perm('administer', profile, profile.realm) if profile.has_perm('administer', profile.realm):
print "Done!" remove_perm('administer', profile, profile.realm)
print "Done!"
else:
raise CommandError("User did not have permission for this realm!")