31 lines
1 KiB
Python
31 lines
1 KiB
Python
"""API Decorators"""
|
|
from functools import wraps
|
|
from typing import Callable, Optional
|
|
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
|
|
def permission_required(perm: Optional[str] = None, other_perms: Optional[list[str]] = None):
|
|
"""Check permissions for a single custom action"""
|
|
|
|
def wrapper_outter(func: Callable):
|
|
"""Check permissions for a single custom action"""
|
|
|
|
@wraps(func)
|
|
def wrapper(self: ModelViewSet, request: Request, *args, **kwargs) -> Response:
|
|
if perm:
|
|
obj = self.get_object()
|
|
if not request.user.has_perm(perm, obj):
|
|
return self.permission_denied(request)
|
|
if other_perms:
|
|
for other_perm in other_perms:
|
|
if not request.user.has_perm(other_perm):
|
|
return self.permission_denied(request)
|
|
return func(self, request, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
return wrapper_outter
|