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 BroadcastToolsSettingsForm class BroadcastToolsLowerThirdsView(TemplateView): template_name = "pretalx_broadcast_tools/lower_thirds.html" class BroadcastToolsOrgaView(PermissionRequired, FormView): form_class = BroadcastToolsSettingsForm permission_required = "orga.change_settings" template_name = "pretalx_broadcast_tools/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 BroadcastToolsEventInfoView(TemplateView): def get(self, request, *args, **kwargs): color = self.request.event.primary_color or "#3aa57c" return JsonResponse( { "slug": self.request.event.slug, "name": str(self.request.event.name), "no_talk": str(self.request.event.settings.broadcast_tools_lower_thirds_no_talk_info), "color": color, }, ) class BroadcastToolsScheduleView(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.broadcast_tools_lower_thirds_info_string or "") return JsonResponse( { "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"] ], }, )