videoportal.py

import os, re, sys
from datetime import date, timedelta
import urllib, urllib2, HTMLParser
import xbmcgui, xbmcplugin, xbmcaddon
from mindmade import *
import simplejson
from BeautifulSoup import BeautifulSoup

__author__     = "Andreas Wetzel"
__copyright__  = "Copyright 2011, 2012, mindmade.org"
__credits__    = [ "Francois Marbot" ]
__maintainer__ = "Andreas Wetzel"
__email__      = "xbmc@mindmade.org"

#
# constants definition
############################################
PLUGINID = "plugin.video.sf-videoportal"

# plugin handle
pluginhandle = int(sys.argv[1])

# plugin modes
MODE_SENDUNGEN_AZ         = "sendungen_az"
MODE_SENDUNGEN_MOSTVIEWED = "mostviewed"
MODE_SENDUNGEN_LATEST     = "latest"
MODE_SENDUNGEN_LAST24H    = "last24h"
MODE_SENDUNGEN_THEMA      = "sendungen_thema"
MODE_SENDUNG              = "sendung"
MODE_PLAY                 = "play"

# parameter keys
PARAMETER_KEY_MODE  = "mode"
PARAMETER_KEY_ID    = "id"
PARAMETER_KEY_URL   = "url"
PARAMETER_KEY_TITLE = "title"
PARAMETER_KEY_POS   = "pos"

ITEM_TYPE_FOLDER, ITEM_TYPE_VIDEO = range(2)
BASE_URL = "http://www.srf.ch/"
BASE_URL_PLAYER = "http://www.srf.ch/play/tv/"
# for some reason, it only works with the old player version.
FLASH_PLAYER = "http://www.videoportal.sf.tv/flash/videoplayer.swf"
#FLASH_PLAYER = "http://www.srf.ch/player/tv/flash/videoplayer.swf"

settings = xbmcaddon.Addon( id=PLUGINID)

LIST_FILE = os.path.join( settings.getAddonInfo( "path"), "resources", "list.dat")
listItems = []

# DEBUGGER
#REMOTE_DBG = False 

# append pydev remote debugger
#if REMOTE_DBG:
#    # Make pydev debugger works for auto reload.
#    # Note pydevd module need to be copied in XBMC\system\python\Lib\pysrc
#    try:
#        import pysrc.pydevd as pydevd
#    # stdoutToServer and stderrToServer redirect stdout and stderr to eclipse console
#        pydevd.settrace('localhost', stdoutToServer=True, stderrToServer=True)
#    except ImportError:
#        sys.stderr.write("Error: " +
#            "You must add org.python.pydev.debug.pysrc to your PYTHONPATH.")
#        sys.exit(1)

#
# utility functions
############################################

# Log NOTICE

def parameters_string_to_dict( parameters):
    ''' Convert parameters encoded in a URL to a dict. '''
    paramDict = {}
    if parameters:
        paramPairs = parameters[1:].split("&")
        for paramsPair in paramPairs:
            paramSplits = paramsPair.split('=')
            if (len(paramSplits)) == 2:
                paramDict[paramSplits[0]] = urllib.unquote( paramSplits[1])
    return paramDict


def addDirectoryItem( type, name, params={}, image="", total=0):
    '''Add a list item to the XBMC UI.'''
    if (type == ITEM_TYPE_FOLDER):
        img = "DefaultFolder.png"
    elif (type == ITEM_TYPE_VIDEO):
        img = "DefaultVideo.png"

    name = htmldecode( name)
    params[ PARAMETER_KEY_TITLE] = name
    li = xbmcgui.ListItem( name, iconImage=img, thumbnailImage=image)
            
    if (type == ITEM_TYPE_VIDEO):
#        li.setProperty( "IsPlayable", "true")
        li.setProperty( "Video", "true")
        global listItems
        listItems.append( (name, params, image))
    
    params_encoded = dict()
    for k in params.keys():
        params_encoded[k] = params[k].encode( "utf-8")
    url = sys.argv[0] + '?' + urllib.urlencode( params_encoded)
    
    return xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]), url=url, listitem=li, isFolder = (type == ITEM_TYPE_FOLDER), totalItems=total)


#
# parsing functions
############################################

def getIdFromUrl( url):
    return re.compile( '[\?|\&]id=([0-9a-z\-]+)').findall( url)[0]

def getUrlWithoutParams( url):
    return url.split('?')[0]


def getJSONForId( id):
    json_url = BASE_URL + "/webservice/cvis/segment/" + id + "/.json?nohttperr=1;omit_video_segments_validity=1;omit_related_segments=1"
    url = fetchHttp( json_url).split( "\n")[1]
    json = simplejson.loads( url)
    return json


def getVideoFromJSON( json):
    streams = json["playlists"]["playlist"]
    index = 2 * int(settings.getSetting( id="quality"))
    sortedstreams = sorted( streams, key=lambda el: int(el["quality"]))
    if (index >= len(sortedstreams)):
        index = len(sortedstreams)-2
    
    return sortedstreams[index]["url"]

