From a3b94f18efa6101fd9b70cdfe934488749337b9f Mon Sep 17 00:00:00 2001 From: Redume Date: Sat, 14 Dec 2024 23:36:57 +0300 Subject: [PATCH] fix(chart): Fixed an error if the date in the request was different from the YYYYY-MM-DD format --- chart/function/create_chart.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/chart/function/create_chart.py b/chart/function/create_chart.py index 1bf7b35..a376fa4 100644 --- a/chart/function/create_chart.py +++ b/chart/function/create_chart.py @@ -8,6 +8,9 @@ from ..database.server import create_pool async def create_chart(from_currency: str, conv_currency: str, start_date: str, end_date: str) -> (str, None): pool = await create_pool() + if not validate_date(start_date) or not validate_date(end_date): + return None + start_date_obj = datetime.strptime(start_date, '%Y-%m-%d').date() end_date_obj = datetime.strptime(end_date, '%Y-%m-%d').date() @@ -52,3 +55,11 @@ async def create_chart(from_currency: str, conv_currency: str, start_date: str, fig.clear() return name + + +def validate_date(date_str: str) -> bool: + try: + datetime.strptime(date_str, '%Y-%m-%d') + return True + except ValueError: + return False