utils.py

# -*- coding: utf-8 -*-
# Author: cache-sk
# Created on: 15.4.2019
import xbmcaddon
import xbmcgui
import skylink
import exports
import requests
import logger
import os

try:
    # Python 3.x
    from urllib.parse import parse_qs as comp_parse_qs, urlencode as comp_urlencode
except ImportError:
    # Python 2.5+
    from urlparse import parse_qs as comp_parse_qs
    from urllib import urlencode as comp_urlencode


_addon = xbmcaddon.Addon()
_skylink_logos = 'false' != _addon.getSetting('a_sl_logos')
if not _skylink_logos:
    _remote_logos = '1' == _addon.getSetting('a_logos_location')
    if _remote_logos:
        _logos_base_url = _addon.getSetting('a_logos_base_url')
        if not _logos_base_url.endswith("/"):
            _logos_base_url = _logos_base_url + "/"
    else:
        _logos_folder = _addon.getSetting('a_logos_folder')


def get_logo(title, api_url):
    if _skylink_logos:
        return api_url + exports.logo_sl_location(title)

    if _remote_logos:
        return _logos_base_url + exports.logo_id(title)

    return os.path.join(_logos_folder, exports.logo_id(title))


def strip_devices(devices):
    strip = _addon.getSetting('device_web_only') == 'true'
    if not strip:
        return devices
    stripped = []
    for device in devices:
        if device['type'] == 'web':
            stripped.append(device)
    if stripped == []:
        dialog = xbmcgui.Dialog()
        dialog.ok(_addon.getAddonInfo('name'), _addon.getLocalizedString(30506))

    return stripped


def select_device(devices):
    devices = strip_devices(devices)
    dialog = xbmcgui.Dialog()
    items = []
    for device in devices:
        items.append(device['name'].replace("+", " "))
    d = dialog.select(_addon.getLocalizedString(30403), items)
    return devices[d]['id'] if d > -1 else ''


def get_last_used_device(devices):
    devices = strip_devices(devices)
    la = 9999999999999
    device = ''
    for d in devices:
        if d['lastactivity'] < la:
            device = d['id']
            la = d['lastactivity']
    return device


def call(sl, fn):
    result = None
    try:
        result = fn()
    except skylink.TooManyDevicesException as e:
        if _addon.getSetting('reuse_last_device') == 'true':
            device = get_last_used_device(e.devices)
        else:
            device = select_device(e.devices)

        if device != '':
            logger.log.info('reconnecting as: ' + device)
            sl.reconnect(device)
            result = fn()
    except requests.exceptions.ConnectionError:
        dialog = xbmcgui.Dialog()
        dialog.ok(_addon.getAddonInfo('name'), _addon.getLocalizedString(30506))

    return result


def ask_for_pin(sl):
    pin = sl.pin_info()
    if pin is not None:
        dialog = xbmcgui.Dialog()
        d = dialog.input(_addon.getLocalizedString(30508), type=xbmcgui.INPUT_NUMERIC)
        if d != pin:
            dialog.ok(_addon.getAddonInfo('name'), _addon.getLocalizedString(30509))
            return False
    return True


def dec_utf8(str):
    try:
        return str.decode("utf-8")  # Python 2.x
    except AttributeError:
        return str  # Python 3.x


def parse_qs(qs):
    return comp_parse_qs(qs)


def urlencode(query):
    return comp_urlencode(query)