68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
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 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]
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
data = request.data
|
|
if isinstance(data, list): # Sprawdzenie, czy dane to lista
|
|
serializer = self.get_serializer(data=data, many=True) # Ustawienie many=True
|
|
serializer.is_valid(raise_exception=True)
|
|
self.perform_create(serializer)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
else:
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
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 |