From cda1fcc4380c73103ebde72f78feb65a9c3c93a8 Mon Sep 17 00:00:00 2001
From: happycastle <41810556+happycastle114@users.noreply.github.com>
Date: Fri, 1 Dec 2023 00:34:08 +0900
Subject: [PATCH] Add: Chart Data year

---
 routers/scrapResultNational.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/routers/scrapResultNational.py b/routers/scrapResultNational.py
index 9eaec53..0b71c96 100644
--- a/routers/scrapResultNational.py
+++ b/routers/scrapResultNational.py
@@ -316,7 +316,7 @@ async def getNationalTemplateData(
 
 @router.get("/chart-data")
 async def getNationalChartData(
-    factor: FactorType,
+    factor: FactorType, year: int = 2020
 ) -> ErrorResponse | ChartData[GenderChartDataPoint] | ChartData[
     AgeChartDataPoint
 ] | ChartData[PartyChartDataPoint]:
@@ -328,11 +328,10 @@ async def getNationalChartData(
                     {
                         "councilorType": "national_councilor",
                         "level": 0,
+                        "year": year,
                         "is_elected": True,
                     }
                 )
-                .sort({"year": -1})
-                .limit(1)
                 .to_list(5)
             )[0]
 
@@ -354,10 +353,9 @@ async def getNationalChartData(
                         "level": 0,
                         "is_elected": True,
                         "method": "equal",
+                        "year": year,
                     }
                 )
-                .sort({"year": -1})
-                .limit(1)
                 .to_list(5)
             )[0]
             age_list = [
@@ -385,10 +383,9 @@ async def getNationalChartData(
                         "councilorType": "national_councilor",
                         "level": 0,
                         "is_elected": True,
+                        "year": year,
                     }
                 )
-                .sort({"year": -1})
-                .limit(1)
                 .to_list(5)
             )[0]
             return ChartData[PartyChartDataPoint].model_validate(