product dynanmic serializer

backend/product/serializers.py

@@ -3,12 +3,39 @@ from rest_framework import serializers
 from django.utils import timezone
 from datetime import  timedelta
 
-class ProductChatSerializer(serializers.ModelSerializer):
+
+class InStuckColorsSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = InStuckColors
+        fields = ['color', 'in_stuck']
+
+
+class DynamicProductSerializer(serializers.ModelSerializer):
+    colors = InStuckColorsSerializer(many=True, read_only=True)
     price = serializers.SerializerMethodField()
     is_new = serializers.SerializerMethodField()
+    related_products = serializers.SerializerMethodField()
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        view_type = self.context.get('view_type', 'all')
+        if view_type != 'all':
+            allowed_fields = self.Meta.view_type[view_type]
+            allowed = set(allowed_fields)
+            existing = set(self.fields.keys())
+            
+            for field_name in existing - allowed:
+                self.fields.pop(field_name)
+
+
     class Meta:
         model = ProductModel
-        fields = ['name', 'description', 'price', 'in_stock', 'discount', 'is_new']
+        fields = "__all__"
+        view_type = {
+            'list': ['name', 'price', 'image1', 'video', 'rating', 'discount', 'slug', 'category', 'colors'],
+            'instance': ['name', 'description', 'price', 'image1', 'image2', 'image3', 'video', 'rating', 'discount', 'slug', 'meta_description', 'meta_keywords', 'meta_rating', 'category', 'colors', 'related_products'],
+            'chat': ['name', 'description', 'price', 'in_stock', 'discount', 'colors']
+        }
+        
     def get_price(self, obj):
         dollor_price = self.context.get('dollor_price')
         dollar_to_dirham = 0.27
@@ -21,27 +48,25 @@ class ProductChatSerializer(serializers.ModelSerializer):
         elif obj.currency == 'derham':
             toman_price =  obj.price * dollor_price * dollar_to_dirham
         return "{:,.0f} تومان".format(toman_price)
+
     def get_is_new(self, obj):
         return timezone.now() < obj.created_at + timedelta(days=7)
 
-class ProductSerializer(ProductChatSerializer):
-    class Meta:
-        model = ProductModel
-        fields = "__all__"
-    #     many_fields = ['name', 'price']
-    #     one_fields = ['description',]
-    # def __init__(self, *args, **kwargs):
+    def get_related_products(self, obj):  
+        if obj.related_products.all().count() >= 5:
+            related_products = obj.related_products.all()
+        else:
+            related_products = obj.category.products
+        serializer = DynamicProductSerializer(
+            related_products,
+            many=True,
+            context={
+                'view_type': 'list',
+                'dollor_price': self.context.get('dollor_price')
+            }
+        )
+        return serializer.data
 
-    #     many = kwargs.pop('many', True)
-    #     super().__init__(*args, **kwargs)
-
-    #     allowed_fields = self.Meta.many_fields if many else self.Meta.one_fields
-
-    #     allowed = set(allowed_fields)
-    #     existing = set(self.fields.keys())
-        
-    #     for field_name in existing - allowed:
-    #         self.fields.pop(field_name)
 
 class CommentSerializer(serializers.ModelSerializer):
     class Meta: