如何在通用 UpdateView 中為 django 表單小部件指定查詢集

Question

我的模型：

class TarifGe(models.Model):
    mise_a_disposition = models.ForeignKey(MiseADisposition)
    tarif_son = models.ForeignKey("TarifGe", on_delete=models.CASCADE, null=True, default=None, blank=True)

我的看法 ：

class TarifGeUpdate(LoginRequiredMixin, UpdateView):
    model = TarifGe
    template_name = "tarifs_form.html"
    fields = ['tarif_son',]

tarif_fils 是一個 TarifGe 相關對象（父 -> 子）

我想過濾 de tarif_son字段的小部件的初始值，只擁有具有相同mise_a_disposition字段的 TarifGe 對象，而不是所有 TarifGe 對象。

我在哪里以及如何為“tarif_son”字段的小部件指定查詢集？

在此先感謝您的幫助。

Answer 1

您可以覆蓋.get_form(…)方法 [Django-doc] ：

class TarifGeUpdate(LoginRequiredMixin, UpdateView):
    model = TarifGe
    template_name = "tarifs_form.html"
    fields = ['tarif_son',]

    def get_form(self, *args, **kwargs):
        form = super().get_form(*args, **kwargs)
        form.fields['tarif_son'].queryset = TarifGe.objects.filter(
            mise_a_disposition=self.object.mise_a_disposition
        )
        return form