add granted_projects function, simplify middleware

This commit is contained in:
Élie Bouttier 2014-09-19 17:40:06 -07:00
parent a889c3f0ad
commit 10dc2d499c
2 changed files with 48 additions and 45 deletions

View file

@ -1,7 +1,52 @@
from django.utils.safestring import mark_safe
from django.db.models import Q
from markdown import markdown
from tracker.models import Project
from permissions.models import GlobalPermission
from permissions.models import PermissionModel as PermModel
def granted_projects(user):
if user.is_authenticated() and user.is_staff:
return Project.objects.all()
elif user.is_authenticated():
teams = user.teams.values_list('id')
groups = user.groups.values_list('id')
# check for a global permission allowing access
if GlobalPermission.objects.filter(access_project=True) \
.filter(
# directly
Q(grantee_type=PermModel.GRANTEE_USER,
grantee_id=user.id)
# through a group
| Q(grantee_type=PermModel.GRANTEE_GROUP,
grantee_id__in=groups)
# through a team
| Q(grantee_type=PermModel.GRANTEE_TEAM,
grantee_id__in=teams)
).exists():
return Project.objects.all()
# searching project reachable throught project permission
else:
# public project
query = Q(access=Project.ACCESS_PUBLIC)
# project reserved to logged users
query |= Q(access=Project.ACCESS_REGISTERED)
# access granted through a team
query |= Q(permissions__grantee_type=PermModel.GRANTEE_TEAM,
permissions__grantee_id__in=teams)
# access granted through a group
query |= Q(permissions__grantee_type=PermModel.GRANTEE_GROUP,
permissions__grantee_id__in=groups)
# access granted by specific permission
query |= Q(permissions__grantee_type=PermModel.GRANTEE_USER,
permissions__grantee_id=user.id)
return Project.objects.filter(query).distinct()
else:
# only public projects
return Project.objects.filter(access=Project.ACCESS_PUBLIC)
def markdown_to_html(value):
# set extensions here if needed