from django.db.models import Sum, Count, Q
from django.db.models.functions import Coalesce

from apps.herd.models import Rancher, Herd


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(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),
        )

        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}