vb

backend/shop/serializers.py

@@ -1,5 +1,5 @@
 from rest_framework import serializers
-from .models import ESP32Config, Order, Salesperson, Service, ARService, ProductFeature, ServiceOrder, WeChatUser, Distributor, Withdrawal
+from .models import ESP32Config, Order, Salesperson, Service, VBCourse, ProductFeature, ServiceOrder, WeChatUser, Distributor, Withdrawal
 
 class WeChatUserSerializer(serializers.ModelSerializer):
     class Meta:
@@ -101,14 +101,16 @@ class ServiceOrderSerializer(serializers.ModelSerializer):
 
         return super().create(validated_data)
 
-class ARServiceSerializer(serializers.ModelSerializer):
+class VBCourseSerializer(serializers.ModelSerializer):
     """
-    AR服务序列化器
+    VB课程序列化器
     """
     display_cover_image = serializers.SerializerMethodField()
+    display_detail_image = serializers.SerializerMethodField()
+    course_type_display = serializers.CharField(source='get_course_type_display', read_only=True)
 
     class Meta:
-        model = ARService
+        model = VBCourse
         fields = '__all__'
     
     def get_display_cover_image(self, obj):
@@ -118,6 +120,13 @@ class ARServiceSerializer(serializers.ModelSerializer):
             return obj.cover_image.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 ESP32ConfigSerializer(serializers.ModelSerializer):
     """
     ESP32配置序列化器