65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
|
from rest_framework import serializers
|
||
|
|
from .models import Article
|
||
|
|
|
||
|
|
|
||
|
|
class ArticleSerializer(serializers.ModelSerializer):
|
||
|
|
"""Serializer for Article model."""
|
||
|
|
|
||
|
|
article_type_display = serializers.CharField(source='get_article_type_display', read_only=True)
|
||
|
|
status_display = serializers.CharField(source='get_publish_status_display', read_only=True)
|
||
|
|
moderator_status_display = serializers.CharField(source='get_moderator_status_display', read_only=True)
|
||
|
|
ai_status_display = serializers.CharField(source='get_ai_status_display', read_only=True)
|
||
|
|
author_username = serializers.CharField(source='author.username', read_only=True)
|
||
|
|
region_name = serializers.CharField(source='region.name', read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Article
|
||
|
|
fields = ('id', 'title', 'content', 'region', 'region_name', 'article_type', 'article_type_display',
|
||
|
|
'author', 'author_username', 'moderator_status', 'moderator_status_display',
|
||
|
|
'moderator_reviewer', 'moderator_reviewed_at', 'moderator_rejection_reason',
|
||
|
|
'ai_status', 'ai_status_display', 'ai_reviewed_at', 'ai_rejection_reason',
|
||
|
|
'publish_status', 'status_display', 'published_at', 'created_at', 'updated_at')
|
||
|
|
read_only_fields = ('id', 'author', 'moderator_reviewer', 'moderator_reviewed_at',
|
||
|
|
'ai_reviewed_at', 'published_at', 'created_at', 'updated_at')
|
||
|
|
|
||
|
|
|
||
|
|
class ArticleCreateSerializer(serializers.ModelSerializer):
|
||
|
|
"""Serializer for creating articles."""
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Article
|
||
|
|
fields = ('title', 'content', 'region', 'article_type')
|
||
|
|
|
||
|
|
def create(self, validated_data):
|
||
|
|
validated_data['author'] = self.context['request'].user
|
||
|
|
return super().create(validated_data)
|
||
|
|
|
||
|
|
|
||
|
|
class ArticleUpdateSerializer(serializers.ModelSerializer):
|
||
|
|
"""Serializer for updating articles."""
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Article
|
||
|
|
fields = ('title', 'content', 'article_type')
|
||
|
|
|
||
|
|
|
||
|
|
class ArticleReviewSerializer(serializers.Serializer):
|
||
|
|
"""Serializer for article review actions."""
|
||
|
|
|
||
|
|
action = serializers.ChoiceField(choices=['approve', 'reject'])
|
||
|
|
reason = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
|
||
|
|
|
||
|
|
class ArticleListSerializer(serializers.ModelSerializer):
|
||
|
|
"""Simplified serializer for article list."""
|
||
|
|
|
||
|
|
article_type_display = serializers.CharField(source='get_article_type_display', read_only=True)
|
||
|
|
author_username = serializers.CharField(source='author.username', read_only=True)
|
||
|
|
region_name = serializers.CharField(source='region.name', read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Article
|
||
|
|
fields = ('id', 'title', 'article_type', 'article_type_display',
|
||
|
|
'author', 'author_username', 'region', 'region_name',
|
||
|
|
'publish_status', 'created_at')
|
||
|
|
read_only_fields = ('id', 'created_at')
|