pandora/pandora/tv/models.py

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division, with_statement
from datetime import datetime, timedelta
from random import randint

from django.db import models
from django.db.models import Max
from django.conf import settings

from item.models import Item


class Channel(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    run = models.IntegerField(default=0)
    list = models.OneToOneField('itemlist.List', related_name='channel', null=True, blank=True)
    #list = models.ForeignKey('itemlist.List', related_name='channel', null=True, unique=True, blank=True)

    def __unicode__(self):
        return u"%s %s" % (self.list or 'All', self.run)

    def update_program(self, now=None):
        if not now:
            now = datetime.now()
        cansee = settings.CONFIG['capabilities']['canSeeItem']['guest']
        if self.list:
            items = self.list.get_items(self.list.user).filter(rendered=True, level__lte=cansee, sort__duration__gt=0)
        else:
            items = Item.objects.filter(rendered=True, level__lte=cansee, sort__duration__gt=0)
        if items.count() == 0:
            return None

        program = self.program.order_by('-start')
        changed = False
        while program.count() < 1 or program[0].end < now:
            not_played = items.exclude(program__in=self.program.filter(run=self.run))
            not_played_count = not_played.count()
            if not_played_count == 0:
                self.run += 1
                changed = True
                not_played = items
                not_played_count = not_played.count()
                if not_played_count > 1:
                    not_played = not_played.exclude(id=program[0].id)
                    not_played_count = not_played.count()
            item = not_played[randint(0, not_played_count-1)]
            if program.count() > 0:
                start = program.aggregate(Max('end'))['end__max']
            else:
                start = now
            p = Program()
            p.item = item
            p.run = self.run
            p.start = start
            p.end = start + timedelta(seconds=item.sort.duration)
            p.channel = self
            p.save()
            program = self.program.order_by('-start')
        if changed:
            self.save()
        return program[0]

    def json(self, user):
        now = datetime.now()
        program = self.update_program(now)
        if not program:
            return {}
        else:
            return program.json(user, now)

class Program(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)
    run = models.IntegerField(default=0)
    start = models.DateTimeField()
    end = models.DateTimeField()
    item = models.ForeignKey('item.Item', related_name='program')
    channel = models.ForeignKey(Channel, related_name='program')

    def __unicode__(self):
        return u"%s %s" % (self.item, self.start)

    def json(self, user, current=False):
        item_json = self.item.get_json()
        r = {
            'item': self.item.public_id,
        }
        for key in ('title', 'director', 'year', 'durations', 'parts', 'rightslevel'):
            r[key] = item_json.get(key, '')
        r['layers'] = self.item.get_layers(user)
        if current:
            #requires python2.7
            #r['position'] = (current - self.start).total_seconds()
            td = current - self.start
            r['position'] = td.seconds + td.days * 24 * 3600 + float(td.microseconds)/10**6
        return r