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

🚨 Fix mypy warnings about primary_key in fieldInfo

Browse Source
This commit is contained in:
Esteban Maya Cadavid 2024-07-03 08:47:27 -05:00
parent 76d72cd32e
commit 79a2ba84e5
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
sqlmodel/_compat.py
View File

@ -240,7 +240,11 @@ if IS_PYDANTIC_V2:
if name != "model_fields":
model_fields = object.__getattribute__(self, "model_fields")
field = model_fields.get(name)
if field is not None and isinstance(field, FieldInfo):
if (
field is not None
and isinstance(field, FieldInfo)
and hasattr(field, "primary_key")
):
if field.primary_key and field.annotation is int and value is None:
raise ValueError(
f"Primary key attribute '{name}' has not been set, please commit() it first."
@ -547,7 +551,11 @@ else:
if name != "__fields__":
fields = object.__getattribute__(self, "__fields__")
field = fields.get(name)
if field is not None and isinstance(field.field_info, FieldInfo):
if (
field is not None
and isinstance(field.field_info, FieldInfo)
and hasattr(field.field_info, "primary_key")
):
if (
field.field_info.primary_key
and field.annotation is int