Source code for radioco.apps.programmes.models

# Radioco - Broadcasting Radio Recording Scheduling system.
# Copyright (C) 2014  Iago Veloso Abalo
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
import pytz
from bs4 import BeautifulSoup
from ckeditor_uploader.fields import RichTextUploadingField
from django.conf import settings
from django.contrib.auth.models import User
from django.core.exceptions import FieldError
from django.core.urlresolvers import reverse
from django.core.validators import MinValueValidator
from django.db import models
from django.db.models import Q
from django.db.models.signals import post_save, pre_save
from django.template.defaultfilters import slugify
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _
import datetime

from radioco.apps.radioco.utils import field_has_changed
from radioco.apps.schedules.utils import next_dates

if hasattr(settings, 'PROGRAMME_LANGUAGES'):
    PROGRAMME_LANGUAGES = settings.PROGRAMME_LANGUAGES
else:
    PROGRAMME_LANGUAGES = settings.LANGUAGES


[docs]class Programme(models.Model): class Meta: verbose_name = _('programme') verbose_name_plural = _('programmes') permissions = (("see_all_programmes", "Can see all programmes"),) CATEGORY_CHOICES = ( ('Arts', _('Arts')), ('Business', _('Business')), ('Comedy', _('Comedy')), ('Education', _('Education')), ('Games & Hobbies', _('Games & Hobbies')), ('Government & Organizations', _('Government & Organizations')), ('Health', _('Health')), ('Kids & Family', _('Kids & Family')), ('Music', _('Music')), ('News & Politics', _('News & Politics')), ('Religion & Spirituality', _('Religion & Spirituality')), ('Science & Medicine', _('Science & Medicine')), ('Society & Culture', _('Society & Culture')), ('Sports & Recreation', _('Sports & Recreation')), ('Technology', _('Technology')), ('TV & Film', _('TV & Film')), ) name = models.CharField( max_length=100, unique=True, verbose_name=_("name") ) announcers = models.ManyToManyField( User, blank=True, through='Role', verbose_name=_("announcers") ) synopsis = RichTextUploadingField(blank=True, verbose_name=_("synopsis")) photo = models.ImageField( upload_to='photos/', default='defaults/default-programme-photo.jpg', verbose_name=_("photo") ) language = models.CharField( default=PROGRAMME_LANGUAGES[0][0], verbose_name=_("language"), choices=map(lambda (k, v): (k, _(v)), PROGRAMME_LANGUAGES), max_length=7 ) # XXX ensure not decreasing current_season = models.PositiveIntegerField( validators=[MinValueValidator(1)], verbose_name=_("current season") ) category = models.CharField( blank=True, null=True, max_length=50, choices=CATEGORY_CHOICES, verbose_name=_("category") ) slug = models.SlugField(max_length=100, unique=True, help_text=_("Please DON'T change this value. It's used to build URL's.")) _runtime = models.PositiveIntegerField( validators=[MinValueValidator(1)], verbose_name=_("runtime"), help_text=_("In minutes.")) start_date = models.DateField(blank=True, null=True, verbose_name=_('start date')) end_date = models.DateField(blank=True, null=True, verbose_name=_('end date')) @property def synopsis_text(self): return BeautifulSoup(self.synopsis, "html.parser").text @property def runtime(self): if not self._runtime: raise FieldError(_('Runtime not set')) return datetime.timedelta(minutes=self._runtime) @runtime.setter def runtime(self, value): self._runtime = value @property def start_dt(self): if not self.start_date: return None tz = timezone.get_default_timezone() return tz.localize(datetime.datetime.combine(self.start_date, datetime.time())).astimezone(pytz.utc) @property def end_dt(self): if not self.end_date: return None tz = timezone.get_default_timezone() return tz.localize(datetime.datetime.combine(self.end_date, datetime.time(23, 59, 59))).astimezone(pytz.utc)
[docs] def save(self, *args, **kwargs): if not self.slug: self.slug = slugify(self.name) super(Programme, self).save(*args, **kwargs)
[docs] def rearrange_episodes(self, after, calendar): """ Update the issue_date of episodes from a given date """ episodes = Episode.objects.unfinished(self, after) dates = next_dates(calendar, self, after) # Further dates and episodes available -> re-order while True: try: date = dates.next() episode = episodes.next() except StopIteration: break else: episode.issue_date = date episode.save() # No further dates available -> unschedule while True: try: episode = episodes.next() except StopIteration: break else: episode.issue_date = None episode.save()
[docs] def get_absolute_url(self): return reverse('programmes:detail', args=[self.slug])
def __unicode__(self): return u"%s" % (self.name)
[docs]def update_schedule_performance(programme): for schedule in programme.schedule_set.all(): # schedule.effective_start_dt = calculate_effective_schedule_start_dt(schedule) # schedule.effective_end_dt = calculate_effective_schedule_end_dt(schedule) schedule.save() # TODO: improve performance, update objects in bulk? We need custom logic on save
[docs]def update_schedule_if_dt_has_changed(sender, instance, **kwargs): if field_has_changed(instance, 'start_date') or field_has_changed(instance, 'end_date'): # TODO: improve update_schedule_performance(instance)
pre_save.connect( update_schedule_if_dt_has_changed, sender=Programme, dispatch_uid='update_schedule_if_dt_has_changed')
[docs]class EpisodeManager(models.Manager): # XXX this is not atomic, transaction?
[docs] def create_episode(self, date, programme, last_episode=None, episode=None): if not last_episode: last_episode = self.last(programme) season = programme.current_season if last_episode and last_episode.season == season: number_in_season = last_episode.number_in_season + 1 else: number_in_season = 1 if episode: episode.programme = programme episode.issue_date = date episode.season = season episode.number_in_season = number_in_season else: episode = Episode( programme=programme, issue_date=date, season=season, number_in_season=number_in_season ) episode.save() for role in Role.objects.filter(programme=programme): Participant.objects.create( person=role.person, episode=episode, role=role.role, description=role.description ) return episode
@staticmethod
[docs] def last(programme): return (programme.episode_set .order_by("-season", "-number_in_season") .first())
@staticmethod
[docs] def unfinished(programme, after=None): if not after: after = timezone.now() return programme.episode_set.filter( Q(issue_date__gte=after) | Q(issue_date=None)).order_by("season", "number_in_season").iterator()
[docs]class Episode(models.Model): class Meta: unique_together = (('season', 'number_in_season', 'programme'),) verbose_name = _('episode') verbose_name_plural = _('episodes') permissions = (("see_all_episodes", "Can see all episodes"),) objects = EpisodeManager() title = models.CharField(max_length=100, blank=True, null=True, verbose_name=_("title")) people = models.ManyToManyField(User, blank=True, through='Participant', verbose_name=_("people")) programme = models.ForeignKey(Programme, verbose_name=_("programme")) summary = RichTextUploadingField(blank=True, verbose_name=_("summary")) issue_date = models.DateTimeField(blank=True, null=True, db_index=True, verbose_name=_('issue date')) season = models.PositiveIntegerField(validators=[MinValueValidator(1)], verbose_name=_("season")) number_in_season = models.PositiveIntegerField(validators=[MinValueValidator(1)], verbose_name=_("No. in season")) @property def summary_text(self): return BeautifulSoup(self.summary, "html.parser").text # FIXME: this is not true for archived episodes @property def runtime(self): return self.programme.runtime
[docs] def get_absolute_url(self): return reverse('programmes:episode_detail', args=[self.programme.slug, self.season, self.number_in_season])
def __unicode__(self): if self.title: return u"%sx%s %s" % (self.season, self.number_in_season, self.title) return u"%sx%s %s" % (self.season, self.number_in_season, self.programme)
[docs]class Participant(models.Model): person = models.ForeignKey(User, verbose_name=_("person")) episode = models.ForeignKey(Episode, verbose_name=_("episode")) role = models.CharField(max_length=60, blank=True, null=True, verbose_name=_("role")) description = models.TextField(blank=True, verbose_name=_("description")) class Meta: unique_together = ('person', 'episode') verbose_name = _('contributor') verbose_name_plural = _('contributors') permissions = ( ("see_all_participants", "Can see all participants"), ) def __unicode__(self): return u"%s: %s" % (self.episode, self.person.username)
[docs]class Role(models.Model): person = models.ForeignKey(User, verbose_name=_("person")) programme = models.ForeignKey(Programme, verbose_name=_("programme")) role = models.CharField(max_length=60, blank=True, null=True, verbose_name=_("role")) description = models.TextField(blank=True, verbose_name=_("description")) date_joined = models.DateField(auto_now_add=True) class Meta: unique_together = ('person', 'programme') verbose_name = _('role') verbose_name_plural = _('roles') permissions = ( ("see_all_roles", "Can see all roles"), ) def __unicode__(self): return u"%s: %s" % (self.programme.name, self.person.username)
[docs]class Podcast(models.Model): episode = models.OneToOneField(Episode, primary_key=True, related_name='podcast') url = models.CharField(max_length=2048) mime_type = models.CharField(max_length=20) length = models.PositiveIntegerField() # bytes duration = models.PositiveIntegerField(validators=[MinValueValidator(1)])
[docs] def get_absolute_url(self): return self.episode.get_absolute_url()