72 lines
2.9 KiB
Python
72 lines
2.9 KiB
Python
from rest_framework import serializers
|
|
from .models import Template
|
|
|
|
class TemplateListSerializer(serializers.ModelSerializer):
|
|
"""模板列表序列化器（简化版）"""
|
|
mission_display = serializers.CharField(source='get_mission_display', read_only=True)
|
|
platform_display = serializers.CharField(source='get_platform_display', read_only=True)
|
|
collaboration_type_display = serializers.CharField(source='get_collaboration_type_display', read_only=True)
|
|
service_display = serializers.CharField(source='get_service_display', read_only=True)
|
|
|
|
class Meta:
|
|
model = Template
|
|
fields = [
|
|
'id', 'title', 'preview',
|
|
'mission', 'mission_display',
|
|
'platform', 'platform_display',
|
|
'service', 'service_display',
|
|
'collaboration_type', 'collaboration_type_display',
|
|
'is_public', 'created_at', 'updated_at'
|
|
]
|
|
read_only_fields = ['created_at', 'updated_at', 'preview']
|
|
|
|
class TemplateDetailSerializer(serializers.ModelSerializer):
|
|
"""模板详情序列化器"""
|
|
mission_display = serializers.CharField(source='get_mission_display', read_only=True)
|
|
platform_display = serializers.CharField(source='get_platform_display', read_only=True)
|
|
collaboration_type_display = serializers.CharField(source='get_collaboration_type_display', read_only=True)
|
|
service_display = serializers.CharField(source='get_service_display', read_only=True)
|
|
|
|
class Meta:
|
|
model = Template
|
|
fields = [
|
|
'id', 'title', 'content', 'preview',
|
|
'mission', 'mission_display',
|
|
'platform', 'platform_display',
|
|
'service', 'service_display',
|
|
'collaboration_type', 'collaboration_type_display',
|
|
'is_public', 'created_at', 'updated_at'
|
|
]
|
|
read_only_fields = ['created_at', 'updated_at', 'preview']
|
|
|
|
class TemplateCreateUpdateSerializer(serializers.ModelSerializer):
|
|
"""模板创建和更新序列化器"""
|
|
class Meta:
|
|
model = Template
|
|
fields = [
|
|
'id', 'title', 'content',
|
|
'mission', 'platform',
|
|
'service', 'collaboration_type',
|
|
'is_public'
|
|
]
|
|
read_only_fields = ['preview']
|
|
|
|
def validate(self, data):
|
|
"""验证数据，处理测试期间可能缺失的字段"""
|
|
# 确保其他必填字段有默认值
|
|
if 'mission' not in data:
|
|
data['mission'] = 'initial_contact'
|
|
|
|
if 'platform' not in data:
|
|
data['platform'] = 'tiktok'
|
|
|
|
if 'service' not in data:
|
|
data['service'] = 'text'
|
|
|
|
if 'collaboration_type' not in data:
|
|
data['collaboration_type'] = 'paid_promotion'
|
|
|
|
if 'is_public' not in data:
|
|
data['is_public'] = True
|
|
|
|
return data |