from rest_framework import viewsets, permissions, authentication from rest_framework.response import Response from jobposting.models import JobOffer, CompanyLogo, Skill from core.models import MyUser, AnonymousUserData from rest_framework import status from oauth2_provider.contrib.rest_framework import TokenHasReadWriteScope, OAuth2Authentication from .auth import OAuth2ClientCredentialAuthentication from rest_framework.views import APIView from django.conf import settings from jobposting.serializers import ( JobOfferSerializer, MyUserSerializer, CompanyLogoSerializer, SkillSerializer, SlimJobOffersSerializer ) class MyUserViewSet(viewsets.ModelViewSet): # authentication_classes = [OAuth2Authentication, authentication.SessionAuthentication] # permission_classes = [permissions.IsAuthenticated|TokenHasReadWriteScope] queryset = MyUser.objects.all() serializer_class = MyUserSerializer required_scope = ['main'] authentication_classes = [] ### test only - to be changed permission_classes = [] ### # # permission_classes = [permissions.IsAuthenticated] class JobOfferViewSet(viewsets.ModelViewSet): # authentication_classes = [OAuth2Authentication, authentication.SessionAuthentication] # permission_classes = [permissions.IsAuthenticated|TokenHasReadWriteScope] # test only queryset = JobOffer.objects.all() serializer_class = JobOfferSerializer required_scope = ['main'] authentication_classes = [] permission_classes = [] # # 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 CompanyLogoViewSet(viewsets.ModelViewSet): # authentication_classes = [OAuth2Authentication, authentication.SessionAuthentication] # permission_classes = [permissions.IsAuthenticated|TokenHasReadWriteScope] queryset = CompanyLogo.objects.all() serializer_class = CompanyLogoSerializer required_scope = ['main'] authentication_classes = [] permission_classes = [] class JobOfferListView(APIView): authentication_classes = [] permission_classes = [] def get(self, request, format=None): start = request.query_params.get('start', 0) end = request.query_params.get('end', 12) job_offers = JobOffer.objects.all()[start:end] serializer = SlimJobOffersSerializer(job_offers, many=True) return Response(serializer.data) class SkillViewset(viewsets.ModelViewSet): # authentication_classes = [OAuth2Authentication, authentication.SessionAuthentication] # permission_classes = [permissions.IsAuthenticated|TokenHasReadWriteScope] queryset = Skill.objects.all() serializer_class = SkillSerializer required_scope = ['main'] authentication_classes = [] permission_classes = [] class SkillMainView(APIView): authentication_classes = [] permission_classes = [] def get_object(self, pk): try: return Skill.objects.get(pk=pk) except Skill.DoesNotExist: return Response({'error': 'Not Found'}, status=status.HTTP_404_NOT_FOUND) def get(self, request, pk=None, format=None): if pk: skill = self.get_object(pk) serializer = SkillSerializer(skill) else: skill = Skill.objects.all() serializer = SkillSerializer(skill, many=True) return Response(serializer.data) def post(self, request, format=None): serializer = SkillSerializer(data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) def put(self, request, pk, format=None): skill = self.get_object(pk) serializer = SkillSerializer(skill, data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) def delete(self, request, pk, format=None): skill = self.get_object(pk) skill.delete() return Response(status=status.HTTP_204_NO_CONTENT) class JobOfferDetailView(APIView): authentication_classes = [] permission_classes = [] def get_object(self, pk): try: return JobOffer.objects.get(pk=pk) except JobOffer.DoesNotExist: return Response({'error': 'Not Found'}, status=status.HTTP_404_NOT_FOUND) def get(self, request, pk, format=None): job_listing = self.get_object(pk) serializer = JobOfferSerializer(job_listing,) return Response(serializer.data) def put(self, request, pk, format=None): job_offer = self.get_object(pk) serializer = JobOfferSerializer(job_offer, data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) def post(self, request, format=None): serializer = JobOfferSerializer(data=request.data) if serializer.is_valid(): if request.user.is_authenticated: serializer.save(created_by=request.user) else: # Pobieranie danych z formularza first_name = request.data.get('first_name') last_name = request.data.get('last_name') contact_email = 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) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) def delete(self, request, pk, format=None): job_listing = self.get_object(pk) job_listing.delete() return Response(status=status.HTTP_204_NO_CONTENT)