25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.

26 lines
823 B

  1. from rest_framework import viewsets
  2. from .models import Employee
  3. from .serializers import EmployeeSerializer
  4. from .forms import EmployeeForm
  5. class EmployeeViewSet(viewsets.ModelViewSet):
  6. queryset = Employee.objects.all()
  7. serializer_class = EmployeeSerializer
  8. def perform_create(self, serializer):
  9. form_data = self.request.data.get('form')
  10. form = EmployeeForm(form_data)
  11. if form.is_valid():
  12. employee = serializer.save()
  13. form.instance = employee
  14. form.save()
  15. def perform_update(self, serializer):
  16. form_data = self.request.data.get('form')
  17. form = EmployeeForm(form_data, instance=serializer.instance)
  18. if form.is_valid():
  19. employee = serializer.save()
  20. form.instance = employee
  21. form.save()