chunyu_project/article/serializers.py

from rest_framework import serializers
from .models import Article, ArticleComment, ArticleLike, ArticleFavorite, ArticleCommentLike


class ArticleCommentSerializer(serializers.ModelSerializer):
    user_id = serializers.IntegerField(source='user.id', read_only=True)
    user_name = serializers.CharField(source='user.first_name', read_only=True)
    user_avatar = serializers.ImageField(source='user.avatar', read_only=True)
    user_location = serializers.CharField(source='user.location', read_only=True)
    parent_user_name = serializers.SerializerMethodField()
    replies = serializers.SerializerMethodField()
    is_liked = serializers.SerializerMethodField()
    created_at = serializers.DateTimeField(format='%Y-%m-%d %H:%M')

    class Meta:
        model = ArticleComment
        fields = ['id', 'user_id', 'user_name', 'user_avatar', 'user_location', 'content', 'parent', 'parent_user_name', 'likes', 'created_at', 'replies', 'is_liked']
        read_only_fields = ['likes', 'created_at']

    def get_parent_user_name(self, obj):
        if obj.parent:
            return obj.parent.user.first_name or obj.parent.user.username
        return ''

    def get_replies(self, obj):
        if obj.parent is None:
            replies = obj.replies.all()
            return ArticleCommentSerializer(replies, many=True, context=self.context).data
        return []

    def get_is_liked(self, obj):
        request = self.context.get('request')
        if request and request.user.is_authenticated:
            return ArticleCommentLike.objects.filter(user=request.user, comment=obj).exists()
        return False


