mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-11-10 04:06:11 +00:00
136 lines
4 KiB
Python
136 lines
4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import division, absolute_import, print_function, unicode_literals
|
|
|
|
#################################################################################################
|
|
|
|
from kodi_six import xbmcgui
|
|
|
|
from helper import LazyLogger
|
|
|
|
from .utils import should_stop
|
|
from .exceptions import LibraryException
|
|
from .translate import translate
|
|
|
|
#################################################################################################
|
|
|
|
LOG = LazyLogger(__name__)
|
|
|
|
#################################################################################################
|
|
|
|
|
|
def progress(message=None):
|
|
|
|
''' Will start and close the progress dialog.
|
|
'''
|
|
def decorator(func):
|
|
def wrapper(self, item=None, *args, **kwargs):
|
|
|
|
dialog = xbmcgui.DialogProgressBG()
|
|
|
|
if item and type(item) == dict:
|
|
|
|
dialog.create(translate('addon_name'), "%s %s" % (translate('gathering'), item['Name']))
|
|
LOG.info("Processing %s: %s", item['Name'], item['Id'])
|
|
else:
|
|
dialog.create(translate('addon_name'), message)
|
|
LOG.info("Processing %s", message)
|
|
|
|
if item:
|
|
args = (item,) + args
|
|
|
|
result = func(self, dialog=dialog, *args, **kwargs)
|
|
dialog.close()
|
|
|
|
return result
|
|
|
|
return wrapper
|
|
return decorator
|
|
|
|
|
|
def stop(func):
|
|
|
|
''' Wrapper to catch exceptions and return using catch
|
|
'''
|
|
def wrapper(*args, **kwargs):
|
|
|
|
try:
|
|
if should_stop(): # ??? TODO: Fixme
|
|
raise Exception
|
|
|
|
except Exception as error:
|
|
LOG.exception(error)
|
|
|
|
raise LibraryException("StopCalled")
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def jellyfin_item(func):
|
|
|
|
''' Wrapper to retrieve the jellyfin_db item.
|
|
'''
|
|
def wrapper(self, item, *args, **kwargs):
|
|
e_item = self.jellyfin_db.get_item_by_id(item['Id'] if type(item) == dict else item)
|
|
|
|
return func(self, item, e_item=e_item, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def library_check(func):
|
|
|
|
''' Wrapper to retrieve the library
|
|
'''
|
|
def wrapper(self, item, *args, **kwargs):
|
|
|
|
''' TODO: Rethink this one... songs and albums cannot be found by library. expensive.
|
|
'''
|
|
from database import get_sync
|
|
|
|
if kwargs.get('library') is None:
|
|
sync = get_sync()
|
|
|
|
if 'e_item' in kwargs:
|
|
try:
|
|
view_id = kwargs['e_item'][6]
|
|
view_name = self.jellyfin_db.get_view_name(view_id)
|
|
view = {'Name': view_name, 'Id': view_id}
|
|
except Exception:
|
|
view = None
|
|
|
|
if view is None:
|
|
ancestors = self.server.jellyfin.get_ancestors(item['Id'])
|
|
|
|
if not ancestors:
|
|
if item['Type'] == 'MusicArtist':
|
|
|
|
try:
|
|
views = self.jellyfin_db.get_views_by_media('music')[0]
|
|
except Exception as error:
|
|
LOG.exception(error)
|
|
return
|
|
|
|
view = {'Id': views[0], 'Name': views[1]}
|
|
else: # Grab the first music library
|
|
return
|
|
else:
|
|
for ancestor in ancestors:
|
|
if ancestor['Type'] == 'CollectionFolder':
|
|
|
|
view = self.jellyfin_db.get_view_name(ancestor['Id'])
|
|
view = {'Id': None, 'Name': None} if view is None else {'Name': ancestor['Name'], 'Id': ancestor['Id']}
|
|
|
|
break
|
|
|
|
if view['Id'] not in [x.replace('Mixed:', "") for x in sync['Whitelist'] + sync['Libraries']]:
|
|
LOG.info("Library %s is not synced. Skip update.", view['Id'])
|
|
|
|
return
|
|
|
|
kwargs['library'] = view
|
|
|
|
return func(self, item, *args, **kwargs)
|
|
|
|
return wrapper
|