在一个Django视图中组合两个表单

时间:2015-01-20 20:56:17

标签: python django

我正在为官方Django教程中的民意调查应用添加更多功能。我正在处理的事情之一是通过登录用户创建投票/选择(而不是在管理屏幕中,教程离开我们)。

我希望创建一个用户可以创建投票的视图,然后还包括一些与投票相关联的选项。 Django Admin自动完成它,我不知道如何在视图中写出来。

首先,这些是我的相关文件:

models.py

import datetime

from django.db import models
from django.utils import timezone


class Poll(models.Model):
    question_text = models.CharField(max_length=200)
    pub_date = models.DateTimeField('date published')

    def __unicode__(self):
        return self.question_text

    def was_published_recently(self):
        return self.pub_date >= timezone.now() - datetime.timedelta(days=1)

    was_published_recently.admin_order_field = 'pub_date'
    was_published_recently.boolean = True
    was_published_recently.short_description = 'Published recently?'

class Choice(models.Model):
    question = models.ForeignKey(Poll)
    choice_text = models.CharField(max_length=200)
    votes = models.IntegerField(default=0)

    def __unicode__(self):
        return self.choice_text

forms.py

from django import forms
from .models import Poll, Choice
from datetime import datetime

class PollForm(forms.ModelForm):
    question_text = forms.CharField(max_length=200, help_text="Please enter the question.")
    pub_date = forms.DateTimeField(widget=forms.HiddenInput(), initial = datetime.now())

    class Meta:
        model = Poll
        fields = ("__all__")

class ChoiceForm(forms.ModelForm):
    choice_text = forms.CharField(max_length=200, help_text="Please enter choices.")
    votes = forms.IntegerField(widget=forms.HiddenInput(),initial=0)
    exclude = ('poll',)

views.py

def add_poll(request):
    # A HTTP POST?
    if request.method == 'POST':
        form = PollForm(request.POST)

        # Have we been provided with a valid form?
        if form.is_valid():
            # Save the new category to the database.
            form.save(commit=True)

            # Now call the index() view.
            # The user will be shown the homepage.
            return render(request, 'polls/index.html', {})
        else:
            # The supplied form contained errors - just print them to the terminal.
            print form.errors
    else:
        # If the request was not a POST, display the form to enter details.
        form = PollForm()

    # Bad form (or form details), no form supplied...
    # Render the form with error messages (if any).
    return render(request, 'polls/add_poll.html', {'form': form})

目前,我的视图允许用户添加投票。我只是不确定如何调整它以将输入的文本作为轮询模型的question_text传递给Choice模型,然后再传递给ChoiceForm。

任何指导和建议总是受到赞赏!

干杯, 保罗

1 个答案:

答案 0 :(得分:19)

Formsets是在django中实现它的方法。

首先为default字段添加Poll.pub_date值:

class Poll(models.Model):
    question_text = models.CharField(max_length=200)
    pub_date = models.DateTimeField('date published', default=timezone.now)

然后使表单更简单:

class PollForm(forms.ModelForm):
    class Meta:
        model = Poll
        fields = ('question_text', )

class ChoiceForm(forms.ModelForm):
    class Meta:
        model = Choice
        fields = ('choice_text',)

为您的视图添加formset支持:

from django.forms.formsets import formset_factory

def add_poll(request):
    ChoiceFormSet = formset_factory(ChoiceForm, extra=3,
                                    min_num=2, validate_min=True)
    if request.method == 'POST':
        form = PollForm(request.POST)
        formset = ChoiceFormSet(request.POST)
        if all([form.is_valid(), formset.is_valid()]):
            poll = form.save()
            for inline_form in formset:
                if inline_form.cleaned_data:
                    choice = inline_form.save(commit=False)
                    choice.question = poll
                    choice.save()
            return render(request, 'polls/index.html', {})
    else:
        form = PollForm()
        formset = ChoiceFormSet()

    return render(request, 'polls/add_poll.html', {'form': form,
                                                   'formset': formset})

最后是你的模板:

<form method="post">

    {% csrf_token %}

    <table>
        {{ form }}
        {{ formset }}
    </table>

    <button>Add</button>

</form>