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

fix polymorphic_on check

Browse Source
This commit is contained in:
John Lyu 2024-11-26 10:54:26 +08:00
parent 589237b80f
commit 4071b0fc64
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sqlmodel/_compat.py
View File

@ -294,6 +294,7 @@ if IS_PYDANTIC_V2:
# Override polymorphic_on default value # Override polymorphic_on default value
mapper = inspect(cls) mapper = inspect(cls)
polymorphic_on = mapper.polymorphic_on polymorphic_on = mapper.polymorphic_on
if polymorphic_on:
polymorphic_property = mapper.get_property_by_column(polymorphic_on) polymorphic_property = mapper.get_property_by_column(polymorphic_on)
field_info = cls.model_fields.get(polymorphic_property.key) field_info = cls.model_fields.get(polymorphic_property.key)
if field_info: if field_info:
@ -302,7 +303,9 @@ if IS_PYDANTIC_V2:
# set the polymorphic_on by default # set the polymorphic_on by default
if mapper.inherits or v is None: if mapper.inherits or v is None:
setattr( setattr(
self_instance, polymorphic_property.key, mapper.polymorphic_identity self_instance,
polymorphic_property.key,
mapper.polymorphic_identity,
) )
return self_instance return self_instance