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()