def getThumbnailForId( id):
    thumb = BASE_URL + "webservice/cvis/videogroup/thumbnail/" + id
    return thumb


#
# content functions
############################################


#
# mode handlers
############################################

def show_root_menu():
    addDirectoryItem( ITEM_TYPE_FOLDER, "Sendungen A-Z", {PARAMETER_KEY_MODE: MODE_SENDUNGEN_AZ})
    addDirectoryItem( ITEM_TYPE_FOLDER, "Meistgesehen", {PARAMETER_KEY_MODE: MODE_SENDUNGEN_MOSTVIEWED})
    addDirectoryItem( ITEM_TYPE_FOLDER, "Die neuesten Videos", {PARAMETER_KEY_MODE: MODE_SENDUNGEN_LATEST})
    addDirectoryItem( ITEM_TYPE_FOLDER, "TV-Sendungen der letzten 24 Stunden", {PARAMETER_KEY_MODE: MODE_SENDUNGEN_LAST24H})
    xbmcplugin.endOfDirectory(handle=pluginhandle, succeeded=True)


def show_sendungen_abisz():
    url = BASE_URL_PLAYER + "/sendungen"
    soup = BeautifulSoup( fetchHttp( url))
    
    for show in soup.findAll( "li", "az_item"):
        url = show.find( "a")['href']
        title = show.find( "img", "az_thumb")['alt']
        id = getIdFromUrl( url)
        image = getThumbnailForId( id)
        addDirectoryItem( ITEM_TYPE_FOLDER, title, {PARAMETER_KEY_MODE: MODE_SENDUNG, PARAMETER_KEY_ID: id, PARAMETER_KEY_URL: url }, image)

    xbmcplugin.endOfDirectory(handle=pluginhandle, succeeded=True)


def show_sendungen_dynamic(location):
    url = BASE_URL_PLAYER + "carouselvideosajax/" + location
    soup = BeautifulSoup( fetchHttp( url, {"count": 25, "mode": "tabcontainer"}))

    for show in soup.findAll( "div", "carousel_item"):
        title = show.find( "a", "headline").text
        show_title = show.find( "div", "show").find( "a").text
        image = getUrlWithoutParams( show.find( "img")['src'])
        a = show.findAll( "a")[1]
        id = getIdFromUrl( a['href'])
        addDirectoryItem( ITEM_TYPE_VIDEO, title + " - " + show_title, {PARAMETER_KEY_MODE: MODE_PLAY, PARAMETER_KEY_ID: id }, image)

    xbmcplugin.endOfDirectory(handle=pluginhandle, succeeded=True)


def show_sendung( params):
    sendid = params.get( PARAMETER_KEY_ID)
    urlParam = params.get( PARAMETER_KEY_URL)
    url = BASE_URL_PLAYER + "episodesfromshow"
    soup = BeautifulSoup( fetchHttp( url, {"id": sendid, "pageNumber": 1}))

    for show in soup.findAll( "li", "sendung_item"):
        title = show.find( "h3", "title").text
        titleDate = show.find( "div", "title_date").text
        image = getUrlWithoutParams( show.find( "img")['src'])
        a = show.findAll( "a")[1]
        id = getIdFromUrl( a['href'])
        addDirectoryItem( ITEM_TYPE_VIDEO, title + " - " + titleDate, {PARAMETER_KEY_MODE: MODE_PLAY, PARAMETER_KEY_ID: id }, image)

    xbmcplugin.endOfDirectory(handle=pluginhandle, succeeded=True)


#
# xbmc entry point
############################################

sayHi()

# read parameters and mode
params = parameters_string_to_dict(sys.argv[2])

mode = params.get(PARAMETER_KEY_MODE, "0")

# depending on the mode, call the appropriate function to build the UI.
if not sys.argv[2]:
    # new start
    ok = show_root_menu()
elif mode == MODE_SENDUNGEN_AZ:
    ok = show_sendungen_abisz()
elif mode == MODE_SENDUNGEN_MOSTVIEWED:
    ok = show_sendungen_dynamic(MODE_SENDUNGEN_MOSTVIEWED)
elif mode == MODE_SENDUNGEN_LATEST:
    ok = show_sendungen_dynamic(MODE_SENDUNGEN_LATEST)
elif mode == MODE_SENDUNGEN_LAST24H:
    ok = show_sendungen_dynamic(MODE_SENDUNGEN_LAST24H)
elif mode == MODE_SENDUNG:
    ok = show_sendung(params)
elif mode == MODE_PLAY:
    id = params["id"]
    json = getJSONForId( id)
    url = getVideoFromJSON( json)
    if "mark_in" in json.keys( ):
        start = json["mark_in"]
    elif "mark_in" in json["video"]["segments"][0].keys():
        start = json["video"]["segments"][0]["mark_in"]
    else: start = 0
    li = xbmcgui.ListItem( params[ PARAMETER_KEY_TITLE])
    li.setProperty( "IsPlayable", "true")
    li.setProperty( "Video", "true")
    li.setProperty( "startOffset", "%f" % (start))
    xbmc.Player().play( url, li)