From d0918ae953b840b06a329ee41782748d556cb228 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Wed, 14 Aug 2024 22:19:56 +0200 Subject: [PATCH] Changed timelinestatistic to allow dates & ints as key, as to allow a simple 'year' key --- src/classes/statistic_classes.py | 2 +- src/main.py | 2 +- src/modules/db_connection.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/classes/statistic_classes.py b/src/classes/statistic_classes.py index 79c01cb..a09a713 100644 --- a/src/classes/statistic_classes.py +++ b/src/classes/statistic_classes.py @@ -21,7 +21,7 @@ class GameOrderStatistic(StatisticBase): } class TimeLineStatistic(StatisticBase): - result: Dict[date, int] + result: Dict[Union[date, int], int] model_config = { 'validate_assignment':True diff --git a/src/main.py b/src/main.py index 9bbe29e..38b150b 100644 --- a/src/main.py +++ b/src/main.py @@ -190,7 +190,7 @@ def get_amount_of_games_played_per_year(query: ExpansionFilteringParams = Depend for year in range(first_year_played, current_year + 1): plays_in_year = list(filter(lambda x: x.play_date.year == year, all_plays)) - years_plays_dict[datetime(year, 1,1)] = len(plays_in_year) + years_plays_dict[year] = len(plays_in_year) diff --git a/src/modules/db_connection.py b/src/modules/db_connection.py index 838e084..235e56a 100644 --- a/src/modules/db_connection.py +++ b/src/modules/db_connection.py @@ -8,7 +8,7 @@ sqlite_url = definitions.SQLITE_URL connect_args = {"check_same_thread": False} -engine = create_engine(sqlite_url, echo=False, connect_args=connect_args) +engine = create_engine(sqlite_url, echo=True, connect_args=connect_args) def get_session(): with Session(engine) as session: