from rest_framework import serializers
from .models import User, Partner, PartnerAdmin

class UserSerializer(serializers.ModelSerializer):
    partner = serializers.SerializerMethodField()
    partner_id = serializers.SerializerMethodField()
    
    class Meta:
        model = User
        fields = [
            'id', 'email', 'username', 'first_name', 'last_name', 
            'role', 'date_joined', 'password', 'phone', 'address', 'avatar',
            'partner', 'partner_id'
        ]
        extra_kwargs = {
            'password': {'write_only': True},
            'date_joined': {'read_only': True}
        }
    
    def get_partner(self, obj):
        # Check if user is an admin for any partner
        partner_admin = PartnerAdmin.objects.filter(user=obj).first()
        if partner_admin:
            return partner_admin.partner.name
        return None
    
    def get_partner_id(self, obj):
        # Check if user is an admin for any partner
        partner_admin = PartnerAdmin.objects.filter(user=obj).first()
        if partner_admin:
            return partner_admin.partner.id
        return None
    
    def create(self, validated_data):
        password = validated_data.pop('password', None)
        user = User.objects.create(**validated_data)
        if password:
            user.set_password(password)
            user.save()
        return user
    
    def update(self, instance, validated_data):
        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()
        return instance

class PartnerSerializer(serializers.ModelSerializer):
    class Meta:
        model = Partner
        fields = ['id', 'name', 'description', 'logo', 'website', 'managed_by']

class PartnerAdminSerializer(serializers.ModelSerializer):
    class Meta:
        model = PartnerAdmin
        fields = ['id', 'user', 'partner']