def save_banner(self, show_obj):
"""
Downloads a banner image and saves it to the filename specified by banner_name
inside the show's root folder.
show_obj: a TVShow object for which to download a banner
"""
banner_path = self.get_banner_path(show_obj)
if not banner_path:
logger.log(
"Banner path for show {} came back blank, skipping this image".
format(show_obj.name), logger.DEBUG)
return False
banner_url = sickchill.indexer.series_banner_url(show_obj)
if not banner_url:
banner_url = self._retrieve_show_image_urls_from_fanart(
show_obj, 'banner')
if not banner_url:
logger.log(
"Banner url not found for show {}, skipping this image".
format(show_obj.name), logger.DEBUG)
return False
banner_data = metadata_helpers.getShowImage(banner_url)
if not banner_data:
logger.log(
"No show banner image was retrieved, unable to write banner",
logger.DEBUG)
return False
return self._write_image(banner_data, banner_path)
def save_thumbnail(self, ep_obj):
"""
Retrieves a thumbnail and saves it to the correct spot. This method should not need to
be overridden by implementing classes, changing get_episode_thumb_path and
_get_episode_thumb_url should suffice.
ep_obj: a TVEpisode object for which to generate a thumbnail
"""
file_path = self.get_episode_thumb_path(ep_obj)
if not file_path:
logger.log(u"Unable to find a file path to use for this thumbnail, not generating it", logger.DEBUG)
return False
thumb_url = self._get_episode_thumb_url(ep_obj)
# if we can't find one then give up
if not thumb_url:
logger.log("No thumb is available for this episode, not creating a thumb", logger.DEBUG)
return False
thumb_data = metadata_helpers.getShowImage(thumb_url)
result = self._write_image(thumb_data, file_path)
if not result:
return False
for cur_ep in [ep_obj] + ep_obj.relatedEps:
cur_ep.hastbn = True
return True
def _retrieve_show_image(self, image_type, show_obj, which=None):
"""
Gets an image URL from theTVDB.com and TMDB.com, downloads it and returns the data.
image_type: type of image to retrieve (currently supported: fanart, poster, banner)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
Returns: the binary image data if available, or else None
"""
image_url = None
indexer_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = sickbeard.indexerApi(show_obj.indexer).api_params.copy()
lINDEXER_API_PARMS['banners'] = True
if indexer_lang and not indexer_lang == sickbeard.INDEXER_DEFAULT_LANGUAGE:
lINDEXER_API_PARMS['language'] = indexer_lang
if show_obj.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = sickbeard.indexerApi(show_obj.indexer).indexer(**lINDEXER_API_PARMS)
indexer_show_obj = t[show_obj.indexerid]
except (sickbeard.indexer_error, IOError) as e:
logger.log(u"Unable to look up show on " + sickbeard.indexerApi(
show_obj.indexer).name + ", not downloading images: " + ex(e), logger.WARNING)
logger.log(u"%s may be experiencing some problems. Try again later." % sickbeard.indexerApi(show_obj.indexer).name, logger.DEBUG)
return None
if image_type not in ('fanart', 'poster', 'banner', 'poster_thumb', 'banner_thumb'):
logger.log(u"Invalid image type " + str(image_type) + ", couldn't find it in the " + sickbeard.indexerApi(
show_obj.indexer).name + " object", logger.ERROR)
return None
if image_type == 'poster_thumb':
if getattr(indexer_show_obj, 'poster', None):
image_url = re.sub('posters', '_cache/posters', indexer_show_obj['poster'])
if not image_url:
# Try and get images from TMDB
image_url = self._retrieve_show_images_from_tmdb(show_obj, image_type)
elif image_type == 'banner_thumb':
if getattr(indexer_show_obj, 'banner', None):
image_url = re.sub('graphical', '_cache/graphical', indexer_show_obj['banner'])
else:
if getattr(indexer_show_obj, image_type, None):
image_url = indexer_show_obj[image_type]
if not image_url:
# Try and get images from TMDB
image_url = self._retrieve_show_images_from_tmdb(show_obj, image_type)
if image_url:
image_data = metadata_helpers.getShowImage(image_url, which)
return image_data
return None
def save_season_poster(self, show_obj, season):
"""
Saves a specific season poster to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
"""
season_poster_url = sickchill.indexer.season_poster_url(
show_obj, season)
if not season_poster_url:
logger.log(
"Season poster url not found for season {}, skipping this season"
.format(season), logger.DEBUG)
return False
season_poster_file_path = self.get_season_poster_path(show_obj, season)
if not season_poster_file_path:
logger.log(
"Path for season {} came back blank, skipping this season".
format(season), logger.DEBUG)
return False
image_data = metadata_helpers.getShowImage(season_poster_url)
if not image_data:
logger.log("No season poster data available, skipping this season",
logger.DEBUG)
return False
return self._write_image(image_data, season_poster_file_path)
def save_poster(self, show_obj):
"""
Downloads a poster image and saves it to the filename specified by poster_name
inside the show's root folder.
show_obj: a TVShow object for which to download a poster
"""
# use the default poster name
poster_path = self.get_poster_path(show_obj)
if not poster_path:
logger.log(
"Banner path for show {} came back blank, skipping this image".
format(show_obj.name), logger.DEBUG)
return False
poster_url = sickchill.indexer.series_poster_url(show_obj)
if not poster_url:
logger.log(
"Poster url not found for show {}, skipping this image".format(
show_obj.name), logger.DEBUG)
return False
poster_data = metadata_helpers.getShowImage(poster_url)
if not poster_data:
logger.log(
"No show poster image was retrieved, unable to write poster",
logger.DEBUG)
return False
return self._write_image(poster_data, poster_path)
def save_season_all_banner(self, show_obj):
banner_url = sickchill.indexer.series_banner_url(show_obj)
if not banner_url:
banner_url = self._retrieve_show_image_urls_from_fanart(
show_obj, 'season_banner', season=0)
if not banner_url:
logger.log(
"Url for season all banner came back blank, skipping this season",
logger.DEBUG)
return False
season_banner_file_path = self.get_season_all_banner_path(show_obj)
if not season_banner_file_path:
logger.log(
"Path for season all banner came back blank, skipping this season",
logger.DEBUG)
return False
image_data = metadata_helpers.getShowImage(banner_url)
if not image_data:
logger.log(
"No season all banner data available, skipping this season",
logger.DEBUG)
return False
return self._write_image(image_data, season_banner_file_path)
def save_thumbnail(self, ep_obj):
"""
Retrieves a thumbnail and saves it to the correct spot. This method should not need to
be overridden by implementing classes, changing get_episode_thumb_path and
_get_episode_thumb_url should suffice.
ep_obj: a TVEpisode object for which to generate a thumbnail
"""
file_path = self.get_episode_thumb_path(ep_obj)
if not file_path:
logger.log(u"Unable to find a file path to use for this thumbnail, not generating it", logger.DEBUG)
return False
thumb_url = self._get_episode_thumb_url(ep_obj)
# if we can't find one then give up
if not thumb_url:
logger.log(u"No thumb is available for this episode, not creating a thumb", logger.DEBUG)
return False
thumb_data = metadata_helpers.getShowImage(thumb_url)
result = self._write_image(thumb_data, file_path)
if not result:
return False
for cur_ep in [ep_obj] + ep_obj.relatedEps:
cur_ep.hastbn = True
return True
def save_fanart(self, show_obj):
"""
Downloads a fanart image and saves it to the filename specified by fanart_name
inside the show's root folder.
show_obj: a TVShow object for which to download fanart
"""
# use the default fanart name
fanart_path = self.get_fanart_path(show_obj)
if not fanart_path:
logger.log(
"Fanart path for show {} came back blank, skipping this image".
format(show_obj.name), logger.DEBUG)
return False
fanart_url = sickchill.indexer.series_fanart_url(show_obj)
if not fanart_url:
fanart_url = self._retrieve_show_image_urls_from_fanart(
show_obj, 'fanart')
if not fanart_url:
logger.log(
"Fanart url not found for show {}, skipping this image".
format(show_obj.name), logger.DEBUG)
return False
fanart_data = metadata_helpers.getShowImage(fanart_url)
if not fanart_data:
logger.log("No fanart image was retrieved, unable to write fanart",
logger.DEBUG)
return False
return self._write_image(fanart_data, fanart_path)
def save_season_banner(self, show_obj, season):
"""
Saves the first season banner for a season to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
"""
season_banner_url = sickchill.indexer.season_banner_url(
show_obj, season)
if not season_banner_url:
season_banner_url = self._retrieve_show_image_urls_from_fanart(
show_obj, 'season_banner', season=season)
if not season_banner_url:
logger.log(
"Url for season banner {} came back blank, skipping this season"
.format(season), logger.DEBUG)
return False
season_banner_file_path = self.get_season_banner_path(show_obj, season)
if not season_banner_file_path:
logger.log(
"Path for season {} came back blank, skipping this season".
format(season), logger.DEBUG)
return False
image_data = metadata_helpers.getShowImage(season_banner_url)
if not image_data:
logger.log("No season banner data available, skipping this season",
logger.DEBUG)
return False
return self._write_image(image_data, season_banner_file_path)
def save_season_banners(self, show_obj, season):
"""
Saves all season banners to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
Cycles through all seasons and saves the season banners if possible. This
method should not need to be overridden by implementing classes, changing
_season_banners_dict and get_season_banner_path should be good enough.
"""
season_dict = self._season_banners_dict(show_obj, season)
result = []
# Returns a nested dictionary of season art with the season
# number as primary key. It's really overkill but gives the option
# to present to user via ui to pick down the road.
for cur_season in season_dict:
cur_season_art = season_dict[cur_season]
if len(cur_season_art) == 0:
continue
# Just grab whatever's there for now
art_id, season_url = cur_season_art.popitem() # @UnusedVariable
season_banner_file_path = self.get_season_banner_path(
show_obj, cur_season)
if not season_banner_file_path:
logger.log(
u"Path for season " + str(cur_season) +
" came back blank, skipping this season", logger.DEBUG)
continue
seasonData = metadata_helpers.getShowImage(season_url)
if not seasonData:
logger.log(
u"No season banner data available, skipping this season",
logger.DEBUG)
continue
result = result + [
self._write_image(seasonData, season_banner_file_path,
show_obj)
]
if result:
return all(result)
else:
return False
return True
def save_season_banners(self, show_obj, season):
"""
Saves all season banners to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
Cycles through all seasons and saves the season banners if possible. This
method should not need to be overridden by implementing classes, changing
_season_banners_dict and get_season_banner_path should be good enough.
"""
season_dict = self._season_banners_dict(show_obj, season)
result = []
# Returns a nested dictionary of season art with the season
# number as primary key. It's really overkill but gives the option
# to present to user via ui to pick down the road.
for cur_season in season_dict:
cur_season_art = season_dict[cur_season]
if len(cur_season_art) == 0:
continue
# Just grab whatever's there for now
art_id, season_url = cur_season_art.popitem() # @UnusedVariable
season_banner_file_path = self.get_season_banner_path(
show_obj, cur_season)
if not season_banner_file_path:
logger.log(
u"Path for season " + str(cur_season) +
" came back blank, skipping this season", logger.DEBUG)
continue
seasonData = metadata_helpers.getShowImage(season_url)
if not seasonData:
logger.log(
u"No season banner data available, skipping this season",
logger.DEBUG)
continue
result = result + [
self._write_image(seasonData, season_banner_file_path)
]
if result:
return all(result)
else:
return False
return True
def save_season_banners(self, show_obj, season):
"""
Saves all season banners to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
Cycles through all seasons and saves the season banners if possible.
"""
season_dict = self._season_image_dict(show_obj, season, 'seasonwides')
result = []
# Returns a nested dictionary of season art with the season
# number as primary key. It's really overkill but gives the option
# to present to user via ui to pick down the road.
for cur_season in season_dict:
cur_season_art = season_dict[cur_season]
if 0 == len(cur_season_art):
continue
# Just grab whatever's there for now
art_id, season_url = cur_season_art.popitem() # @UnusedVariable
season_banner_file_path = self.get_season_banner_path(
show_obj, cur_season)
if not season_banner_file_path:
logger.log(
u'Path for season ' + str(cur_season) +
' came back blank, skipping this season', logger.DEBUG)
continue
season_data = metadata_helpers.getShowImage(season_url,
showName=show_obj.name)
if not season_data:
logger.log(
u'No season banner data available, skipping this season',
logger.DEBUG)
continue
result = result + [
self._write_image(season_data, season_banner_file_path)
]
if result:
return all(result)
return False
def save_season_all_poster(self, show_obj):
poster_url = sickchill.indexer.series_poster_url(show_obj)
if not poster_url:
logger.log("Url for season all poster came back blank, skipping this season", logger.DEBUG)
return False
season_poster_file_path = self.get_season_all_poster_path(show_obj)
if not season_poster_file_path:
logger.log("Path for season all poster came back blank, skipping this season", logger.DEBUG)
return False
image_data = metadata_helpers.getShowImage(poster_url)
if not image_data:
logger.log("No season all poster data available, skipping this season", logger.DEBUG)
return False
return self._write_image(image_data, season_poster_file_path)
def _retrieve_show_image(self, image_type, show_obj, which=None, all_images=False):
"""
Gets an image URL from theTVDB.com or fanart.tv (depending on type),
downloads it, and returns the data.
image_type: type of image to retrieve (currently supported: poster, fanart, banner, landscape, logo, clearart, character)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
all_images: optional, if all_images is true, it will return a list of dictionary objects containing the keys 'id' and 'data'
representing all_images images of that type that could be found (only for fanart.tv images).
Returns: the binary image data if available, or else None
"""
TVDB_TYPES = ('fanart', 'poster', 'banner')
FANARTTV_TYPES = ('landscape', 'logo', 'clearart', 'character', 'extra_fanart')
if image_type in TVDB_TYPES:
api = 'tvdb'
elif image_type in FANARTTV_TYPES:
api = 'fanart.tv'
else:
logger.log(u"Invalid image type "+str(image_type), logger.ERROR)
return None
image_data = None
if api is 'tvdb':
tvdb_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
ltvdb_api_parms = sickbeard.TVDB_API_PARMS.copy()
if tvdb_lang and not tvdb_lang == 'en':
ltvdb_api_parms['language'] = tvdb_lang
t = tvdb_api.Tvdb(banners=True, **ltvdb_api_parms)
tvdb_show_obj = t[show_obj.tvdbid]
except (tvdb_exceptions.tvdb_error, IOError), e:
logger.log(u"Unable to look up show on TVDB, not downloading images: "+ex(e), logger.ERROR)
return None
image_url = tvdb_show_obj[image_type]
image_data = metadata_helpers.getShowImage(image_url, which)
def save_season_banners(self, show_obj, season):
"""
Saves all season banners to disk for the given show.
show_obj: a TVShow object for which to save the season thumbs
Cycles through all seasons and saves the season banners if possible.
"""
season_dict = self._season_image_dict(show_obj, season, 'seasonwides')
result = []
# Returns a nested dictionary of season art with the season
# number as primary key. It's really overkill but gives the option
# to present to user via ui to pick down the road.
for cur_season in season_dict:
cur_season_art = season_dict[cur_season]
if 0 == len(cur_season_art):
continue
# Just grab whatever's there for now
art_id, season_url = cur_season_art.popitem() # @UnusedVariable
season_banner_file_path = self.get_season_banner_path(show_obj, cur_season)
if not season_banner_file_path:
logger.log(u'Path for season ' + str(cur_season) + ' came back blank, skipping this season',
logger.DEBUG)
continue
season_data = metadata_helpers.getShowImage(season_url, showName=show_obj.name)
if not season_data:
logger.log(u'No season banner data available, skipping this season', logger.DEBUG)
continue
result = result + [self._write_image(season_data, season_banner_file_path)]
if result:
return all(result)
return False
if image_type not in ('fanart', 'poster', 'banner', 'poster_thumb', 'banner_thumb'):
logger.log(u"Invalid image type "+str(image_type)+", couldn't find it in the TVDB object", logger.ERROR)
return None
try:
if image_type == 'poster_thumb':
image_url = re.sub('posters', '_cache/posters', tvdb_show_obj['poster'])
elif image_type == 'banner_thumb':
image_url = re.sub('graphical', '_cache/graphical', tvdb_show_obj['banner'])
else:
image_url = tvdb_show_obj[image_type]
except:
return None
image_data = metadata_helpers.getShowImage(image_url, which)
return image_data
def _season_thumb_dict(self, show_obj):
"""
Should return a dict like:
result = {<season number>:
{1: '<url 1>', 2: <url 2>, ...},}
"""
# This holds our resulting dictionary of season art
result = {}
tvdb_lang = show_obj.lang
", couldn't find it in the TVDB object", logger.ERROR)
return None
try:
if image_type == 'poster_thumb':
image_url = re.sub('posters', '_cache/posters',
tvdb_show_obj['poster'])
elif image_type == 'banner_thumb':
image_url = re.sub('graphical', '_cache/graphical',
tvdb_show_obj['banner'])
else:
image_url = tvdb_show_obj[image_type]
except:
return None
image_data = metadata_helpers.getShowImage(image_url, which)
return image_data
def _season_thumb_dict(self, show_obj):
"""
Should return a dict like:
result = {<season number>:
{1: '<url 1>', 2: <url 2>, ...},}
"""
# This holds our resulting dictionary of season art
result = {}
tvdb_lang = show_obj.lang
def _retrieve_show_image(self, image_type, show_obj, which=None):
"""
Gets an image URL from theTVDB.com and TMDB.com, downloads it and returns the data.
image_type: type of image to retrieve (currently supported: fanart, poster, banner)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
Returns: the binary image data if available, or else None
"""
image_url = None
indexer_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = sickbeard.indexerApi(
show_obj.indexer).api_params.copy()
lINDEXER_API_PARMS['banners'] = True
lINDEXER_API_PARMS[
'language'] = indexer_lang or sickbeard.INDEXER_DEFAULT_LANGUAGE
if show_obj.dvdorder:
lINDEXER_API_PARMS['dvdorder'] = True
t = sickbeard.indexerApi(
show_obj.indexer).indexer(**lINDEXER_API_PARMS)
indexer_show_obj = t[show_obj.indexerid]
except (sickbeard.indexer_error, IOError) as e:
logger.log(
"Unable to look up show on " +
sickbeard.indexerApi(show_obj.indexer).name +
", not downloading images: " + ex(e), logger.WARNING)
logger.log(
"{0} may be experiencing some problems. Try again later.".
format(sickbeard.indexerApi(show_obj.indexer).name),
logger.DEBUG)
return None
if image_type not in ('fanart', 'poster', 'banner', 'poster_thumb',
'banner_thumb'):
logger.log(
"Invalid image type " + str(image_type) +
", couldn't find it in the " +
sickbeard.indexerApi(show_obj.indexer).name + " object",
logger.ERROR)
return None
if image_type == 'poster_thumb':
if getattr(indexer_show_obj, 'poster', None):
image_url = re.sub('posters', '_cache/posters',
indexer_show_obj['poster'])
if not image_url:
# Try and get images from Fanart.TV
image_url = self._retrieve_show_images_from_fanart(
show_obj, image_type)
if not image_url:
# Try and get images from TMDB
image_url = self._retrieve_show_images_from_tmdb(
show_obj, image_type)
elif image_type == 'banner_thumb':
if getattr(indexer_show_obj, 'banner', None):
image_url = re.sub('graphical', '_cache/graphical',
indexer_show_obj['banner'])
if not image_url:
# Try and get images from Fanart.TV
image_url = self._retrieve_show_images_from_fanart(
show_obj, image_type)
else:
if getattr(indexer_show_obj, image_type, None):
image_url = indexer_show_obj[image_type]
if not image_url:
# Try and get images from Fanart.TV
image_url = self._retrieve_show_images_from_fanart(
show_obj, image_type)
if not image_url:
# Try and get images from TMDB
image_url = self._retrieve_show_images_from_tmdb(
show_obj, image_type)
if image_url:
image_data = metadata_helpers.getShowImage(image_url, which)
return image_data
return None
def _retrieve_show_image(self, image_type, show_obj, which=None):
"""
Gets an image URL from theTVDB.com, fanart.tv and TMDB.com, downloads it and returns the data.
If type is fanart, multiple image src urls are returned instead of a single data image.
image_type: type of image to retrieve (currently supported: fanart, poster, banner, poster_thumb, banner_thumb)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
Returns: the binary image data if available, or else None
"""
indexer_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = sickbeard.indexerApi(
show_obj.indexer).api_params.copy()
if image_type.startswith('fanart'):
lINDEXER_API_PARMS['fanart'] = True
elif image_type.startswith('poster'):
lINDEXER_API_PARMS['posters'] = True
else:
lINDEXER_API_PARMS['banners'] = True
lINDEXER_API_PARMS['dvdorder'] = 0 != show_obj.dvdorder
if indexer_lang and not 'en' == indexer_lang:
lINDEXER_API_PARMS['language'] = indexer_lang
t = sickbeard.indexerApi(
show_obj.indexer).indexer(**lINDEXER_API_PARMS)
indexer_show_obj = t[show_obj.indexerid, False]
except (sickbeard.indexer_error, IOError) as e:
logger.log(
u"Unable to look up show on " +
sickbeard.indexerApi(show_obj.indexer).name +
", not downloading images: " + ex(e), logger.WARNING)
return None
if not self._valid_show(indexer_show_obj, show_obj):
return None
return_links = False
if 'fanart_all' == image_type:
return_links = True
image_type = 'fanart'
if image_type not in ('poster', 'banner', 'fanart', 'poster_thumb',
'banner_thumb'):
logger.log(
u"Invalid image type " + str(image_type) +
", couldn't find it in the " +
sickbeard.indexerApi(show_obj.indexer).name + " object",
logger.ERROR)
return None
image_urls = []
init_url = None
alt_tvdb_urls = []
def build_url(s_o, image_mode):
_urls = [[], []]
_url = s_o[image_mode]
if _url and _url.startswith('http'):
if 'poster' == image_mode:
_url = re.sub('posters', '_cache/posters', _url)
elif 'banner' == image_mode:
_url = re.sub('graphical', '_cache/graphical', _url)
_urls[0].append(_url)
try:
alt_url = '%swww.%s%s' % re.findall(
r'(https?://)(?:artworks\.)?(thetvdb\.[^/]+/banners/[^\d]+[^.]+)(?:_t)(.*)',
_url)[0][0:3]
if alt_url not in _urls[0]:
_urls[1].append(alt_url)
except (IndexError, Exception):
try:
alt_url = '%sartworks.%s_t%s' % re.findall(
r'(https?://)(?:www\.)?(thetvdb\.[^/]+/banners/[^\d]+[^.]+)(.*)',
_url)[0][0:3]
if alt_url not in _urls[0]:
_urls[1].append(alt_url)
except (IndexError, Exception):
pass
return _urls
if 'poster_thumb' == image_type:
if None is not getattr(indexer_show_obj, image_type, None):
image_urls, alt_tvdb_urls = build_url(indexer_show_obj,
image_type)
elif None is not getattr(indexer_show_obj, 'poster', None):
image_urls, alt_tvdb_urls = build_url(indexer_show_obj,
'poster')
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang, True) or []:
image_urls.append(item[2])
if 0 == len(image_urls):
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append(item[2])
elif 'banner_thumb' == image_type:
if None is not getattr(indexer_show_obj, image_type, None):
image_urls, alt_tvdb_urls = build_url(indexer_show_obj,
image_type)
elif None is not getattr(indexer_show_obj, 'banner', None):
image_urls, alt_tvdb_urls = build_url(indexer_show_obj,
'banner')
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang, True) or []:
image_urls.append(item[2])
else:
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang) or []:
image_urls.append(item[2])
if getattr(indexer_show_obj, image_type, None) is not None:
image_url = indexer_show_obj[image_type]
if image_url:
image_urls.append(image_url)
if 'poster' == image_type:
init_url = image_url
# check extra provided images in '_banners' key
if None is not getattr(indexer_show_obj, '_banners', None) and \
isinstance(indexer_show_obj['_banners'].get(image_type, None), (list, dict)):
for res, value in iteritems(
indexer_show_obj['_banners'][image_type]):
for item in itervalues(value):
image_urls.append(item['bannerpath'])
if 0 == len(image_urls) or 'fanart' == image_type:
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append('%s?%s' % (item[2], item[0]))
image_data = len(image_urls) or None
if image_data:
# remove duplicates, keeping order
image_urls = list(OrderedDict.fromkeys(image_urls))
if return_links:
return image_urls
else:
image_data = metadata_helpers.getShowImage(
(init_url, image_urls[0])[None is init_url], which,
show_obj.name, bool(len(alt_tvdb_urls)))
if None is image_data and len(alt_tvdb_urls):
for url in alt_tvdb_urls:
image_data = metadata_helpers.getShowImage(
(init_url, url)[None is init_url], which,
show_obj.name)
if None is not image_data:
break
if None is not image_data:
return image_data
return None
if len(image_urls) is 0:
logger.log(u"Unable to find any suitable extra fanart images for "+tvshow.name+" on Fanart.tv.")
return None
else:
best = TvShow.get_best_leaf([background for background in tvshow.backgrounds if background.language == show_obj.lang])
if best is not None:
image_url = best.url
else:
logger.log(u"Unable to find any suitable extra fanart images for "+tvshow.name+" on Fanart.tv")
return None
# Build image data to return
if all_images:
image_data = []
for tup in image_urls:
image_data.append(dict(id=tup[0], data=metadata_helpers.getShowImage(tup[1])))
else:
image_data = metadata_helpers.getShowImage(image_url)
else:
# This should be impossible
raise RuntimeError()
return image_data
def _season_thumb_dict(self, show_obj):
"""
Should return a dict like:
result = {<season number>:
{1: '<url 1>', 2: <url 2>, ...},}
"""
def _retrieve_show_image(self, image_type, show_obj, which=None):
"""
Gets an image URL from theTVDB.com, fanart.tv and TMDB.com, downloads it and returns the data.
If type is fanart, multiple image src urls are returned instead of a single data image.
image_type: type of image to retrieve (currently supported: fanart, poster, banner, poster_thumb, banner_thumb)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
Returns: the binary image data if available, or else None
"""
indexer_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = sickbeard.indexerApi(
show_obj.indexer).api_params.copy()
if image_type.startswith('fanart'):
lINDEXER_API_PARMS['fanart'] = True
elif image_type.startswith('poster'):
lINDEXER_API_PARMS['posters'] = True
else:
lINDEXER_API_PARMS['banners'] = True
lINDEXER_API_PARMS['dvdorder'] = 0 != show_obj.dvdorder
if indexer_lang and not 'en' == indexer_lang:
lINDEXER_API_PARMS['language'] = indexer_lang
t = sickbeard.indexerApi(
show_obj.indexer).indexer(**lINDEXER_API_PARMS)
indexer_show_obj = t[show_obj.indexerid, False]
except (sickbeard.indexer_error, IOError) as e:
logger.log(
u"Unable to look up show on " +
sickbeard.indexerApi(show_obj.indexer).name +
", not downloading images: " + ex(e), logger.WARNING)
return None
if not self._valid_show(indexer_show_obj, show_obj):
return None
return_links = False
if 'fanart_all' == image_type:
return_links = True
image_type = 'fanart'
if image_type not in ('poster', 'banner', 'fanart', 'poster_thumb',
'banner_thumb'):
logger.log(
u"Invalid image type " + str(image_type) +
", couldn't find it in the " +
sickbeard.indexerApi(show_obj.indexer).name + " object",
logger.ERROR)
return None
image_urls = []
init_url = None
if 'poster_thumb' == image_type:
if getattr(indexer_show_obj, 'poster', None) is not None:
image_url = re.sub('posters', '_cache/posters',
indexer_show_obj['poster'])
if image_url:
image_urls.append(image_url)
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang, True) or []:
image_urls.append(item[2])
if 0 == len(image_urls):
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append(item[2])
elif 'banner_thumb' == image_type:
if getattr(indexer_show_obj, 'banner', None) is not None:
image_url = re.sub('graphical', '_cache/graphical',
indexer_show_obj['banner'])
if image_url:
image_urls.append(image_url)
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang, True) or []:
image_urls.append(item[2])
else:
for item in self._fanart_urls_from_show(show_obj, image_type,
indexer_lang) or []:
image_urls.append(item[2])
if getattr(indexer_show_obj, image_type, None) is not None:
image_url = indexer_show_obj[image_type]
if image_url:
image_urls.append(image_url)
if 'poster' == image_type:
init_url = image_url
if 0 == len(image_urls) or 'fanart' == image_type:
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append('%s?%s' % (item[2], item[0]))
image_data = len(image_urls) or None
if image_data:
if return_links:
return image_urls
else:
image_data = metadata_helpers.getShowImage(
(init_url, image_urls[0])[None is init_url], which,
show_obj.name)
if None is not image_data:
return image_data
return None
def _retrieve_show_image(self, image_type, show_obj, which=None):
"""
Gets an image URL from theTVDB.com, fanart.tv and TMDB.com, downloads it and returns the data.
If type is fanart, multiple image src urls are returned instead of a single data image.
image_type: type of image to retrieve (currently supported: fanart, poster, banner, poster_thumb, banner_thumb)
show_obj: a TVShow object to use when searching for the image
which: optional, a specific numbered poster to look for
Returns: the binary image data if available, or else None
"""
indexer_lang = show_obj.lang
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = sickbeard.indexerApi(show_obj.indexer).api_params.copy()
if image_type.startswith('fanart'):
lINDEXER_API_PARMS['fanart'] = True
elif image_type.startswith('poster'):
lINDEXER_API_PARMS['posters'] = True
else:
lINDEXER_API_PARMS['banners'] = True
lINDEXER_API_PARMS['dvdorder'] = 0 != show_obj.dvdorder
if indexer_lang and not 'en' == indexer_lang:
lINDEXER_API_PARMS['language'] = indexer_lang
t = sickbeard.indexerApi(show_obj.indexer).indexer(**lINDEXER_API_PARMS)
indexer_show_obj = t[show_obj.indexerid, False]
except (sickbeard.indexer_error, IOError) as e:
logger.log(u"Unable to look up show on " + sickbeard.indexerApi(
show_obj.indexer).name + ", not downloading images: " + ex(e), logger.WARNING)
return None
if not self._valid_show(indexer_show_obj, show_obj):
return None
return_links = False
if 'fanart_all' == image_type:
return_links = True
image_type = 'fanart'
if image_type not in ('poster', 'banner', 'fanart', 'poster_thumb', 'banner_thumb'):
logger.log(u"Invalid image type " + str(image_type) + ", couldn't find it in the " + sickbeard.indexerApi(
show_obj.indexer).name + " object", logger.ERROR)
return None
image_urls = []
init_url = None
if 'poster_thumb' == image_type:
if getattr(indexer_show_obj, 'poster', None) is not None:
image_url = re.sub('posters', '_cache/posters', indexer_show_obj['poster'])
if image_url:
image_urls.append(image_url)
for item in self._fanart_urls_from_show(show_obj, image_type, indexer_lang, True) or []:
image_urls.append(item[2])
if 0 == len(image_urls):
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append(item[2])
elif 'banner_thumb' == image_type:
if getattr(indexer_show_obj, 'banner', None) is not None:
image_url = re.sub('graphical', '_cache/graphical', indexer_show_obj['banner'])
if image_url:
image_urls.append(image_url)
for item in self._fanart_urls_from_show(show_obj, image_type, indexer_lang, True) or []:
image_urls.append(item[2])
else:
for item in self._fanart_urls_from_show(show_obj, image_type, indexer_lang) or []:
image_urls.append(item[2])
if getattr(indexer_show_obj, image_type, None) is not None:
image_url = indexer_show_obj[image_type]
if image_url:
image_urls.append(image_url)
if 'poster' == image_type:
init_url = image_url
if 0 == len(image_urls) or 'fanart' == image_type:
for item in self._tmdb_image_url(show_obj, image_type) or []:
image_urls.append('%s?%s' % (item[2], item[0]))
image_data = len(image_urls) or None
if image_data:
if return_links:
return image_urls
else:
image_data = metadata_helpers.getShowImage((init_url, image_urls[0])[None is init_url], which, show_obj.name)
if None is not image_data:
return image_data
return None
