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()