7

在提交表单时，我很难弄清楚如何为 ForeignKey 字段自动创建模型实例。这是一个说明问题的简单玩具网站：

我有两个模型，Model1 和 Model2。Model2 包含 Model1 的 ForeignKey。我希望用户能够通过专门选择要存储在 ForeignKey 中的 Model1 的实例，或者将该值留空并让 Model1 的实例自动生成来创建 Model2 的实例。

这就是我觉得该代码应该是什么样子的。我的 models.py 代码非常简单：

# models.py
from django.db import models
from django.core.validators import MinValueValidator


class Model1(models.Model):

    # Note this field cannot be negative
    my_field1 = models.IntegerField(validators=[MinValueValidator(0)])


class Model2(models.Model):
    # blank = True will make key_to_model1 not required on the form,
    # but since null = False, I will still require the ForeignKey
    # to be set in the database.
    related_model1 = models.ForeignKey(Model1, blank=True)

    # Note this field cannot be negative
    my_field2 = models.IntegerField(validators=[MinValueValidator(0)])

forms.py 有点涉及，但发生的事情非常简单。如果 Model2Form 没有收到 Model1 的实例，它会尝试在 clean 方法中自动创建一个，验证它，如果它有效，它会保存它。如果它无效，则会引发异常。

#forms.py
from django import forms
from django.forms.models import model_to_dict

from .models import Model1, Model2


# A ModelForm used for validation purposes only.
class Model1Form(forms.ModelForm):
    class Meta:
        model = Model1


class Model2Form(forms.ModelForm):
    class Meta:
        model = Model2

    def clean(self):
        cleaned_data = super(Model2Form, self).clean()

        if not cleaned_data.get('related_model1', None):

            # Don't instantiate field2 if it doesn't exist.
            val = cleaned_data.get('my_field2', None)
            if not val:
                raise forms.ValidationError("My field must exist")

            # Generate a new instance of Model1 based on Model2's data
            new_model1 = Model1(my_field1=val)

            # validate the Model1 instance with a form form
            validation_form_data = model_to_dict(new_model1)
            validation_form = Model1Form(validation_form_data)

            if not validation_form.is_valid():
                raise forms.ValidationError("Could not create a proper instance of Model1.")

            # set the model1 instance to the related model and save it to the database.
            new_model1.save()
            cleaned_data['related_model1'] = new_model1

        return cleaned_data

但是，这种方法行不通。如果我在表单中输入有效数据，它就可以正常工作。但是，如果我没有为 ForeignKey 输入任何内容并为整数输入一个负值，我会得到一个 ValueError。

