OpenSlides/openslides/poll/forms.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
    openslides.poll.forms
    ~~~~~~~~~~~~~~~~~~~~~

    Forms for the poll app.

    :copyright: 2011, 2012 by OpenSlides team, see AUTHORS.
    :license: GNU GPL, see LICENSE for more details.
"""

from django import forms

from utils.forms import CssClassMixin
from utils.translation_ext import ugettext as _
from models import Vote


class OptionForm(forms.Form, CssClassMixin):
    def __init__(self, *args, **kwargs):
        extra = kwargs.pop('extra')
        formid = kwargs.pop('formid')
        kwargs['prefix'] = "option-%s" % formid
        super(OptionForm, self).__init__(*args, **kwargs)

        for vote in extra:
            if type(vote) is Vote:
                key = vote.value
                value = vote.get_value()
                weight = vote.get_weight(raw = True)
            else:
                key = vote
                value = _(vote)
                weight = None
            self.fields[key] = forms.IntegerField(
                label=value,
                initial=weight,
            )