59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
from openslides.core.exceptions import ProjectorException
|
|
from openslides.core.views import TagViewSet
|
|
from openslides.utils.projector import ProjectorElement, ProjectorRequirement
|
|
|
|
from .models import Assignment
|
|
from .views import AssignmentViewSet
|
|
|
|
|
|
class AssignmentSlide(ProjectorElement):
|
|
"""
|
|
Slide definitions for Assignment model.
|
|
|
|
Set 'id' to get a detail slide. Omit it to get a list slide.
|
|
"""
|
|
name = 'assignments/assignment'
|
|
|
|
def get_context(self):
|
|
pk = self.config_entry.get('id')
|
|
if pk is not None:
|
|
# Detail slide.
|
|
if not Assignment.objects.filter(pk=pk).exists():
|
|
raise ProjectorException('Election does not exist.')
|
|
|
|
def get_requirements(self, config_entry):
|
|
pk = config_entry.get('id')
|
|
if pk is None:
|
|
# List slide. Related objects like users and tags are not unlocked.
|
|
yield ProjectorRequirement(
|
|
view_class=AssignmentViewSet,
|
|
view_action='list')
|
|
else:
|
|
# Detail slide.
|
|
try:
|
|
assignment = Assignment.objects.get(pk=pk)
|
|
except Assignment.DoesNotExist:
|
|
# Assignment does not exist. Just do nothing.
|
|
pass
|
|
else:
|
|
yield ProjectorRequirement(
|
|
view_class=AssignmentViewSet,
|
|
view_action='retrieve',
|
|
pk=str(assignment.pk))
|
|
for user in assignment.related_users.all():
|
|
yield ProjectorRequirement(
|
|
view_class=user.get_view_class(),
|
|
view_action='retrieve',
|
|
pk=str(user.pk))
|
|
for poll in assignment.polls.all().prefetch_related('options'):
|
|
for option in poll.options.all():
|
|
yield ProjectorRequirement(
|
|
view_class=option.candidate.get_view_class(),
|
|
view_action='retrieve',
|
|
pk=str(option.candidate_id))
|
|
for tag in assignment.tags.all():
|
|
yield ProjectorRequirement(
|
|
view_class=TagViewSet,
|
|
view_action='retrieve',
|
|
pk=str(tag.pk))
|