OpenSlides/openslides/assignments/apps.py

98 lines
3.1 KiB
Python
Raw Normal View History

from typing import Any, Dict, Set
2017-08-24 12:26:55 +02:00
from django.apps import AppConfig
class AssignmentsAppConfig(AppConfig):
2019-01-06 16:22:33 +01:00
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
2019-10-18 14:18:49 +02:00
from .views import (
AssignmentViewSet,
AssignmentPollViewSet,
AssignmentOptionViewSet,
2019-10-18 14:18:49 +02:00
AssignmentVoteViewSet,
)
# Define projector elements.
register_projector_slides()
2017-02-21 09:34:24 +01:00
# Connect signals.
permission_change.connect(
get_permission_change_data,
2019-01-06 16:22:33 +01:00
dispatch_uid="assignments_get_permission_change_data",
)
2017-02-21 09:34:24 +01:00
# Register viewsets.
2019-01-06 16:22:33 +01:00
router.register(
self.get_model("Assignment").get_collection_string(), AssignmentViewSet
)
2019-10-18 14:18:49 +02:00
router.register(
self.get_model("AssignmentPoll").get_collection_string(),
AssignmentPollViewSet,
)
router.register(
self.get_model("AssignmentOption").get_collection_string(),
AssignmentOptionViewSet,
)
2019-10-18 14:18:49 +02:00
router.register(
self.get_model("AssignmentVote").get_collection_string(),
AssignmentVoteViewSet,
)
2017-01-14 12:29:42 +01:00
# Register required_users
2019-01-06 16:22:33 +01:00
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
2019-01-06 16:22:33 +01:00
return get_config_variables()
2017-01-14 12:29:42 +01:00
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",
):
2019-10-18 14:18:49 +02:00
yield self.get_model(model_name)
async 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.
"""
from openslides.assignments.models import AssignmentPoll, AssignmentOption
from openslides.utils.cache import element_cache
2019-01-06 16:22:33 +01:00
candidates = set(
related_user["user_id"] for related_user in element["assignment_related_users"]
)
for poll_id in element["polls_id"]:
poll = await element_cache.get_element_data(
AssignmentPoll.get_collection_string(), poll_id
)
if poll:
for option_id in poll["options_id"]:
option = await element_cache.get_element_data(
AssignmentOption.get_collection_string(), option_id
)
if option:
candidates.add(option["user_id"])
return candidates