utils.py

# -*- coding: utf-8 -*-

import xbmc, xbmcaddon, xbmcvfs
import os, time, datetime, sys, json

UTF8 = 'utf-8'

addon = xbmcaddon.Addon()
__addonid__   = addon.getAddonInfo('id')
__addonname__ = addon.getAddonInfo('name')
__addonicon__ = addon.getAddonInfo('icon')
cacheSec = int(addon.getSetting('cachesec'))


def log(txt):
    message = '[%s]: %s' % (__addonid__, txt.encode('ascii', 'ignore'))
    xbmc.log(msg=message, level=xbmc.LOGDEBUG)


def logerr(txt):
    message = '[%s]: %s' % (__addonid__, txt.encode('ascii', 'ignore'))
    xbmc.log(msg=message, level=xbmc.LOGERROR)


def loginfo(txt):
    message = '[%s]: %s' % (__addonid__, txt.encode('ascii', 'ignore'))
    xbmc.log(msg=message, level=xbmc.LOGINFO)


def dequote(u):
    try:
        u = urllib.unquote_plus(u)
    except:
        pass
    return u


def boolean(s):
    if s == 'true':
        return True
    else:
        return False


def popup(txt):
    xbmc.executebuiltin('Notification(%s, %s, %d, %s)' % (__addonname__, txt.encode(UTF8), 5000, __addonicon__))


def cacheSave(fn, data, cacheTime=cacheSec, forceClear=False):
    log('Save cache: %s' % fn)

    path = os.path.dirname(os.path.abspath(fn))
    if not os.path.exists(path):
        log('Create path: %s' % path)
        xbmcvfs.mkdirs(path)

    if xbmcvfs.exists(fn):
        eTime = int(xbmcvfs.Stat(fn).st_mtime()) + cacheTime
        cTime = int(time.time())
        log('cTime=%s, eTime=%s' % (cTime, eTime))
        if cTime > eTime:
            log('Cache reset: expired')
            xbmcvfs.delete(fn)
        elif forceClear:
            log('Cache reset: force clear')
            xbmcvfs.delete(fn)

    try:
        fh = xbmcvfs.File(fn, 'w')
        fh.write(data)
        fh.close()
        log('Cache saved')
    except Exception as e:
        logerr('Error saving cache %s, err=%s' % (fn, e))


def cacheLoad(fn, cacheTime=cacheSec):
    log('Load cache: %s' % fn)
    data = None
    if xbmcvfs.exists(fn):
        eTime = int(xbmcvfs.Stat(fn).st_mtime()) + cacheTime
        cTime = int(time.time())
        log('cTime=%s, eTime=%s' % (cTime, eTime))
        if cTime > eTime:
            log('Cache reset: expired')
            xbmcvfs.delete(fn)
        else:
            fh = xbmcvfs.File(fn)
            data = fh.read()
            fh.close()

    return data


def xbmcJsonRequest(params):
    data = json.dumps(params)
    req = xbmc.executeJSONRPC(data)
    resp = json.loads(req)

    try:
        if 'result' in resp:
            return resp['result']
        return None
    except KeyError:
        logerr('[%s] %s' % (params['method'], resp['error']['message']))
        return None


def secondSinceEpoch(dtime):
    epoch = datetime.datetime(1970,1,1,8,0,0)       # assume HK GMT+8
    td = dtime - epoch
    if sys.version_info >= (2, 7):
       t = td.total_seconds()
    else:
       t = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
    return int(t)