from rest_framework import serializers
from .models import CustomUser
from django.contrib.auth.models import Group
from api.customs.models import Importador

class CustomUserSerializer(serializers.ModelSerializer):
    """
    Serializer for the CustomUser model.
    """

    password = serializers.CharField(write_only=True, required=False)
    groups = serializers.PrimaryKeyRelatedField(queryset=Group.objects.all(), many=True, required=False)
    rfc = serializers.PrimaryKeyRelatedField(
        queryset=Importador.objects.all(),
        many=True,
        required=False,
        pk_field=serializers.CharField(),
    )

    class Meta:
        model = CustomUser
        fields = ['id', 'username', 'email', 'first_name', 'last_name', 'password', 'profile_picture', 'organizacion', 'is_importador', 'rfc', 'is_active', 'is_superuser', 'groups']
        read_only_fields = ['id', 'organizacion', 'is_superuser']

    def validate_password(self, value):
        if not value or not value.strip():
            raise serializers.ValidationError("La contraseña no puede estar vacía o contener solo espacios.")
        return value

    def validate(self, attrs):
        # En create, la contraseña es obligatoria
        if self.instance is None and not attrs.get('password'):
            raise serializers.ValidationError({"password": "Este campo es requerido."})
        return attrs

    def create(self, validated_data):
        groups = validated_data.pop('groups', [])
        rfcs = validated_data.pop('rfc', [])
        password = validated_data.pop('password')
        user = CustomUser(**validated_data)
        user.set_password(password)
        user.save()
        if groups:
            user.groups.set(groups)
        if rfcs:
            user.rfc.set(rfcs)
        return user

    def update(self, instance, validated_data):
        groups = validated_data.pop('groups', None)
        rfcs = validated_data.pop('rfc', None)
        password = validated_data.pop('password', None)
        for attr, value in validated_data.items():
            setattr(instance, attr, value)
        if password:
            instance.set_password(password)
        instance.save()
        if groups is not None:
            instance.groups.set(groups)
        if rfcs is not None:
            instance.rfc.set(rfcs)
        return instance