MPTT tree копирование ветвей?

Question

[Владимир Куц]

Может кто сталкивался с проблемой:
Есть дерево PolymorphicMPTTModel. Появилась задача - если меняется родитель произвольной ветки, то копируем данные ветки на новое место, оставляя на старом месте копию ветки со старыми id с пометками объектов "deleted".
Я перехватываю сохранение объекта, смотрю - если parent у него меняется - то создаю ветку с топологией старой ветки, копируя данные с помощью генератора self.get_descendants(include_self=True). А parent оригинальной ветки меняю на новый.
Все хорошо - только теперь копия ветки deleted - с новыми id, - то есть ветка перенеслась на новое место, а на ее месте возникла ветка deleted с НОВЫМИ id.
Как наименьшей кровью у ветки deleted оставить СТАРЫЕ id, а у перемещенной - создать новые?

Answer 1

[Anatoly Scherbakov]

Как я понимаю, проблема у вас в том, что вы перехватываете сохранение, когда parent уже изменён. Я бы делал это примерно так.

class MyModel(models.Model):
    def __init__(self, *args, **kwargs):
        super(MyModel, self).__init__(*args, **kwargs)
        self._initial_parent = self.parent

    def save(self, *args, **kwargs):
        new_parent = self.parent
        if new_parent != self._initial_parent:
            # Marking old branch as obsolete
            self.descendants().update(is_deleted=True)

            # Now, creating a new branch (with new IDs) at the destination node
            root = MyModel.objects.create(parent=new_parent, ...)
            [MyModel.objects.create(...) ...]

            # And cancelling the moving of the current element
            self.parent = self._initial_parent
            self.is_deleted = True

        return super(MyModel, self).save(*args, **kwargs)

Comments

[Владимир Куц]

Перехватываю я вовремя. Дело в том что объекты сильно развесистые - на несколько десятков полей, полиморфные - в зависимости от типа имеют разную структуру. Плюс к тому имеют поля с переводами на разные языки. Этот участок кода выливается в головняк:
# Now, creating a new branch (with new IDs) at the destination node
root = MyModel.objects.create(parent=new_parent, ...)
[MyModel.objects.create(...) ...]
Поэтому моя задумка - перемещать ветку, а на ее место вставлять ветку-заглушку из нод, в которых будет только статус "удалено" и время удаления с сохранением топологии - старые id и старые связи на родительские ноды. Контент не нужен - он в перемещенной ветке. Засада пока только в том, что при перемещении ветки ее айдишники старые остаются, а у ветки заглушки-новые. А нужно наоборот.

[Владимир Куц]

Привязка к старым айдишникам нужна, так как дерево завязано на REST-api

[Anatoly Scherbakov]

А. Теперь я понял вашу проблему. Ну, поле `id` переписывать нельзя, но тогда есть гораздо более простое решение. Добавьте ещё одно поле, IntegerField, назовите его `permanent_id` или ещё как-нибудь. Используйте его в вашем REST API и обновляйте его при `save()`, задавая по умолчанию равным `id`.

[Владимир Куц]

Anatoly Scherbakov: да - я тоже уже склоняюсь к подобному решению, - единственно много кода перелопатить придется, но похоже - пока это самое оптимальное.