def clean_artwork(mediaitem): updated_art = dict(_get_clean_art(*art) for art in mediaitem['art'].iteritems()) for basetype in iter_base_arttypes(updated_art): remove_duplicate_fanart = basetype == 'fanart' updated_art.update(_arrange_multiart(updated_art, basetype, remove_duplicate_fanart)) return updated_art
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