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

from apps.authentication.models import Organization
from apps.authentication.services.service import get_all_org_child
from apps.core.services.filter.search import DynamicSearchService
from apps.warehouse.models import InventoryEntry


class InventoryEntryDashboardService:
    """
    Inventory Entry Dashboard Services
    """

    @staticmethod
    def get_dashboard(
            org: Organization,
            start_date: str = None,
            end_date: str = None,
            search_fields: list[str] = None,
            query_string: str = None
    ):
        """ dashboard of inventory entry page """

        child_orgs = get_all_org_child(org)
        child_orgs.append(org)

        if org.type.key == 'ADM':
            inventory_entries = InventoryEntry.objects.filter(quota__is_closed=False)
        else:
            inventory_entries = InventoryEntry.objects.filter(
                quota__is_closed=False,
                organization__in=child_orgs
            )

        if (start_date and end_date) or query_string:
            inventory_entries = DynamicSearchService(
                queryset=inventory_entries,
                start=start_date,
                end=end_date,
                search_fields=search_fields,
                date_field="create_date",
                query_string=query_string
            ).apply()

        dashboard = inventory_entries.aggregate(
            total_entries=Count("id"),
            total_weight=Coalesce(Sum("weight"), 0),
        )

        return dashboard