回溯：get_response 111 中的文件“/Library/Python/2.7/site-packages/django/core/handlers/base.py”。响应=回调（请求，*callback_args，**callback_kwargs）文件“/Library/Python/2.7 /site-packages/django/views/generic/base.py" 在视图 48. return self.dispatch(request, *args, **kwargs) File "/Library/Python/2.7/site-packages/django/views/ generic/base.py” 在 dispatch 69. return handler(request, *args, **kwargs) File "/Library/Python/2.7/site-packages/django/views/generic/edit.py" in post 172. return super(BaseCreateView, self).post(request, *args, **kwargs) 文件 "/Library/Python/2.7/site-packages/django/views/generic/edit.py" 在帖子 137. if form.is_valid( ): 文件“/库/Python/2.7/site-packages/django/forms/forms.py" in is_valid 124. 返回 self.is_bound 而不是 bool(self.errors) 文件 "/Library/Python/2.7/site-packages/django/forms/forms.py " 在 _get_errors 115. self.full_clean() 文件 "/Library/Python/2.7/site-packages/django/forms/forms.py" 在 full_clean 272. self._post_clean() 文件 "/Library/Python/2.7/site -packages/django/forms/models.py" in _post_clean 309. self.instance = construction_instance(self, self.instance, opts.fields, opts.exclude) 文件 "/Library/Python/2.7/site-packages/django/ forms/models.py”在construct_instance 51. f.save_form_data(instance,cleaned_data[f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/返回 self.is_bound 而不是 bool(self.errors) File "/Library/Python/2.7/site-packages/django/forms/forms.py" in _get_errors 115. self.full_clean() File "/Library/Python/2.7 /site-packages/django/forms/forms.py" 在 full_clean 272. self._post_clean() 文件 "/Library/Python/2.7/site-packages/django/forms/models.py" 在 _post_clean 309. self.instance =construct_instance（self，self.instance，opts.fields，opts.exclude）文件“/Library/Python/2.7/site-packages/django/forms/models.py”在construct_instance 51. f.save_form_data（instance，cleaned_data [ f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/返回 self.is_bound 而不是 bool(self.errors) File "/Library/Python/2.7/site-packages/django/forms/forms.py" in _get_errors 115. self.full_clean() File "/Library/Python/2.7 /site-packages/django/forms/forms.py" 在 full_clean 272. self._post_clean() 文件 "/Library/Python/2.7/site-packages/django/forms/models.py" 在 _post_clean 309. self.instance =construct_instance（self，self.instance，opts.fields，opts.exclude）文件“/Library/Python/2.7/site-packages/django/forms/models.py”在construct_instance 51. f.save_form_data（instance，cleaned_data [ f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/py" in _get_errors 115. self.full_clean() 文件 "/Library/Python/2.7/site-packages/django/forms/forms.py" 在 full_clean 272. self._post_clean() 文件 "/Library/Python/2.7/ site-packages/django/forms/models.py" in _post_clean 309. self.instance = construction_instance(self, self.instance, opts.fields, opts.exclude) 文件 "/Library/Python/2.7/site-packages/django /forms/models.py”在construct_instance 51. f.save_form_data(instance,cleaned_data[f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/py" in _get_errors 115. self.full_clean() 文件 "/Library/Python/2.7/site-packages/django/forms/forms.py" 在 full_clean 272. self._post_clean() 文件 "/Library/Python/2.7/ site-packages/django/forms/models.py" in _post_clean 309. self.instance = construction_instance(self, self.instance, opts.fields, opts.exclude) 文件 "/Library/Python/2.7/site-packages/django /forms/models.py”在construct_instance 51. f.save_form_data(instance,cleaned_data[f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/7/site-packages/django/forms/models.py" in _post_clean 309. self.instance = construction_instance(self, self.instance, opts.fields, opts.exclude) 文件 "/Library/Python/2.7/site-packages /django/forms/models.py”在construct_instance 51. f.save_form_data(instance,clean_data[f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/7/site-packages/django/forms/models.py" in _post_clean 309. self.instance = construction_instance(self, self.instance, opts.fields, opts.exclude) 文件 "/Library/Python/2.7/site-packages /django/forms/models.py”在construct_instance 51. f.save_form_data(instance,clean_data[f.name]) 文件“/Library/Python/2.7/site-packages/django/db/models/fields/init .py" in save_form_data 454. setattr(instance, self.name, data) 文件 "/Library/Python/2.7/site-packages/django/db/models/fields/related.py" 在set 362. (instance. _meta.object_name，self.field.name))

异常类型：ValueError at /add/ 异常值：无法分配无：“Model2.related_model1”不允许空值。

所以，发生的事情是 Django 正在捕获我的 ValidationError 并仍然创建 Model2 的实例，即使验证失败。

如果出现错误，我可以通过覆盖 _post_clean 方法来解决此问题，以便不创建 Model2 的实例。但是，这个解决方案是丑陋的。特别是，_post_clean 的行为通常非常有帮助——在更复杂的项目中，我需要 _post_clean 来运行其他原因。

我也可以允许 ForeignKey 为空，但在实践中从不将其设置为空。但是，再一次，这似乎是个坏主意。

我什至可以设置一个虚拟的 Model1，每当对尝试的新 Model1 的验证失败时，我都会使用它，但这似乎也很骇人听闻。

一般来说，我可以想出很多技巧来解决这个问题，但我不知道如何以一种相当干净、pythonic 的方式来解决这个问题。

4

1 回答 1

1

根据 karthikr 在评论中的讨论，我找到了一个我认为可能可以接受的解决方案。我绝对仍然对替代品持开放态度。

这个想法是使用视图中的逻辑在两种表单之间进行选择来进行验证：一种是标准模型表单，一种是没有 ForeignKey 字段的模型表单。

所以，我的 models.py 是相同的。

我的 forms.py 有两个 Model2 表单……一个非常简单，一个没有 ForeignKey 字段，并且具有新的逻辑来为 ForeignKey 动态生成 Model1 的新实例。新表单的干净逻辑就是我以前放在 Model2Form 中的干净逻辑：

#forms.py
from django import forms
from django.forms.models import model_to_dict

from .models import Model1, Model2


# A ModelForm used for validation purposes only.
class Model1Form(forms.ModelForm):
    class Meta:
        model = Model1


class Model2Form(forms.ModelForm):
    class Meta:
        model = Model2

# This inherits from Model2Form so that any additional logic that I put in Model2Form
# will apply to it.
class Model2FormPrime(Model2Form):
    class Meta:
        model = Model2
        exclude = ('related_model1',)

    def clean(self):
        cleaned_data = super(Model2Form, self).clean()

        if cleaned_data.get('related_model1', None):
            raise Exception('Huh? This should not happen...')

        # Don't instantiate field2 if it doesn't exist.
        val = cleaned_data.get('my_field2', None)
        if not val:
            raise forms.ValidationError("My field must exist")

        # Generate a new instance of Model1 based on Model2's data
        new_model1 = Model1(my_field1=val)

        # validate the Model1 instance with a form form
        validation_form_data = model_to_dict(new_model1)
        validation_form = Model1Form(validation_form_data)

        if not validation_form.is_valid():
            raise forms.ValidationError("Could not create a proper instance of Model1.")

        # set the Model1 instance to the related model and save it to the database.
        cleaned_data['related_model1'] = new_model1

        return cleaned_data

    def save(self, commit=True):
        # Best to wait til save is called to save the instance of Model1
        # so that instances aren't created when the Model2Form is invalid
        self.cleaned_data['related_model1'].save()

        # Need to handle saving this way because otherwise related_model1 is excluded
        # from the save due to Meta.excludes
        instance = super(Model2FormPrime, self).save(False)
        instance.related_model1 = self.cleaned_data['related_model1']
        instance.save()

        return instance

然后我的视图逻辑使用两种形式之一进行验证，具体取决于发布数据。如果它使用 Model2FormPrime 并且验证失败，它会将数据和错误移动到常规 Model2Form 以向用户显示：

# Create your views here.
from django.views.generic.edit import CreateView
from django.http import HttpResponseRedirect

from .forms import Model2Form, Model2FormPrime


class Model2CreateView(CreateView):
    form_class = Model2Form
    template_name = 'form_template.html'
    success_url = '/add/'

    def post(self, request, *args, **kwargs):
        if request.POST.get('related_model', None):
            # Complete data can just be sent to the standard CreateView form
            return super(Model2CreateView, self).post(request, *args, **kwargs)
        else:
            # super does this, and I won't be calling super.
            self.object = None

            # use Model2FormPrime to validate the post data without the related model.
            validation_form = Model2FormPrime(request.POST)
            if validation_form.is_valid():
                return self.form_valid(validation_form)
            else:
                # Create a normal instance of Model2Form to be displayed to the user
                # Insantiate it with post data and validation_form's errors
                form = Model2Form(request.POST)
                form._errors = validation_form._errors
                return self.form_invalid(form)

此解决方案有效，而且非常灵活。我可以将逻辑添加到我的模型和基础 Model2Form 中，而不必担心会破坏它或违反 DRY。

不过，它有点难看，因为它需要我使用两种表单来完成一个工作，在表单之间传递错误。因此，如果有人可以提出任何建议，我绝对愿意接受替代解决方案。

于 2013-04-05T06:29:57.267 回答