1395 lines
44 KiB
Python
1395 lines
44 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Created on: 03.06.2015
|
|
"""
|
|
SimplePlugin micro-framework for Kodi content plugins
|
|
|
|
**Author**: Roman Miroshnychenko aka Roman V.M.
|
|
|
|
**License**: `GPL v.3 <https://www.gnu.org/copyleft/gpl.html>`_
|
|
"""
|
|
|
|
from __future__ import unicode_literals
|
|
from future.builtins import (zip, super,
|
|
bytes, dict, int, list, object, str)
|
|
from future.utils import (PY2, PY3, iteritems, itervalues,
|
|
python_2_unicode_compatible)
|
|
# from future.standard_library import install_aliases
|
|
# install_aliases()
|
|
|
|
if PY3:
|
|
basestring = str
|
|
long = int
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import inspect
|
|
import time
|
|
import hashlib
|
|
import pickle
|
|
from collections import MutableMapping, namedtuple
|
|
from copy import deepcopy
|
|
from functools import wraps
|
|
from shutil import copyfile
|
|
from contextlib import contextmanager
|
|
from pprint import pformat
|
|
from platform import uname
|
|
if PY3:
|
|
from urllib.parse import urlencode, quote_plus, urlparse, unquote_plus, parse_qs
|
|
else:
|
|
from future.backports.urllib.parse import urlencode, quote_plus, urlparse, unquote_plus
|
|
from urlparse import parse_qs
|
|
import xbmcaddon
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcvfs
|
|
|
|
__all__ = ['SimplePluginError', 'Storage', 'MemStorage', 'Addon', 'Plugin',
|
|
'RoutedPlugin', 'Params', 'log_exception', 'py2_encode',
|
|
'py2_decode', 'translate_path']
|
|
|
|
if PY3:
|
|
getargspec = inspect.getfullargspec
|
|
else:
|
|
getargspec = inspect.getargspec
|
|
|
|
Route = namedtuple('Route', ['pattern', 'func'])
|
|
|
|
|
|
class SimplePluginError(Exception):
|
|
"""Custom exception"""
|
|
pass
|
|
|
|
|
|
class TimeoutError(SimplePluginError):
|
|
pass
|
|
|
|
|
|
def _format_vars(variables):
|
|
"""
|
|
Format variables dictionary
|
|
|
|
:param variables: variables dict
|
|
:type variables: dict
|
|
:return: formatted string with sorted ``var = val`` pairs
|
|
:rtype: str
|
|
"""
|
|
var_list = [(var, val) for var, val in iteritems(variables)]
|
|
lines = []
|
|
for var, val in sorted(var_list, key=lambda i: i[0]):
|
|
if not (var.startswith('__') or var.endswith('__')):
|
|
lines.append('{0} = {1}'.format(var, pformat(val)))
|
|
return '\n'.join(lines)
|
|
|
|
|
|
def py2_encode(s, encoding='utf-8'):
|
|
"""
|
|
Encode Python 2 ``unicode`` to ``str``
|
|
|
|
In Python 3 the string is not changed.
|
|
"""
|
|
if PY2 and isinstance(s, str):
|
|
s = s.encode(encoding)
|
|
return s
|
|
|
|
|
|
def py2_decode(s, encoding='utf-8'):
|
|
"""
|
|
Decode Python 2 ``str`` to ``unicode``
|
|
|
|
In Python 3 the string is not changed.
|
|
"""
|
|
if PY2 and isinstance(s, bytes):
|
|
s = s.decode(encoding)
|
|
return s
|
|
|
|
def _kodi_major_version():
|
|
kodi_version = xbmc.getInfoLabel('System.BuildVersion').split(' ')[0]
|
|
return kodi_version.split('.')[0]
|
|
|
|
def translate_path(*args, **kwargs):
|
|
if _kodi_major_version() < '19':
|
|
return xbmc.translatePath(*args, **kwargs)
|
|
else:
|
|
return xbmcvfs.translatePath(*args, **kwargs)
|
|
|
|
|
|
@contextmanager
|
|
def log_exception(logger=None):
|
|
"""
|
|
Diagnostic helper context manager
|
|
|
|
It controls execution within its context and writes extended
|
|
diagnostic info to the Kodi log if an unhandled exception
|
|
happens within the context. The info includes the following items:
|
|
|
|
- System info
|
|
- Kodi version
|
|
- Module path.
|
|
- Code fragment where the exception has happened.
|
|
- Global variables.
|
|
- Local variables.
|
|
|
|
After logging the diagnostic info the exception is re-raised.
|
|
|
|
Example::
|
|
|
|
with log_exception():
|
|
# Some risky code
|
|
raise RuntimeError('Fatal error!')
|
|
|
|
:param logger: logger function which must accept a single argument
|
|
which is a log message. By default it is :func:`xbmc.log`
|
|
with ``ERROR`` level.
|
|
"""
|
|
try:
|
|
yield
|
|
except:
|
|
if logger is None:
|
|
logger = lambda msg: xbmc.log(py2_encode(msg), xbmc.LOGERROR)
|
|
frame_info = inspect.trace(5)[-1]
|
|
logger('Unhandled exception detected!')
|
|
logger('*** Start diagnostic info ***')
|
|
logger('System info: {0}'.format(uname()))
|
|
logger('OS info: {0}'.format(py2_decode(xbmc.getInfoLabel('System.OSVersionInfo'))))
|
|
logger('Kodi version: {0}'.format(
|
|
xbmc.getInfoLabel('System.BuildVersion'))
|
|
)
|
|
logger('File: {0}'.format(frame_info[1]))
|
|
context = ''
|
|
if frame_info[4] is not None:
|
|
for i, line in enumerate(frame_info[4], frame_info[2] - frame_info[5]):
|
|
if i == frame_info[2]:
|
|
context += '{0}:>{1}'.format(str(i).rjust(5), line)
|
|
else:
|
|
context += '{0}: {1}'.format(str(i).rjust(5), line)
|
|
logger('Code context:\n' + context)
|
|
logger('Global variables:\n' + _format_vars(frame_info[0].f_globals))
|
|
logger('Local variables:\n' + _format_vars(frame_info[0].f_locals))
|
|
logger('**** End diagnostic info ****')
|
|
raise
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Params(dict):
|
|
"""
|
|
Params(**kwargs)
|
|
|
|
A class that stores parsed plugin call parameters
|
|
|
|
Parameters can be accessed both through :class:`dict` keys and
|
|
instance properties.
|
|
|
|
.. note:: For a missing parameter an instance property returns ``None``.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
@plugin.action('foo')
|
|
def action(params):
|
|
foo = params['foo'] # Access by key
|
|
bar = params.bar # Access through property. Both variants are equal
|
|
"""
|
|
def __getattr__(self, key):
|
|
return self.get(key)
|
|
|
|
def __str__(self):
|
|
return '<Params {0}>'.format(super(Params, self).__str__())
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Storage(MutableMapping):
|
|
"""
|
|
Storage(storage_dir, filename='storage.pcl')
|
|
|
|
Persistent storage for arbitrary data with a dictionary-like interface
|
|
|
|
It is designed as a context manager and better be used
|
|
with 'with' statement.
|
|
|
|
:param storage_dir: directory for storage
|
|
:type storage_dir: str
|
|
:param filename: the name of a storage file (optional)
|
|
:type filename: str
|
|
|
|
Usage::
|
|
|
|
with Storage('/foo/bar/storage/') as storage:
|
|
storage['key1'] = value1
|
|
value2 = storage['key2']
|
|
|
|
.. note:: After exiting :keyword:`with` block a :class:`Storage` instance
|
|
is invalidated. Storage contents are saved to disk only for
|
|
a new storage or if the contents have been changed.
|
|
"""
|
|
def __init__(self, storage_dir, filename='storage.pcl'):
|
|
"""
|
|
Class constructor
|
|
|
|
:type storage_dir: str
|
|
:type filename: str
|
|
"""
|
|
self._storage = {}
|
|
self._hash = None
|
|
self._filename = os.path.join(storage_dir, filename)
|
|
try:
|
|
with open(self._filename, 'rb') as fo:
|
|
contents = fo.read()
|
|
self._storage = pickle.loads(contents)
|
|
self._hash = hashlib.md5(contents).hexdigest()
|
|
except (IOError, pickle.PickleError, EOFError, AttributeError):
|
|
pass
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, t, v, tb):
|
|
self.flush()
|
|
|
|
def __getitem__(self, key):
|
|
return self._storage[key]
|
|
|
|
def __setitem__(self, key, value):
|
|
self._storage[key] = value
|
|
|
|
def __delitem__(self, key):
|
|
del self._storage[key]
|
|
|
|
def __iter__(self):
|
|
return iter(self._storage)
|
|
|
|
def __len__(self):
|
|
return len(self._storage)
|
|
|
|
def __str__(self):
|
|
return '<Storage {0}>'.format(self._storage)
|
|
|
|
def flush(self):
|
|
"""
|
|
Save storage contents to disk
|
|
|
|
This method saves new and changed :class:`Storage` contents to disk
|
|
and invalidates the Storage instance. Unchanged Storage is not saved
|
|
but simply invalidated.
|
|
"""
|
|
contents = pickle.dumps(self._storage, protocol=2)
|
|
if self._hash is None or hashlib.md5(contents).hexdigest() != self._hash:
|
|
tmp = self._filename + '.tmp'
|
|
start = time.time()
|
|
while os.path.exists(tmp):
|
|
if time.time() - start > 2.0:
|
|
raise TimeoutError(
|
|
'Exceeded timeout for saving {0} contents!'.format(self)
|
|
)
|
|
xbmc.sleep(100)
|
|
try:
|
|
with open(tmp, 'wb') as fo:
|
|
fo.write(contents)
|
|
copyfile(tmp, self._filename)
|
|
finally:
|
|
os.remove(tmp)
|
|
del self._storage
|
|
|
|
def copy(self):
|
|
"""
|
|
Make a copy of storage contents
|
|
|
|
.. note:: this method performs a *deep* copy operation.
|
|
|
|
:return: a copy of storage contents
|
|
:rtype: dict
|
|
"""
|
|
return deepcopy(self._storage)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class MemStorage(MutableMapping):
|
|
"""
|
|
MemStorage(storage_id)
|
|
|
|
In-memory storage with dict-like interface
|
|
|
|
The data is stored in the Kodi core so contents of a MemStorage instance
|
|
with the same ID can be shared between different Python processes.
|
|
|
|
.. note:: Keys are case-insensitive
|
|
|
|
.. warning:: :class:`MemStorage` does not allow to modify mutable objects
|
|
in place! You need to assign them to variables first, modify and
|
|
store them back to a MemStorage instance.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
storage = MemStorage('foo')
|
|
some_list = storage['bar']
|
|
some_list.append('spam')
|
|
storage['bar'] = some_list
|
|
|
|
:param storage_id: ID of this storage instance
|
|
:type storage_id: str
|
|
:param window_id: the ID of a Kodi Window object where storage contents
|
|
will be stored.
|
|
:type window_id: int
|
|
"""
|
|
def __init__(self, storage_id, window_id=10000):
|
|
"""
|
|
:type storage_id: str
|
|
:type window_id: int
|
|
"""
|
|
self._id = storage_id
|
|
self._window = xbmcgui.Window(window_id)
|
|
try:
|
|
self['__keys__']
|
|
except KeyError:
|
|
self['__keys__'] = []
|
|
|
|
def _check_key(self, key):
|
|
"""
|
|
:type key: str
|
|
"""
|
|
if not isinstance(key, basestring):
|
|
raise TypeError('Storage key must be of str type!')
|
|
|
|
def _format_contents(self):
|
|
"""
|
|
:rtype: str
|
|
"""
|
|
lines = []
|
|
for key, val in iteritems(self):
|
|
lines.append('{0}: {1}'.format(repr(key), repr(val)))
|
|
return ', '.join(lines)
|
|
|
|
def __str__(self):
|
|
return '<MemStorage {{{0}}}>'.format(self._format_contents())
|
|
|
|
def __getitem__(self, key):
|
|
self._check_key(key)
|
|
full_key = py2_encode('{0}__{1}'.format(self._id, key))
|
|
raw_item = self._window.getProperty(full_key)
|
|
if raw_item:
|
|
try:
|
|
return pickle.loads(bytes(raw_item))
|
|
except TypeError as e:
|
|
return pickle.loads(bytes(raw_item, 'utf-8', errors='surrogateescape'))
|
|
else:
|
|
raise KeyError(key)
|
|
|
|
def __setitem__(self, key, value):
|
|
self._check_key(key)
|
|
full_key = py2_encode('{0}__{1}'.format(self._id, key))
|
|
# protocol=0 is needed for safe string handling in Python 3
|
|
self._window.setProperty(full_key, pickle.dumps(value, protocol=0))
|
|
if key != '__keys__':
|
|
keys = self['__keys__']
|
|
keys.append(key)
|
|
self['__keys__'] = keys
|
|
|
|
def __delitem__(self, key):
|
|
self._check_key(key)
|
|
full_key = py2_encode('{0}__{1}'.format(self._id, key))
|
|
item = self._window.getProperty(full_key)
|
|
if item:
|
|
self._window.clearProperty(full_key)
|
|
if key != '__keys__':
|
|
keys = self['__keys__']
|
|
keys.remove(key)
|
|
self['__keys__'] = keys
|
|
else:
|
|
raise KeyError(key)
|
|
|
|
def __contains__(self, key):
|
|
self._check_key(key)
|
|
full_key = py2_encode('{0}__{1}'.format(self._id, key))
|
|
item = self._window.getProperty(full_key)
|
|
return bool(item)
|
|
|
|
def __iter__(self):
|
|
return iter(self['__keys__'])
|
|
|
|
def __len__(self):
|
|
return len(self['__keys__'])
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Addon(object):
|
|
"""
|
|
Base addon class
|
|
|
|
Provides access to basic addon parameters
|
|
|
|
:param id_: addon id, e.g. 'plugin.video.foo' (optional)
|
|
:type id_: str
|
|
"""
|
|
def __init__(self, id_=''):
|
|
"""
|
|
Class constructor
|
|
|
|
:type id_: str
|
|
"""
|
|
self._addon = xbmcaddon.Addon(id_)
|
|
self._profile_dir = py2_decode(
|
|
translate_path(self._addon.getAddonInfo('profile'))
|
|
)
|
|
self._ui_strings_map = None
|
|
if not os.path.exists(self._profile_dir):
|
|
os.mkdir(self._profile_dir)
|
|
|
|
def __str__(self):
|
|
return '<Addon [{0}]>'.format(self.id)
|
|
|
|
@property
|
|
def addon(self):
|
|
"""
|
|
Kodi Addon instance that represents this Addon
|
|
|
|
:return: Addon instance
|
|
:rtype: xbmcaddon.Addon
|
|
"""
|
|
return self._addon
|
|
|
|
@property
|
|
def id(self):
|
|
"""
|
|
Addon ID
|
|
|
|
:return: Addon ID, e.g. 'plugin.video.foo'
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('id')
|
|
|
|
@property
|
|
def path(self):
|
|
"""
|
|
Addon path
|
|
|
|
:return: path to the addon folder
|
|
:rtype: unicode
|
|
"""
|
|
return py2_decode(self._addon.getAddonInfo('path'))
|
|
|
|
@property
|
|
def icon(self):
|
|
"""
|
|
Addon icon
|
|
|
|
:return: path to the addon icon image
|
|
:rtype: str
|
|
"""
|
|
icon = os.path.join(self.path, self._addon.getAddonInfo('icon'))
|
|
if not icon:
|
|
icon = os.path.join(self.path, 'icon.png')
|
|
if os.path.exists(icon):
|
|
return icon
|
|
else:
|
|
return ''
|
|
|
|
@property
|
|
def fanart(self):
|
|
"""
|
|
Addon fanart
|
|
|
|
:return: path to the addon fanart image
|
|
:rtype: str
|
|
"""
|
|
fanart = self._addon.getAddonInfo('fanart')
|
|
if not fanart :
|
|
fanart = os.path.join(self.path, 'fanart.jpg')
|
|
if os.path.exists(fanart):
|
|
return fanart
|
|
else:
|
|
return ''
|
|
|
|
@property
|
|
def profile_dir(self):
|
|
"""
|
|
Addon config dir
|
|
|
|
:return: path to the addon profile dir
|
|
:rtype: str
|
|
"""
|
|
return self._profile_dir
|
|
|
|
@property
|
|
def version(self):
|
|
"""
|
|
Addon version
|
|
|
|
:return: addon version
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('version')
|
|
|
|
@property
|
|
def name(self):
|
|
"""
|
|
Addon name
|
|
|
|
:return: addon name
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('name')
|
|
|
|
@property
|
|
def author(self):
|
|
"""
|
|
Addon author
|
|
|
|
:return: addon author
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('author')
|
|
|
|
@property
|
|
def changelog(self):
|
|
"""
|
|
Addon changelog
|
|
|
|
:return: addon changelog
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('changelog')
|
|
|
|
@property
|
|
def description(self):
|
|
"""
|
|
Addon description
|
|
|
|
:return: addon description
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('description')
|
|
|
|
@property
|
|
def disclaimer(self):
|
|
"""
|
|
Addon disclaimer
|
|
|
|
:return: addon disclaimer
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('disclaimer')
|
|
|
|
@property
|
|
def stars(self):
|
|
"""
|
|
Addon stars
|
|
|
|
:return: addon stars
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('stars')
|
|
|
|
@property
|
|
def summary(self):
|
|
"""
|
|
Addon summary
|
|
|
|
:return: addon summary
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('summary')
|
|
|
|
@property
|
|
def type(self):
|
|
"""
|
|
Addon type
|
|
|
|
:return: addon type
|
|
:rtype: str
|
|
"""
|
|
return self._addon.getAddonInfo('type')
|
|
|
|
def get_localized_string(self, id_):
|
|
"""
|
|
Get localized UI string
|
|
|
|
:param id_: UI string ID
|
|
:type id_: int
|
|
:return: UI string in the current language
|
|
:rtype: unicode
|
|
"""
|
|
return self._addon.getLocalizedString(id_)
|
|
|
|
def get_setting(self, id_, convert=True):
|
|
"""
|
|
Get addon setting
|
|
|
|
If ``convert=True``, 'bool' settings are converted to Python
|
|
:class:`bool` values, and numeric strings to Python :class:`long` or
|
|
:class:`float` depending on their format.
|
|
|
|
.. note:: Settings can also be read via :class:`Addon` instance
|
|
poperties named as the respective settings. I.e. ``addon.foo``
|
|
is equal to ``addon.get_setting('foo')``.
|
|
|
|
:param id_: setting ID
|
|
:type id_: str
|
|
:param convert: try to guess and convert the setting to an appropriate type
|
|
E.g. ``'1.0'`` will be converted to float ``1.0`` number,
|
|
``'true'`` to ``True`` and so on.
|
|
:type convert: bool
|
|
:return: setting value
|
|
"""
|
|
setting = py2_decode(self._addon.getSetting(id_))
|
|
if convert:
|
|
if setting == 'true':
|
|
return True # Convert boolean strings to bool
|
|
elif setting == 'false':
|
|
return False
|
|
elif re.search(r'^-?\d+$', setting) is not None:
|
|
return long(setting) # Convert numeric strings to long
|
|
elif re.search(r'^-?\d+\.\d+$', setting) is not None:
|
|
return float(setting) # Convert numeric strings with a dot to float
|
|
return setting
|
|
|
|
def set_setting(self, id_, value):
|
|
"""
|
|
Set addon setting
|
|
|
|
Python :class:`bool` type are converted to ``'true'`` or ``'false'``
|
|
Non-string/non-unicode values are converted to strings.
|
|
|
|
.. warning:: Setting values via :class:`Addon` instance properties
|
|
is not supported! Values can only be set using
|
|
:meth:`Addon.set_setting` method.
|
|
|
|
:param id_: setting ID
|
|
:type id_: str
|
|
:param value: setting value
|
|
"""
|
|
if isinstance(value, bool):
|
|
value = 'true' if value else 'false'
|
|
elif not isinstance(value, basestring):
|
|
value = str(value)
|
|
self._addon.setSetting(id_, py2_encode(value))
|
|
|
|
def log(self, message, level=xbmc.LOGDEBUG):
|
|
"""
|
|
Add message to Kodi log starting with Addon ID
|
|
|
|
:param message: message to be written into the Kodi log
|
|
:type message: str
|
|
:param level: log level. :mod:`xbmc` module provides the necessary
|
|
symbolic constants. Default: ``xbmc.LOGDEBUG``
|
|
:type level: int
|
|
"""
|
|
xbmc.log(
|
|
py2_encode('{0} [v.{1}]: {2}'.format(self.id, self.version, message)),
|
|
level
|
|
)
|
|
|
|
def log_notice(self, message):
|
|
"""
|
|
Add NOTICE message to the Kodi log
|
|
|
|
:param message: message to write to the Kodi log
|
|
:type message: str
|
|
"""
|
|
if _kodi_major_version() < '19':
|
|
self.log(message, xbmc.LOGNOTICE)
|
|
else:
|
|
self.log(message, xbmc.LOGINFO)
|
|
|
|
def log_warning(self, message):
|
|
"""
|
|
Add WARNING message to the Kodi log
|
|
|
|
:param message: message to write to the Kodi log
|
|
:type message: str
|
|
"""
|
|
self.log(message, xbmc.LOGWARNING)
|
|
|
|
def log_error(self, message):
|
|
"""
|
|
Add ERROR message to the Kodi log
|
|
|
|
:param message: message to write to the Kodi log
|
|
:type message: str
|
|
"""
|
|
self.log(message, xbmc.LOGERROR)
|
|
|
|
def log_debug(self, message):
|
|
"""
|
|
Add debug message to the Kodi log
|
|
|
|
:param message: message to write to the Kodi log
|
|
:type message: str
|
|
"""
|
|
self.log(message, xbmc.LOGDEBUG)
|
|
|
|
def get_storage(self, filename='storage.pcl'):
|
|
"""
|
|
Get a persistent :class:`Storage` instance for storing arbitrary values
|
|
between addon calls.
|
|
|
|
A :class:`Storage` instance can be used as a context manager.
|
|
|
|
Example::
|
|
|
|
with plugin.get_storage() as storage:
|
|
storage['param1'] = value1
|
|
value2 = storage['param2']
|
|
|
|
.. note:: After exiting :keyword:`with` block a :class:`Storage`
|
|
instance is invalidated.
|
|
|
|
:param filename: the name of a storage file (optional)
|
|
:type filename: str
|
|
:return: Storage object
|
|
:rtype: Storage
|
|
"""
|
|
return Storage(self.profile_dir, filename)
|
|
|
|
def get_mem_storage(self, storage_id='', window_id=10000):
|
|
"""
|
|
Creates an in-memory storage for this addon with :class:`dict`-like
|
|
interface
|
|
|
|
The storage can store picklable Python objects as long as
|
|
Kodi is running and storage contents can be shared between
|
|
Python processes. Different addons have separate storages,
|
|
so storages with the same names created with this method
|
|
do not conflict.
|
|
|
|
Example::
|
|
|
|
addon = Addon()
|
|
storage = addon.get_mem_storage()
|
|
foo = storage['foo']
|
|
storage['bar'] = bar
|
|
|
|
:param storage_id: optional storage ID (case-insensitive).
|
|
:type storage_id: str
|
|
:param window_id: the ID of a Kodi Window object where storage contents
|
|
will be stored.
|
|
:type window_id: int
|
|
:return: in-memory storage for this addon
|
|
:rtype: MemStorage
|
|
"""
|
|
if storage_id:
|
|
storage_id = '{0}_{1}'.format(self.id, storage_id)
|
|
return MemStorage(storage_id, window_id)
|
|
|
|
def _get_cached_data(self, cache, func, duration, *args, **kwargs):
|
|
"""
|
|
Get data from a cache object
|
|
|
|
:param cache: cache object
|
|
:param func: function to cache
|
|
:param duration: cache duration
|
|
:param args: function args
|
|
:param kwargs: function kwargs
|
|
:return: function return data
|
|
"""
|
|
if duration <= 0:
|
|
raise ValueError('Caching duration cannot be zero or negative!')
|
|
current_time = time.time()
|
|
key = func.__name__ + str(args) + str(kwargs)
|
|
try:
|
|
data, timestamp = cache[key]
|
|
if current_time - timestamp > duration * 60:
|
|
raise KeyError
|
|
self.log_debug('Cache hit: {0}'.format(key))
|
|
except KeyError:
|
|
self.log_debug('Cache miss: {0}'.format(key))
|
|
data = func(*args, **kwargs)
|
|
cache[key] = (data, current_time)
|
|
return data
|
|
|
|
def cached(self, duration=10):
|
|
"""
|
|
Cached decorator
|
|
|
|
Used to cache function return data
|
|
|
|
Usage::
|
|
|
|
@plugin.cached(30)
|
|
def my_func(*args, **kwargs):
|
|
# Do some stuff
|
|
return value
|
|
|
|
:param duration: caching duration in min (positive values only)
|
|
:type duration: int
|
|
:raises ValueError: if duration is zero or negative
|
|
"""
|
|
def outer_wrapper(func):
|
|
@wraps(func)
|
|
def inner_wrapper(*args, **kwargs):
|
|
with self.get_storage('__cache__.pcl') as cache:
|
|
return self._get_cached_data(cache, func, duration,
|
|
*args, **kwargs)
|
|
return inner_wrapper
|
|
return outer_wrapper
|
|
|
|
def mem_cached(self, duration=10):
|
|
"""
|
|
In-memory cache decorator
|
|
|
|
Usage::
|
|
|
|
@plugin.mem_cached(30)
|
|
def my_func(*args, **kwargs):
|
|
# Do some stuff
|
|
return value
|
|
|
|
:param duration: caching duration in min (positive values only)
|
|
:type duration: int
|
|
:raises ValueError: if duration is zero or negative
|
|
"""
|
|
def outer_wrapper(func):
|
|
@wraps(func)
|
|
def inner_wrapper(*args, **kwargs):
|
|
cache = self.get_mem_storage('***cache***')
|
|
return self._get_cached_data(cache, func, duration,
|
|
*args, **kwargs)
|
|
return inner_wrapper
|
|
return outer_wrapper
|
|
|
|
def gettext(self, ui_string):
|
|
"""
|
|
Get a translated UI string from addon localization files.
|
|
|
|
This function emulates GNU Gettext for more convenient access
|
|
to localized addon UI strings. To reduce typing this method object
|
|
can be assigned to a ``_`` (single underscore) variable.
|
|
|
|
For using gettext emulation :meth:`Addon.initialize_gettext` method
|
|
needs to be called first. See documentation for that method for more
|
|
info about Gettext emulation.
|
|
|
|
:param ui_string: a UI string from English :file:`strings.po`.
|
|
:type ui_string: str
|
|
:return: a UI string from translated :file:`strings.po`.
|
|
:rtype: unicode
|
|
:raises SimplePluginError: if :meth:`Addon.initialize_gettext`
|
|
wasn't called first or if a string is not found in
|
|
English :file:`strings.po`.
|
|
"""
|
|
if self._ui_strings_map is not None:
|
|
try:
|
|
return self.get_localized_string(
|
|
self._ui_strings_map['strings'][ui_string]
|
|
)
|
|
except KeyError:
|
|
raise SimplePluginError(
|
|
'UI string "{0}" is not found in strings.po!'.format(
|
|
ui_string)
|
|
)
|
|
else:
|
|
raise SimplePluginError('Addon localization is not initialized!')
|
|
|
|
def initialize_gettext(self):
|
|
"""
|
|
Initialize GNU gettext emulation in addon
|
|
|
|
Kodi localization system for addons is not very convenient
|
|
because you need to operate with numeric string codes instead
|
|
of UI strings themselves which reduces code readability and
|
|
may lead to errors. The :class:`Addon` class provides facilities
|
|
for emulating GNU Gettext localization system.
|
|
|
|
This allows to use UI strings from addon's English :file:`strings.po`
|
|
file instead of numeric codes to return localized strings from
|
|
respective localized :file:`.po` files.
|
|
|
|
This method returns :meth:`Addon.gettext` method object that
|
|
can be assigned to a short alias to reduce typing. Traditionally,
|
|
``_`` (a single underscore) is used for this purpose.
|
|
|
|
Example::
|
|
|
|
addon = simpleplugin.Addon()
|
|
_ = addon.initialize_gettext()
|
|
|
|
xbmcgui.Dialog().notification(_('Warning!'), _('Something happened'))
|
|
|
|
In the preceding example the notification strings will be replaced
|
|
with localized versions if these strings are translated.
|
|
|
|
:return: :meth:`Addon.gettext` method object
|
|
:raises SimplePluginError: if the addon's English :file:`strings.po`
|
|
file is missing
|
|
"""
|
|
strings_po = os.path.join(self.path, 'resources', 'language',
|
|
'resource.language.en_gb', 'strings.po')
|
|
if not os.path.exists(strings_po):
|
|
strings_po = os.path.join(self.path, 'resources', 'language',
|
|
'English', 'strings.po')
|
|
if os.path.exists(strings_po):
|
|
with open(strings_po, 'rb') as fo:
|
|
raw_strings = fo.read()
|
|
raw_strings_hash = hashlib.md5(raw_strings).hexdigest()
|
|
ui_strings_map = self.get_mem_storage('__gettext__')
|
|
if raw_strings_hash != ui_strings_map.get('hash', ''):
|
|
ui_strings = self._parse_po(
|
|
raw_strings.decode('utf-8').split('\n')
|
|
)
|
|
self._ui_strings_map = {
|
|
'hash': raw_strings_hash,
|
|
'strings': ui_strings
|
|
}
|
|
ui_strings_map['hash'] = raw_strings_hash
|
|
ui_strings_map['strings'] = ui_strings.copy()
|
|
else:
|
|
self._ui_strings_map = {}
|
|
self._ui_strings_map.update(ui_strings_map)
|
|
else:
|
|
raise SimplePluginError('Unable to initialize localization because '
|
|
'of missing English strings.po!')
|
|
return self.gettext
|
|
|
|
def _parse_po(self, strings):
|
|
"""
|
|
Parses ``strings.po`` file into a dict of {'string': id} items.
|
|
"""
|
|
ui_strings = {}
|
|
string_id = None
|
|
for string in strings:
|
|
if string_id is None and 'msgctxt' in string:
|
|
string_id = int(re.search(r'"#(\d+)"', string, re.U).group(1))
|
|
elif string_id is not None and 'msgid' in string:
|
|
ui_strings[re.search(r'"(.*?)"', string, re.U).group(1)] = string_id
|
|
string_id = None
|
|
return ui_strings
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Plugin(Addon):
|
|
"""
|
|
Plugin class with URL query string routing.
|
|
|
|
It provides a simplified plugin call routing mechanism using URL query strings.
|
|
A URL query string must contain "action" parameter that defines which function
|
|
will be invoked during this plugin call.
|
|
|
|
:param id_: plugin's id, e.g. 'plugin.video.foo' (optional)
|
|
:type id_: str
|
|
"""
|
|
def __init__(self, id_=''):
|
|
"""
|
|
Class constructor
|
|
|
|
:type id_: str
|
|
"""
|
|
super(Plugin, self).__init__(id_)
|
|
self._url = 'plugin://{0}/'.format(self.id)
|
|
self._handle = None
|
|
self.actions = {}
|
|
self._params = None
|
|
|
|
def __str__(self):
|
|
return '<Plugin {0}>'.format(sys.argv)
|
|
|
|
@property
|
|
def params(self):
|
|
"""
|
|
Get plugin call parameters
|
|
|
|
:return: plugin call parameters
|
|
:rtype: Params
|
|
"""
|
|
return self._params
|
|
|
|
@property
|
|
def handle(self):
|
|
"""
|
|
Get plugin handle
|
|
|
|
:return: plugin handle
|
|
:rtype: int
|
|
"""
|
|
return self._handle
|
|
|
|
@staticmethod
|
|
def get_params(paramstring):
|
|
"""
|
|
Convert a URL-encoded paramstring to a Python dict
|
|
|
|
:param paramstring: URL-encoded paramstring
|
|
:type paramstring: str
|
|
:return: parsed paramstring
|
|
:rtype: Params
|
|
"""
|
|
raw_params = parse_qs(paramstring)
|
|
params = Params()
|
|
for key, value in iteritems(raw_params):
|
|
param_value = value[0] if len(value) == 1 else value
|
|
params[key] = py2_decode(param_value)
|
|
return params
|
|
|
|
def get_url(self, plugin_url='', **kwargs):
|
|
"""
|
|
Construct a callable URL for a virtual directory item
|
|
|
|
If plugin_url is empty, a current plugin URL is used.
|
|
kwargs are converted to a URL-encoded string of plugin call parameters
|
|
To call a plugin action, 'action' parameter must be used,
|
|
if 'action' parameter is missing, then the plugin root action is called
|
|
If the action is not added to :class:`Plugin` actions,
|
|
:class:`PluginError` will be raised.
|
|
|
|
:param plugin_url: plugin URL with trailing / (optional)
|
|
:type plugin_url: str
|
|
:param kwargs: pairs of key=value items
|
|
:return: a full plugin callback URL
|
|
:rtype: str
|
|
"""
|
|
url = plugin_url or self._url
|
|
if kwargs:
|
|
return '{0}?{1}'.format(url, urlencode(kwargs, doseq=True))
|
|
return url
|
|
|
|
def action(self, name=None):
|
|
"""
|
|
Action decorator
|
|
|
|
Defines plugin callback action. If action's name is not defined
|
|
explicitly, then the action is named after the decorated function.
|
|
|
|
.. warning:: Action's name must be unique.
|
|
|
|
A plugin must have at least one action named ``'root'``
|
|
implicitly or explicitly.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
# The action is implicitly named 'root' after the decorated function
|
|
@plugin.action()
|
|
def root(params):
|
|
pass
|
|
|
|
@plugin.action('foo') # The action name is set explicitly
|
|
def foo_action(params):
|
|
pass
|
|
|
|
:param name: action's name (optional).
|
|
:type name: str
|
|
:raises SimplePluginError: if the action with such name is already defined.
|
|
"""
|
|
def wrap(func, name=name):
|
|
if name is None:
|
|
name = func.__name__
|
|
if name in self.actions:
|
|
raise SimplePluginError(
|
|
'Action "{0}" already defined!'.format(name)
|
|
)
|
|
self.actions[name] = func
|
|
return func
|
|
return wrap
|
|
|
|
def run(self):
|
|
"""
|
|
Run plugin
|
|
|
|
:raises SimplePluginError: if unknown action string is provided.
|
|
"""
|
|
self._handle = int(sys.argv[1])
|
|
self._params = self.get_params(sys.argv[2][1:])
|
|
self.log_debug(str(self))
|
|
result = self._resolve_function()
|
|
if result is not None:
|
|
raise SimplePluginError(
|
|
'A decorated function must not return any value! '
|
|
'It returned {0} instead.'.format(result)
|
|
)
|
|
|
|
def _resolve_function(self):
|
|
"""
|
|
Resolve action from plugin call params and call the respective callable
|
|
function
|
|
|
|
:return: action callable's return value
|
|
"""
|
|
self.log_debug('Actions: {0}'.format(str(list(self.actions.keys()))))
|
|
action = self._params.get('action', 'root')
|
|
self.log_debug('Called action {0} with params {1}'.format(
|
|
action, str(self._params))
|
|
)
|
|
try:
|
|
action_callable = self.actions[action]
|
|
except KeyError:
|
|
raise SimplePluginError('Invalid action: "{0}"!'.format(action))
|
|
else:
|
|
with log_exception(self.log_error):
|
|
# inspect.isfunction is needed for tests
|
|
if (inspect.isfunction(action_callable) and
|
|
not getargspec(action_callable).args):
|
|
return action_callable()
|
|
else:
|
|
return action_callable(self._params)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class RoutedPlugin(Plugin):
|
|
"""
|
|
Plugin class that implements "pretty URL" routing similar to Flask and Bottle
|
|
web-frameworks
|
|
|
|
:param id_: plugin's id, e.g. 'plugin.video.foo' (optional)
|
|
:type id_: str
|
|
"""
|
|
def __init__(self, id_=''):
|
|
"""
|
|
:param id_: plugin's id, e.g. 'plugin.video.foo' (optional)
|
|
:type id_: str
|
|
"""
|
|
super(RoutedPlugin, self).__init__(id_)
|
|
self._routes = {}
|
|
|
|
def __str__(self):
|
|
return '<RoutedPlugin {0}>'.format(sys.argv)
|
|
|
|
def url_for(self, func_, *args, **kwargs):
|
|
"""
|
|
Build a URL for a plugin route
|
|
|
|
This method performs reverse resolving a plugin callback URL for
|
|
the named route. If route's name is not set explicitly, then the name
|
|
of a decorated function is used as the name of the corresponding route.
|
|
The method can optionally take positional args and kwargs.
|
|
If any positional args are provided their values replace
|
|
variable placeholders by position.
|
|
|
|
.. warning:: The number of positional args must not exceed
|
|
the number of variable placeholders!
|
|
|
|
If any kwargs are provided their values replace variable placeholders
|
|
by name. If the number of kwargs provided exceeds the number of variable
|
|
placeholders, then the rest of the kwargs are added to the URL
|
|
as a query string.
|
|
|
|
.. note:: All :class:`unicode` arguments are encoded with UTF-8 encoding.
|
|
|
|
Let's assume that the ID of our plugin is ``plugin.acme``.
|
|
The following examples will show how to use this method to resolve
|
|
callback URLs for this plugin.
|
|
|
|
Example 1::
|
|
|
|
@plugin.route('/foo')
|
|
def foo():
|
|
pass
|
|
url = plugin.url_for('foo')
|
|
# url = 'plugin://plugin.acme/foo'
|
|
|
|
Example 2::
|
|
|
|
@plugin.route('/foo/<param>')
|
|
def foo(param):
|
|
pass
|
|
url = plugin.url_for('foo', param='bar')
|
|
# url = 'plugin://plugin.acme/foo/bar'
|
|
|
|
Example 3::
|
|
|
|
plugin.route('/foo/<param>')
|
|
def foo(param):
|
|
pass
|
|
url = plugin.url_for('foo', param='bar', ham='spam')
|
|
# url = 'plugin://plugin.acme/foo/bar?ham=spam
|
|
|
|
:param func_: route's name or a decorated function object.
|
|
:type func_: str or types.FunctionType
|
|
:param args: positional arguments.
|
|
:param kwargs: keyword arguments.
|
|
:return: full plugin call URL for the route.
|
|
:rtype: str
|
|
:raises simpleplugin.SimplePluginError: if a route with such name
|
|
does not exist or on arguments mismatch.
|
|
"""
|
|
if isinstance(func_, basestring):
|
|
name = func_
|
|
elif inspect.isfunction(func_) or inspect.ismethod(func_):
|
|
name = func_.__name__
|
|
else:
|
|
raise TypeError('The first argument to url_for must be '
|
|
'a route\'s name or a route function object!')
|
|
try:
|
|
pattern = self._routes[name].pattern
|
|
except KeyError:
|
|
raise SimplePluginError('Route "{0}" does not exist!'.format(name))
|
|
matches = re.findall(r'/(<\w+?>)', pattern)
|
|
if len(args) + len(kwargs) < len(matches) or len(args) > len(matches):
|
|
raise SimplePluginError(
|
|
'Arguments for the route "{0}" '
|
|
'do not match placeholders!'.format(name)
|
|
)
|
|
if matches:
|
|
for arg, match in zip(args, matches):
|
|
pattern = pattern.replace(
|
|
match,
|
|
quote_plus(py2_encode(str(arg)))
|
|
)
|
|
# list allows to manipulate the dict during iteration
|
|
for key, value in list(iteritems(kwargs)):
|
|
for match in matches[len(args):]:
|
|
|
|
match_string = match[1:-1]
|
|
match_parts = match_string.split('__')
|
|
if len(match_parts) > 1:
|
|
match_string = match_parts[1]
|
|
|
|
if key == match_string:
|
|
pattern = pattern.replace(
|
|
match, quote_plus(py2_encode(str(value)))
|
|
)
|
|
del kwargs[key]
|
|
url = 'plugin://{0}{1}'.format(self.id, pattern)
|
|
if kwargs:
|
|
url += '?' + urlencode(kwargs, doseq=True)
|
|
return url
|
|
|
|
get_url = url_for
|
|
|
|
def route(self, pattern, name=None):
|
|
"""
|
|
Route decorator for plugin callback routes
|
|
|
|
The route decorator is used to define plugin callback routes
|
|
similar to a URL routing mechanism in Flask and Bottle Python
|
|
web-frameworks. The plugin routing mechanism calls decorated functions
|
|
by matching a path in a plugin callback URL (passed as ``sys.argv[0]``)
|
|
to a route pattern. A route pattern *must* start with a forward slash
|
|
``/``. An end slash is optional. A plugin must have at least the root
|
|
route with ``'/'`` pattern. Bu default a route is named by the decorated
|
|
function, but route's name can be set explicitly by providing
|
|
the 2nd optional ``name`` argument.
|
|
|
|
.. warning:: Route names must be unique!
|
|
|
|
Example 1::
|
|
|
|
@plugin.route('/foo')
|
|
def foo_function():
|
|
pass
|
|
|
|
In the preceding example ``foo_function`` will be called when the plugin
|
|
is invoked with ``plugin://plugin.acme/foo/`` callback URL.
|
|
A route pattern can contain variable placeholders
|
|
(marked with angular brackets ``<>``) that are used to pass arguments
|
|
to a route function.
|
|
|
|
Example 2::
|
|
|
|
@plugin.route('/foo/<param>')
|
|
def foo_function(param):
|
|
pass
|
|
|
|
In the preceding example the part of a callback path marked with
|
|
``<param>`` placeholder will be passed to the function as an argument.
|
|
The name of a placeholder must be the same as the name of
|
|
the corresponding parameter. By default arguments are passed as strings.
|
|
The ``int`` and ``float`` prefixes can be used to pass arguments
|
|
as :class:`int` and :class:`float` numbers, for example ``<int:foo>``
|
|
or ``<float:bar>``.
|
|
|
|
Example 3::
|
|
|
|
@plugin.route('/add/<int:param1>/<int:param2>')
|
|
def addition(param1, param2):
|
|
sum = param1 + param2
|
|
|
|
A function can have multiple route decorators. In this case additional
|
|
routes must have explicitly defined names. If a route has less variable
|
|
placeholders than function parameters, "missing" function parameters
|
|
must have default values.
|
|
|
|
Example 4::
|
|
|
|
@plugin.route('/foo/<param>', name='foo_route')
|
|
@plugin.route('/bar')
|
|
def some_function(param='spam'):
|
|
# Do something
|
|
|
|
In the preceding example ``some_function`` can be called through
|
|
2 possible routes. If the function is called through the 1st route
|
|
(``'foo_route'``) ``<param>`` value will be passed as an argument.
|
|
The 2nd route will call the function with the default argument
|
|
``'spam'`` because this route has no variable placeholders to pass
|
|
arguments to the function. The order of the ``route`` decorators
|
|
does not matter but each route must have a unique name.
|
|
|
|
.. note:: A route pattern must start with a forward slash ``/``
|
|
and must not have a slash at the end.
|
|
|
|
:param pattern: route matching pattern
|
|
:type pattern: str
|
|
:param name: route's name (optional). If no name is provided,
|
|
the route is named after the decorated function.
|
|
The name must be unique.
|
|
:type name: str
|
|
"""
|
|
def wrapper(func, pattern=pattern, name=name):
|
|
if name is None:
|
|
name = func.__name__
|
|
if name in self._routes:
|
|
raise SimplePluginError(
|
|
'The route "{0}" already exists!'.format(name)
|
|
)
|
|
pattern = pattern.replace('int:', 'int__'
|
|
).replace('float:', 'float__')
|
|
self._routes[name] = Route(pattern, func)
|
|
return func
|
|
return wrapper
|
|
|
|
def _resolve_function(self):
|
|
"""
|
|
Resolve route from plugin callback path and call the respective
|
|
route function
|
|
|
|
:return: route function's return value
|
|
"""
|
|
path = urlparse(sys.argv[0]).path
|
|
self.log_debug('Routes: {0}'.format(self._routes))
|
|
for route in itervalues(self._routes):
|
|
if route.pattern == path:
|
|
kwargs = {}
|
|
self.log_debug(
|
|
'Calling {0} with kwargs {1}'.format(route, kwargs))
|
|
with log_exception(self.log_error):
|
|
return route.func(**kwargs)
|
|
|
|
for route in itervalues(self._routes):
|
|
pattern = route.pattern
|
|
if not pattern.count('/') == path.count('/'):
|
|
continue
|
|
while True:
|
|
pattern, count = re.subn(r'/(<\w+?>)', r'/(?P\1.+?)', pattern)
|
|
if not count:
|
|
break
|
|
match = re.search(r'^' + pattern + r'$', path)
|
|
if match is not None:
|
|
kwargs = match.groupdict()
|
|
# list allows to manipulate the dict during iteration
|
|
for key, value in list(iteritems(kwargs)):
|
|
if key.startswith('int__') or key.startswith('float__'):
|
|
del kwargs[key]
|
|
if key.startswith('int__'):
|
|
key = key[5:]
|
|
value = int(value)
|
|
else:
|
|
key = key[7:]
|
|
value = float(value)
|
|
kwargs[key] = value
|
|
else:
|
|
kwargs[key] = py2_decode(unquote_plus(value))
|
|
self.log_debug(
|
|
'Calling {0} with kwargs {1}'.format(route, kwargs))
|
|
with log_exception(self.log_error):
|
|
return route.func(**kwargs)
|
|
raise SimplePluginError(
|
|
'No route matches the path "{0}"!'.format(path)
|
|
)
|
|
|
|
def action(self, name=None):
|
|
raise NotImplementedError(
|
|
'RoutedPlugin does not support action decorator. '
|
|
'Use route decorator instead.'
|
|
)
|