nonstop/models.py

   1 import datetime
   2 import os
   3
   4 import mutagen
   5
   6 from django.conf import settings
   7 from django.core.urlresolvers import reverse
   8 from django.db import models
   9 from django.db.models.signals import post_delete
  10 from django.dispatch import receiver
  11 from django.utils.timezone import now
  12 from django.utils.translation import ugettext_lazy as _
  13
  14 REMOTE_BASE_PATH = '/srv/soma/nonstop/'
  15 LOCAL_BASE_PATH = '/media/nonstop/'
  16
  17 TRANCHE_SLUG_DIR_MAPPING = {
  18     'acouphene': 'Acouphene',
  19     'biodiversite': 'Biodiversite',
  20     'l-heure-de-pointe': 'Heure_de_pointe',
  21     'hop-bop-co': 'Hop_Bop_and_co',
  22     'la-panique': 'la_panique',
  23     'le-mange-disque': 'Mange_Disque',
  24     'matin-tranquille': 'Matins_tranquilles',
  25     'reveries': 'Reveries',
  26     'up-beat-tempo': 'Up_Beat_Tempo',
  27 }
  28
  29 class Artist(models.Model):
  30     name = models.CharField(_('Name'), max_length=255)
  31
  32     class Meta:
  33         ordering = ['name']
  34
  35     def get_absolute_url(self):
  36         return reverse('artist-view', kwargs={'pk': self.id})
  37
  38     def recent_diffusions(self):
  39         return SomaLogLine.objects.filter(filepath__track__artist=self
  40                 ).exclude(on_air=False).order_by('-play_timestamp')
  41
  42
  43 class Album(models.Model):
  44     name = models.CharField(_('Name'), max_length=255)
  45
  46
  47 LANGUAGES = [
  48     ('en', _('English')),
  49     ('fr', _('French')),
  50     ('nl', _('Dutch'))
  51 ]
  52
  53 class Track(models.Model):
  54     title = models.CharField(_('Title'), max_length=255)
  55     artist = models.ForeignKey(Artist, null=True)
  56     album = models.ForeignKey(Album, null=True)
  57     instru = models.BooleanField(_('Instru'), default=False)
  58     language = models.CharField(max_length=3,
  59             choices=LANGUAGES, blank=True)
  60     sabam = models.BooleanField('SABAM', default=True)
  61     cfwb = models.BooleanField('CFWB', default=False)
  62     nonstop_zones = models.ManyToManyField('emissions.Nonstop', blank=True)
  63
  64     creation_timestamp = models.DateTimeField(auto_now_add=True, null=True)
  65     added_to_nonstop_timestamp = models.DateTimeField(null=True)
  66     uploader = models.ForeignKey(settings.AUTH_USER_MODEL, null=True)
  67     duration = models.DurationField(_('Duration'), null=True)
  68
  69     class Meta:
  70         ordering = ['creation_timestamp']
  71
  72     def __str__(self):
  73         return 'Track %s (%s)' % (self.title, self.artist or 'unknown')
  74
  75     def get_absolute_url(self):
  76         return reverse('track-view', kwargs={'pk': self.id})
  77
  78     def recent_diffusions(self):
  79         return SomaLogLine.objects.filter(filepath__track=self
  80                 ).exclude(on_air=False).order_by('-play_timestamp')
  81
  82     def file_path(self):
  83         nfile = None
  84         for nfile in self.nonstopfile_set.all().order_by('creation_timestamp'):
  85             if os.path.exists(nfile.get_local_filepath()):
  86                 return nfile.get_local_filepath()
  87         if nfile:
  88            return nfile.get_local_filepath()
  89         return None
  90
  91     def file_exists(self):
  92         file_path = self.file_path()
  93         if not file_path:
  94             return False
  95         try:
  96             return os.path.exists(file_path)
  97         except AttributeError:
  98             return False
  99
 100     def sync_nonstop_zones(self):
 101         current_zones = self.nonstop_zones.all()
 102         if current_zones.count():
 103             if not self.added_to_nonstop_timestamp:
 104                 self.added_to_nonstop_timestamp = now()
 105                 self.save()
 106         else:
 107             self.added_to_nonstop_timestamp = None
 108             self.save()
 109
 110         if not self.file_exists():
 111             return
 112         nonstop_file = self.nonstopfile_set.order_by('creation_timestamp').last()
 113         filename = nonstop_file.filename
 114         from emissions.models import Nonstop
 115
 116         for zone in Nonstop.objects.all():
 117             if not zone.slug in TRANCHE_SLUG_DIR_MAPPING:
 118                 continue
 119             zone_dir = TRANCHE_SLUG_DIR_MAPPING[zone.slug]
 120             zone_path = os.path.join(LOCAL_BASE_PATH, 'Tranches', zone_dir, filename)
 121             if zone in current_zones:
 122                 if not os.path.exists(zone_path):
 123                     os.symlink(os.path.join('..', '..', nonstop_file.short), zone_path)
 124             else:
 125                 if os.path.exists(zone_path):
 126                     os.unlink(zone_path)
 127
 128
 129 class NonstopFile(models.Model):
 130     filepath = models.CharField(_('Filepath'), max_length=255)
 131     filename = models.CharField(_('Filename'), max_length=255, null=True)
 132     creation_timestamp = models.DateTimeField(auto_now_add=True, null=True)
 133     track = models.ForeignKey(Track, null=True)
 134
 135     @property
 136     def short(self):
 137         return self.filepath[len(REMOTE_BASE_PATH):]
 138
 139     def set_track_filepath(self, filepath):
 140         self.filepath = os.path.join(REMOTE_BASE_PATH, 'tracks', filepath)
 141         self.filename = os.path.basename(filepath)
 142
 143     def get_local_filepath(self):
 144         if not self.short:
 145             return None
 146         return os.path.join(LOCAL_BASE_PATH, self.short)
 147
 148
 149 class SomaLogLine(models.Model):
 150     class Meta:
 151         verbose_name = _('Soma log line')
 152         verbose_name_plural = _('Soma log lines')
 153         ordering = ['play_timestamp']
 154
 155     filepath = models.ForeignKey(NonstopFile)
 156     play_timestamp = models.DateTimeField()
 157     on_air = models.NullBooleanField('On Air')
 158
 159
 160 class Jingle(models.Model):
 161     class Meta:
 162         verbose_name = _('Jingle')
 163         verbose_name_plural = _('Jingles')
 164         ordering = ['label']
 165
 166     label = models.CharField(_('Label'), max_length=100)
 167     filepath = models.CharField(_('File Path'), max_length=255)
 168     duration = models.DurationField(_('Duration'), null=True, blank=True)
 169     default_for_initial_diffusions = models.BooleanField(_('Default for initial diffusions'), default=False)
 170     default_for_reruns = models.BooleanField(_('Default for reruns'), default=False)
 171     default_for_streams = models.BooleanField(_('Default for streams'), default=False)
 172
 173     def __str__(self):
 174         return self.label
 175
 176     def save(self, **kwargs):
 177         for attr in ('default_for_initial_diffusions', 'default_for_reruns', 'default_for_streams'):
 178             if getattr(self, attr):
 179                 Jingle.objects.all().update(**{attr: False})
 180         return super(Jingle, self).save(**kwargs)
 181
 182     @property
 183     def title(self):
 184         return self.label
 185
 186
 187 class Stream(models.Model):
 188     class Meta:
 189         verbose_name = _('Stream')
 190         verbose_name_plural = _('Streams')
 191         ordering = ['label']
 192
 193     label = models.CharField(_('Label'), max_length=100)
 194     url = models.URLField(_('URL'), max_length=255)
 195
 196     def __str__(self):
 197         return self.label
 198
 199
 200 class ScheduledDiffusion(models.Model):
 201     class Meta:
 202         verbose_name = _('Scheduled diffusion')
 203         verbose_name_plural = _('Scheduled diffusions')
 204
 205     diffusion = models.ForeignKey('emissions.Diffusion', null=True, blank=True, on_delete=models.SET_NULL)
 206     jingle = models.ForeignKey(Jingle, null=True, blank=True)
 207     stream = models.ForeignKey(Stream, null=True, blank=True)
 208     creation_timestamp = models.DateTimeField(auto_now_add=True, null=True)
 209     added_to_nonstop_timestamp = models.DateTimeField(null=True)
 210
 211     def __str__(self):
 212         return 'Diffusion of %s' % self.diffusion
 213
 214     @property
 215     def datetime(self):
 216         return self.diffusion.datetime
 217
 218     @property
 219     def end_datetime(self):
 220         if self.is_stream():
 221             return self.diffusion.end_datetime
 222         dt = self.diffusion.datetime
 223         if self.jingle:
 224             dt += self.jingle.duration
 225         dt += datetime.timedelta(seconds=self.soundfile.duration)
 226         return dt
 227
 228     @property
 229     def soundfile(self):
 230         return self.diffusion.episode.soundfile_set.filter(fragment=False).first()
 231
 232     @property
 233     def duration(self):
 234         return (self.end_datetime - self.datetime).seconds
 235
 236     @property
 237     def episode(self):
 238         return self.diffusion.episode
 239
 240     @property
 241     def soma_id(self):
 242         if self.is_stream():
 243             return '[stream:%s]' % self.id
 244         else:
 245             return '[sound:%s]' % self.id
 246
 247     def is_stream(self):
 248         return bool(self.stream_id)
 249
 250
 251 class NonstopZoneSettings(models.Model):
 252     nonstop = models.ForeignKey('emissions.Nonstop', on_delete=models.CASCADE)
 253     intro_jingle = models.ForeignKey(Jingle, blank=True, null=True, related_name='+')
 254     jingles = models.ManyToManyField(Jingle, blank=True)
 255
 256     def __str__(self):
 257         return str(self.nonstop)
 258
 259
 260 @receiver(post_delete)
 261 def remove_soundfile(sender, instance=None, **kwargs):
 262     from emissions.models import SoundFile
 263     if not issubclass(sender, SoundFile):
 264         return
 265     ScheduledDiffusion.objects.filter(
 266             diffusion__episode_id=instance.episode_id,
 267             stream_id=None).update(
 268             diffusion=None)