Final changes to support permissions and roles
This commit is contained in:
70
apps/permissions/api.py
Normal file
70
apps/permissions/api.py
Normal file
@@ -0,0 +1,70 @@
|
||||
from django.http import Http404
|
||||
from django.contrib.auth.models import User
|
||||
from django.contrib.auth.models import Group
|
||||
from django.db.utils import DatabaseError
|
||||
from django.shortcuts import get_object_or_404
|
||||
from django.contrib.contenttypes.models import ContentType
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
from django.utils.translation import ugettext
|
||||
|
||||
from models import Permission, Role
|
||||
|
||||
class Unauthorized(Exception):
|
||||
pass
|
||||
|
||||
|
||||
def register_permissions(namespace, permissions):
|
||||
if permissions:
|
||||
for permission in permissions:
|
||||
try:
|
||||
permission_obj, created = Permission.objects.get_or_create(
|
||||
namespace=namespace, name=permission['name'])
|
||||
permission_obj.label=unicode(permission['label'])
|
||||
permission_obj.save()
|
||||
except DatabaseError:
|
||||
#Special case for ./manage.py syncdb
|
||||
pass
|
||||
|
||||
#TODO: Handle anonymous users
|
||||
def check_permissions(requester, namespace, permission_list):
|
||||
if isinstance(requester, User):
|
||||
if requester.is_superuser:
|
||||
return True
|
||||
|
||||
for permission_item in permission_list:
|
||||
permission = get_object_or_404(Permission,
|
||||
namespace=namespace, name=permission_item)
|
||||
if check_permission(requester, permission):
|
||||
return True
|
||||
|
||||
raise Unauthorized(ugettext(u'Insufficient permissions.'))
|
||||
|
||||
|
||||
def check_permission(requester, permission):
|
||||
for permission_holder in permission.permissionholder_set.all():
|
||||
if check_requester(requester, permission_holder):
|
||||
return True
|
||||
|
||||
|
||||
def check_requester(requester, permission_holder):
|
||||
ct = ContentType.objects.get_for_model(requester)
|
||||
if permission_holder.holder_type == ct and permission_holder.holder_id == requester.id:
|
||||
return True
|
||||
|
||||
if isinstance(permission_holder.holder_object, Role):
|
||||
requester_list = [role_member.member_object for role_member in permission_holder.holder_object.rolemember_set.all()]
|
||||
if check_elements(requester, requester_list):
|
||||
return True
|
||||
|
||||
#Untested
|
||||
if isinstance(permission_holder.holder_object, Group):
|
||||
if check_elements(requester, permission_holder.holder_object.user_set.all()):
|
||||
return True
|
||||
|
||||
|
||||
#TODO: a role may contain groups, make recursive
|
||||
def check_elements(requester, requester_list):
|
||||
ct = ContentType.objects.get_for_model(requester)
|
||||
for requester_object in requester_list:
|
||||
if requester == requester_object:
|
||||
return True
|
||||
Reference in New Issue
Block a user