Sith/election/models.py

76 lines
2.6 KiB
Python
Raw Normal View History

2016-12-05 19:18:03 +00:00
from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.utils import timezone
from django.conf import settings
from datetime import timedelta
from core.models import User
from subscription.models import Subscriber
from subscription.views import get_subscriber
class Election(models.Model):
"""
2016-12-07 22:55:05 +00:00
This class allows to create a new election
2016-12-05 19:18:03 +00:00
"""
title = models.CharField(_('title'), max_length=255)
description = models.TextField(_('description'), null=True, blank=True)
2016-12-07 22:55:05 +00:00
start_proposal = models.DateTimeField(_('start proposal'), blank=False)
end_proposal = models.DateTimeField(_('end proposal'), blank=False)
2016-12-05 19:18:03 +00:00
start_date = models.DateTimeField(_('start date'), blank=False)
end_date = models.DateTimeField(_('end date'), blank=False)
def __str__(self):
return self.title
@property
def is_active(self):
now = timezone.now()
return bool(now <= self.end_date and now >= self.start_date)
2016-12-07 22:55:05 +00:00
@property
def is_proposal_active(self):
now = timezone.now()
return bool(now <= self.end_proposal and now >= self.start_proposal)
2016-12-05 22:49:13 +00:00
def has_voted(self, user):
2016-12-07 22:55:05 +00:00
return self.vote.filter(user__id=user.id).exists()
2016-12-05 22:49:13 +00:00
2016-12-05 19:18:03 +00:00
def get_results(self):
pass
class Responsability(models.Model):
"""
2016-12-07 22:55:05 +00:00
This class allows to create a new responsability
2016-12-05 19:18:03 +00:00
"""
2016-12-05 22:49:13 +00:00
election = models.ForeignKey(Election, related_name='responsability', verbose_name=_("election"))
2016-12-05 19:18:03 +00:00
title = models.CharField(_('title'), max_length=255)
description = models.TextField(_('description'), null=True, blank=True)
def __str__(self):
return ("%s : %s") % (self.election.title, self.title)
class Candidate(models.Model):
"""
2016-12-07 22:55:05 +00:00
This class is a component of responsability
2016-12-05 19:18:03 +00:00
"""
2016-12-05 22:49:13 +00:00
responsability = models.ForeignKey(Responsability, related_name='candidate', verbose_name=_("responsability"))
2016-12-07 22:55:05 +00:00
user = models.ForeignKey(User, verbose_name=_('user'), related_name='candidate', blank=True)
2016-12-05 19:18:03 +00:00
program = models.TextField(_('description'), null=True, blank=True)
def __str__(self):
2016-12-07 22:55:05 +00:00
return ("%s : %s -> %s") % (self.election.title, self.title, self.user.get_full_name())
class Vote(models.Model):
"""
This class allows to vote for candidates
"""
election = models.ForeignKey(Election, related_name='vote', verbose_name=_("election"))
candidate = models.ManyToManyField(Candidate, related_name='vote', verbose_name=_("candidate"))
user = models.ForeignKey(User, related_name='vote', verbose_name=_("user"))
def __str__(self):
return "Vote"