import strawberry
import strawberry_django

from main import models
from main.api.types.shop import Order, OrderOrdering, OrderStats, OrderStatsByTime, Product


@strawberry.type
class ShopQueries:
    all_products: list[Product] = strawberry_django.field()
    all_orders: list[Order] = strawberry_django.field()
    order_list: list[Order] = strawberry_django.field(pagination=True, order=OrderOrdering)

    @strawberry.field
    def order_stats(self) -> OrderStats:
        return OrderStats(
            **{
                period: [
                    OrderStatsByTime(**stat)
                    for stat in models.Order.objects.count_by_created_period(period).order_by("period")
                ]
                for period in ["yearly", "monthly", "daily", "hourly"]
            }
        )