ctactic/aircox-radiocampus
1
0
Fork 0
forked from rc/aircox
Code Issues 10 Pull Requests Activity

schedule & diffusions check/update + cleanup Schedule methods

Browse Source
This commit is contained in:
bkfox
2020-05-30 14:50:07 +02:00
parent 687238752c
commit dfdcf78344
8 changed files with 46 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
aircox/models/episode.py
View File

@ -9,7 +9,7 @@ from django.utils.functional import cached_property
from aircox import settings, utils
from .program import Program, ProgramChildQuerySet, \
BaseRerun, BaseRerunQuerySet
BaseRerun, BaseRerunQuerySet, Schedule
from .page import Page, PageQuerySet
@ -145,6 +145,10 @@ class Diffusion(BaseRerun):
episode = models.ForeignKey(
Episode, models.CASCADE, verbose_name=_('episode'),
)
schedule = models.ForeignKey(
Schedule, models.CASCADE, verbose_name=_('schedule'),
blank=True, null=True,
)
type = models.SmallIntegerField(
verbose_name=_('type'), default=TYPE_ON_AIR, choices=TYPE_CHOICES,
)

63
aircox/models/program.py
View File

@ -229,7 +229,7 @@ class BaseRerun(models.Model):
})
# BIG FIXME: self.date is still used as datetime
# ? BIG FIXME: self.date is still used as datetime
class Schedule(BaseRerun):
"""
A Schedule defines time slots of programs' diffusions. It can be an initial
@ -284,7 +284,6 @@ class Schedule(BaseRerun):
}[x]) for x, y in Frequency.__members__.items()],
)
class Meta:
verbose_name = _('Schedule')
verbose_name_plural = _('Schedules')
@ -340,59 +339,10 @@ class Schedule(BaseRerun):
return False
def match(self, date=None, check_time=True):
"""
Return True if the given date(time) matches the schedule.
"""
date = utils.date_or_default(
date, tz.datetime if check_time else datetime.date)
if self.date.weekday() != date.weekday() or \
not self.match_week(date):
return False
# we check against a normalized version (norm_date will have
# schedule's date.
return date == self.normalize(date) if check_time else True
def match_week(self, date=None):
"""
Return True if the given week number matches the schedule, False
otherwise.
If the schedule is ponctual, return None.
"""
if self.frequency == Schedule.Frequency.ponctual:
return False
# since we care only about the week, go to the same day of the week
date = utils.date_or_default(date, datetime.date)
date += tz.timedelta(days=self.date.weekday() - date.weekday())
# FIXME this case
if self.frequency == Schedule.Frequency.one_on_two:
# cf notes in date_of_month
diff = date - utils.cast_date(self.date, datetime.date)
return not (diff.days % 14)
first_of_month = date.replace(day=1)
week = date.isocalendar()[1] - first_of_month.isocalendar()[1]
# weeks of month
if week == 4:
# fifth week: return if for every week
return self.frequency == self.Frequency.every
return (self.frequency & (0b0001 << week) > 0)
def normalize(self, date):
"""
Return a new datetime with schedule time. Timezone is handled
using `schedule.timezone`.
Return a datetime set to schedule's time for the provided date,
handling timezone (based on schedule's timezone).
"""
date = tz.datetime.combine(date, self.time)
return self.tz.normalize(self.tz.localize(date))
@ -412,11 +362,9 @@ class Schedule(BaseRerun):
date_wday = date.weekday()
# end of month before the wanted weekday: move one week back
if date_wday < sched_wday:
date -= tz.timedelta(days=7)
date += tz.timedelta(days=sched_wday - date_wday)
return [self.normalize(date)]
# move to the first day of the month that matches the schedule's weekday
@ -466,7 +414,8 @@ class Schedule(BaseRerun):
# remove dates corresponding to existing diffusions
saved = set(Diffusion.objects.filter(start__in=dates.keys(),
program=self.program)
program=self.program,
schedule=self)
.values_list('start', flat=True))
# make diffs
@ -487,7 +436,7 @@ class Schedule(BaseRerun):
initial = diffusions[initial]
diffusions[date] = Diffusion(
episode=episode, type=Diffusion.TYPE_ON_AIR,
episode=episode, schedule=self, type=Diffusion.TYPE_ON_AIR,
initial=initial, start=date, end=date+duration
)
return episodes.values(), diffusions.values()

30
aircox/models/signals.py
View File

@ -1,6 +1,7 @@
import pytz
from django.contrib.auth.models import User, Group, Permission
from django.db import transaction
from django.db.models import F, signals
from django.dispatch import receiver
from django.utils import timezone as tz
@ -62,7 +63,6 @@ def schedule_pre_save(sender, instance, *args, **kwargs):
instance._initial = Schedule.objects.get(pk=instance.pk)
# TODO
@receiver(signals.post_save, sender=Schedule)
def schedule_post_save(sender, instance, created, *args, **kwargs):
"""
@ -76,27 +76,21 @@ def schedule_post_save(sender, instance, created, *args, **kwargs):
today = tz.datetime.today()
delta = instance.normalize(today) - initial.normalize(today)
qs = Diffusion.objects.program(instance.program).after(tz.now())
pks = [d.pk for d in qs if initial.match(d.date)]
qs.filter(pk__in=pks).update(
start=F('start') + delta,
end=F('start') + delta + utils.to_timedelta(instance.duration)
)
duration = utils.to_timedelta(instance.duration)
with transaction.atomic():
qs = Diffusion.objects.filter(schedule=instance).after(tz.now())
for diffusion in qs:
diffusion.start = diffusion.start + delta
diffusion.end = diffusion.start + duration
diffusion.save()
@receiver(signals.pre_delete, sender=Schedule)
def schedule_pre_delete(sender, instance, *args, **kwargs):
"""
Delete later corresponding diffusion to a changed schedule.
"""
if not instance.program.sync:
return
qs = Diffusion.objects.program(instance.program).after(tz.now())
pks = [d.pk for d in qs if instance.match(d.date)]
qs.filter(pk__in=pks).delete()
""" Delete later corresponding diffusion to a changed schedule. """
Diffusion.objects.filter(schedule=instance).after(tz.now()).delete()
Episode.objects.filter(diffusion__isnull=True, content__isnull=True,
sound__isnull=True).delete()
@receiver(signals.post_delete, sender=Diffusion)
def diffusion_post_delete(sender, instance, *args, **kwargs):