first push

LiveStock/Rancher/serializers.py

@@ -0,0 +1,40 @@
+from rest_framework import serializers
+from LiveStock.models import Rancher, LiveStock, LiveStockProduct
+from authentication.serializer.serializer import BankCardSerializer, SystemUserProfileForInspectionSerializer
+from authentication.serializers import SystemAddressSerializer
+
+
+class RancherSerializer(serializers.ModelSerializer):
+    user = SystemUserProfileForInspectionSerializer(read_only=True)
+    # address = SystemAddressSerializer(read_only=True)
+
+    class Meta:
+        model = Rancher
+        fields = '__all__'
+
+
+class RancherForBazrasiSerializer(serializers.ModelSerializer):
+    heavy_livestock = serializers.SerializerMethodField()
+    light_livestock = serializers.SerializerMethodField()
+
+    class Meta:
+        model = Rancher
+        fields = [
+            'herd_code',
+            'fullname',
+            'mobile',
+            'lot',
+            'lng',
+            'heavy_livestock',
+            'light_livestock',
+            'type'
+
+        ]
+
+    def get_heavy_livestock(self, obj):
+        heavy_counts = self.context.get('heavy_counts') or {}
+        return heavy_counts.get(obj.herd_code, 0)
+
+    def get_light_livestock(self, obj):
+        light_counts = self.context.get('light_counts') or {}
+        return light_counts.get(obj.herd_code, 0)