DH-4732 Datetime data types should be parsed as a string in query_res…

…ults
Dataherald · Oct 2, 2023 · b32e931 · b32e931
1 parent 52dd426
commit b32e931
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/dataherald/sql_generator/create_sql_query_status.py b/dataherald/sql_generator/create_sql_query_status.py
@@ -1,4 +1,4 @@
-from datetime import date
+from datetime import date, datetime
 from decimal import Decimal
 
 from sqlalchemy import text
@@ -30,9 +30,10 @@ def create_sql_query_status(
                 for row in result:
                     modified_row = {}
                     for key, value in zip(row.keys(), row, strict=True):
-                        if (
-                            type(value) is date
-                        ):  # Check if the value is an instance of datetime.date
+                        if type(value) in [
+                            date,
+                            datetime,
+                        ]:  # Check if the value is an instance of datetime.date
                             modified_row[key] = str(value)
                         elif (
                             type(value) is Decimal