from datetime import datetime, timedelta
from urllib.parse import urlparse
import pytz
from django.db import models
from django.utils.functional import cached_property
from django.utils.translation import ugettext_lazy as _
from django_scopes import ScopedManager
from pretalx.common.mixins import LogMixin
from pretalx.common.urls import get_base_url
[docs]class TalkSlot(LogMixin, models.Model):
"""The TalkSlot object is the scheduled version of a :class:`~pretalx.submission.models.submission.Submission`.
TalkSlots always belong to one submission and one :class:`~pretalx.schedule.models.schedule.Schedule`.
:param is_visible: This parameter is set on schedule release. Only confirmed talks will be visible.
"""
submission = models.ForeignKey(
to='submission.Submission', on_delete=models.PROTECT, related_name='slots'
)
room = models.ForeignKey(
to='schedule.Room',
on_delete=models.PROTECT,
related_name='talks',
null=True,
blank=True,
)
schedule = models.ForeignKey(
to='schedule.Schedule', on_delete=models.PROTECT, related_name='talks'
)
is_visible = models.BooleanField(default=False)
start = models.DateTimeField(null=True)
end = models.DateTimeField(null=True)
objects = ScopedManager(event='submission__event')
def __str__(self):
"""Help when debugging."""
return f'TalkSlot(event={self.submission.event.slug}, submission={self.submission.title}, schedule={self.schedule.version})'
@cached_property
def event(self):
return self.submission.event
@cached_property
def duration(self) -> int:
"""Returns the actual duration in minutes if the talk is scheduled, and the planned duration in minutes otherwise."""
if self.start and self.end:
return int((self.end - self.start).total_seconds() / 60)
return self.submission.get_duration()
@cached_property
def export_duration(self):
from pretalx.common.serialize import serialize_duration
return serialize_duration(minutes=self.duration)
@cached_property
def pentabarf_export_duration(self):
duration = timedelta(minutes=self.duration)
days = duration.days
hours = duration.total_seconds() // 3600 - days * 24
minutes = duration.seconds // 60 % 60
return f'{hours:02}{minutes:02}00'
@cached_property
def real_end(self):
"""Guaranteed to provide a useful end datetime if ``start`` is set, even if ``end`` is empty."""
return self.end or (
self.start + timedelta(minutes=self.duration) if self.start else None
)
@cached_property
def as_availability(self):
""" 'Casts' a slot as :class:`~pretalx.schedule.models.availability.Availability`, useful for availability arithmetics. """
from pretalx.schedule.models import Availability
return Availability(
start=self.start, end=self.real_end, event=self.submission.event
)
@cached_property
def warnings(self) -> list:
"""A list of warnings that apply to this slot.
Warnings are dictionaries with a ``type`` (``room`` or ``speaker``, for
now) and a ``message`` fit for public display. This property only
shows availability based warnings.
"""
if not self.start:
return []
warnings = []
availability = self.as_availability
if self.room:
if not any(
room_availability.contains(availability)
for room_availability in self.room.availabilities.all()
):
warnings.append(
{
'type': 'room',
'message': _(
'The room is not available at the scheduled time.'
),
}
)
for speaker in self.submission.speakers.all():
profile = speaker.event_profile(event=self.submission.event)
if profile.availabilities.exists() and not any(
speaker_availability.contains(availability)
for speaker_availability in profile.availabilities.all()
):
warnings.append(
{
'type': 'speaker',
'speaker': {
'name': speaker.get_display_name(),
'id': speaker.pk,
},
'message': _(
'A speaker is not available at the scheduled time.'
),
}
)
return warnings
[docs] def copy_to_schedule(self, new_schedule, save=True):
"""Create a new slot for the given :class:`~pretalx.schedule.models.schedule.Schedule` with all other fields identical to this one."""
new_slot = TalkSlot(schedule=new_schedule)
for field in [f for f in self._meta.fields if f.name not in ('id', 'schedule')]:
setattr(new_slot, field.name, getattr(self, field.name))
if save:
new_slot.save()
return new_slot
[docs] def is_same_slot(self, other_slot) -> bool:
"""Checks if both slots have the same room and start time."""
return self.room == other_slot.room and self.start == other_slot.start
def build_ical(self, calendar, creation_time=None, netloc=None):
if not self.start or not self.end or not self.room:
return
creation_time = creation_time or datetime.now(pytz.utc)
netloc = netloc or urlparse(get_base_url(self.event)).netloc
tz = pytz.timezone(self.submission.event.timezone)
vevent = calendar.add('vevent')
vevent.add(
'summary'
).value = f'{self.submission.title} - {self.submission.display_speaker_names}'
vevent.add('dtstamp').value = creation_time
vevent.add('location').value = str(self.room.name)
vevent.add('uid').value = 'pretalx-{}-{}@{}'.format(
self.submission.event.slug, self.submission.code, netloc
)
vevent.add('dtstart').value = self.start.astimezone(tz)
vevent.add('dtend').value = self.end.astimezone(tz)
vevent.add('description').value = self.submission.abstract or ""
vevent.add('url').value = self.submission.urls.public.full()
