rezometz/django-paiji2-survey

from django import template
from django.db.models import Count
from django.utils.translation import ugettext as _

from graphos.renderers import morris
from graphos.sources.model import ModelDataSource

from ..models import Poll
from ..forms import SurveyVoteForm


register = template.Library()


@register.inclusion_tag(
    'paiji2_survey/chart.html',
    takes_context=True,
)
def display_poll_chart(context, poll, choices=None):
    choices = poll.choices.annotate(nb_votes=Count('votes__user'))
    data_source = ModelDataSource(choices, fields=[
        'stripped_value',
        'nb_votes'
    ])
    chart = morris.DonutChart(data_source, width=250, height=250, options={
        'title': _('Votes'),
    })
    if choices is not None:
        context['choices'] = choices
    return {
        'chart': chart,
    }


@register.inclusion_tag(
    'paiji2_survey/form.html',
    takes_context=True,
)
def display_survey_form(context):
    try:
        poll = Poll.objects.current()
    except Poll.DoesNotExist:
        return {
            'no_poll': True,
        }

    # We get the vote for the current user
    vote = poll.vote_for(context['request'].user)
    if vote is not None:
        return {
            'has_voted': True,
            'poll': poll,
        }
    else:
        return {
            'form': SurveyVoteForm(poll=poll),
            'poll': poll,
        }