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

from apps.herd.models import Rancher, Herd
from apps.warehouse.models import InventoryQuotaSaleItem


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
        """

        transaction_sale_items = InventoryQuotaSaleItem.objects.select_related(
            'transaction', 'quota_stat'
        ).filter(
            transaction__rancher=rancher,
            transaction__transaction_status='success',
        ).values('quota_stat')

        print(list(set(transaction_sale_items)))

        return {}