class ArticleListSerializer(serializers.ModelSerializer):
    author_name = serializers.CharField(source='author.first_name', read_only=True)
    author_avatar = serializers.ImageField(source='author.avatar', read_only=True)
    comments_count = serializers.SerializerMethodField()
    read_time = serializers.SerializerMethodField()
    is_hot = serializers.SerializerMethodField()
    is_new = serializers.SerializerMethodField()
    publish_date = serializers.DateTimeField(source='created_at', format='%Y-%m-%d', read_only=True)
    image = serializers.SerializerMethodField()

    class Meta:
        model = Article
        fields = [
            'id', 'title', 'excerpt', 'category', 'author_name', 'author_avatar',
            'views', 'likes', 'comments_count', 'publish_date', 'image', 'tags',
            'read_time', 'is_featured', 'is_hot', 'is_new', 'is_top', 'status',
        ]

    def get_comments_count(self, obj):
        return obj.comments.count()

    def get_read_time(self, obj):
        if obj.content:
            word_count = len(obj.content)
            minutes = max(1, word_count // 300)
            return f'{minutes} 分钟'
        return '1 分钟'

    def get_is_hot(self, obj):
        return obj.views > 1000 or obj.likes > 100

    def get_is_new(self, obj):
        from django.utils import timezone
        import datetime
        return (timezone.now() - obj.created_at).days < 7

    def get_image(self, obj):
        if obj.cover_image:
            request = self.context.get('request')
            if request:
                return request.build_absolute_uri(obj.cover_image.url)
            return obj.cover_image.url
        return ''


class ArticleDetailSerializer(serializers.ModelSerializer):
    author_name = serializers.CharField(source='author.first_name', read_only=True)
    author_avatar = serializers.ImageField(source='author.avatar', read_only=True)
    author_bio = serializers.CharField(source='author.bio', read_only=True, default='')
    author_user_id = serializers.PrimaryKeyRelatedField(source='author', read_only=True)
    comments_count = serializers.SerializerMethodField()
    favorites_count = serializers.SerializerMethodField()
    read_time = serializers.SerializerMethodField()
    publish_date = serializers.DateTimeField(source='created_at', format='%Y-%m-%d %H:%M', read_only=True)
    update_date = serializers.DateTimeField(source='updated_at', format='%Y-%m-%d %H:%M', read_only=True)
    image = serializers.SerializerMethodField()
    is_liked = serializers.SerializerMethodField()
    is_favorited = serializers.SerializerMethodField()
    is_following = serializers.SerializerMethodField()
    related_articles = serializers.SerializerMethodField()
    comments = serializers.SerializerMethodField()
    comments_pagination = serializers.SerializerMethodField()

    class Meta:
        model = Article
        fields = [
            'id', 'title', 'excerpt', 'content', 'category', 'tags', 'image',
            'author_name', 'author_avatar', 'author_bio', 'author_user_id',
            'views', 'likes', 'favorites_count', 'comments_count', 'publish_date',
            'update_date', 'read_time', 'is_featured', 'is_top', 'status', 'editor_mode',
            'is_liked', 'is_favorited', 'is_following', 'related_articles', 'comments', 'comments_pagination',
        ]

    def get_comments_count(self, obj):
        return obj.comments.count()

    def get_favorites_count(self, obj):
        return obj.article_favorites.count()

    def get_read_time(self, obj):
        if obj.content:
            word_count = len(obj.content)
            minutes = max(1, word_count // 300)
            return f'{minutes} 分钟'
        return '1 分钟'

    def get_image(self, obj):
        if obj.cover_image:
            request = self.context.get('request')
            if request:
                return request.build_absolute_uri(obj.cover_image.url)
            return obj.cover_image.url
        return ''

    def get_is_liked(self, obj):
        request = self.context.get('request')
        if request and request.user.is_authenticated:
            return ArticleLike.objects.filter(user=request.user, article=obj).exists()
        return False

    def get_is_favorited(self, obj):
        request = self.context.get('request')
        if request and request.user.is_authenticated:
            return ArticleFavorite.objects.filter(user=request.user, article=obj).exists()
        return False

    def get_is_following(self, obj):
        request = self.context.get('request')
        if request and request.user.is_authenticated:
            from user.models import Follow
            return Follow.objects.filter(follower=request.user, following=obj.author).exists()
        return False

    def get_related_articles(self, obj):
        queryset = Article.objects.filter(status='published').exclude(id=obj.id)
        related = queryset.filter(category=obj.category)[:4]
        if related.count() < 4:
            remaining = 4 - related.count()
            extra = queryset.filter(tags__overlap=obj.tags or []).exclude(
                id__in=list(related.values_list('id', flat=True))
            )[:remaining]
            related = list(related) + list(extra)
        return ArticleListSerializer(related, many=True, context=self.context).data

    def get_comments(self, obj):
        request = self.context.get('request')
        page_size = int(request.query_params.get('comment_page_size', 10)) if request else 10
        page = int(request.query_params.get('comment_page', 1)) if request else 10
        top_comments = obj.comments.filter(parent__isnull=True)
        start = (page - 1) * page_size
        end = start + page_size
        comments_page = top_comments[start:end]
        serializer = ArticleCommentSerializer(comments_page, many=True, context=self.context)
        return serializer.data

    def get_comments_pagination(self, obj):
        request = self.context.get('request')
        page_size = int(request.query_params.get('comment_page_size', 10)) if request else 10
        top_count = obj.comments.filter(parent__isnull=True).count()
        return {
            'count': top_count,
            'next': top_count > page_size,
        }


class ArticleCreateUpdateSerializer(serializers.ModelSerializer):
    tags = serializers.JSONField(required=False, default=list)

    class Meta:
        model = Article
        fields = ['title', 'excerpt', 'content', 'category', 'tags', 'cover_image', 'status', 'editor_mode']
        read_only_fields = ['status']

    def create(self, validated_data):
        validated_data['author'] = self.context['request'].user
        return super().create(validated_data)


class ArticleManageSerializer(serializers.ModelSerializer):
    author_name = serializers.CharField(source='author.first_name', read_only=True)
    comments_count = serializers.SerializerMethodField()
    favorites_count = serializers.SerializerMethodField()
    summary = serializers.CharField(source='excerpt', read_only=True)
    publish_date = serializers.DateTimeField(source='created_at', format='%Y-%m-%d', read_only=True)
    update_date = serializers.DateTimeField(source='updated_at', format='%Y-%m-%d', read_only=True)

    class Meta:
        model = Article
        fields = [
            'id', 'title', 'summary', 'status', 'views', 'likes',
            'comments_count', 'favorites_count', 'publish_date', 'update_date',
            'category', 'tags', 'is_top', 'author_name',
        ]

    def get_comments_count(self, obj):
        return obj.comments.count()

    def get_favorites_count(self, obj):
        return obj.article_favorites.count()