from rest_framework import viewsets, permissions
from jobposting.models import JobListing, SkillLevels, CompanyLogo, Skill
from core.models import MyUser, AnonymousUserData

from jobposting.serializers import (
    JobListingSerializer,
    SkillLevelsSerializer,
    MyUserSerializer,
    CompanyLogoSerializer,
    SkillSerializer
)

class MyUserViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.AllowAny] # test only
    queryset = MyUser.objects.all()
    serializer_class = MyUserSerializer
    # permission_classes = [permissions.IsAuthenticated]

class JobListingViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.AllowAny] # test only
    queryset = JobListing.objects.all()
    serializer_class = JobListingSerializer
    # permission_classes = [permissions.IsAuthenticatedOrReadOnly]
    
    def perform_create(self, serializer):
        if self.request.user.is_authenticated:
            serializer.save(created_by=self.request.user)
        else:
            # Pobieranie danych z formularza
            first_name = self.request.data.get('first_name')
            last_name = self.request.data.get('last_name')
            contact_email = self.request.data.get('contact_email')

            # Tworzenie rekordu AnonymousUserData
            anonymous_user_data = AnonymousUserData.objects.create(
                first_name=first_name, 
                last_name=last_name, 
                contact_email=contact_email
            )

            serializer.save(anonymous_user_data=anonymous_user_data)

class SkillLevelsViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.AllowAny] # test only
    queryset = SkillLevels.objects.all()
    serializer_class = SkillLevelsSerializer
    # permission_classes = [permissions.IsAuthenticatedOrReadOnly]

class CompanyLogoViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.AllowAny]
    queryset = CompanyLogo.objects.all()
    serializer_class = CompanyLogoSerializer

class SkillViewset(viewsets.ModelViewSet):
    permission_classes = [permissions.AllowAny]
    queryset = Skill.objects.all()
    serializer_class = SkillSerializer