def prompt_for_artwork(mediatype, medialabel, availableart, monitor):
    if not availableart:
        return None, None

    addon = pykodi.get_main_addon()
    arttypes = []
    for arttype, artlist in availableart.iteritems():
        if arttype.startswith('season.-1.'):
            # Ignore 'all' seasons artwork, as I can't set artwork for it with JSON
            continue
        label = arttype if not arttype.startswith('season.') else get_seasonlabel(arttype)
        for image in artlist:
            if image.get('existing'):
                arttypes.append({'arttype': arttype, 'label': label, 'count': len(artlist), 'url': image['url']})
                break
        if arttype not in (at['arttype'] for at in arttypes):
            arttypes.append({'arttype': arttype, 'label': label, 'count': len(artlist)})
    arttypes.sort(key=lambda art: sort_arttype(art['arttype']))
    typeselectwindow = ArtworkTypeSelector('DialogSelect.xml', addon.path, arttypes=arttypes, medialabel=medialabel)
    hqpreview = addon.get_setting('highquality_preview')
    singletype = arttypes[0]['arttype'] if len(arttypes) == 1 else None
    selectedarttype = None
    selectedart = None
    while selectedart == None and not monitor.abortRequested():
        # The loop shows the first window if viewer backs out of the second
        if singletype:
            selectedarttype = singletype
        else:
            selectedarttype = typeselectwindow.prompt()
        if not selectedarttype:
            break
        artlist = availableart[selectedarttype]
        if selectedarttype.startswith('season.'):
            stype = selectedarttype.rsplit('.', 1)[1]
            stype = mediatypes.artinfo[mediatypes.SEASON].get(stype)
        else:
            stype = mediatypes.artinfo[mediatype].get(selectedarttype)
        multi = stype['multiselect'] if stype else False
        artselectwindow = ArtworkSelector('DialogSelect.xml', addon.path, artlist=artlist, arttype=selectedarttype,
            medialabel=medialabel, multi=multi, hqpreview=hqpreview)
        selectedart = artselectwindow.prompt()
        if singletype and selectedart is None:
            selectedarttype = None
            break
    return selectedarttype, selectedart
Exemplo n.º 2
0
def remove_otherartwork(mediaitem):
    ''' Remove artwork not enabled in add-on settings. '''
    keep_types = pykodi.get_main_addon().get_setting('save_additional_arttypes')
    keep_types = [addon.strip() for addon in keep_types.split(',')]
    keep_types = dict(arttype.split(':', 2) if ':' in arttype else (arttype, sys.maxsize) for arttype in keep_types)
    finalart = {}

    for basetype in iter_base_arttypes(mediaitem['art']):
        if basetype in keep_types:
            try:
                max_allowed = int(keep_types[basetype])
            except ValueError:
                max_allowed = sys.maxsize
        else:
            max_allowed = mediatypes.artinfo[mediaitem['mediatype']].get(basetype, {}).get('autolimit', 0)
        finalart.update(_arrange_multiart(mediaitem['art'], basetype, limit=max_allowed))

    finalart.update((arttype, None) for arttype in mediaitem['art'] if arttype not in finalart)
    return finalart
Exemplo n.º 3
0
import re
import xbmc
from collections import namedtuple
from os.path import split, basename, dirname

from devhelper import pykodi
from devhelper.pykodi import log

addon = pykodi.get_main_addon()

SortedDisplay = namedtuple('SortedDisplay', ['sort', 'display'])

def localize(messageid):
    if isinstance(messageid, basestring):
        return messageid
    if messageid >= 32000 and messageid < 33000:
        return addon.getLocalizedString(messageid)
    return xbmc.getLocalizedString(messageid)

def natural_sort(string, split_regex=re.compile(r'([0-9]+)')):
    return [int(text) if text.isdigit() else text.lower() for text in re.split(split_regex, string)]

# TODO: Load from advancedsettings.xml
moviestacking = [re.compile(r'(.*?)([ _.-]*(?:cd|dvd|p(?:ar)?t|dis[ck])[ _.-]*[0-9]+)(.*?)(\.[^.]+)$', re.IGNORECASE),
    re.compile(r'(.*?)([ _.-]*(?:cd|dvd|p(?:ar)?t|dis[ck])[ _.-]*[a-d])(.*?)(\.[^.]+)$', re.IGNORECASE),
    re.compile(r'(.*?)([ ._-]*[a-d])(.*?)(\.[^.]+)$', re.IGNORECASE)
]
def get_movie_path_list(stackedpath):
    """Returns a list of filenames that can be used to find a movie's supporting files.
    The list includes both possible filenames for stacks, and the VIDEO_TS/BDMV parent directory.
    If neither applies, returns a list of one item, the original path.