Moved PermissionManager and RoleMemberManager to a separate manager.py file

This commit is contained in:
Roberto Rosario
2011-05-24 23:39:10 -04:00
parent 3fd7a3181e
commit 581f8cd52d
2 changed files with 15 additions and 11 deletions

View File

@@ -0,0 +1,14 @@
from django.db import models
from django.contrib.contenttypes.models import ContentType
class RoleMemberManager(models.Manager):
def get_roles_for_member(self, member_obj):
member_type = ContentType.objects.get_for_model(member_obj)
return [role_member.role for role_member in self.model.objects.filter(member_type=member_type, member_id=member_obj.pk)]
class PermissionManager(models.Manager):
def get_for_holder(self, holder):
ct = ContentType.objects.get_for_model(holder)
return self.model.objects.filter(permissionholder__holder_type=ct).filter(permissionholder__holder_id=holder.pk)