Save new foreign key with django form

前端 未结 2 1311
傲寒
傲寒 2020-12-24 00:08

I have two models:

class Studio(models.Model):
    name = models.CharField(\"Studio\", max_length=30, unique=True)

class Film(models.Model):
    studio = mo         


        
2条回答
  •  臣服心动
    2020-12-24 00:32

    Really, your only problem is that you've used a standard Form instead of a ModelForm. Form doesn't have a save method because it's not inherently tied to anything (i.e. it doesn't know what to save or where to save to).

    However, if you use a ModelForm you need to take care of all logic involved in creating a new studio in the form. This is actually better, though, because then you can just use the form and not worry about anything else: the form holds all the logic needed to properly save itself.

    class FilmForm(forms.ModelForm):
        class Meta:
            model = Film
    
        # only need to define `new_studio`, other fields come automatically from model
        new_studio = forms.CharField(max_length=30, required=False, label = "New Studio Name")
    
        def __init__(self, *args, **kwargs):
            super(FilmForm, self).__init__(*args, **kwargs)
            # make `studio` not required, we'll check for one of `studio` or `new_studio` in the `clean` method
            self.fields['studio'].required = False
    
        def clean(self):
            studio = self.cleaned_data.get('studio')
            new_studio = self.cleaned_data.get('new_studio')
            if not studio and not new_studio:
                # neither was specified so raise an error to user
                raise forms.ValidationError('Must specify either Studio or New Studio!')
            elif not studio:
                # get/create `Studio` from `new_studio` and use it for `studio` field
                studio, created = Studio.objects.get_or_create(name=new_studio)
                self.cleaned_data['studio'] = studio
    
            return super(FilmForm, self).clean()
    

    Then, in your view, all you need is:

    if form.is_valid():
        form.save()
    

提交回复
热议问题