model: check for duplicate objects
return empty dict for detached instances inside dict_recursive
This commit is contained in:
parent
848b87070c
commit
3253088700
@ -261,19 +261,32 @@ class Model(Base, sqlmodel.SQLModel): # pyright: ignore [reportGeneralTypeIssue
|
|||||||
super().__init_subclass__()
|
super().__init_subclass__()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _dict_recursive(cls, value):
|
def _dict_recursive(cls, value, seen_objects: set | None = None):
|
||||||
"""Recursively serialize the relationship object(s).
|
"""Recursively serialize the relationship object(s).
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
value: The value to serialize.
|
value: The value to serialize.
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
The serialized value.
|
The serialized value,
|
||||||
|
or Empty value when the relationship cannot be followed,
|
||||||
|
or None if the value has already been serialized in this descent.
|
||||||
"""
|
"""
|
||||||
if hasattr(value, "dict"):
|
if seen_objects is None:
|
||||||
return value.dict()
|
seen_objects = set()
|
||||||
elif isinstance(value, list):
|
if id(value) in seen_objects:
|
||||||
return [cls._dict_recursive(item) for item in value]
|
return None
|
||||||
|
seen_objects.add(id(value))
|
||||||
|
try:
|
||||||
|
if hasattr(value, "dict"):
|
||||||
|
return value.dict()
|
||||||
|
except sqlalchemy.orm.exc.DetachedInstanceError:
|
||||||
|
return {}
|
||||||
|
if isinstance(value, list):
|
||||||
|
return [
|
||||||
|
cls._dict_recursive(item, seen_objects=seen_objects)
|
||||||
|
for item in value
|
||||||
|
]
|
||||||
return value
|
return value
|
||||||
|
|
||||||
def dict(self, **kwargs):
|
def dict(self, **kwargs):
|
||||||
|
Loading…
Reference in New Issue
Block a user