from django.shortcuts import render
from rest_framework import viewsets
from .models import Company
from .form import CompanyForm
from .serializers import CompanySerializer
# Create your views here.
class CompanyViewSet(viewsets.ModelViewSet):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer

    def perform_create(self, serializer):
        form_data = self.request.data.get('form')
        form = CompanyForm(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 = CompanyForm(form_data, instance=serializer.instance)
        if form.is_valid():
            employee = serializer.save()
            form.instance = employee
            form.save()