addon.py

#
#      Copyright (C) 2014 Tommy Winther
#      http://tommy.winther.nu
#
#  This Program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2, or (at your option)
#  any later version.
#
#  This Program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this Program; see the file LICENSE.txt.  If not, write to
#  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
#  http://www.gnu.org/copyleft/gpl.html
#
import sys
import os
import urlparse
import urllib2

import xbmcaddon
import xbmcgui
import xbmcplugin

import buggalo

from channels import CHANNELS, CATEGORIES

TITLE_OFFSET = 30500


def showChannels(category=None):
    if category is not None:
        channels = CATEGORIES[category]
    else:
        channels = CHANNELS

    for c in channels:
        channel = c
        """:type : channels.Channel"""
        icon = os.path.join(ADDON.getAddonInfo('path'), 'resources', 'logos', '%d.png' % channel.channel_id)
        if not os.path.exists(icon):
            icon = ICON

        title = ADDON.getLocalizedString(TITLE_OFFSET + channel.channel_id)
        item = xbmcgui.ListItem(title, iconImage=icon, thumbnailImage=icon)
        item.setInfo('video', infoLabels={
            'title': title,
            'studio': ADDON.getLocalizedString(channel.category)
        })
        if channel.fanart is not None:
            item.setProperty('Fanart_Image', channel.fanart)
        else:
            item.setProperty('Fanart_Image', FANART)
        item.setProperty('IsPlayable', 'true')
        xbmcplugin.addDirectoryItem(HANDLE, PATH + '?playChannel=%d' % channel.channel_id, item)

    xbmcplugin.endOfDirectory(HANDLE)


def showCategories():
    for category in CATEGORIES:
        title = ADDON.getLocalizedString(category)
        item = xbmcgui.ListItem(title, iconImage=ICON, thumbnailImage=ICON)
        item.setProperty('Fanart_Image', FANART)
        url = PATH + '?category=%d' % category
        xbmcplugin.addDirectoryItem(HANDLE, url, item, isFolder=True)

    xbmcplugin.endOfDirectory(HANDLE)


def playChannel(channel_id):
    for c in CHANNELS:
        channel = c
        """:type : channels.Channel"""
        if str(channel.channel_id) == channel_id:
            url = channel.get_url()
            if url:
                icon = os.path.join(ADDON.getAddonInfo('path'), 'resources', 'logos', '%d.png' % channel.channel_id)
                if not os.path.exists(icon):
                    icon = ICON

                title = ADDON.getLocalizedString(TITLE_OFFSET + channel.channel_id)
                item = xbmcgui.ListItem(title, iconImage=icon, thumbnailImage=icon, path=url)
                item.setProperty('Fanart_Image', FANART)
                item.setProperty('IsLive', 'true')

                xbmcplugin.setResolvedUrl(HANDLE, True, item)
                break


def imInDenmark():
    if ADDON.getSetting('warn.if.not.in.denmark') != 'true':
        return  # Don't bother checking

    try:
        u = urllib2.urlopen('http://www.dr.dk/nu/api/estoyendinamarca.json', timeout=30)
        response = u.read()
        u.close()
        inDenmark = 'true' == response
    except Exception:
        # If an error occurred assume we are not in Denmark
        inDenmark = False

    if not inDenmark:
        heading = ADDON.getLocalizedString(30970)
        line1 = ADDON.getLocalizedString(30971)
        line2 = ADDON.getLocalizedString(30972)
        line3 = ADDON.getLocalizedString(30973)
        nolabel = ADDON.getLocalizedString(30974)
        yeslabel = ADDON.getLocalizedString(30975)
        if xbmcgui.Dialog().yesno(heading, line1, line2, line3, nolabel, yeslabel):
            ADDON.setSetting('warn.if.not.in.denmark', 'false')


if __name__ == '__main__':
    ADDON = xbmcaddon.Addon()
    PATH = sys.argv[0]
    HANDLE = int(sys.argv[1])
    PARAMS = urlparse.parse_qs(sys.argv[2][1:])

    FANART = os.path.join(ADDON.getAddonInfo('path'), 'fanart.jpg')
    ICON = os.path.join(ADDON.getAddonInfo('path'), 'icon.png')

    buggalo.SUBMIT_URL = 'http://tommy.winther.nu/exception/submit.php'
    try:
        if 'playChannel' in PARAMS:
            playChannel(PARAMS['playChannel'][0])
        elif 'category' in PARAMS:
            showChannels(int(PARAMS['category'][0]))
        elif ADDON.getSetting('group.by.category') == 'true':
            imInDenmark()
            showCategories()
        else:
            imInDenmark()
            showChannels()
    except Exception:
        buggalo.onExceptionRaised()