from rest_framework import serializers
from common.serializers import BankDetailsSerializer, AddressSerializer
from .models import Employee,Role
from common.models import BankDetails, Address


class EmployeeSerializer(serializers.ModelSerializer):
    bank_details = BankDetailsSerializer()
    address = AddressSerializer()

    class Meta:
        model = Employee
        fields = '__all__'

    def create(self, validated_data):
        bank_details_data = validated_data.pop('bank_details')
        address_data = validated_data.pop('address')

        bank_details = BankDetails.objects.create(**bank_details_data)
        address = Address.objects.create(**address_data)

        employee = Employee.objects.create(bank_details=bank_details, address=address, **validated_data)
        return employee


class RoleSerializer(serializers.ModelSerializer):
    class Meta:
        model = Role
        fields = '__all__'