mirror of
https://github.com/lemeow125/DocManagerBackend.git
synced 2025-01-19 01:23:02 +08:00
31 lines
773 B
Python
31 lines
773 B
Python
from rest_framework.permissions import BasePermission
|
|
|
|
|
|
class IsStaff(BasePermission):
|
|
"""
|
|
Allows access only to users with staff role
|
|
"""
|
|
|
|
def has_permission(self, request, view):
|
|
return bool(
|
|
request.user and request.user.role in (
|
|
"head", "admin", "planning", "staff")
|
|
)
|
|
|
|
|
|
class IsPlanning(BasePermission):
|
|
"""
|
|
Allows access only to users with head, admin planning role
|
|
"""
|
|
|
|
def has_permission(self, request, view):
|
|
return bool(request.user and request.user.role == "planning")
|
|
|
|
|
|
class IsHead(BasePermission):
|
|
"""
|
|
Allows access only to users with staff role
|
|
"""
|
|
|
|
def has_permission(self, request, view):
|
|
return bool(request.user and request.user.role == "head")
|