109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
from rest_framework import serializers
|
||
from .models import ESP32Config, Order, Salesperson, Service, ARService
|
||
|
||
class ServiceSerializer(serializers.ModelSerializer):
|
||
"""
|
||
AI服务序列化器
|
||
"""
|
||
features_list = serializers.SerializerMethodField()
|
||
display_icon = serializers.SerializerMethodField()
|
||
display_detail_image = serializers.SerializerMethodField()
|
||
|
||
class Meta:
|
||
model = Service
|
||
fields = '__all__'
|
||
|
||
def get_features_list(self, obj):
|
||
if obj.features:
|
||
return [line.strip() for line in obj.features.split('\n') if line.strip()]
|
||
return []
|
||
|
||
def get_display_icon(self, obj):
|
||
if obj.icon_url:
|
||
return obj.icon_url
|
||
if obj.icon:
|
||
return obj.icon.url
|
||
return None
|
||
|
||
def get_display_detail_image(self, obj):
|
||
if obj.detail_image_url:
|
||
return obj.detail_image_url
|
||
if obj.detail_image:
|
||
return obj.detail_image.url
|
||
return None
|
||
|
||
class ARServiceSerializer(serializers.ModelSerializer):
|
||
"""
|
||
AR服务序列化器
|
||
"""
|
||
display_cover_image = serializers.SerializerMethodField()
|
||
|
||
class Meta:
|
||
model = ARService
|
||
fields = '__all__'
|
||
|
||
def get_display_cover_image(self, obj):
|
||
if obj.cover_image_url:
|
||
return obj.cover_image_url
|
||
if obj.cover_image:
|
||
return obj.cover_image.url
|
||
return None
|
||
|
||
class ESP32ConfigSerializer(serializers.ModelSerializer):
|
||
"""
|
||
ESP32配置序列化器
|
||
"""
|
||
display_detail_image = serializers.SerializerMethodField()
|
||
|
||
class Meta:
|
||
model = ESP32Config
|
||
fields = '__all__'
|
||
|
||
def get_display_detail_image(self, obj):
|
||
if obj.detail_image_url:
|
||
return obj.detail_image_url
|
||
if obj.detail_image:
|
||
return obj.detail_image.url
|
||
return None
|
||
|
||
|
||
class OrderSerializer(serializers.ModelSerializer):
|
||
"""
|
||
订单序列化器
|
||
"""
|
||
config_name = serializers.CharField(source='config.name', read_only=True)
|
||
# 接收前端传来的 ref_code，用于查找 Salesperson
|
||
ref_code = serializers.CharField(write_only=True, required=False, allow_blank=True)
|
||
|
||
class Meta:
|
||
model = Order
|
||
fields = ['id', 'config', 'config_name', 'quantity', 'total_price', 'status', 'created_at', 'wechat_trade_no',
|
||
'customer_name', 'phone_number', 'shipping_address', 'ref_code']
|
||
read_only_fields = ['total_price', 'status', 'wechat_trade_no', 'created_at']
|
||
extra_kwargs = {
|
||
'customer_name': {'required': True},
|
||
'phone_number': {'required': True},
|
||
'shipping_address': {'required': True},
|
||
}
|
||
|
||
def create(self, validated_data):
|
||
"""
|
||
重写创建方法，自动计算总价并关联销售员
|
||
"""
|
||
config = validated_data.get('config')
|
||
quantity = validated_data.get('quantity', 1)
|
||
ref_code = validated_data.pop('ref_code', None)
|
||
|
||
validated_data['total_price'] = config.price * quantity
|
||
|
||
# 尝试关联销售员
|
||
if ref_code:
|
||
try:
|
||
salesperson = Salesperson.objects.get(code=ref_code)
|
||
validated_data['salesperson'] = salesperson
|
||
except Salesperson.DoesNotExist:
|
||
# 如果找不到对应的销售员，忽略该推广码，仍继续创建订单（算作自然流量）
|
||
pass
|
||
|
||
return super().create(validated_data)
|