OpenSlides/openslides/assignments/apps.py
2020-03-17 07:24:38 +01:00

88 lines
2.7 KiB
Python

from typing import Any, Dict, Set
from django.apps import AppConfig
class AssignmentsAppConfig(AppConfig):
name = "openslides.assignments"
verbose_name = "OpenSlides Assignments"
def ready(self):
# Import all required stuff.
from ..core.signals import permission_change
from ..utils.access_permissions import required_user
from ..utils.rest_api import router
from . import serializers # noqa
from .projector import register_projector_slides
from .signals import get_permission_change_data
from .views import (
AssignmentViewSet,
AssignmentPollViewSet,
AssignmentOptionViewSet,
AssignmentVoteViewSet,
)
# Define projector elements.
register_projector_slides()
# Connect signals.
permission_change.connect(
get_permission_change_data,
dispatch_uid="assignments_get_permission_change_data",
)
# Register viewsets.
router.register(
self.get_model("Assignment").get_collection_string(), AssignmentViewSet
)
router.register(
self.get_model("AssignmentPoll").get_collection_string(),
AssignmentPollViewSet,
)
router.register(
self.get_model("AssignmentOption").get_collection_string(),
AssignmentOptionViewSet,
)
router.register(
self.get_model("AssignmentVote").get_collection_string(),
AssignmentVoteViewSet,
)
# Register required_users
required_user.add_collection_string(
self.get_model("Assignment").get_collection_string(), required_users
)
def get_config_variables(self):
from .config_variables import get_config_variables
return get_config_variables()
def get_startup_elements(self):
"""
Yields all Cachables required on startup i. e. opening the websocket
connection.
"""
for model_name in (
"Assignment",
"AssignmentPoll",
"AssignmentVote",
"AssignmentOption",
):
yield self.get_model(model_name)
def required_users(element: Dict[str, Any]) -> Set[int]:
"""
Returns all user ids that are displayed as candidates (including poll
options) in the assignment element.
TODO: Adapt this method for new poll structure!!
"""
candidates = set(
related_user["user_id"] for related_user in element["assignment_related_users"]
)
for poll in element["polls"]:
candidates.update(option["candidate_id"] for option in poll["options"])
return candidates