from rest_framework import viewsets, permissions from rest_framework.response import Response from jobposting.models import JobListing, SkillLevels, CompanyLogo, Skill from core.models import MyUser, AnonymousUserData from rest_framework import status from jobposting.permissions import ClientCredentialPermission from oauth2_provider.contrib.rest_framework import TokenHasReadWriteScope, OAuth2Authentication from jobposting.auth import OAuth2ClientCredentialAuthentication from jobposting.serializers import ( JobListingSerializer, SkillLevelsSerializer, MyUserSerializer, CompanyLogoSerializer, SkillSerializer ) class MyUserViewSet(viewsets.ModelViewSet): authentication_classes = [OAuth2ClientCredentialAuthentication] # permission_classes = [TokenHasReadWriteScope] # test only queryset = MyUser.objects.all() serializer_class = MyUserSerializer # required_scope = ['main'] # permission_classes = [permissions.IsAuthenticated] class JobListingViewSet(viewsets.ModelViewSet): authentication_classes = [OAuth2ClientCredentialAuthentication] # permission_classes = [TokenHasReadWriteScope] # test only queryset = JobListing.objects.all() serializer_class = JobListingSerializer required_scope = ['main'] # 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): authentication_classes = [OAuth2ClientCredentialAuthentication] # permission_classes = [TokenHasReadWriteScope] # test only queryset = SkillLevels.objects.all() serializer_class = SkillLevelsSerializer # required_scope = ['main'] # permission_classes = [permissions.IsAuthenticatedOrReadOnly] def create(self, request, *args, **kwargs): data = request.data print(type(data)) if isinstance(data, list): # Sprawdzenie, czy dane to lista serializers = [self.get_serializer(data=item) for item in data] for serializer in serializers: serializer.is_valid(raise_exception=True) self.perform_create(serializer) return Response([serializer.data for serializer in serializers], status=status.HTTP_201_CREATED) else: return super().create(request, *args, **kwargs) class CompanyLogoViewSet(viewsets.ModelViewSet): authentication_classes = [OAuth2ClientCredentialAuthentication] # permission_classes = [TokenHasReadWriteScope] queryset = CompanyLogo.objects.all() serializer_class = CompanyLogoSerializer # required_scope = ['main'] class SkillViewset(viewsets.ModelViewSet): authentication_classes = [OAuth2ClientCredentialAuthentication] # permission_classes = [TokenHasReadWriteScope] queryset = Skill.objects.all() serializer_class = SkillSerializer # required_scope = ['main']