Django ModelForm如何根据URL参数预填OneToOneField字段

问题场景 #

使用包含OneToOneField的ModelForm时，希望通过URL中的product_pk参数预填ModelChoiceField（比如访问catalogue/location/add/7时自动关联PK=7的产品）；若URL无该参数，则表单不预填。

现有代码 #

models.py #

class Product(models.Model):
    pass

class ProductLocation(geo_models.Model):
    product = geo_models.OneToOneField(Product, verbose_name=_('Product'), on_delete=geo_models.CASCADE, null=True,)
    geo_point = geo_models.PointField(verbose_name=_('Location'), blank=True)
    address = geo_models.CharField(verbose_name=_('Address'), max_length=255)

urls.py #

urls += [
    re_path(r"^location/add/(?P<product_pk>\d+)/$",
            self.add_update_location.as_view(),
            name='catalogue-product-location-add'),  # 预填产品字段

    path("location/add",
         self.add_update_location.as_view(),
         name='catalogue-location-create'),  # 产品字段为空

    path("location/<int:pk>",
         self.add_update_location.as_view(),
         name='catalogue-location-update')  # 编辑已有位置
]

views.py #

class LocationCreateUpdateView(generic.UpdateView):
    form_class = ProductLocationForm

    creating = True
    product = None

    def get_object(self, queryset=None):
        """
        扩展get_object以获取URL中指定的产品
        """
        self.creating = "pk" not in self.kwargs
        if self.creating:
            if self.kwargs.get("product_pk"):
                self.product = get_object_or_404(Product, pk=self.kwargs["product_pk"])
            return None
        else:
            return super().get_object(queryset)

    # 尝试方法1
    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        if self.product and not self.request.POST.get('product'):
            kwargs["product"] = self.product
        return kwargs

    # 尝试方法2
    def get_initial(self):
        initial = super().get_initial()
        if self.product:
            initial["product"] = self.product
        return initial

forms.py #

class ProductLocationForm(forms.ModelForm):
    def __init__(self, product=None, instance=None, *args, **kwargs):
        # 尝试过的操作
        if not instance and product:
            instance = ProductLocation(product=product)
        super().__init__(instance=instance, *args, **kwargs)

遇到的问题 #

• 重写get_initial()传递产品到表单后，无法通过验证，仍要求选择产品，且初始值被清除
• 使用get_form_kwargs()传递产品到表单__init__时，尝试过以下操作但均无效：
  • 将产品关联到ProductLocation实例
  • 在__init__中设置初始数据
  • 设置self.fields['product'].initial = product

解决方案 #

1. 调整表单逻辑 #

修改ProductLocationForm，如果传入product参数，直接设置字段初始值，还可根据需求隐藏该字段避免用户修改：

class ProductLocationForm(forms.ModelForm):
    class Meta:
        model = ProductLocation
        fields = ['product', 'geo_point', 'address']

    def __init__(self, product=None, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 新建场景下传入产品时设置初始值
        if product and not self.instance.pk:
            self.initial['product'] = product.pk
            # 可选：隐藏字段，禁止用户修改预填产品
            self.fields['product'].widget = forms.HiddenInput()

2. 优化视图的get_form_kwargs #

确保product参数仅在GET渲染表单时传递，避免覆盖POST提交的内容：

class LocationCreateUpdateView(generic.UpdateView):
    form_class = ProductLocationForm
    model = ProductLocation
    creating = True
    product = None

    def get_object(self, queryset=None):
        self.creating = "pk" not in self.kwargs
        if self.creating:
            if self.kwargs.get("product_pk"):
                self.product = get_object_or_404(Product, pk=self.kwargs["product_pk"])
            return None
        else:
            return super().get_object(queryset)

    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        # 仅在新建、有product且为GET请求时传递参数
        if self.creating and self.product and self.request.method != 'POST':
            kwargs['product'] = self.product
        return kwargs

3. 验证逻辑说明 #

ModelChoiceField的初始值需要传入主键值而非对象实例，设置后表单验证会自动识别该值；如果隐藏了字段，也不会影响验证流程。

内容的提问来源于stack exchange，提问作者Hugo Delannoy