Added extra check

This commit is contained in:
Raynaldo Rivera
2014-10-12 17:19:12 -07:00
parent 392c752eb8
commit b869c465bb
3 changed files with 17 additions and 7 deletions

View File

@@ -22,4 +22,11 @@ def remove_member_permission(user, permission):
user = User.objects.get(username=user.username)
if user.has_perm(permission):
user.user_permissions.remove(stored_permission)
user.save()
user.save()
def check_if_user_has_permission(user, permission):
ct = ContentType.objects.get_for_model(User)
stored_permission, created = Permission.objects.get_or_create(codename=permission,
content_type=ct, name=permission)
return user.has_perm(stored_permission)