根据同一模型中的另一个外键动态限制 Django 模型中外键的​​选择

Question

我有这些模型：

class UserProfile(models.Model):
    name = models.CharField(max_length=100)

class Dialog(models.Model):
    belong_to = models.ManyToManyField(UserProfile)

class Message(models.Model):
    # Dialog to which this message belongs
    part_of = models.ForeignKey(Dialog)

    # User who sends message
    sender = models.ForeignKey(UserProfile, related_name='sender')
    # User who receives message 
    receiver = models.ForeignKey(UserProfile, related_name='receiver')

我想要做的是限制发送者和接收者字段的选择，以便他们只能是整个对话框所属的用户。 我试过这个：

sender = models.ForeignKey(UserProfile,
                           related_name='sender',
                           limit_choices_to={'dialog':1})

这限制了选择，但仅限于 id=1 的对话成员。我想知道这是否可以动态完成？

Answer 1

如果Message 的实例都属于Dialog，为什么不在Dialog 模型上创建一个字段messages？然后，您可以将发送者和接收者附加到每个Dialog。简而言之，大致如下：

class Dialog(models.Model):
    messages = models.ManyToManyField(Message)
    sender = models.ForeignKey(UserProfile)
    receiver = models.ForeignKey(UserProfile)

class Message(models.Model):
    # Other fields

Message 的发送者和接收者始终是 Dialog 所属的对象。

Answer 2

我不相信有任何方法可以像使用 limit_choices_to 那样动态过滤，因为您无法访问所需的对象来在那里形成这样的查询。

相反，您可能应该为消息创建自己的模型表单并在那里为这些字段设置查询集。像下面这样...

class MessageForm(forms.ModelForm):
    class Meta:
        model = Message

    def __init__(self, *args, **kwargs):
        super(MessageForm, self).__init__(*args, **kwargs)

        if self.instance.part_of and self.instance.part_of.id:
            users = self.instance.part_of.belong_to.all()
            self.fields['sender'].queryset = users
            self.fields['receiver'].queryset = users

此外，为什么 limit_choices_to 适用于您的示例，但在动态方面没有用处。

Django 只是将 limit_choices_to 表达式作为一个额外的过滤器处理，以应用于 ModelForm 字段查询集。您的表达式 {dialog: 1} 在语义上与在我的示例中将 UserProfile.objects.filter(dialog=1) 的结果分配给查询集时没有区别。

Django 不知道具有该 id 的对话框是否作为 UserProfile 上的关系存在，它只是应用过滤器。在这种情况下，存在一个 id 为 1 的对话框，因此可以正常工作。如果您在示例中粘贴了无效的对话框 id..它将评估为空查询集，您将在表单中获得 0 个选项。

它不能是动态的，因为在limit_choices_to 中您只能为 UserProfile 模型创建过滤器表达式。您无权访问该字段所属的 Message 实例，也无权访问该消息所属的 Dialog 模型...因此您无法创建过滤器来动态限制这些实例。

创建您自己的 ModelForm 并限制那里的字段的查询集，您可以在其中获得所需的信息，这是正确的方法。