import - rancher page main dashboard

apps/herd/services/rancher_dashboard_service.py

@@ -1,47 +1,45 @@
-from django.db import models
-from django.db.models import Count, Sum
+from django.db.models import Sum, Count, Q
+from django.db.models.functions import Coalesce
 
 from apps.herd.models import Rancher, Herd
-from apps.livestock.models import LiveStock
 
 
-class FarmerDashboardService:
-    """ Dashboard aggregation service for Rancher """
+class RancherDashboardService:
+    """
+    Rancher Dashboard Service
+    """
 
     @staticmethod
-    def get_basic_info(rancher: Rancher):
-        return {
-            "id": rancher.id,
-            "name": f"{rancher.first_name} {rancher.last_name}",
-            "mobile": rancher.mobile,
-            "national_code": rancher.national_code,
-            "province": rancher.province.name if rancher.province else None,
-            "city": rancher.city.name if rancher.city else None,
-        }
+    def get_main_dashboard(
+            self,
+            rancher_search_fields: list[str] = None,
+            herd_search_fields: list[str] = None,
+            query_string: str = None
+    ):
+        """
+        get ranchers main page dashboard
+        """
+        ranchers = Rancher.objects.all()
+        herds = Herd.objects.all()
 
-    @staticmethod
-    def get_herd_counts(rancher_id):
-        return Herd.objects.filter(rancher_id=rancher_id).aggregate(
-            total_herds=Count('id'),
-            total_heavy=Sum('heavy_livestock_number'),
-            total_light=Sum('light_livestock_number'),
+        ranchers_data = ranchers.aggregate(
+            total_ranchers_count=Coalesce(Count("id"), 0),
+            total_with_herd=Coalesce(Count("id", filter=Q(without_herd=False)), 0),
+            total_without_herd=Coalesce(Count("id", filter=Q(without_herd=True)), 0),
+            total_natural_ranchers=Coalesce(Count("id", filter=Q(rancher_type='N')), 0),
+            total_legal_ranchers=Coalesce(Count("id", filter=Q(rancher_type='L')), 0),
+            total_industrial_ranchers=Coalesce(Count("id", filter=Q(rancher_type='I')), 0),
+            total_village_ranchers=Coalesce(Count("id", filter=Q(rancher_type='V')), 0),
+            total_nomadic_ranchers=Coalesce(Count("id", filter=Q(rancher_type='N')), 0),
         )
 
-    @staticmethod
-    def get_livestock_counts(rancher_id):
-        return LiveStock.objects.filter(herd__rancher_id=rancher_id, archive=False).aggregate(
-            total=Count('id'),
-            heavy=Count('id', filter=models.Q(weight_type='H')),
-            light=Count('id', filter=models.Q(weight_type='L')),
+        herds_data = herds.aggregate(
+            total_herds_count=Coalesce(Count("id"), 0),
+            total_industrial_herds_count=Coalesce(Count("id", filter=Q(activity='I')), 0),
+            total_Village_herds_count=Coalesce(Count("id", filter=Q(activity='V')), 0),
+            total_nomadic_herds_count=Coalesce(Count("id", filter=Q(activity='N')), 0),
+            total_heavy_livestock_count=Coalesce(Sum("heavy_livestock_number"), 0),
+            total_light_livestock_count=Coalesce(Sum("light_livestock_number"), 0),
         )
 
-    @staticmethod
-    def get_livestock_by_type(rancher_id):
-        return (
-            LiveStock.objects.filter(herd__rancher_id=rancher_id, archive=False)
-            .values("type__name")
-            .annotate(count=Count("id"))
-        )
-
-# @staticmethod
-# def get_dashboard(rancher_id):
+        return {"rancher_dashboard": ranchers_data, "herd_dashboard": herds_data}

apps/herd/services/services.py

@@ -90,6 +90,7 @@ def rancher_quota_weight(
     :param inventory_entry: InventoryEntry instance
     :param distribution: QuotaDistribution instance
     :param quota: Quota instance
+    :param quota_stat: OrganizationQuotaStat instance
     :return: dict {total, by_type}
     """
 
@@ -115,7 +116,6 @@ def rancher_quota_weight(
     # list of rancher herds
     herds = rancher.herd.all()
     herd_livestock_gropes = [herd.activity if herd.activity else 'V' for herd in herds]
-    print(herd_livestock_gropes)
     livestock_counts_list, livestock_counts_dict = get_rancher_statistic_by_herd(rancher)
 
     total_weight = 0
@@ -153,14 +153,12 @@ def rancher_quota_weight(
             # incentive plan quantity by this livestock type
             rancher_plan_weight = rancher_plans.first().allowed_quantity * item.quantity_kg
             total_weight += rancher_plan_weight
-            print(total_weight)
     # get rancher remaining usage of quota for purchase
     rancher_remaining_usage = RancherService.get_total_used_weight(
         rancher,
         InventoryQuotaSaleItem,
         quota_stat=quota_stat
     )
-    print(rancher_remaining_usage)
 
     if total_weight - rancher_remaining_usage < 0:
         remaining_weight = 0