# usuarios/serializers.py
from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
from rest_framework import serializers
from .models import CustomUser

class CustomTokenObtainPairSerializer(TokenObtainPairSerializer):
    @classmethod
    def get_token(cls, user):
        token = super().get_token(user)

        # Puedes agregar datos extra al token si quieres
        token['tipo_usuario'] = user.tipo_usuario
        return token

    

class UserProfileSerializer(serializers.ModelSerializer):
    
    class Meta:
        model = CustomUser
        fields = ['id', 'username', 'email', 'first_name', 'last_name', 'tipo_usuario']


from rest_framework import serializers
from .models import CustomUser

class UserSerializer(serializers.ModelSerializer):
    # Declara 'nombre_completo' como un SerializerMethodField
    nombre = serializers.SerializerMethodField()

    class Meta:
        model = CustomUser
        fields = ['id', 'nombre']

    # Método para obtener el nombre completo
    def get_nombre(self, obj):
        return f"{obj.first_name} {obj.last_name}"