我有三个模型: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修复...我还没有发现是否有任何东西现在更好了。
答案 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对象从空列表变为填充。