from rest_framework.exceptions import PermissionDenied
from rest_framework.permissions import BasePermission

from core.apps.accounts.choices import RoleChoice


class IsAdminRole(BasePermission):
    def has_permission(self, request, view):
        if request.user and request.user.role in [RoleChoice.ADMIN, RoleChoice.SUPERUSER]:
            return True
        return False