from rest_framework import serializers
from user_management.models import OperatorAccount, PlatformAccount, Video, KnowledgeBase, KnowledgeBaseDocument
import uuid


class OperatorAccountSerializer(serializers.ModelSerializer):
    id = serializers.UUIDField(read_only=False, required=False)  # 允许前端不提供ID，但如果提供则必须是有效的UUID
    
    class Meta:
        model = OperatorAccount
        fields = ['id', 'username', 'password', 'real_name', 'email', 'phone', 'position', 'department', 'is_active', 'created_at', 'updated_at']
        read_only_fields = ['created_at', 'updated_at']
        extra_kwargs = {
            'password': {'write_only': True}
        }
    
    def create(self, validated_data):
        # 如果没有提供ID，则生成一个UUID
        if 'id' not in validated_data:
            validated_data['id'] = uuid.uuid4()
            
        password = validated_data.pop('password', None)
        instance = self.Meta.model(**validated_data)
        if password:
            instance.password = password  # 在实际应用中应该加密存储密码
        instance.save()
        return instance


class PlatformAccountSerializer(serializers.ModelSerializer):
    operator_name = serializers.CharField(source='operator.real_name', read_only=True)
    
    class Meta:
        model = PlatformAccount
        fields = ['id', 'operator', 'operator_name', 'platform_name', 'account_name', 'account_id', 
                 'status', 'followers_count', 'account_url', 'description', 
                 'tags', 'profile_image', 'last_posting',
                 'created_at', 'updated_at', 'last_login']
        read_only_fields = ['id', 'created_at', 'updated_at']
    
    def to_internal_value(self, data):
        # 处理operator字段，可能是字符串格式的UUID
        if 'operator' in data and isinstance(data['operator'], str):
            try:
                # 尝试获取对应的运营账号对象
                operator = OperatorAccount.objects.get(id=data['operator'])
                data['operator'] = operator.id  # 确保使用正确的ID格式
            except OperatorAccount.DoesNotExist:
                # 如果找不到对应的运营账号，保持原值，让验证器捕获此错误
                pass
            except Exception as e:
                # 其他类型的错误，如ID格式不正确等
                pass
        
        return super().to_internal_value(data)


class VideoSerializer(serializers.ModelSerializer):
    platform_account_name = serializers.CharField(source='platform_account.account_name', read_only=True)
    platform_name = serializers.CharField(source='platform_account.platform_name', read_only=True)
    
    class Meta:
        model = Video
        fields = ['id', 'platform_account', 'platform_account_name', 'platform_name', 'title', 
                 'description', 'video_url', 'local_path', 'thumbnail_url', 'status', 
                 'views_count', 'likes_count', 'comments_count', 'shares_count', 'tags', 
                 'publish_time', 'scheduled_time', 'created_at', 'updated_at']
        read_only_fields = ['id', 'created_at', 'updated_at', 'views_count', 'likes_count', 
                           'comments_count', 'shares_count']
                           
    def to_internal_value(self, data):
        # 处理platform_account字段，可能是字符串格式的UUID
        if 'platform_account' in data and isinstance(data['platform_account'], str):
            try:
                # 尝试获取对应的平台账号对象
                platform_account = PlatformAccount.objects.get(id=data['platform_account'])
                data['platform_account'] = platform_account.id  # 确保使用正确的ID格式
            except PlatformAccount.DoesNotExist:
                # 如果找不到对应的平台账号，保持原值，让验证器捕获此错误
                pass
            except Exception as e:
                # 其他类型的错误，如ID格式不正确等
                pass
        
        return super().to_internal_value(data)


class KnowledgeBaseSerializer(serializers.ModelSerializer):
    class Meta:
        model = KnowledgeBase
        fields = ['id', 'user_id', 'name', 'desc', 'type', 'department', 'group', 
                 'external_id', 'create_time', 'update_time']
        read_only_fields = ['id', 'create_time', 'update_time']


class KnowledgeBaseDocumentSerializer(serializers.ModelSerializer):
    class Meta:
        model = KnowledgeBaseDocument
        fields = ['id', 'knowledge_base', 'document_id', 'document_name', 
                 'external_id', 'uploader_name', 'status', 'create_time', 'update_time']
        read_only_fields = ['id', 'create_time', 'update_time']