OpenSlides/openslides/assignment/models.py

242 lines
8.1 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.
"""
from django.db import models
2012-06-11 13:43:48 +02:00
from django.core.urlresolvers import reverse
2011-07-31 10:46:29 +02:00
from config.models import config
2011-07-31 10:46:29 +02:00
from participant.models import Profile
2012-03-12 18:27:13 +01:00
from projector.projector import SlideMixin
2012-02-06 22:22:16 +01:00
from projector.api import register_slidemodel
2012-03-03 11:16:10 +01:00
from poll.models import BasePoll, CountInvalid, CountVotesCast, BaseOption, PublishPollMixin
2012-04-14 14:31:09 +02:00
from utils.translation_ext import ugettext as _
from agenda.models import Item
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"))
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 = _("Short description (for ballot paper)"))
2011-07-31 10:46:29 +02:00
profile = models.ManyToManyField(Profile, null=True, blank=True)
elected = models.ManyToManyField(Profile, null=True, blank=True, related_name='elected_set')
2011-07-31 10:46:29 +02:00
status = models.CharField(max_length=1, choices=STATUS, default='sea')
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:
raise NameError(_('The assignment status is already %s.') % self.status)
self.status = status
self.save()
def run(self, profile):
"""
run for a vote
"""
if self.is_candidate(profile):
raise NameError(_('<b>%s</b> is already a candidate.') % profile)
self.profile.add(profile)
def delrun(self, profile):
"""
stop running for a vote
"""
if self.is_candidate(profile):
self.profile.remove(profile)
else:
raise NameError(_('%s is no candidate') % profile)
def is_candidate(self, profile):
if profile in self.profile.get_query_set():
return True
else:
return False
@property
def candidates(self):
2012-02-19 19:27:00 +01:00
return self.profile.get_query_set()
def set_elected(self, profile, value=True):
if profile in self.candidates:
if value and not self.is_elected(profile):
self.elected.add(profile)
elif not value:
self.elected.remove(profile)
def is_elected(self, profile):
if profile in self.elected.all():
return True
return False
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()
candidates = list(self.profile.all())
for elected in self.elected.all():
try:
candidates.remove(elected)
except ValueError:
pass
poll.set_options([{'candidate': profile} for profile in candidates])
2011-07-31 10:46:29 +02:00
return poll
@property
def vote_results(self):
votes = []
publish_winner_results_only = config["assignment_publish_winner_results_only"]
# list of votes
votes = []
for candidate in self.candidates:
tmplist = [[candidate, self.is_elected(candidate)], []]
for poll in self.poll_set.all():
if poll.published:
if poll.get_options().filter(candidate=candidate).exists():
# check config option 'publish_winner_results_only'
if not publish_winner_results_only \
or publish_winner_results_only and self.is_elected(candidate):
option = AssignmentOption.objects.filter(poll=poll).get(candidate=candidate)
try:
tmplist[1].append(option.get_votes()[0])
except IndexError:
tmplist[1].append('')
else:
tmplist[1].append("")
else:
tmplist[1].append("-")
votes.append(tmplist)
return votes
2012-06-23 10:27:58 +02:00
def get_agenda_title(self):
return self.name
def delete(self):
for item in Item.objects.filter(releated_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.all()
data['votes'] = self.vote_results
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-23 10:27:58 +02:00
return reverse('assignment_view', args=(str(self.id)))
2012-06-11 13:43:48 +02:00
if link == 'edit':
2012-06-23 10:27:58 +02:00
return reverse('assignment_edit', args=(str(self.id)))
2011-07-31 10:46:29 +02:00
if link == 'delete':
2012-06-23 10:27:58 +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 = (
('can_see_assignment', _("Can see assignment", fixstr=True)),
('can_nominate_other', _("Can nominate another person", fixstr=True)),
('can_nominate_self', _("Can nominate themselves", fixstr=True)),
('can_manage_assignment', _("Can manage assignment", fixstr=True)),
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 AssignmentOption(BaseOption):
candidate = models.ForeignKey(Profile)
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')
yesnoababstain = models.NullBooleanField()
2012-02-19 19:27:00 +01:00
def get_assignment(self):
return self.assignment
def get_vote_values(self):
if not self.yesnoababstain:
if config['assignment_poll_vote_values'] == 'votes':
self.yesnoababstain = False
elif config['assignment_poll_vote_values'] == 'yesnoabstain':
self.yesnoababstain = True
else:
# candidates <= available posts -> yes/no/abstain
if self.assignment.candidates.count() <= self.assignment.posts - self.assignment.elected.count():
self.yesnoababstain = True
else:
self.yesnoababstain = False
self.save()
if self.yesnoababstain:
return ['yes', 'no', 'abstain']
else:
return ['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
from django.dispatch import receiver
from openslides.config.signals import default_config_value
@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': '1',
2012-04-22 16:55:16 +02:00
'assignment_pdf_ballot_papers_number': '1',
'assignment_pdf_title': _('Elections'),
2012-04-22 16:55:16 +02:00
'assignment_pdf_preamble': '',
'assignment_poll_vote_values': 'auto',
}.get(key)