Validate date type in processors

crate · Jul 18, 2022 · c364c4a · c364c4a
1 parent 9a24409
commit c364c4a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/crate/client/sqlalchemy/dialect.py b/src/crate/client/sqlalchemy/dialect.py
@@ -89,6 +89,8 @@ def result_processor(self, dialect, coltype):
         def process(value):
             if not value:
                 return
+            if isinstance(value, datetime):
+                return value.date()
             try:
                 return datetime.utcfromtimestamp(value / 1e3).date()
             except TypeError:
@@ -128,6 +130,8 @@ def result_processor(self, dialect, coltype):
         def process(value):
             if not value:
                 return
+            if isinstance(value, datetime):
+                return value
             try:
                 return datetime.utcfromtimestamp(value / 1e3)
             except TypeError: