Django:在保存覆盖中创建并保存另一个模型

时间:2014-12-02 18:29:55

标签: django

我有三个模型:Node,Link和Path。链接是两个节点之间的关系,路径是节点列表。我试图覆盖路径保存功能,以在路径中的所有相邻节点之间创建链接。我在Path模型中编写了一个add_link函数,并在Path模型中为所有相邻对调用save函数。虽然Path正确保存,并且我可以通过控制台中的add_link函数创建链接,但它们不是在Path的保存功能中创建的。我错过了什么?

以下是模型:

class Node(models.Model):
    title = models.CharField(max_length=200, blank=True)
    links = models.ManyToManyField('self', through='Link',
                                           symmetrical=False,
                                           related_name='related_to+')
    def add_link(self, other, symm=True):
        link, created = Link.objects.get_or_create(
            from_node=self,
            to_node=other)
        if symm:
            # avoid recursion by passing `symm=False`
            other.add_link(self, False)
        return link

class Link(models.Model):
    from_node = models.ForeignKey(Node, related_name="from")
    to_node = models.ForeignKey(Node, related_name="to")

class Path(models.Model):
    nodes = models.ManyToManyField(Node, related_name="nodes",through='PathNodeRelationship')

    def save(self, *args, **kwargs):
        super(Path, self).save(*args, **kwargs)

        # save all not-existent links on this path
        nodes = self.nodes.all()
        if nodes:
            f = nodes[0]
            i = 1
            while i < len(nodes):
                s = nodes[i]

                f.add_link(s)
                f = s
                i += 1

class PathNodeRelationship(models.Model):
    node = models.ForeignKey(Node)
    path = models.ForeignKey(Path)
    order_index = models.IntegerField()

**编辑:在控制台中调用path.save()时会创建链接,但在使用管理界面时则不会。这就是我在做管理员的方式。 **

class NodeInline(admin.TabularInline):
    model = Path.nodes.through
    extra = 1

class PathAdmin(admin.ModelAdmin):
      inlines = (NodeInline,)
admin.site.register(Path, PathAdmin)

第二次修改:大约3 - 4年前看起来这是an issue,管理员m2m有一些hacky修复...我还没有发现是否有任何东西现在更好了。

1 个答案:

答案 0 :(得分:3)

我不确定你可能会出现什么问题,因为当我测试你的代码时它会起作用。

对于它的价值,一个建议就是让这段代码更优雅......

    def save(self, *args, **kwargs):
        super(Path, self).save(*args, **kwargs)

        # save all not-existent links on this path
        previous_node = None
        for node in self.nodes.all():
            if previous_node is not None:
                previous_node.add_link(node)
            previous_node = node

这是我通过shell运行的测试...

>>> Link.objects.all()
[]
>>> path = Path.objects.get(id=2)
>>> path.save()
>>> Link.objects.all()
[<Link: Link object>, <Link: Link object>, <Link: Link object>, <Link: Link object>, <Link: Link object>, <Link: Link object>]

正如您所看到的,在对具有某些PathNodeRelationships设置的Path对象运行save之后,Link对象从空列表变为填充。