added bytes type for Pydantic (#31)

* added bytes type for Pydantic * Testing a byte type field for a pydantic model * removed autogenerated models.py * W292 no newline at end of file * W292 no newline at end of file Co-authored-by: typhoon <[email protected]>
xnuinside · Apr 21, 2022 · 7a2a960 · 7a2a960
1 parent a523933
commit 7a2a960
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/omymodels/models/pydantic/types.py b/omymodels/models/pydantic/types.py
@@ -7,6 +7,7 @@
     numeric_types,
     boolean_types,
     datetime_types,
+    binary_types,
     json_types,
     text_types,
 )
@@ -21,6 +22,7 @@
     datetime_types: "datetime.datetime",
     json_types: "Json",
     text_types: "str",
+    binary_types: "bytes"
 }
 
 types_mapping = populate_types_mapping(mapper)

diff --git a/omymodels/types.py b/omymodels/types.py
@@ -24,6 +24,13 @@
     "date",
 )
 
+binary_types = (
+    "BINARY",
+    "VARBINARY",
+    "binary",
+    "varbinary",
+)
+
 json_types = ("union[dict, list]", "json", "union")
 
 integer_types = ("integer", "int", "serial")

diff --git a/tests/functional/generator/test_pydantic_models.py b/tests/functional/generator/test_pydantic_models.py
@@ -177,3 +177,23 @@ class Material(BaseModel):
     updated_at: Optional[datetime.datetime]
 """
     assert expected == result
+
+
+def test_pydantic_with_bytes():
+    expected = """from pydantic import BaseModel
+
+
+class User(BaseModel):
+
+    avatar: bytes
+"""
+
+    ddl = """
+CREATE TABLE "User" (
+    "avatar" BINARY  NOT NULL
+);
+"""
+    result = create_models(ddl, models_type="pydantic")
+
+    assert expected == result["code"]
+