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]>

Author: vortex14

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)

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")

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"]
+