Mercurial > public > sg101
view polls/models.py @ 1160:b84ce2d7e93d
New template for displaying django messages.
New common template for rendering django messages in the new V3 Foundation
style.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Tue, 24 Jan 2017 19:22:29 -0600 (2017-01-25) |
parents | eeaf387803c6 |
children |
line wrap: on
line source
""" Models for the Polls application. """ import datetime from django.contrib.auth.models import User from django.core.urlresolvers import reverse from django.db import models from django.db.models import Q class PollManager(models.Manager): """Manager for the Poll model""" def get_current_polls(self): now = datetime.datetime.now() return self.filter( Q(is_enabled=True), Q(start_date__lte=now), Q(end_date__isnull=True) | Q(end_date__gte=now)) def get_old_polls(self): now = datetime.datetime.now() return self.filter( is_enabled=True, end_date__isnull=False, end_date__lt=now) class Poll(models.Model): """Model to represent polls""" start_date = models.DateTimeField(db_index=True, help_text='Date/time the poll will be eligible for voting.',) end_date = models.DateTimeField(blank=True, null=True, db_index=True, help_text='Date/time the poll will be ineligible for voting. ' 'Leave blank for an open ended poll.') is_enabled = models.BooleanField(default=True, db_index=True, help_text='Check to allow the poll to be viewed on the site.') question = models.CharField(max_length=200) objects = PollManager() def __unicode__(self): return self.question class Meta: ordering = ('-start_date', ) get_latest_by = 'start_date' def get_absolute_url(self): return reverse('polls-detail', kwargs={'poll_id': str(self.pk)}) def results(self): """ Returns a tuple; element 0 is the total votes, element 1 is a list of {choice, votes, pct} """ choices = [] total_votes = 0 for choice in self.choice_set.all(): total_votes += choice.votes choices.append({'choice': choice.choice, 'votes': choice.votes, 'pct': 0.0}) if total_votes > 0: for choice in choices: choice['pct'] = float(choice['votes']) / total_votes * 100.0 return (total_votes, choices) def _total_votes(self): """ Returns the number of votes cast in this poll to date. """ if not hasattr(self, '_total_votes_cache'): self._total_votes_cache = sum(choice.votes for choice in self.choice_set.all()) return self._total_votes_cache total_votes = property(_total_votes) def is_open(self): now = datetime.datetime.now() return self.start_date <= now and (not self.end_date or now <= self.end_date) def can_comment_on(self): return self.is_open() class Choice(models.Model): """Model for poll choices""" poll = models.ForeignKey(Poll) choice = models.CharField(max_length=200) votes = models.IntegerField(default=0) voters = models.ManyToManyField(User, blank=True) def __unicode__(self): return self.choice