57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
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 |