from rest_framework.permissions import BasePermission
from .models import UserInformation

class IsAdmin(BasePermission):
    def has_permission(self, request, view):
        user_information = UserInformation.objects.filter(user=request.user.id).first()
        if user_information and user_information.role == "Admin":
            return True
        return False
        

class IsOrgAdmin(BasePermission):
    def has_permission(self, request, view):
        user_information = UserInformation.objects.filter(user=request.user.id).first()
        if user_information and user_information.role == "OrgAdmin":
            return True
        return False
                

class IsInstructor(BasePermission):
    def has_permission(self, request, view):
        user_information = UserInformation.objects.filter(user=request.user.id).first()
        if user_information and user_information.role == "Instructor":
            return True
        return False
                        

class IsStudent(BasePermission):        
    def has_permission(self, request, view):
        user_information = UserInformation.objects.filter(user=request.user.id).first()
        if user_information and user_information.role == "Student":
            return True
        return False
                                

class IsGuardian(BasePermission):
    def has_permission(self, request, view):
        user_information = UserInformation.objects.filter(user=request.user.id).first()
        if user_information and user_information.role == "Guardian":
            return True
        return False