GitHub-Proxy/sqlmodel-fix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
John Lyu 2024-12-03 18:04:46 +08:00
parent 015601cd5b
commit 66c1d93cfe
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sqlmodel/_compat.py
View File

@ -66,12 +66,16 @@ def _is_union_type(t: Any) -> bool:
finish_init: ContextVar[bool] = ContextVar("finish_init", default=True) finish_init: ContextVar[bool] = ContextVar("finish_init", default=True)
def set_polymorphic_default_value(self_instance, values): def set_polymorphic_default_value(
self_instance: _TSQLModel,
values: Dict[str, Any],
) -> bool:
"""By defalut, when init a model, pydantic will set the polymorphic_on """By defalut, when init a model, pydantic will set the polymorphic_on
value to field default value. But when inherit a model, the polymorphic_on value to field default value. But when inherit a model, the polymorphic_on
should be set to polymorphic_identity value by default.""" should be set to polymorphic_identity value by default."""
cls = type(self_instance) cls = type(self_instance)
mapper = inspect(cls) mapper = inspect(cls)
ret = False
if isinstance(mapper, Mapper): if isinstance(mapper, Mapper):
polymorphic_on = mapper.polymorphic_on polymorphic_on = mapper.polymorphic_on
if polymorphic_on is not None: if polymorphic_on is not None:
@ -87,6 +91,8 @@ def set_polymorphic_default_value(self_instance, values):
polymorphic_property.key, polymorphic_property.key,
mapper.polymorphic_identity, mapper.polymorphic_identity,
) )
ret = True
return ret
@contextmanager @contextmanager