some apis of pos

apps/herd/pos/api/v1/api.py

@@ -1,6 +1,11 @@
 from apps.herd.web.api.v1.serializers import HerdSerializer, RancherSerializer
-from apps.core.mixins.search_mixin import DynamicSearchMixin
 from apps.livestock.web.api.v1.serializers import LiveStockSerializer
+from apps.herd.services.services import (
+    get_rancher_statistics,
+    rancher_quota_weight
+)
+from apps.warehouse.models import InventoryEntry
+from apps.core.mixins.search_mixin import DynamicSearchMixin
 from rest_framework.exceptions import APIException
 from rest_framework.permissions import AllowAny
 from rest_framework.response import Response
@@ -110,8 +115,9 @@ class HerdViewSet(viewsets.ModelViewSet):
 
 
 class RancherViewSet(viewsets.ModelViewSet, DynamicSearchMixin):
-    queryset = Rancher.objects.all()
+    queryset = Rancher.objects.all()  # noqa
     serializer_class = RancherSerializer
+    permission_classes = [AllowAny]
     search_fields = [
         "ranching_farm",
         "first_name",
@@ -125,30 +131,28 @@ class RancherViewSet(viewsets.ModelViewSet, DynamicSearchMixin):
         "city__name",
     ]
 
-    def list(self, request, *args, **kwargs):
-        """ list of ranchers """
-
-        search = self.filter_query(self.queryset.order_by('-modify_date'))  # search & filter
-        page = self.paginate_queryset(search)
-        if page is not None:
-            serializer = self.get_serializer(page, many=True)
-            return self.get_paginated_response(serializer.data)
-
     @action(
-        methods=['get'],
-        detail=True,
-        url_path='herds',
-        url_name='herds',
-        name='herds'
+        methods=['post'],
+        detail=False,
+        url_name='check_national_code',
+        url_path='check_national_code',
+        name='check_national_code'
     )
-    def herds_by_rancher(self, request, pk=None):
-        """ list of rancher herds """
+    @transaction.atomic
+    def check_national_code(self, request):
+        """ check national code & existence of rancher """
 
-        rancher = self.get_object()
-        queryset = rancher.herd.all().order_by('-modify_date')  # get rancher herds
+        rancher = self.queryset.filter(national_code=request.data['national_code'])
+        inventory = InventoryEntry.objects.get(id=43)
 
-        # paginate queryset
-        page = self.paginate_queryset(queryset)
-        if page is not None:
-            serializer = HerdSerializer(page, many=True)
-            return self.get_paginated_response(serializer.data)
+        # get rancher live stocks information ant total quota for rancher
+        rancher_quota_by_live_stock = rancher_quota_weight(rancher.first(), inventory)
+
+        if rancher.exists():
+            serializer = self.serializer_class(rancher.first())
+
+            return Response(serializer.data, status=status.HTTP_200_OK)
+        else:
+            return Response({
+                "message": "rancher has not existence"
+            }, status=status.HTTP_204_NO_CONTENT)

apps/herd/services/services.py

@@ -0,0 +1,74 @@
+from decimal import Decimal
+from apps.herd.models import Rancher
+from apps.livestock.models import LiveStock
+from apps.warehouse.models import InventoryEntry
+from apps.product.models import Quota
+import typing
+
+
+def get_rancher_statistics(rancher: Rancher = None) -> typing.Any:
+    """ get statistics of a rancher """  # noqa
+
+    herds = rancher.herd.all()  # noqa
+    herd_count = herds.count()
+
+    livestocks = LiveStock.objects.filter(herd__in=herds)  # noqa
+
+    light_count = livestocks.filter(weight_type='L').count()
+    heavy_count = livestocks.filter(weight_type='H').count()
+
+    sheep_count = livestocks.filter(type__name="گوسفند").count()  # noqa
+    goat_count = livestocks.filter(type__name="بز").count()
+    cow_count = livestocks.filter(type__name="گاو").count()
+    camel_count = livestocks.filter(type__name="شتر").count()
+    horse_count = livestocks.filter(type__name="اسب").count()
+
+    return {
+        "herd_count": herd_count,
+        "light_count": light_count,
+        "heavy_count": heavy_count,
+        "sheep_count": sheep_count,
+        "goat_count": goat_count,
+        "cow_count": cow_count,
+        "camel_count": camel_count,
+        "horse_count": horse_count,
+    }
+
+
+def rancher_quota_weight(rancher, inventory_entry: InventoryEntry):
+    """
+    :param rancher: Rancher instance
+    :param inventory_entry: InventoryEntry instance
+    :return: dict {total, by_type}
+    """
+
+    live_stock_meta = {
+        "گوسفند": "sheep_count",  # noqa
+        "بز": "goat_count",
+        "گاو": "cow_count",
+        "شتر": "camel_count",
+        "اسب": "horse_count"
+    }
+
+    quota: Quota = inventory_entry.distribution.quota
+    allocations = quota.livestock_allocations.all()
+
+    livestock_counts = get_rancher_statistics(rancher)
+
+    total_weight = Decimal(0)
+    details = {}
+
+    for alloc in allocations:
+        animal_type = alloc.livestock_type.name
+        per_head = Decimal(alloc.quantity_kg)
+        count = livestock_counts.get(live_stock_meta.get(animal_type), 0)
+
+        weight = per_head * Decimal(count)
+        print(weight)
+        details[animal_type] = weight
+        total_weight += weight
+
+    return {
+        "total": total_weight,
+        "by_type": details
+    }