pretalx-plugin-broadcast-tools/pretalx_lower_thirds/views.py

import datetime as dt

import pytz
from django.http import JsonResponse
from django.views.generic import FormView
from django.views.generic.base import TemplateView
from pretalx.agenda.views.schedule import ScheduleMixin
from pretalx.common.mixins.views import EventPermissionRequired, PermissionRequired
from pretalx.schedule.exporters import ScheduleData

from .forms import LowerThirdsSettingsForm


class LowerThirdsView(TemplateView):
    template_name = "pretalx_lower_thirds/lower_thirds.html"


class LowerThirdsOrgaView(PermissionRequired, FormView):
    form_class = LowerThirdsSettingsForm
    permission_required = "orga.change_settings"
    template_name = "pretalx_lower_thirds/orga.html"

    def get_success_url(self):
        return self.request.path

    def form_valid(self, form):
        form.save()
        return super().form_valid(form)

    def get_object(self):
        return self.request.event

    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        return {
            "obj": self.request.event,
            "attribute_name": "settings",
            "locales": self.request.event.locales,
            **kwargs,
        }


class ScheduleView(EventPermissionRequired, ScheduleMixin, TemplateView):
    permission_required = "agenda.view_schedule"

    def get(self, request, *args, **kwargs):
        schedule = ScheduleData(
            event=self.request.event,
            schedule=self.schedule,
        )
        tz = pytz.timezone(schedule.event.timezone)
        infoline = str(schedule.event.settings.infoline or "")
        return JsonResponse(
            {
                "conference": {
                    "slug": schedule.event.slug,
                    "name": str(schedule.event.name),
                    "no_talk": str(schedule.event.settings.lower_thirds_no_talk_info),
                },
                "rooms": sorted(
                    {
                        str(room["name"])
                        for day in schedule.data
                        for room in day["rooms"]
                    }
                ),
                "talks": [
                    {
                        "id": talk.submission.id,
                        "start": talk.start.astimezone(tz).isoformat(),
                        "end": (talk.start + dt.timedelta(minutes=talk.duration))
                        .astimezone(tz)
                        .isoformat(),
                        "slug": talk.frab_slug,
                        "title": talk.submission.title,
                        "persons": sorted(
                            {
                                person.get_display_name()
                                for person in talk.submission.speakers.all()
                            }
                        ),
                        "track": {
                            "color": talk.submission.track.color,
                            "name": str(talk.submission.track.name),
                        }
                        if talk.submission.track
                        else None,
                        "room": str(room["name"]),
                        "infoline": infoline.format(
                            EVENT_SLUG=str(schedule.event.slug),
                            TALK_SLUG=talk.frab_slug,
                            CODE=talk.submission.code,
                        ),
                    }
                    for day in schedule.data
                    for room in day["rooms"]
                    for talk in room["talks"]
                ],
            },
            json_dumps_params={
                "indent": 4,
            },
        )