This commit is contained in:
Khaleel Al-Adhami 2024-10-02 11:22:41 -07:00
parent c39de662c8
commit 96e5181e14

View File

@ -69,3 +69,21 @@ def pydantic_v1_patch():
with pydantic_v1_patch(): with pydantic_v1_patch():
import sqlmodel as sqlmodel import sqlmodel as sqlmodel
def sqlmodel_field_has_primary_key(field) -> bool:
"""Determines if a field is a priamary.
Args:
field: a rx.model field
Returns:
If field is a primary key (Bool)
"""
if getattr(field.field_info, "primary_key", None) is True:
return True
if getattr(field.field_info, "sa_column", None) is None:
return False
if getattr(field.field_info.sa_column, "primary_key", None) is True:
return True
return False