82 lines
3.7 KiB
Python
82 lines
3.7 KiB
Python
|
from itertools import chain
|
||
|
|
|
||
|
|
from django.apps import apps
|
||
|
|
|
||
|
|
from c3nav.api.schema import BaseSchema
|
||
|
|
from c3nav.editor.operations import DatabaseOperationCollection, CreateObjectOperation, UpdateObjectOperation, \
|
||
|
|
DeleteObjectOperation, ClearManyToManyOperation, FieldValuesDict, ObjectReference, PreviousObjectCollection
|
||
|
|
from c3nav.mapdata.fields import I18nField
|
||
|
|
|
||
|
|
|
||
|
|
class ChangedManyToMany(BaseSchema):
|
||
|
|
cleared: bool = False
|
||
|
|
added: list[str] = []
|
||
|
|
removed: list[str] = []
|
||
|
|
|
||
|
|
|
||
|
|
class ChangedObject(BaseSchema):
|
||
|
|
obj: ObjectReference
|
||
|
|
titles: dict[str, str] | None
|
||
|
|
created: bool = False
|
||
|
|
deleted: bool = False
|
||
|
|
fields: FieldValuesDict = {}
|
||
|
|
m2m_changes: dict[str, ChangedManyToMany] = {}
|
||
|
|
|
||
|
|
|
||
|
|
class ChangedObjectCollection(BaseSchema):
|
||
|
|
"""
|
||
|
|
A collection of ChangedObject instances, sorted by model and id.
|
||
|
|
Also stores a PreviousObjectCollection for comparison with the current state.
|
||
|
|
Iterable as a list of ChangedObject instances.
|
||
|
|
"""
|
||
|
|
prev: PreviousObjectCollection = PreviousObjectCollection()
|
||
|
|
objects: dict[str, dict[int, ChangedObject]] = {}
|
||
|
|
|
||
|
|
def __iter__(self):
|
||
|
|
yield from chain(*(objects.keys() for model, objects in self.objects.items()))
|
||
|
|
|
||
|
|
def add_operations(self, operations: DatabaseOperationCollection):
|
||
|
|
"""
|
||
|
|
Add the given operations, creating/updating changed objects to represent the resulting state.
|
||
|
|
"""
|
||
|
|
# todo: merge prev
|
||
|
|
for operation in operations.operations:
|
||
|
|
changed_object = self.objects.setdefault(operation.obj.model, {}).get(operation.obj.id, None)
|
||
|
|
if changed_object is None:
|
||
|
|
changed_object = ChangedObject(obj=operation.obj,
|
||
|
|
titles=self.prev.get(operation.obj).titles)
|
||
|
|
self.objects[operation.obj.model][operation.obj.id] = changed_object
|
||
|
|
if isinstance(operation, CreateObjectOperation):
|
||
|
|
changed_object.created = True
|
||
|
|
changed_object.fields.update(operation.fields)
|
||
|
|
elif isinstance(operation, UpdateObjectOperation):
|
||
|
|
model = apps.get_model('mapdata', operation.obj.model)
|
||
|
|
for field_name, value in operation.fields.items():
|
||
|
|
field = model._meta.get_field(field_name)
|
||
|
|
if isinstance(field, I18nField) and field_name in changed_object.fields:
|
||
|
|
changed_object.fields[field_name] = {
|
||
|
|
lang: val for lang, val in {**changed_object.fields[field_name], **value}.items()
|
||
|
|
}
|
||
|
|
else:
|
||
|
|
changed_object.fields[field_name] = value
|
||
|
|
elif isinstance(operation, DeleteObjectOperation):
|
||
|
|
changed_object.deleted = False
|
||
|
|
else:
|
||
|
|
changed_m2m = changed_object.m2m_changes.get(operation.field, None)
|
||
|
|
if changed_m2m is None:
|
||
|
|
changed_m2m = ChangedManyToMany()
|
||
|
|
changed_object.m2m_changes[operation.field] = changed_m2m
|
||
|
|
if isinstance(operation, ClearManyToManyOperation):
|
||
|
|
changed_m2m.cleared = True
|
||
|
|
changed_m2m.added = []
|
||
|
|
changed_m2m.removed = []
|
||
|
|
else:
|
||
|
|
changed_m2m.added = sorted((set(changed_m2m.added) | operation.add_values)
|
||
|
|
- operation.remove_values)
|
||
|
|
changed_m2m.removed = sorted((set(changed_m2m.removed) - operation.add_values)
|
||
|
|
| operation.remove_values)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def as_operations(self) -> DatabaseOperationCollection:
|
||
|
|
|
||
|
|
pass # todo: implement
|