addon.py

# coding: utf-8
from __future__ import print_function, division, absolute_import

import operator
import sys

import routing
from xbmcgui import ListItem
from xbmcplugin import (addDirectoryItem, endOfDirectory, setResolvedUrl,
    setContent)

from resources.lib import http, kodi, settings
from resources.lib.helpers import maybe_json, calc_aspect, json_date_to_info

plugin = routing.Plugin()


@plugin.route('/')
@plugin.route('/dir/<path:subdir>')
def show_dir(subdir=''):
    try:
        data = get_index_data()
    except http.FetchError:
        return

    subdirs = set()
    if subdir == '':
        depth = 0

        addDirectoryItem(plugin.handle, plugin.url_for(show_live),
                         ListItem('Live Streaming'), True)
    else:
        depth = len(subdir.split('/'))

    for event in sorted(data, key=operator.itemgetter('title')):
        top, down, children = split_pathname(event['slug'], depth)

        if top != subdir or down in subdirs:
            continue
        if children:
            addDirectoryItem(plugin.handle, plugin.url_for(show_dir,
                             subdir=build_path(top, down)),
                             ListItem(down.title()), True)
            subdirs.add(down)
        else:
            item = ListItem(event['title'])
            item.setLabel2(event['acronym'])
            item.setArt({'thumb': event['logo_url']})
            url = plugin.url_for(show_conference,
                                 conf=event['url'].rsplit('/', 1)[1])
            addDirectoryItem(plugin.handle, url, item, True)

    endOfDirectory(plugin.handle)


@plugin.route('/conference/<conf>')
def show_conference(conf):
    data = None
    relive = None
    try:
        data = http.fetch_data('conferences/' + conf)
    except http.FetchError:
        return

    try:
        relive = http.fetch_relive_index().by_conference(conf)
    except http.FetchError:
        pass # gui error shown by http class

    setContent(plugin.handle, 'movies')

    aspect = calc_aspect(maybe_json(data, 'aspect_ratio', '16:9'))

    for event in sorted(data['events'], key=operator.itemgetter('title')):
        item = ListItem(event['title'])
        item.setArt({'thumb': event['thumb_url']})
        item.setProperty('IsPlayable', 'true')

        info = {
            'cast': maybe_json(event, 'persons', []),
            'credits': ", ".join(maybe_json(event, 'persons', [])),
            'genre': " / ".join(maybe_json(event, 'tags', [])),
            'plot': maybe_json(event, 'description', ''),
            'tagline': maybe_json(event, 'subtitle', '')
        }
        json_date_to_info(event, 'date', info)
        item.setInfo('video', info)

        streamInfo = {}
        length = maybe_json(event, 'length', 0)
        if length > 0:
            streamInfo['duration'] = length
        if aspect:
            streamInfo['aspect'] = aspect
        item.addStreamInfo('video', streamInfo)

        url = plugin.url_for(resolve_event,
                             event=event['url'].rsplit('/', 1)[1])
        addDirectoryItem(plugin.handle, url, item, False)

    if relive is not None:
        relive_item = ListItem('ReLive (unreleased)')
        url = plugin.url_for(show_relive, conf=conf)
        addDirectoryItem(plugin.handle, url, relive_item, True)

    endOfDirectory(plugin.handle)


@plugin.route('/relive/<conf>')
def show_relive(conf):
    data = None
    try:
        relive = http.fetch_relive_index().by_conference(conf)
        data = http.fetch_relive_recordings(relive.get_url())
    except http.FetchError:
        return

    setContent(plugin.handle, 'movies')

    data = data.unreleased()

    for recording in data:
        item = ListItem(recording.title)
        item.setArt({'thumb':  recording.get_thumb_url()})
        item.setProperty('IsPlayable', 'true')

        item.setInfo('video', {
            'plot': recording.room,
        })

        item.addStreamInfo('video', {
            'duration': recording.duration
        })

        addDirectoryItem(plugin.handle, recording.get_video_url(), item, False)

    endOfDirectory(plugin.handle)


@plugin.route('/event/<event>')
@plugin.route('/event/<event>/<quality>/<format>')
def resolve_event(event, quality=None, format=None):
    if quality not in settings.QUALITY:
        quality = settings.get_quality(plugin)
    if format not in settings.FORMATS:
        format = settings.get_format(plugin)

    data = None
    try:
        data = http.fetch_recordings(event)
    except http.FetchError:
        return
    want = data.recordings_sorted(quality, format)

    if len(want) > 0:
        http.count_view(event, want[0].url)
        setResolvedUrl(plugin.handle, True, ListItem(path=want[0].url))


@plugin.route('/live')
def show_live():
    quality = settings.get_quality(plugin)
    format = settings.get_format(plugin)
    prefer_dash = settings.prefer_dash(plugin)

    data = None
    try:
        data = http.fetch_live()
    except http.FetchError:
        return

    if len(data.conferences) == 0:
        entry = ListItem('No live event currently, go watch some recordings!')
        addDirectoryItem(plugin.handle, plugin.url_for(show_dir), entry, True)

    for conference in data.conferences:
        for room in conference.rooms:
            want = room.streams_sorted(quality, format, prefer_dash)

            # Assumption: want now starts with the "best" alternative,
            # followed by an arbitrary number of translations, after which
            # the first "worse" non-translated stream follows.

            for id, stream in enumerate(want):
                if id > 0 and not stream.translated:
                    break
                extra = ''
                if stream.translated:
                    extra = ' (Translated %i)' % id if id > 1 else ' (Translated)'
                talk_title = ' >> ' + room.current_talk_title if room.current_talk_title != '' else ''
                item = ListItem(conference.name + ': ' + room.display + talk_title + extra)
                item.setProperty('IsPlayable', 'true')
                if stream.type == 'dash':
                    dashproperty = 'inputstream'
                    if kodi.major_version() < 19:
                        dashproperty += 'addon'
                    item.setProperty(dashproperty, 'inputstream.adaptive')
                    item.setProperty('inputstream.adaptive.manifest_type', 'mpd')

                addDirectoryItem(plugin.handle, stream.url, item, False)

                # MPEG-DASH includes all translated streams
                if stream.type == 'dash':
                    break

    endOfDirectory(plugin.handle)


# FIXME: @plugin.cached()
def get_index_data():
    return http.fetch_data('conferences')['conferences']


def split_pathname(name, depth):
    path = name.split('/')
    top = '/'.join(path[0:depth])
    if depth < len(path):
        down = path[depth]
    else:
        down = None
    children = len(path) - 1 > depth
    return (top, down, children)


def build_path(top, down):
    if top == '':
        return down
    else:
        return '/'.join((top, down))


if __name__ == '__main__':
    plugin.run(argv=sys.argv)