Fixed date and date-time types convert rules (#4)

pydantic_glue/handler.py

@@ -18,6 +18,10 @@ def dispatch(v: dict[str, Any]) -> str:
         return handle_array(v)
 
     if t == "string":
+        if v.get("format") == "date-time":
+            return "timestamp"
+        if v.get("format") == "date":
+            return "date"
         return "string"
 
     if t == "boolean":

pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "pydantic-glue"
 keywords = ["pydantic", "glue", "athena", "types", "convert"]
-version = "0.2.1"
+version = "0.3.0"
 description = "Convert pydantic model to aws glue schema for terraform"
 authors = ["Serhii Dimchenko <[email protected]>"]
 readme = "README.md"

tests/unit/test_convert.py

@@ -1,3 +1,4 @@
+import datetime
 import json
 from typing import Optional, Union
 
@@ -33,6 +34,22 @@ class A(BaseModel):
     assert convert(json.dumps(A.model_json_schema())) == expected
 
 
+def test_single_date_column():
+    class A(BaseModel):
+        modifiedOn: datetime.date
+
+    expected = [("modifiedOn", "date")]
+    assert convert(json.dumps(A.model_json_schema())) == expected
+
+
+def test_single_datetime_column():
+    class A(BaseModel):
+        modifiedOn: datetime.datetime
+
+    expected = [("modifiedOn", "timestamp")]
+    assert convert(json.dumps(A.model_json_schema())) == expected
+
+
 def test_multiple_string_column():
     class A(BaseModel):
         hey: str