Проблема привязки имени к классу в Django - PullRequest
0 голосов
/ 20 апреля 2010

У меня есть функция просмотра, которая должна решить, какую форму использовать в зависимости от некоторых условий. Две формы выглядят так:

class OpenExtraForm(forms.ModelForm):
    class Meta:
        model = Extra

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

        self.fields['opening_challenge'].label = "lame translation"

    def clean_opening_challenge(self):
        challenge = self.cleaned_data['opening_challenge']
        if challenge is None:
            raise forms.ValidationError('Укажите шаг, открывающий данную доп. возможность')
        return challenge

class HiddenExtraForm(forms.ModelForm):
    class Meta:
        model = Extra
        exclude = ('opening_challenge')

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

Код вида выглядит так:

@login_required
def manage_extra(request, extra_id=None, hidden=False):
    if not_admin(request.user):
        raise Http404
    if extra_id is None:
        # Adding a new extra
        extra = Extra()
        if hidden:
            FormClass = HiddenExtraForm
        else:
            FormClass = OpenExtraForm
    else:
        # Editing an extra
        extra = get_object_or_404(Extra, pk=extra_id)
        if extra.is_hidden():
            FromClass = HiddenExtraForm
        else:
            FormClass = OpenExtraForm

    if request.POST:
        form = FormClass(request.POST, instance=extra)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect(reverse(view_extra, args=[extra.id]))
    else:
        form = FormClass(instance=extra)
    return render_to_response('form.html', { 'form' : form,
                                              }, context_instance=RequestContext(request) )

Проблема в том, что почему-то, если extra.is_hidden() возвращает True, оператор FromClass = HiddenExtraForm не работает. Я имею в виду, что во всех других условиях, которые используются в коде, все работает нормально: правильные классы Form созданы, и все работает. Но если extra.is_hidden(), то отладчик показывает, что условие выполнено, переходит на следующую строку и ничего не делает! В результате я получаю ошибку UnboundLocalVar, которая говорит, что FormClass вообще не был назначен.

Есть идеи о том, что происходит?

1 Ответ

4 голосов
/ 20 апреля 2010

Вам нужно выбрать между FromClass и FormClass. Вы используете FormClass везде, кроме:

    if extra.is_hidden():
        FromClass = HiddenExtraForm
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...