在管理员外键中指定Django代理模型

时间:2019-02-06 11:02:31

标签: python django

在Django 1.11中,我有一个模型Friend和一个代理模型Relative

class FriendManager(models.Manager):
  def get_queryset(self):
    return super(RelativeManager, self).get_queryset().filter(is_relative=False)

class Friend(models.Model):
  # Model fields defined here
  objects = FriendManager()

class RelativeManager(models.Manager):
  def get_queryset(self):
    return super(RelativeManager, self).get_queryset().filter(is_relative=True)

class Relative(Friend):
  class Meta:
    proxy = True

  objects = RelativeManager()

  def save(self, *args, **kwargs):
    self.is_relative = True
    super(Relative, self).save(*args, **kwargs)

我还有一个模型FriendPortrait,该模型具有一个外键字段friend

class FriendPortrait(models.Model):
  friend = models.ForeignKey(Friend)

还有一个代理:

class RelativePortrait(FriendPortrait):
  class Meta:
    proxy = True

现在,我希望RelativePortraits的详细信息视图仅在friend的下拉列表中显示亲戚。

admin.py:

@admin.register(RelativePortrait)
class RelativePortraitAdmin(admin.ModelAdmin):
  fields = ('friend')

  def formfield_for_foreignkey(self, db_field, request, **kwargs):
    if db_field.name == 'friend':
      kwargs['queryset'] = Relative.objects.all()
    return super(RelativePortraitAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)

这有效,因为在friend下拉列表中仅显示亲戚。但是,当我尝试保存肖像时,Django管理员给了我一个验证错误:

  

ID为14的朋友实例不存在。

如何为RelativePortraitAdmin中的外键指定我要使用代理模型?

1 个答案:

答案 0 :(得分:1)

这里的问题是您的ForeignKey指向Friend模型。该模型的默认管理器会过滤掉所有亲戚,因此这将不起作用。

解决此问题的简单方法是稍微重构模型。引入类似通用Person模型的方法,并使用Friend使Relativeproxy=True从其继承。 Person模型不应具有预先过滤实例的管理器;那么您可以将ForeignKey指向某人。