pandora/pandora/tv/models.py

98 lines
3.5 KiB
Python

# -*- 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.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