src/polls/templatetags/polls_tags.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from django import template
   5 from ..forms import PollForm
   6
   7
   8 register = template.Library()
   9
  10
  11 @register.inclusion_tag('polls/tags/poll.html', takes_context=True)
  12 def poll(context, poll, show_results=True, redirect_to=''):
  13     form = None
  14     voted_already = poll.voted(context.get('request').session)
  15     if not voted_already:
  16         form = PollForm(poll=poll, initial=dict(redirect_to=redirect_to))
  17     return {'poll': poll, 'form': form, 'voted_already': voted_already, 'vote_count': poll.vote_count,
  18             'show_results': show_results, 'request': context.get('request')}