def update_art_for_items(items, start):
changedcount = 0
for i, item in enumerate(items):
if fg:
progress.update(start + i * stepsize // len(items),
item['label'])
else:
progress.update(start + i * stepsize // len(items))
item = info.MediaItem(item)
if item.mediatype == mediatypes.SEASON:
item.file = info.get_cached_tvshow(item.tvshowid)['file']
updates = function(item)
if isinstance(updates, int):
changedcount += updates
else:
processed = utils.get_simpledict_updates(item.art, updates)
if processed:
info.update_art_in_library(item.mediatype, item.dbid,
processed)
changedcount += len(processed)
if monitor.abortRequested() or fg and progress.iscanceled():
break
return changedcount
def _process_item(self, mediaitem, singleitem=False, auto=True):
log("Processing {0} '{1}' automatically.".format(
mediaitem.mediatype, mediaitem.label))
mediatype = mediaitem.mediatype
onlyfs = self.localmode or mediatypes.only_filesystem(
mediaitem.mediatype)
if not mediaitem.uniqueids and not onlyfs:
mediaitem.missingid = True
if singleitem:
header = L(NO_IDS_MESSAGE)
message = "{0} '{1}'".format(mediatype, mediaitem.label)
log(header + ": " + message, xbmc.LOGNOTICE)
xbmcgui.Dialog().notification("Artwork Beef: " + header,
message,
xbmcgui.NOTIFICATION_INFO)
if auto:
cleaned = get_simpledict_updates(mediaitem.art,
cleaner.clean_artwork(mediaitem))
if cleaned:
if not self.debug:
add_art_to_library(mediatype, mediaitem.seasons,
mediaitem.dbid, cleaned)
mediaitem.art.update(cleaned)
mediaitem.art = dict(item
for item in mediaitem.art.iteritems()
if item[1])
mediaitem.missingart = list(
info.iter_missing_arttypes(mediaitem, mediaitem.art))
services_hit, error = self.gatherer.getartwork(mediaitem, onlyfs, auto)
if auto:
existingart = dict(mediaitem.art)
selectedart = dict(
(key, image['url'])
for key, image in mediaitem.forcedart.iteritems())
existingart.update(selectedart)
# Then add the rest of the missing art
selectedart.update(
self.get_top_missing_art(
info.iter_missing_arttypes(mediaitem, existingart),
mediatype, existingart, mediaitem.availableart))
selectedart = get_simpledict_updates(mediaitem.art, selectedart)
mediaitem.selectedart = selectedart
toset = dict(selectedart)
if not self.localmode and mediatypes.downloadanyartwork(
mediaitem.mediatype):
sh, er = self.downloader.downloadfor(mediaitem)
services_hit = services_hit or sh
error = error or er
toset.update(mediaitem.downloadedart)
if toset:
mediaitem.updatedart = list(
set(mediaitem.updatedart + toset.keys()))
if not self.debug:
add_art_to_library(mediatype, mediaitem.seasons,
mediaitem.dbid, toset)
self.cachelocal(mediaitem, toset)
if error:
if isinstance(error, dict):
header = L(PROVIDER_ERROR_MESSAGE).format(
error['providername'])
error = '{0}: {1}'.format(header, error['message'])
mediaitem.error = error
log(error, xbmc.LOGWARNING)
self.notify_warning(error)
elif auto and not self.debug and not self.localmode:
if not (mediatype == mediatypes.EPISODE and 'fanart' in mediaitem.skip_artwork) and \
mediatype != mediatypes.SONG:
self.processed.set_nextdate(
mediaitem.dbid, mediatype, mediaitem.label,
datetime_now() +
timedelta(days=self.get_nextcheckdelay(mediaitem)))
if mediatype == mediatypes.TVSHOW:
self.processed.set_data(mediaitem.dbid, mediatype,
mediaitem.label, mediaitem.season)
if mediaitem.borked_filename:
msg = L(FILENAME_ENCODING_ERROR).format(mediaitem.file)
if not mediaitem.error:
mediaitem.error = msg
log(msg, xbmc.LOGWARNING)
if self.debug:
log(mediaitem, xbmc.LOGNOTICE)
return services_hit
def _manual_item_process(self, mediaitem, busy):
self._process_item(mediaitem, True, False)
busy.close()
if mediaitem.availableart or mediaitem.forcedart:
availableart = dict(mediaitem.availableart)
if mediaitem.mediatype == mediatypes.TVSHOW and 'fanart' in availableart:
# add unseasoned backdrops as manual-only options for each season fanart
unseasoned_backdrops = [
dict(art) for art in availableart['fanart']
if not art.get('hasseason')
]
if unseasoned_backdrops:
for season in mediaitem.seasons.keys():
key = 'season.{0}.fanart'.format(season)
if key in availableart:
availableart[key].extend(unseasoned_backdrops)
else:
availableart[key] = list(unseasoned_backdrops)
if mediaitem.mediatype in (mediatypes.MOVIE, mediatypes.MOVIESET
) and 'poster' in availableart:
# add no-language posters from TMDB as manual-only options for 'keyart'
nolang_posters = [
dict(art) for art in availableart['poster']
if not art['language']
]
for art in nolang_posters:
if art['provider'].sort == 'themoviedb.org':
if 'keyart' not in availableart:
availableart['keyart'] = []
availableart['keyart'].append(art)
tag_forcedandexisting_art(availableart, mediaitem.forcedart,
mediaitem.art)
selectedarttype, selectedart = prompt_for_artwork(
mediaitem.mediatype, mediaitem.label, availableart,
self.monitor)
if selectedarttype and selectedarttype not in availableart:
self.manual_id(mediaitem)
return
if selectedarttype and selectedart:
if mediatypes.get_artinfo(mediaitem.mediatype,
selectedarttype)['multiselect']:
selectedart = info.fill_multiart(mediaitem.art,
selectedarttype,
selectedart)
else:
selectedart = {selectedarttype: selectedart}
selectedart = get_simpledict_updates(mediaitem.art,
selectedart)
mediaitem.selectedart = selectedart
toset = dict(selectedart)
if settings.remove_deselected_files:
self.downloader.remove_deselected_files(mediaitem)
if mediatypes.downloadanyartwork(mediaitem.mediatype):
try:
self.downloader.downloadfor(mediaitem, False)
except FileError as ex:
mediaitem.error = ex.message
log(ex.message, xbmc.LOGERROR)
xbmcgui.Dialog().notification(
"Artwork Beef", ex.message,
xbmcgui.NOTIFICATION_ERROR)
toset.update(mediaitem.downloadedart)
if toset:
mediaitem.updatedart = toset.keys()
add_art_to_library(mediaitem.mediatype, mediaitem.seasons,
mediaitem.dbid, toset)
self.cachelocal(mediaitem, toset)
reporting.report_item(mediaitem, True, True,
self.downloader.size)
if not mediaitem.error:
notifycount(len(toset))
else:
xbmcgui.Dialog().notification(
L(NOT_AVAILABLE_MESSAGE),
L(SOMETHING_MISSING) + ' ' + L(FINAL_MESSAGE), '-', 8000)
self.finish_run()