如何在嵌套的ModelForm中包含OneToOne子项

如何解决如何在嵌套的ModelForm中包含OneToOne子项

我正在尝试创建一个包含条件的OneToOne子模型的ModelForm。对于上下文，我将模型的相关部分总结如下：

models.py

class Quote(models.Model):
    ...

class ScreenOperationType(models.Model):
    ...

class Screen(models.Model):
    quote = models.ForeignKey(Quote,on_delete=models.PROTECT)
    operation_type = models.ForeignKey(ScreenOperationType,on_delete=models.PROTECT)
    ...

class ScreenMotor(models.Model):
    screen = models.OneToOneField(Screen,on_delete=models.PROTECT)
    ...

class ScreenCrank(models.Model):
   screen = models.OneToOneField(Screen,on_delete=models.PROTECT)
   ...

报价可以具有任意数量的屏幕。屏幕具有以下四种操作类型之一：电动机，曲柄，固定或无。仅当operation_type为'motor'时存在ScreenMotor子记录，仅当screen operation_type为'crank'时存在ScreenCrank子记录。如果操作固定或不执行，则不存在子记录。

我想以相同的形式创建所有嵌套记录。我已经为带子屏幕内联表单的报价创建了一个表单，但是如何在Screen ModelForm中包括ScreenMotor和ScreenCrank的字段？我曾尝试为ScreenCrank和ScreenMotor编写一个ModelForm，但是将它们嵌套为另一个表单集似乎并不会产生预期的结果，因为它们是OneToOne。

我知道我将不得不编写javascript来显示/隐藏给孩子的输入，但是现在我很高兴只一次显示Screen，ScreenMotor和ScreenCrank的输入，并让QuoteCreateView form_valid函数确定哪些字段相关。

forms.py：

import re

from crispy_forms.helper import FormHelper
from crispy_forms.layout import Layout,Field,Fieldset,Div,HTML,ButtonHolder,Submit,Row
from django import forms
from django.forms.models import inlineformset_factory

from .custom_layout_object import Formset
from .models import Screen,Quote


class ScreenMotorForm(forms.ModelForm):

    class Meta:
        model = ScreenMotor
        fields = ('color','motor_type',)

class ScreenCrankForm(forms.ModelForm):

    class Meta:
        model = ScreenCrank
        fields = ('position','length',)

class ScreenForm(forms.ModelForm):

    # How do I add fields for ScreenMotorForm and ScreenCrankForm to this?

    class Meta:
        model = Screen
        fields = ('fabric_color','quote',)

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

        formtag_prefix = re.sub('-[0-9]+$','',kwargs.get('prefix',''))

        self.helper = FormHelper(self)
        self.helper.form_tag = False
        self.helper.render_hidden_fields = True
        self.helper.layout = Layout(
            Row(
                Field('fabric_color'),Field('DELETE'),css_class='formset_row-{}'.format(formtag_prefix)
            )
        )

ScreenFormSet = inlineformset_factory(
    Quote,Screen,form=ScreenForm,fields=['fabric_color'],extra=1,can_delete=True
)

class QuoteForm(forms.ModelForm):

    class Meta:
        model = Quote
        fields = ('status',**kwargs):
        super(QuoteForm,self).__init__(*args,**kwargs)
        self.helper = FormHelper()
        self.helper.form_tag = True
        self.helper.layout = Layout(
            Div(
                Fieldset(
                    'Add screens',Formset('screens')
                ),Field('status'),HTML('<br>'),ButtonHolder(Submit('submit','save')),)
        )

views.py

from django.views.generic import CreateView
from django.urls import reverse
from django.contrib.auth.mixins import LoginRequiredMixin,PermissionRequiredMixin
from django.db import transaction


class QuoteCreateView(LoginRequiredMixin,PermissionRequiredMixin,CreateView):

    permission_required = 'quotes.add_quote'
    model = Quote
    form_class = QuoteForm

    def get_context_data(self,**kwargs):
        data = super(QuoteCreateView,self).get_context_data(**kwargs)
        if self.request.POST:
            data['screens'] = ScreenFormSet(self.request.POST)
        else:
            data['screens'] = ScreenFormSet()
        return data

    def form_valid(self,form):
        context = self.get_context_data()
        screens = context['screens']
        with transaction.atomic():
            form.instance.written_by = self.request.user
            self.object = form.save()
            if screens.is_valid():
                screens.instance = self.object
                screens.save()
        return super(QuoteCreateView,self).form_valid(form)

    def get_success_url(self):
        return reverse('quotes:detail',args=[str(self.object.id)])

quote_form.html

{% extends 'main/base.html' %}

{% load crispy_forms_tags %}

{% block content %}
  <form method="post">
    {% csrf_token %}
    {% crispy form %}
  </form>
{% endblock %}

这可能无关紧要，但我应该注意，我基于发现的here解决方案使用了第三方django-crispy-forms和django-dynamic-formset。我是Django的新手，所以能提供的任何帮助将不胜感激。