fix: pydantic v2 pydantic_model_creator nullable field not optional. (#…

…1454)

CHANGELOG.rst

@@ -18,6 +18,7 @@ Added
 
 Fixed
 ^^^^^
+- Fix pydantic v2 pydantic_model_creator nullable field not optional. (#1454)
 - Fix foreign key constraint not generated on MSSQL Server. (#1400)
 - Fix testcase error with python3.11 (#1308)
 

examples/blacksheep/models.py

@@ -1,6 +1,10 @@
+import tortoise
 from tortoise import fields, models
 from tortoise.contrib.pydantic import pydantic_model_creator
 
+if tortoise.__version__ >= "0.20":
+    raise RuntimeError("blacksheep not support pydantic V2, use tortoise-orm<0.20 instead!")
+
 
 class Users(models.Model):
     id = fields.UUIDField(pk=True)

examples/fastapi/main.py

@@ -22,7 +22,7 @@ async def get_users():
 
 @app.post("/users", response_model=User_Pydantic)
 async def create_user(user: UserIn_Pydantic):
-    user_obj = await Users.create(**user.dict(exclude_unset=True))
+    user_obj = await Users.create(**user.model_dump(exclude_unset=True))
     return await User_Pydantic.from_tortoise_orm(user_obj)
 
 
@@ -33,7 +33,7 @@ async def get_user(user_id: int):
 
 @app.put("/user/{user_id}", response_model=User_Pydantic)
 async def update_user(user_id: int, user: UserIn_Pydantic):
-    await Users.filter(id=user_id).update(**user.dict(exclude_unset=True))
+    await Users.filter(id=user_id).update(**user.model_dump(exclude_unset=True))
     return await User_Pydantic.from_queryset_single(Users.get(id=user_id))
 
 

examples/pydantic/tutorial_1.py

@@ -5,7 +5,7 @@
 * Creating a Pydantic model from a Tortoise model
 * Docstrings & doc-comments are used
 * Evaluating the generated schema
-* Simple serialisation with both .dict() and .json()
+* Simple serialisation with both .model_dump() and .json()
 """
 from tortoise import Tortoise, fields, run_async
 from tortoise.contrib.pydantic import pydantic_model_creator
@@ -38,7 +38,7 @@ async def run():
     tourpy = await Tournament_Pydantic.from_tortoise_orm(tournament)
 
     # As Python dict with Python objects (e.g. datetime)
-    print(tourpy.dict())
+    print(tourpy.model_dump())
     # As serialised JSON (e.g. datetime is ISO8601 string representation)
     print(tourpy.json(indent=4))
 

examples/pydantic/tutorial_2.py

@@ -46,7 +46,7 @@ async def run():
 
     # As Python dict with Python objects (e.g. datetime)
     # Note that the root element is '__root__' that contains the root element.
-    print(tourpy.dict())
+    print(tourpy.model_dump())
     # As serialised JSON (e.g. datetime is ISO8601 string representation)
     print(tourpy.json(indent=4))
 

tortoise/contrib/pydantic/creator.py

@@ -8,7 +8,7 @@
 
 from tortoise.contrib.pydantic.base import PydanticListModel, PydanticModel
 from tortoise.contrib.pydantic.utils import get_annotations
-from tortoise.fields import JSONField, relational
+from tortoise.fields import IntField, JSONField, TextField, relational
 
 if TYPE_CHECKING:  # pragma: nocoverage
     from tortoise.models import Model
@@ -348,11 +348,13 @@ def get_submodel(_model: "Type[Model]") -> Optional[Type[PydanticModel]]:
             return pmodel
 
         # Foreign keys and OneToOne fields are embedded schemas
+        is_to_one_relation = False
         if (
             field_type is relational.ForeignKeyFieldInstance
             or field_type is relational.OneToOneFieldInstance
             or field_type is relational.BackwardOneToOneRelation
         ):
+            is_to_one_relation = True
             model = get_submodel(fdesc["python_type"])
             if model:
                 if fdesc.get("nullable"):
@@ -410,6 +412,15 @@ def get_submodel(_model: "Type[Model]") -> Optional[Type[PydanticModel]]:
             if field_default is not None and not callable(field_default):
                 properties[fname] = (ftype, Field(default=field_default, **fconfig))
             else:
+                if (j := fconfig.get("json_schema_extra")) and (
+                    (
+                        j.get("nullable")
+                        and not is_to_one_relation
+                        and field_type not in (IntField, TextField)
+                    )
+                    or (exclude_readonly and j.get("readOnly"))
+                ):
+                    fconfig["default_factory"] = lambda: None
                 properties[fname] = (ftype, Field(**fconfig))
 
     # Here we endure that the name is unique, but complete objects are still labeled verbatim