Files
RasadDam_Backend/apps/herd/services/rancher_dashboard_service.py

147 lines
5.7 KiB
Python

from django.db.models import Sum, Count, Q
from django.db.models.functions import Coalesce
from apps.herd.models import Rancher, Herd
from apps.herd.services.services import rancher_quota_weight
from apps.product.models import OrganizationQuotaStats
from apps.warehouse.models import InventoryQuotaSaleItem
from apps.warehouse.services.services import can_buy_from_inventory
class RancherDashboardService:
"""
Rancher Dashboard Service
"""
@staticmethod
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()
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(activity='I')), 0),
total_village_ranchers=Coalesce(Count("id", filter=Q(activity='V')), 0),
total_nomadic_ranchers=Coalesce(Count("id", filter=Q(activity='N')), 0),
)
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),
)
return {"rancher_dashboard": ranchers_data, "herd_dashboard": herds_data}
@staticmethod
def get_rancher_dashboard(self, rancher: Rancher):
"""
get rancher dashboard
"""
rancher_data = rancher.herd.aggregate(
total_herds_count=Coalesce(Count("id"), 0),
total_heavy_livestock_count=Coalesce(Sum("heavy_livestock_number"), 0),
total_light_livestock_count=Coalesce(Sum("light_livestock_number"), 0),
)
transaction_sale_items = InventoryQuotaSaleItem.objects.select_related(
'transaction', 'quota_stat'
).filter(
transaction__rancher=rancher,
# transaction__transaction_status='success',
).aggregate(
total_purchased_weight=Coalesce(Sum('weight'), 0),
)
rancher_data.update(transaction_sale_items)
return rancher_data
@staticmethod
def get_rancher_dashboard_by_quota_usage(self, rancher: Rancher):
"""
get rancher dashboard by quota usage
"""
# get organization quota stats of transaction item
quota_stat = OrganizationQuotaStats.objects.filter(stat_type='quota', quota__is_closed=False)
available_stats = [
stat for stat in quota_stat if (
can_buy_from_inventory(rancher, quota_stat=stat) and rancher is not None
)
]
rancher_data_by_quota_usage = []
for stat in available_stats:
rancher_quota_data = rancher_quota_weight(rancher=rancher, quota=stat.quota, quota_stat=stat)
rancher_quota_data.update({'product': stat.quota.product.name})
rancher_data_by_quota_usage.append(rancher_quota_data)
return rancher_data_by_quota_usage
@staticmethod
def get_rancher_dashboard_by_product_usage(self, rancher: Rancher):
"""
get rancher dashboard by product usage
"""
# get organization quota stats of transaction item
quota_stat = OrganizationQuotaStats.objects.filter(stat_type='quota', quota__is_closed=False)
available_stats = [
stat for stat in quota_stat if (
can_buy_from_inventory(rancher, quota_stat=stat) and rancher is not None
)
]
rancher_data_by_quota_usage = []
for stat in available_stats:
rancher_quota_data = rancher_quota_weight(rancher=rancher, quota=stat.quota, quota_stat=stat)
rancher_quota_data.update({'product': stat.quota.product.name, 'product_id': stat.quota.product.id})
rancher_data_by_quota_usage.append(rancher_quota_data)
product_list = set(stat.quota.product.id for stat in quota_stat)
rancher_data_by_product_usage_list = []
for prod in product_list:
product_summary = {
'product_id': prod,
'product': None,
'total_weight': 0,
'remaining_weight': 0,
'free_sale': 0,
'total_purchase': 0,
'items': []
}
for item in rancher_data_by_quota_usage:
if item['product_id'] == prod:
product_summary['product'] = item['product']
product_summary['total_weight'] += item['total_weight']
product_summary['remaining_weight'] += item['remaining_weight']
product_summary['free_sale'] += item['free_sale']
product_summary['total_purchase'] += item['total_purchase']
product_summary['items'].append(item)
rancher_data_by_product_usage_list.append(product_summary)
return rancher_data_by_product_usage_list