From cfa8ae6835b3e1ae4c3083df207e3896a8e371ad Mon Sep 17 00:00:00 2001
From: Ruslan Bel'kov <ruslan.belckov@yandex.ru>
Date: Sat, 23 Nov 2024 14:29:31 +0300
Subject: [PATCH] feat: add counter by months route

---
 backend/src/modules/events/routes.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/backend/src/modules/events/routes.py b/backend/src/modules/events/routes.py
index 2c4fd30..6ac21ea 100644
--- a/backend/src/modules/events/routes.py
+++ b/backend/src/modules/events/routes.py
@@ -1,9 +1,11 @@
+from datetime import datetime
+
 from fastapi import APIRouter
 from pydantic import BaseModel
 
 from src.api.exceptions import IncorrectCredentialsException
 from src.modules.events.repository import events_repository
-from src.modules.events.schemas import Filters, Pagination, Sort
+from src.modules.events.schemas import DateFilter, Filters, Pagination, Sort
 from src.modules.sports.repository import sports_repository
 from src.storages.mongo.events import Event
 
@@ -82,6 +84,30 @@ async def count_events(filters: Filters) -> int:
     return count
 
 
+@router.post("/search/count-by-month", responses={200: {"description": "Count events by months"}})
+async def count_events_by_month(filters: Filters) -> dict[str, int]:
+    """
+    Count filtered events by months.
+    """
+
+    counts = {}
+    current_year = datetime.now().year
+    for i in range(1, 13):
+        date_filter = DateFilter()
+        date_filter.start_date = datetime(current_year, month=i, day=1)
+        if i == 12:
+            date_filter.end_date = datetime(current_year + 1, month=1, day=1)
+        else:
+            date_filter.end_date = datetime(current_year, month=i + 1, day=1)
+        filters.date = date_filter
+        count = await events_repository.read_with_filters(
+            filters, Sort(), Pagination(page_size=0, page_no=0), count=True
+        )
+        counts[f"{current_year}-{i:02d}"] = count
+
+    return counts
+
+
 class RegionsFilterVariants(BaseModel):
     region: str | None
     "Название региона"