feat: new templatetag to check if user has role
This commit is contained in:
parent
4d82c76fbd
commit
af773a0afd
|
@ -5,6 +5,7 @@ from django import template
|
||||||
from django.utils.html import mark_safe
|
from django.utils.html import mark_safe
|
||||||
|
|
||||||
from ..sites import site
|
from ..sites import site
|
||||||
|
from ..models import UserRole
|
||||||
|
|
||||||
register = template.Library()
|
register = template.Library()
|
||||||
|
|
||||||
|
@ -59,3 +60,10 @@ def pagination(request, page):
|
||||||
@register.filter
|
@register.filter
|
||||||
def to_json(data):
|
def to_json(data):
|
||||||
return mark_safe(json.dumps(data))
|
return mark_safe(json.dumps(data))
|
||||||
|
|
||||||
|
|
||||||
|
@register.filter
|
||||||
|
def has_role(user, role):
|
||||||
|
if user.is_superuser:
|
||||||
|
return True
|
||||||
|
return UserRole.objects.filter(role__in=(role, "admins"), user=user).exists()
|
||||||
|
|
Loading…
Reference in New Issue