ff840ab3c6
- Reorder assignment ballot tabs in detail template. - Removed logic to set phase automatically to 'finished' if elected candidates reached, because autoupdate makes it unreliable.
445 lines
13 KiB
Python
445 lines
13 KiB
Python
from collections import OrderedDict
|
|
|
|
from django.conf import settings
|
|
from django.contrib.contenttypes.fields import GenericRelation
|
|
from django.db import models
|
|
from django.utils.translation import ugettext as _
|
|
from django.utils.translation import ugettext_noop
|
|
|
|
from openslides.agenda.models import Item, Speaker
|
|
from openslides.core.config import config
|
|
from openslides.core.models import Tag
|
|
from openslides.poll.models import (
|
|
BaseOption,
|
|
BasePoll,
|
|
BaseVote,
|
|
CollectDefaultVotesMixin,
|
|
PublishPollMixin,
|
|
)
|
|
from openslides.utils.autoupdate import inform_changed_data
|
|
from openslides.utils.exceptions import OpenSlidesError
|
|
from openslides.utils.models import RESTModelMixin
|
|
|
|
from .access_permissions import AssignmentAccessPermissions
|
|
|
|
|
|
class AssignmentRelatedUser(RESTModelMixin, models.Model):
|
|
"""
|
|
Many to Many table between an assignment and user.
|
|
"""
|
|
|
|
assignment = models.ForeignKey(
|
|
'Assignment',
|
|
on_delete=models.CASCADE,
|
|
related_name='assignment_related_users')
|
|
"""
|
|
ForeinKey to the assignment.
|
|
"""
|
|
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE)
|
|
"""
|
|
ForeinKey to the user who is related to the assignment.
|
|
"""
|
|
|
|
elected = models.BooleanField(default=False)
|
|
"""
|
|
Saves the election state of each user
|
|
"""
|
|
|
|
weight = models.IntegerField(default=0)
|
|
"""
|
|
The sort order of the candidates.
|
|
"""
|
|
|
|
class Meta:
|
|
default_permissions = ()
|
|
unique_together = ('assignment', 'user')
|
|
|
|
def __str__(self):
|
|
return "%s <-> %s" % (self.assignment, self.user)
|
|
|
|
def get_root_rest_element(self):
|
|
"""
|
|
Returns the assignment to this instance which is the root REST element.
|
|
"""
|
|
return self.assignment
|
|
|
|
|
|
class AssignmentManager(models.Manager):
|
|
"""
|
|
Customized model manager to support our get_full_queryset method.
|
|
"""
|
|
def get_full_queryset(self):
|
|
"""
|
|
Returns the normal queryset with all assignments. In the background
|
|
all related users (candidates), the related agenda item and all
|
|
polls are prefetched from the database.
|
|
"""
|
|
return self.get_queryset().prefetch_related(
|
|
'related_users',
|
|
'agenda_items',
|
|
'polls',
|
|
'tags')
|
|
|
|
|
|
class Assignment(RESTModelMixin, models.Model):
|
|
"""
|
|
Model for assignments.
|
|
"""
|
|
access_permissions = AssignmentAccessPermissions()
|
|
|
|
objects = AssignmentManager()
|
|
|
|
PHASE_SEARCH = 0
|
|
PHASE_VOTING = 1
|
|
PHASE_FINISHED = 2
|
|
|
|
PHASES = (
|
|
(PHASE_SEARCH, 'Searching for candidates'),
|
|
(PHASE_VOTING, 'Voting'),
|
|
(PHASE_FINISHED, 'Finished'),
|
|
)
|
|
|
|
title = models.CharField(
|
|
max_length=100)
|
|
"""
|
|
Title of the assignment.
|
|
"""
|
|
|
|
description = models.TextField(
|
|
blank=True)
|
|
"""
|
|
Text to describe the assignment.
|
|
"""
|
|
|
|
open_posts = models.PositiveSmallIntegerField()
|
|
"""
|
|
The number of members to be elected.
|
|
"""
|
|
|
|
poll_description_default = models.CharField(
|
|
max_length=79,
|
|
blank=True)
|
|
"""
|
|
Default text for the poll description.
|
|
"""
|
|
|
|
phase = models.IntegerField(
|
|
choices=PHASES,
|
|
default=PHASE_SEARCH)
|
|
"""
|
|
Phase in which the assignment is.
|
|
"""
|
|
|
|
related_users = models.ManyToManyField(
|
|
settings.AUTH_USER_MODEL,
|
|
through='AssignmentRelatedUser')
|
|
"""
|
|
Users that are candidates or elected.
|
|
|
|
See AssignmentRelatedUser for more information.
|
|
"""
|
|
|
|
tags = models.ManyToManyField(Tag, blank=True)
|
|
"""
|
|
Tags for the assignment.
|
|
"""
|
|
|
|
# In theory there could be one then more agenda_item. But we support only
|
|
# one. See the property agenda_item.
|
|
agenda_items = GenericRelation(Item, related_name='assignments')
|
|
|
|
class Meta:
|
|
default_permissions = ()
|
|
permissions = (
|
|
('can_see', 'Can see elections'),
|
|
('can_nominate_other', 'Can nominate another participant'),
|
|
('can_nominate_self', 'Can nominate oneself'),
|
|
('can_manage', 'Can manage elections'),
|
|
)
|
|
ordering = ('title', )
|
|
verbose_name = ugettext_noop('Election')
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def get_slide_context(self, **context):
|
|
"""
|
|
Retuns the context to generate the assignment slide.
|
|
"""
|
|
return super().get_slide_context(
|
|
polls=self.polls.filter(published=True),
|
|
vote_results=self.vote_results(only_published=True),
|
|
**context)
|
|
|
|
@property
|
|
def candidates(self):
|
|
"""
|
|
Queryset that represents the candidates for the assignment.
|
|
"""
|
|
return self.related_users.filter(
|
|
assignmentrelateduser__elected=False)
|
|
|
|
@property
|
|
def elected(self):
|
|
"""
|
|
Queryset that represents all elected users for the assignment.
|
|
"""
|
|
return self.related_users.filter(
|
|
assignmentrelateduser__elected=True)
|
|
|
|
def is_candidate(self, user):
|
|
"""
|
|
Returns True if user is a candidate.
|
|
|
|
Costs one database query.
|
|
"""
|
|
return self.candidates.filter(pk=user.pk).exists()
|
|
|
|
def is_elected(self, user):
|
|
"""
|
|
Returns True if the user is elected for this assignment.
|
|
|
|
Costs one database query.
|
|
"""
|
|
return self.elected.filter(pk=user.pk).exists()
|
|
|
|
def set_candidate(self, user):
|
|
"""
|
|
Adds the user as candidate.
|
|
"""
|
|
weight = self.assignment_related_users.aggregate(
|
|
models.Max('weight'))['weight__max'] or 0
|
|
defaults = {
|
|
'elected': False,
|
|
'weight': weight + 1}
|
|
related_user, __ = self.assignment_related_users.update_or_create(
|
|
user=user,
|
|
defaults=defaults)
|
|
|
|
def set_elected(self, user):
|
|
"""
|
|
Makes user an elected user for this assignment.
|
|
"""
|
|
related_user, __ = self.assignment_related_users.update_or_create(
|
|
user=user,
|
|
defaults={'elected': True})
|
|
|
|
def delete_related_user(self, user):
|
|
"""
|
|
Delete the connection from the assignment to the user.
|
|
"""
|
|
self.assignment_related_users.filter(user=user).delete()
|
|
inform_changed_data(self)
|
|
|
|
def set_phase(self, phase):
|
|
"""
|
|
Sets the phase attribute of the assignment.
|
|
|
|
Raises a ValueError if the phase is not valide.
|
|
"""
|
|
if phase not in dict(self.PHASES):
|
|
raise ValueError("Invalid phase %s" % phase)
|
|
|
|
self.phase = phase
|
|
|
|
def create_poll(self):
|
|
"""
|
|
Creates a new poll for the assignment and adds all candidates to all
|
|
lists of speakers of related agenda items.
|
|
"""
|
|
candidates = self.candidates.all()
|
|
|
|
# Find out the method of the election
|
|
if config['assignments_poll_vote_values'] == 'votes':
|
|
pollmethod = 'votes'
|
|
elif config['assignments_poll_vote_values'] == 'yesnoabstain':
|
|
pollmethod = 'yna'
|
|
elif config['assignments_poll_vote_values'] == 'yesno':
|
|
pollmethod = 'yn'
|
|
else:
|
|
# config['assignments_poll_vote_values'] == 'auto'
|
|
# candidates <= available posts -> yes/no/abstain
|
|
if len(candidates) <= (self.open_posts - self.elected.count()):
|
|
pollmethod = 'yna'
|
|
else:
|
|
pollmethod = 'votes'
|
|
|
|
# Create the poll with the candidates.
|
|
poll = self.polls.create(
|
|
description=self.poll_description_default,
|
|
pollmethod=pollmethod)
|
|
options = []
|
|
related_users = AssignmentRelatedUser.objects.filter(assignment__id=self.id).exclude(elected=True)
|
|
for related_user in related_users:
|
|
options.append({
|
|
'candidate': related_user.user,
|
|
'weight': related_user.weight})
|
|
poll.set_options(options)
|
|
|
|
# Add all candidates to list of speakers of related agenda item
|
|
# TODO: Try to do this in a bulk create
|
|
for candidate in self.candidates:
|
|
try:
|
|
Speaker.objects.add(candidate, self.agenda_item)
|
|
except OpenSlidesError:
|
|
# The Speaker is already on the list. Do nothing.
|
|
# TODO: Find a smart way not to catch the error concerning AnonymousUser.
|
|
pass
|
|
|
|
return poll
|
|
|
|
def vote_results(self, only_published):
|
|
"""
|
|
Returns a table represented as a list with all candidates from all
|
|
related polls and their vote results.
|
|
"""
|
|
vote_results_dict = OrderedDict()
|
|
|
|
polls = self.polls.all()
|
|
if only_published:
|
|
polls = polls.filter(published=True)
|
|
|
|
# All PollOption-Objects related to this assignment
|
|
options = []
|
|
for poll in polls:
|
|
options += poll.get_options()
|
|
|
|
for option in options:
|
|
candidate = option.candidate
|
|
if candidate in vote_results_dict:
|
|
continue
|
|
vote_results_dict[candidate] = []
|
|
for poll in polls:
|
|
votes = {}
|
|
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()
|
|
except AssignmentOption.DoesNotExist:
|
|
# candidate not in related to this poll
|
|
votes = None
|
|
vote_results_dict[candidate].append(votes)
|
|
return vote_results_dict
|
|
|
|
def get_agenda_title(self):
|
|
return str(self)
|
|
|
|
def get_agenda_list_view_title(self):
|
|
"""
|
|
Return a title string for the agenda list view.
|
|
|
|
Contains agenda item number, title and assignment verbose name.
|
|
Note: It has to be the same return value like in JavaScript.
|
|
"""
|
|
return '%s (%s)' % (self.title, _(self._meta.verbose_name))
|
|
|
|
@property
|
|
def agenda_item(self):
|
|
"""
|
|
Returns the related agenda item.
|
|
"""
|
|
# We support only one agenda item so just return the first element of
|
|
# the queryset.
|
|
return self.agenda_items.all()[0]
|
|
|
|
@property
|
|
def agenda_item_id(self):
|
|
"""
|
|
Returns the id of the agenda item object related to this object.
|
|
"""
|
|
return self.agenda_item.pk
|
|
|
|
|
|
class AssignmentVote(RESTModelMixin, BaseVote):
|
|
option = models.ForeignKey(
|
|
'AssignmentOption',
|
|
on_delete=models.CASCADE,
|
|
related_name='votes')
|
|
|
|
class Meta:
|
|
default_permissions = ()
|
|
|
|
def get_root_rest_element(self):
|
|
"""
|
|
Returns the assignment to this instance which is the root REST element.
|
|
"""
|
|
return self.option.poll.assignment
|
|
|
|
|
|
class AssignmentOption(RESTModelMixin, BaseOption):
|
|
poll = models.ForeignKey(
|
|
'AssignmentPoll',
|
|
on_delete=models.CASCADE,
|
|
related_name='options')
|
|
candidate = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE)
|
|
weight = models.IntegerField(default=0)
|
|
|
|
vote_class = AssignmentVote
|
|
|
|
class Meta:
|
|
default_permissions = ()
|
|
|
|
def __str__(self):
|
|
return str(self.candidate)
|
|
|
|
def get_root_rest_element(self):
|
|
"""
|
|
Returns the assignment to this instance which is the root REST element.
|
|
"""
|
|
return self.poll.assignment
|
|
|
|
|
|
class AssignmentPoll(RESTModelMixin, CollectDefaultVotesMixin,
|
|
PublishPollMixin, BasePoll):
|
|
option_class = AssignmentOption
|
|
|
|
assignment = models.ForeignKey(
|
|
Assignment,
|
|
on_delete=models.CASCADE,
|
|
related_name='polls')
|
|
pollmethod = models.CharField(
|
|
max_length=5,
|
|
default='yna')
|
|
description = models.CharField(
|
|
max_length=79,
|
|
blank=True)
|
|
|
|
class Meta:
|
|
default_permissions = ()
|
|
|
|
def get_assignment(self):
|
|
return self.assignment
|
|
|
|
def get_vote_values(self):
|
|
if self.pollmethod == 'yna':
|
|
return ['Yes', 'No', 'Abstain']
|
|
elif self.pollmethod == 'yn':
|
|
return ['Yes', 'No']
|
|
else:
|
|
return ['Votes']
|
|
|
|
def get_ballot(self):
|
|
return self.assignment.polls.filter(id__lte=self.pk).count()
|
|
|
|
def get_percent_base_choice(self):
|
|
return config['assignments_poll_100_percent_base']
|
|
|
|
def append_pollform_fields(self, fields):
|
|
fields.append('description')
|
|
super().append_pollform_fields(fields)
|
|
|
|
def get_slide_context(self, **context):
|
|
return super().get_slide_context(poll=self)
|
|
|
|
def get_root_rest_element(self):
|
|
"""
|
|
Returns the assignment to this instance which is the root REST element.
|
|
"""
|
|
return self.assignment
|