from rest_framework.permissions import BasePermission
from .models import UserRoles

class IsAdmin(BasePermission):
    """Allow only Admin users"""
    def has_permission(self, request, view):
        return request.user.is_authenticated and request.user.role == UserRoles.ADMIN

class IsStaff(BasePermission):
    """Allow only Staff users"""
    def has_permission(self, request, view):
        return request.user.is_authenticated and request.user.role == UserRoles.STAFF