2

django-reversion で関連オブジェクトを復元するにはどうすればよいですか。

私は動作しない次のことを試しました...

models.py:

@reversion.register(follow=['bar_set',])
class Foo(models.Model):
    def save(self, *args, **kwargs):
        with transaction.atomic(), reversion.create_revision():
            return super(Foo, self).save(*args, **kwargs)

@reversion.register
class Bar(models.Model):
    foo = models.ForeignKey(Foo)

    def save(self, *args, **kwargs):
        with transaction.atomic(), reversion.create_revision():
            return super(Bar, self).save(*args, **kwargs)

そして次のテスト

test.py:

class TestModels(TestCase):
    def test_soft_delete(self):
        from myapp.models import Foo, Bar
        a = Foo()
        a.save()

        b1 = Bar(foo=a)
        b1.save()
        b2 = Bar(foo=a)
        b2.save()

        a.delete()

        self.assertEqual(0, Foo.objects.count())
        self.assertEqual(0, Bar.objects.count())

        version = reversion.get_deleted(Foo)[0]  # There is only one.
        version.revert()

        self.assertEqual(1, Foo.objects.count())
        self.assertEqual(2, Bar.objects.count()) # HERE IT FAILS: 2 != 0
4

1 に答える 1

0

最後に、次善の方法で手動で行うことができました。

コードをカスタム QuerySet に入れました

class FooQuerySet(models.QuerySet):
    def restore(self, id):
        # restore
        deleted_list = reversion.get_deleted(self.model)
        deleted_list.get(id=id).revision.revert()

        # restore related 
        deleted_list = reversion.get_deleted(Bar)
        bar_set = filter(lambda s: s.field_dict['foo'] == id, deleted_list)

        for bar in bar_set:
            bar.revision.revert()

次に、Fooオブジェクトにはカスタムマネージャーが必要です

class Foo(models.Model):
    ...
    objects = FooQuerySet.as_manager()

関連するフィールドをモデルから自動的に抽出できます。しかし、私の目的には手動で行うだけで十分です。

于 2015-01-02T14:01:46.103 に答える