OpenSlides/openslides/assignment/models.py

298 lines
9.9 KiB
Python
Raw Normal View History

2011-07-31 10:46:29 +02:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
openslides.assignment.models
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Models for the assignment app.
2012-04-25 22:29:19 +02:00
:copyright: 2011, 2012 by OpenSlides team, see AUTHORS.
2011-07-31 10:46:29 +02:00
:license: GNU GPL, see LICENSE for more details.
"""
2012-06-11 13:43:48 +02:00
from django.core.urlresolvers import reverse
2012-07-10 11:27:06 +02:00
from django.db import models
from django.dispatch import receiver
from django.utils.translation import ugettext_lazy as _, ugettext_noop
2011-07-31 10:46:29 +02:00
from openslides.utils.user import UserField
2012-07-10 11:27:06 +02:00
from openslides.config.models import config
from openslides.config.signals import default_config_value
2012-07-10 11:27:06 +02:00
from openslides.projector.api import register_slidemodel
from openslides.projector.projector import SlideMixin
2011-07-31 10:46:29 +02:00
2012-07-10 11:27:06 +02:00
from openslides.participant.models import Profile
from openslides.poll.models import (BasePoll, CountInvalid, CountVotesCast,
BaseOption, PublishPollMixin, BaseVote)
from openslides.agenda.models import Item
class AssignmentCandidate(models.Model):
assignment = models.ForeignKey("Assignment")
user = UserField(db_index=True)
elected = models.BooleanField(default=False)
def __unicode__(self):
return unicode(self.user)
2012-03-03 09:11:56 +01:00
class Assignment(models.Model, SlideMixin):
prefix = 'assignment'
2011-07-31 10:46:29 +02:00
STATUS = (
('sea', _('Searching for candidates')),
('vot', _('Voting')),
('fin', _('Finished')),
)
name = models.CharField(max_length=100, verbose_name=_("Name"))
2012-07-10 11:27:06 +02:00
description = models.TextField(null=True, blank=True,
verbose_name=_("Description"))
posts = models.PositiveSmallIntegerField(
verbose_name=_("Number of available posts"))
polldescription = models.CharField(max_length=100, null=True, blank=True,
verbose_name=_("Comment on the ballot paper"))
status = models.CharField(max_length=3, choices=STATUS, default='sea')
2011-07-31 10:46:29 +02:00
def set_status(self, status):
error = True
for a, b in Assignment.STATUS:
if status == a:
error = False
break
if error:
raise NameError(_('%s is not a valid status.') % status)
if self.status == status:
2012-07-10 11:27:06 +02:00
raise NameError(_('The assignment status is already %s.')
% self.status)
2011-07-31 10:46:29 +02:00
self.status = status
self.save()
def run(self, profile, user=None):
2011-07-31 10:46:29 +02:00
"""
run for a vote
"""
if self.is_candidate(profile):
raise NameError(_('<b>%s</b> is already a candidate.') % profile)
if not user.has_perm("assignment.can_manage_assignment") and self.status != 'sea':
raise NameError(_('The candidate list is already closed.'))
AssignmentCandidate(assignment=self, user=profile, elected=False).save()
2011-07-31 10:46:29 +02:00
def delrun(self, user):
2011-07-31 10:46:29 +02:00
"""
stop running for a vote
"""
if self.status != 'sea':
# TODO: Use an OpenSlides Error
raise Exception(_('The candidate list is already closed.'))
if self.is_candidate(user):
assignment_candidats.get(user=user).delete()
2011-07-31 10:46:29 +02:00
else:
# TODO: Use an OpenSlides Error
raise Exception(_('%s is no candidate') % user)
2011-07-31 10:46:29 +02:00
def is_candidate(self, user):
if self.assignment_candidats.filter(user=user).exists():
2011-07-31 10:46:29 +02:00
return True
else:
return False
@property
def assignment_candidats(self):
return AssignmentCandidate.objects.filter(assignment=self)
@property
def candidates(self):
return self.get_participants(only_candidate=True)
@property
def elected(self):
return self.get_participants(only_elected=True)
def get_participants(self, only_elected=False, only_candidate=False):
candidates = self.assignment_candidats
2012-07-03 00:05:48 +02:00
if only_elected and only_candidate:
# TODO: Use right Exception
raise Exception("only_elected and only_candidate can not both be Treu")
if only_elected:
candidates = candidates.filter(elected=True)
if only_candidate:
candidates = candidates.filter(elected=False)
return [candidate.user for candidate in candidates]
#for candidate in candidates:
# yield candidate.user
def set_elected(self, user, value=True):
candidate = AssignmentCandidate.objects.get(user=user)
candidate.elected = value
candidate.save()
def is_elected(self, user):
return user in self.elected
2011-07-31 10:46:29 +02:00
def gen_poll(self):
2012-02-19 19:27:00 +01:00
poll = AssignmentPoll(assignment=self)
2011-07-31 10:46:29 +02:00
poll.save()
poll.set_options([{'candidate': user} for user in self.candidates])
2011-07-31 10:46:29 +02:00
return poll
2012-07-03 00:05:48 +02:00
def vote_results(self, only_published):
2012-07-03 00:05:48 +02:00
"""
returns a table represented as a list with all candidates from all
2012-07-04 11:00:58 +02:00
related polls and their vote results.
2012-07-03 00:05:48 +02:00
"""
vote_results_dict = {}
2012-07-04 11:00:58 +02:00
# All polls related to this assigment
polls = self.poll_set.all()
if only_published:
polls = polls.filter(published=True)
2012-07-04 11:00:58 +02:00
# All PollOption-Objects related to this assignment
2012-07-03 00:05:48 +02:00
options = []
for poll in polls:
options += poll.get_options()
2012-07-03 00:05:48 +02:00
for option in options:
candidate = option.candidate
if candidate in vote_results_dict:
continue
vote_results_dict[candidate] = []
for poll in polls:
votes = {}
2012-07-03 00:05:48 +02:00
try:
# candidate related to this poll
poll_option = poll.get_options().get(candidate=candidate)
for vote in poll_option.get_votes():
votes[vote.value] = vote.print_weight()
2012-07-03 00:05:48 +02:00
except AssignmentOption.DoesNotExist:
2012-07-04 11:00:58 +02:00
# candidate not in related to this poll
votes = None
vote_results_dict[candidate].append(votes)
2012-07-03 00:05:48 +02:00
return vote_results_dict
2012-06-23 10:27:58 +02:00
def get_agenda_title(self):
return self.name
def delete(self):
# Remove any Agenda-Item, which is related to this application.
2012-07-04 11:00:58 +02:00
for item in Item.objects.filter(related_sid=self.sid):
item.delete()
super(Assignment, self).delete()
2012-02-06 22:08:08 +01:00
def slide(self):
"""
2012-02-06 22:08:08 +01:00
return the slide dict
"""
2012-02-06 22:08:08 +01:00
data = super(Assignment, self).slide()
data['assignment'] = self
data['title'] = self.name
data['polls'] = self.poll_set.filter(published=True)
data['vote_results'] = self.vote_results(only_published=True)
2012-07-10 11:27:06 +02:00
data['assignment_publish_winner_results_only'] = \
config['assignment_publish_winner_results_only']
2012-02-06 22:08:08 +01:00
data['template'] = 'projector/Assignment.html'
return data
2011-07-31 10:46:29 +02:00
def get_absolute_url(self, link='view'):
if link == 'view':
2012-06-28 01:14:51 +02:00
return reverse('assignment_view', args=[str(self.id)])
2012-06-11 13:43:48 +02:00
if link == 'edit':
2012-06-28 01:14:51 +02:00
return reverse('assignment_edit', args=[str(self.id)])
2011-07-31 10:46:29 +02:00
if link == 'delete':
2012-06-28 01:14:51 +02:00
return reverse('assignment_delete', args=[str(self.id)])
2011-07-31 10:46:29 +02:00
def __unicode__(self):
return self.name
class Meta:
permissions = (
2012-07-10 11:27:06 +02:00
('can_see_assignment', ugettext_noop("Can see assignment")),
('can_nominate_other',
ugettext_noop("Can nominate another person")),
('can_nominate_self', ugettext_noop("Can nominate themselves")),
('can_manage_assignment', ugettext_noop("Can manage assignment")),
2011-07-31 10:46:29 +02:00
)
2012-04-14 11:18:47 +02:00
register_slidemodel(Assignment)
2012-03-03 11:16:10 +01:00
2012-02-19 19:27:00 +01:00
class AssignmentVote(BaseVote):
option = models.ForeignKey('AssignmentOption')
2012-02-19 19:27:00 +01:00
class AssignmentOption(BaseOption):
poll = models.ForeignKey('AssignmentPoll')
candidate = UserField()
vote_class = AssignmentVote
2012-02-19 19:27:00 +01:00
def __unicode__(self):
return unicode(self.candidate)
2012-03-03 11:16:10 +01:00
class AssignmentPoll(BasePoll, CountInvalid, CountVotesCast, PublishPollMixin):
2012-02-19 19:27:00 +01:00
option_class = AssignmentOption
assignment = models.ForeignKey(Assignment, related_name='poll_set')
yesnoabstain = models.NullBooleanField()
2012-02-19 19:27:00 +01:00
def get_assignment(self):
return self.assignment
def get_vote_values(self):
if self.yesnoabstain is None:
if config['assignment_poll_vote_values'] == 'votes':
self.yesnoabstain = False
elif config['assignment_poll_vote_values'] == 'yesnoabstain':
self.yesnoabstain = True
else:
# candidates <= available posts -> yes/no/abstain
if self.assignment.assignment_candidats.filter(elected=False).count() <= (self.assignment.posts
- self.assignment.assignment_candidats.filter(elected=True).count()):
self.yesnoabstain = True
else:
self.yesnoabstain = False
self.save()
if self.yesnoabstain:
return [ugettext_noop('Yes'), ugettext_noop('No'),
ugettext_noop('Abstain')]
else:
return [ugettext_noop('Votes')]
2012-02-19 19:27:00 +01:00
def append_pollform_fields(self, fields):
CountInvalid.append_pollform_fields(self, fields)
CountVotesCast.append_pollform_fields(self, fields)
2012-03-03 11:16:10 +01:00
def get_ballot(self):
return self.assignment.poll_set.filter(id__lte=self.id).count()
@models.permalink
def get_absolute_url(self, link='view'):
if link == 'view':
return ('assignment_poll_view', [str(self.id)])
if link == 'delete':
return ('assignment_poll_delete', [str(self.id)])
def __unicode__(self):
return _("Ballot %d") % self.get_ballot()
2012-03-18 17:11:58 +01:00
@receiver(default_config_value, dispatch_uid="assignment_default_config")
def default_config(sender, key, **kwargs):
return {
2012-04-22 16:55:16 +02:00
'assignment_publish_winner_results_only': False,
'assignment_pdf_ballot_papers_selection': 'CUSTOM_NUMBER',
'assignment_pdf_ballot_papers_number': '8',
'assignment_pdf_title': _('Elections'),
2012-04-22 16:55:16 +02:00
'assignment_pdf_preamble': '',
'assignment_poll_vote_values': 'auto',
}.get(key)