Make pointer serialization idempotent
This commit is contained in:
parent
600209d485
commit
f1a47aaed4
|
@ -130,14 +130,19 @@ class Pointer(Field):
|
|||
name: str = ""
|
||||
value_type: type = Record
|
||||
|
||||
def serialize(self, value: value_type) -> str:
|
||||
def serialize(self, value: value_type | str) -> str:
|
||||
return Pointer.reference(value)
|
||||
|
||||
def deserialize(self, value: str, db: TinyDB, recurse: bool = True) -> value_type:
|
||||
return Pointer.dereference(value, db, recurse)
|
||||
|
||||
@classmethod
|
||||
def reference(cls, value: Record):
|
||||
def reference(cls, value: Record | str):
|
||||
# XXX This could be smarter
|
||||
if isinstance(value, str):
|
||||
if '::' not in value:
|
||||
raise PointerReferenceError("Value {value} does not look like a reference!")
|
||||
return value
|
||||
if value:
|
||||
if not value.doc_id:
|
||||
raise PointerReferenceError(value)
|
||||
|
@ -196,7 +201,7 @@ class Collection(Field):
|
|||
for member in record[self.name]:
|
||||
target = Pointer.dereference(member, db=db, recurse=False)
|
||||
for backref in target._metadata.backrefs(type(record)):
|
||||
target[backref.name] = Pointer.reference(record)
|
||||
db.table(target._metadata.table).update(
|
||||
{backref.name: target[backref.name]}, where("uid") == target.uid
|
||||
)
|
||||
target[backref.name] = record
|
||||
db.save(target)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user