2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

allow retrieving all ever created models, even deleted ones

Browse source
This commit is contained in:
Laura Klünder 2017-06-18 15:30:35 +02:00
parent 86de9dd053
commit 0c675ce5aa
1 changed files with 20 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/c3nav/editor/models.py
View file

@ -1,7 +1,6 @@
import json import json
import typing import typing
from collections import OrderedDict from collections import OrderedDict
from itertools import chain
from django.apps import apps from django.apps import apps
from django.conf import settings from django.conf import settings
@ -34,6 +33,7 @@ class ChangeSet(models.Model):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.default_author = None self.default_author = None
self.parsed = False self.parsed = False
self.ever_created_objects = {}
self.created_objects = {} self.created_objects = {}
self.updated_existing = {} self.updated_existing = {}
self.deleted_existing = {} self.deleted_existing = {}
@ -56,7 +56,9 @@ class ChangeSet(models.Model):
model = change.model_class model = change.model_class
if change.action == 'create': if change.action == 'create':
self.created_objects.setdefault(model, {})[change.pk] = {} new = {}
self.created_objects.setdefault(model, {})[change.pk] = new
self.ever_created_objects.setdefault(model, {})[change.pk] = new
return return
elif change.action == 'delete': elif change.action == 'delete':
if change.existing_object_pk is not None: if change.existing_object_pk is not None:
@ -98,17 +100,29 @@ class ChangeSet(models.Model):
r = tuple((pk, values[name]) for pk, values in self.updated_existing.get(model, {}).items() if name in values) r = tuple((pk, values[name]) for pk, values in self.updated_existing.get(model, {}).items() if name in values)
return r return r
def get_created_object(self, model, pk, author=None, get_foreign_objects=False): def get_created_object(self, model, pk, author=None, get_foreign_objects=False, allow_deleted=False):
if is_created_pk(pk): if is_created_pk(pk):
pk = pk[1:] pk = pk[1:]
pk = int(pk) pk = int(pk)
self.parse_changes() self.parse_changes()
if issubclass(model, ModelWrapper): if issubclass(model, ModelWrapper):
model = model._obj model = model._obj
objects = self.ever_created_objects if allow_deleted else self.created_objects
objects = tuple(obj for obj in ((submodel, objects.get(submodel, {}).get(pk, None))
for submodel in ModelWrapper.get_submodels(model)) if obj[1] is not None)
if not objects:
raise model.DoesNotExist
if len(objects) > 1:
raise model.MultipleObjectsReturned
model, data = objects[0]
obj = model() obj = model()
obj.pk = 'c' + str(pk) obj.pk = 'c' + str(pk)
for name, value in chain(*(self.created_objects.get(submodel, {}).get(pk, {}).items()
for submodel in ModelWrapper.get_submodels(model))): for name, value in data.items():
if name.startswith('title_'): if name.startswith('title_'):
obj.titles[name[6:]] = value obj.titles[name[6:]] = value
continue continue