addon.py

# coding=utf-8

import sys
import os
import xbmcaddon
import xbmcgui
import xbmcplugin
import requests
import re
import xmltodict
from six.moves.urllib.parse import parse_qs, urlencode


def build_url(query):
    """Build route url

    :param query: Dictionary to create URL for.
    :type query: dict
    :return: Complete route URL.
    :rtype: str
    """
    base_url = sys.argv[0]
    return base_url + '?' + urlencode(query)


def get_rss(url):
    """Download the source XML for given RSS URL using requests
    and parse the page using xmltodict.

    :param url: URL of RSS page.
    :type url: str
    :return: Dictionary of parsed XML RSS page.
    :rtype: dict
    """
    return xmltodict.parse(requests.get(url).text)


def get_channels():
    """Extract channels from rss.

    :return: Return dictionary of received channels.
    :rtype: dict
    """
    rss = get_rss('http://eco99fm.maariv.co.il/RSS_MusicChannels_Index/')
    channels = {}
    index = 1

    for item in rss["rss"]["channel"]["item"]:
        channels.update({
            index: {
                'album_cover':
                re.search("src='([^']+)'", item['description']).group(1),
                'title':
                item['title'],
                'description':
                item['itunes:summary'],
                'url':
                build_url({
                    'mode': 'playlist',
                    'url': item['link']
                })
            }
        })
        index += 1
    return channels


def get_playlists(url):
    """Get playlists of a channel.

    :param url: Channel rss url.
    :type url: str
    :return: Dictionary containing playlist items.
    :rtype: dict
    """
    rss = get_rss(url)
    playlists = {}
    index = 1
    for item in rss["rss"]["channel"]["item"]:
        playlists.update({
            index: {
                'album_cover':
                re.search("src='([^']+)'", item['description']).group(1),
                'title':
                item['title'],
                'description':
                item['itunes:summary'],
                'url':
                build_url({
                    'mode': 'stream',
                    'url': item['enclosure']['@url']
                })
            }
        })
        index += 1
    return playlists


def build_menu(items, is_folder):
    """Build menu control

    :param items: List of items, can be channels or playlist.
    :type items: list
    :param is_folder: If True the item is channel else a playlist.
    :type is_folder: bool
    """
    items_list = []

    for item in items:
        # create a list item using the song filename for the label
        li = xbmcgui.ListItem(label=items[item]['title'])
        # set the fanart to the album cover
        if not is_folder:
            li.setProperty('IsPlayable', 'true')
        li.setProperty('PlotOutline', items[item]['description'])
        li.setInfo(
            'video', {
                'title': items[item]['title'],
                'genre': 'Podcast',
                'plot': items[item]['description']
            })
        li.setArt({
            'thumb':
            items[item]['album_cover'],
            'poster':
            items[item]['album_cover'],
            'fanart':
            os.path.join(ADDON_FOLDER, 'resources', 'media', 'fanart.jpg')
        })
        url = items[item]['url']
        items_list.append((url, li, is_folder))
    xbmcplugin.addDirectoryItems(ADDON_HANDLE, items_list, len(items_list))
    xbmcplugin.setContent(ADDON_HANDLE, 'songs')
    xbmcplugin.endOfDirectory(ADDON_HANDLE)


def play(url):
    """Play playlist by URL.

    :param url: URL of playlist.
    :type url: str
    """
    play_item = xbmcgui.ListItem(path=url)
    xbmcplugin.setResolvedUrl(ADDON_HANDLE, True, listitem=play_item)


def main():
    """Main method."""
    args = parse_qs(sys.argv[2][1:])
    mode = args.get('mode', None)
    if mode is None:
        items = get_channels()
        build_menu(items, True)
    elif mode[0] == 'playlist':
        items = get_playlists(args['url'][0])
        build_menu(items, False)
    elif mode[0] == 'stream':
        play(args['url'][0].replace('/playlist.m3u8', ''))


if __name__ == '__main__':
    ADDON_FOLDER = xbmcaddon.Addon().getAddonInfo('path')
    ADDON_HANDLE = int(sys.argv[1])
    main()