from rest_framework import viewsets
from rest_framework.authentication import SessionAuthentication
from rest_framework.permissions import IsAuthenticated
from rest_framework_simplejwt.authentication import JWTAuthentication

from .models import CustomUser, EmailAddress
from .permissions import IsCustomPerrmission
from .serializers import CustomUserSerializer, EmailAddressSerializer


class CustomUserViewSet(viewsets.ModelViewSet):
    queryset = CustomUser.objects.all()
    serializer_class = CustomUserSerializer
    authentication_classes = [JWTAuthentication, SessionAuthentication]
    permission_classes = [IsCustomPerrmission]


class EmailAddressViewSet(viewsets.ModelViewSet):
    queryset = EmailAddress.objects.all()
    serializer_class = EmailAddressSerializer
    authentication_classes = [JWTAuthentication, SessionAuthentication]
    permission_classes = [IsCustomPerrmission]