|
12345678910111213141516171819202122232425 |
- from rest_framework import viewsets
- from .models import Employee
- from .serializers import EmployeeSerializer
- from .forms import EmployeeForm
-
- class EmployeeViewSet(viewsets.ModelViewSet):
- queryset = Employee.objects.all()
- serializer_class = EmployeeSerializer
-
- def perform_create(self, serializer):
- form_data = self.request.data.get('form')
- form = EmployeeForm(form_data)
- if form.is_valid():
- employee = serializer.save()
- form.instance = employee
- form.save()
-
- def perform_update(self, serializer):
- form_data = self.request.data.get('form')
- form = EmployeeForm(form_data, instance=serializer.instance)
- if form.is_valid():
- employee = serializer.save()
- form.instance = employee
- form.save()
-
|