mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-22 10:43:38 +00:00
32987134ba
Cleanup most init warnings. Cleanup some vars, pythonic instead of js. Some typos and python var/func names for Scheduler. Remove legacy handlers deprecated in 2020. Remove some legacy tagged stuff. Cleanup ConfigParser and 23.py Change cleanup vendored scandir. Remove redundant pkg_resources.py in favour of the vendor folder. Remove backports. Remove trakt checker. Change remove redundant WindowsSelectorEventLoopPolicy from webserveInit. Cleanup varnames and providers Various minor tidy ups to remove ide warnings.
1248 lines
52 KiB
Python
1248 lines
52 KiB
Python
#
|
|
# This file is part of SickGear.
|
|
#
|
|
# SickGear is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickGear is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from __future__ import with_statement
|
|
from collections import OrderedDict
|
|
|
|
import datetime
|
|
import io
|
|
import os.path
|
|
import re
|
|
|
|
from . import helpers as metadata_helpers
|
|
from .. import logger
|
|
import sg_helpers
|
|
from ..indexers import indexer_config
|
|
from ..indexers.indexer_config import TVINFO_TVDB, TVINFO_TMDB
|
|
from lib.tvinfo_base import TVInfoImage, TVInfoImageType, TVInfoImageSize
|
|
from lib.tvinfo_base.exceptions import *
|
|
import sickgear
|
|
from exceptions_helper import ex
|
|
from lib.fanart.core import Request as fanartRequest
|
|
import lib.fanart as fanart
|
|
from lxml_etree import etree
|
|
|
|
from six import iteritems, itervalues, string_types
|
|
|
|
# noinspection PyUnreachableCode
|
|
if False:
|
|
from typing import AnyStr, Dict, Generator, List, Optional, Tuple, Union
|
|
from lib.tvinfo_base import TVInfoShow
|
|
from ..tv import TVShow
|
|
|
|
|
|
map_image_types = {
|
|
'poster': TVInfoImageType.poster,
|
|
'banner': TVInfoImageType.banner,
|
|
'fanart': TVInfoImageType.fanart,
|
|
'poster_thumb': TVInfoImageType.poster,
|
|
'banner_thumb': TVInfoImageType.banner,
|
|
}
|
|
|
|
|
|
class ShowInfosDict(OrderedDict):
|
|
|
|
def __getitem__(self, k):
|
|
v = OrderedDict.__getitem__(self, k)
|
|
if callable(v):
|
|
v = v(k)
|
|
OrderedDict.__setitem__(self, k, v)
|
|
return v
|
|
|
|
|
|
class GenericMetadata(object):
|
|
"""
|
|
Base class for all metadata providers. Default behavior is meant to mostly
|
|
follow XBMC 12+ metadata standards. Has support for:
|
|
- show metadata file
|
|
- episode metadata file
|
|
- episode thumbnail
|
|
- show fanart
|
|
- show poster
|
|
- show banner
|
|
- season thumbnails (poster)
|
|
- season thumbnails (banner)
|
|
- season all poster
|
|
- season all banner
|
|
"""
|
|
|
|
def __init__(self,
|
|
show_metadata=False, # type: bool
|
|
episode_metadata=False, # type: bool
|
|
use_fanart=False, # type: bool
|
|
use_poster=False, # type: bool
|
|
use_banner=False, # type: bool
|
|
episode_thumbnails=False, # type: bool
|
|
season_posters=False, # type: bool
|
|
season_banners=False, # type: bool
|
|
season_all_poster=False, # type: bool
|
|
season_all_banner=False # type: bool
|
|
):
|
|
|
|
self.name = "Generic" # type: AnyStr
|
|
|
|
self._ep_nfo_extension = "nfo" # type: AnyStr
|
|
self._show_metadata_filename = "tvshow.nfo" # type: AnyStr
|
|
|
|
self.fanart_name = "fanart.jpg" # type: AnyStr
|
|
self.poster_name = "poster.jpg" # type: AnyStr
|
|
self.banner_name = "banner.jpg" # type: AnyStr
|
|
|
|
self.season_all_poster_name = "season-all-poster.jpg" # type: AnyStr
|
|
self.season_all_banner_name = "season-all-banner.jpg" # type: AnyStr
|
|
|
|
self.show_metadata = show_metadata
|
|
self.episode_metadata = episode_metadata
|
|
self.fanart = use_fanart
|
|
self.poster = use_poster
|
|
self.banner = use_banner
|
|
self.episode_thumbnails = episode_thumbnails
|
|
self.season_posters = season_posters
|
|
self.season_banners = season_banners
|
|
self.season_all_poster = season_all_poster
|
|
self.season_all_banner = season_all_banner
|
|
|
|
def get_config(self):
|
|
# type: (...) -> AnyStr
|
|
config_list = [self.show_metadata, self.episode_metadata, self.fanart, self.poster, self.banner,
|
|
self.episode_thumbnails, self.season_posters, self.season_banners, self.season_all_poster,
|
|
self.season_all_banner]
|
|
return '|'.join([str(int(x)) for x in config_list])
|
|
|
|
def get_id(self):
|
|
# type: (...) -> AnyStr
|
|
return GenericMetadata.make_id(self.name)
|
|
|
|
@staticmethod
|
|
def make_id(name):
|
|
# type: (AnyStr) -> AnyStr
|
|
name_id = re.sub("[+]", "plus", name)
|
|
name_id = re.sub(r"[^\w_]", "_", name_id).lower()
|
|
return name_id
|
|
|
|
def set_config(self, string):
|
|
# type: (AnyStr) -> None
|
|
config_list = [bool(int(x)) for x in string.split('|')]
|
|
self.show_metadata = config_list[0]
|
|
self.episode_metadata = config_list[1]
|
|
self.fanart = config_list[2]
|
|
self.poster = config_list[3]
|
|
self.banner = config_list[4]
|
|
self.episode_thumbnails = config_list[5]
|
|
self.season_posters = config_list[6]
|
|
self.season_banners = config_list[7]
|
|
self.season_all_poster = config_list[8]
|
|
self.season_all_banner = config_list[9]
|
|
|
|
def _has_show_metadata(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_show_file_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_show_file_path(show_obj) + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def has_episode_metadata(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> bool
|
|
result = os.path.isfile(self.get_episode_file_path(ep_obj))
|
|
logger.log(u"Checking if " + self.get_episode_file_path(ep_obj) + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_fanart(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_fanart_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_fanart_path(show_obj) + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_poster(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_poster_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_poster_path(show_obj) + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_banner(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_banner_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_banner_path(show_obj) + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def has_episode_thumb(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> bool
|
|
location = self.get_episode_thumb_path(ep_obj)
|
|
result = None is not location and os.path.isfile(location)
|
|
if location:
|
|
logger.log(u"Checking if " + location + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_season_poster(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow,int) -> bool
|
|
location = self.get_season_poster_path(show_obj, season)
|
|
result = None is not location and os.path.isfile(location)
|
|
if location:
|
|
logger.log(u"Checking if " + location + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_season_banner(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow,int) -> bool
|
|
location = self.get_season_banner_path(show_obj, season)
|
|
result = None is not location and os.path.isfile(location)
|
|
if location:
|
|
logger.log(u"Checking if " + location + " exists: " + str(result), logger.DEBUG)
|
|
return result
|
|
|
|
def _has_season_all_poster(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_season_all_poster_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_season_all_poster_path(show_obj) + " exists: " + str(result),
|
|
logger.DEBUG)
|
|
return result
|
|
|
|
def _has_season_all_banner(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
result = os.path.isfile(self.get_season_all_banner_path(show_obj))
|
|
logger.log(u"Checking if " + self.get_season_all_banner_path(show_obj) + " exists: " + str(result),
|
|
logger.DEBUG)
|
|
return result
|
|
|
|
@staticmethod
|
|
def get_show_year(show_obj, show_info, year_only=True):
|
|
# type: (sickgear.tv.TVShow, Dict, bool) -> Optional[AnyStr]
|
|
if None is not getattr(show_info, 'firstaired', None):
|
|
try:
|
|
first_aired = datetime.datetime.strptime(show_info['firstaired'], '%Y-%m-%d')
|
|
if first_aired:
|
|
if year_only:
|
|
return str(first_aired.year)
|
|
return str(first_aired.date())
|
|
except (BaseException, Exception):
|
|
pass
|
|
if isinstance(show_obj, sickgear.tv.TVShow):
|
|
if year_only and show_obj.startyear:
|
|
return '%s' % show_obj.startyear
|
|
if not show_obj.sxe_ep_obj.get(1, {}).get(1, None):
|
|
show_obj.get_all_episodes()
|
|
try:
|
|
first_ep_obj = show_obj.first_aired_regular_episode
|
|
except (BaseException, Exception):
|
|
first_ep_obj = None
|
|
if isinstance(first_ep_obj, sickgear.tv.TVEpisode) \
|
|
and isinstance(first_ep_obj.airdate, datetime.date) and 1900 < first_ep_obj.airdate.year:
|
|
return '%s' % (first_ep_obj.airdate.year, first_ep_obj.airdate)[not year_only]
|
|
|
|
def get_show_file_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self._show_metadata_filename)
|
|
|
|
def get_episode_file_path(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> AnyStr
|
|
return sg_helpers.replace_extension(ep_obj.location, self._ep_nfo_extension)
|
|
|
|
def get_fanart_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self.fanart_name)
|
|
|
|
def get_poster_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self.poster_name)
|
|
|
|
def get_banner_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self.banner_name)
|
|
|
|
def get_episode_thumb_path(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> Optional[AnyStr]
|
|
"""
|
|
Returns the path where the episode thumbnail should be stored.
|
|
ep_obj: a TVEpisode instance for which to create the thumbnail
|
|
"""
|
|
if os.path.isfile(ep_obj.location):
|
|
|
|
tbn_filename = ep_obj.location.rpartition('.')
|
|
|
|
if '' == tbn_filename[0]:
|
|
tbn_filename = ep_obj.location
|
|
else:
|
|
tbn_filename = tbn_filename[0]
|
|
|
|
return tbn_filename + '-thumb.jpg'
|
|
|
|
def get_season_poster_path(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow, int) -> AnyStr
|
|
"""
|
|
Returns the full path to the file for a given season poster.
|
|
|
|
show_obj: a TVShow instance for which to generate the path
|
|
season: a season number to be used for the path. Note that season 0
|
|
means specials.
|
|
"""
|
|
|
|
# Our specials thumbnail is, well, special
|
|
if 0 == season:
|
|
season_poster_filename = 'season-specials'
|
|
else:
|
|
season_poster_filename = 'season' + str(season).zfill(2)
|
|
|
|
return os.path.join(show_obj.location, season_poster_filename + '-poster.jpg')
|
|
|
|
def get_season_banner_path(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow, int) -> AnyStr
|
|
"""
|
|
Returns the full path to the file for a given season banner.
|
|
|
|
show_obj: a TVShow instance for which to generate the path
|
|
season: a season number to be used for the path. Note that season 0
|
|
means specials.
|
|
"""
|
|
|
|
# Our specials thumbnail is, well, special
|
|
if 0 == season:
|
|
season_banner_filename = 'season-specials'
|
|
else:
|
|
season_banner_filename = 'season' + str(season).zfill(2)
|
|
|
|
return os.path.join(show_obj.location, season_banner_filename + '-banner.jpg')
|
|
|
|
def get_season_all_poster_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self.season_all_poster_name)
|
|
|
|
def get_season_all_banner_path(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> AnyStr
|
|
return os.path.join(show_obj.location, self.season_all_banner_name)
|
|
|
|
def _show_data(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> Optional[Union[bool, etree.Element]]
|
|
"""
|
|
This should be overridden by the implementing class. It should
|
|
provide the content of the show metadata file.
|
|
"""
|
|
return None
|
|
|
|
@staticmethod
|
|
def _valid_show(fetched_show_info, show_obj):
|
|
# type: (Dict, sickgear.tv.TVShow) -> bool
|
|
"""
|
|
Test the integrity of fetched show data
|
|
|
|
:param fetched_show_info: the object returned from the tvinfo source
|
|
:param show_obj: Show that the fetched data relates to
|
|
:return: True if fetched_show_obj is valid data otherwise False
|
|
"""
|
|
if not (isinstance(fetched_show_info, dict) and
|
|
isinstance(getattr(fetched_show_info, 'data', None), (list, dict)) and
|
|
'seriesname' in getattr(fetched_show_info, 'data', [])) and \
|
|
not hasattr(fetched_show_info, 'seriesname'):
|
|
logger.log(u'Show %s not found on %s ' %
|
|
(show_obj.name, sickgear.TVInfoAPI(show_obj.tvid).name), logger.WARNING)
|
|
return False
|
|
return True
|
|
|
|
def _ep_data(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> Optional[Union[bool, etree.Element]]
|
|
"""
|
|
This should be overridden by the implementing class. It should
|
|
provide the content of the episode metadata file.
|
|
"""
|
|
return None
|
|
|
|
def create_show_metadata(self, show_obj, force=False):
|
|
# type: (sickgear.tv.TVShow, bool) -> bool
|
|
result = False
|
|
if self.show_metadata and show_obj and (not self._has_show_metadata(show_obj) or force):
|
|
logger.debug('Metadata provider %s creating show metadata for %s' % (self.name, show_obj.unique_name))
|
|
try:
|
|
result = self.write_show_file(show_obj)
|
|
except BaseTVinfoError as e:
|
|
logger.log('Unable to find useful show metadata for %s on %s: %s' % (
|
|
self.name, sickgear.TVInfoAPI(show_obj.tvid).name, ex(e)), logger.WARNING)
|
|
|
|
return result
|
|
|
|
def create_episode_metadata(self, ep_obj, force=False):
|
|
# type: (sickgear.tv.TVEpisode, bool) -> bool
|
|
result = False
|
|
if self.episode_metadata and ep_obj and (not self.has_episode_metadata(ep_obj) or force):
|
|
logger.log('Metadata provider %s creating episode metadata for %s' % (self.name, ep_obj.pretty_name()),
|
|
logger.DEBUG)
|
|
try:
|
|
result = self.write_ep_file(ep_obj)
|
|
except BaseTVinfoError as e:
|
|
logger.log('Unable to find useful episode metadata for %s on %s: %s' % (
|
|
self.name, sickgear.TVInfoAPI(ep_obj.show_obj.tvid).name, ex(e)), logger.WARNING)
|
|
|
|
return result
|
|
|
|
def update_show_indexer_metadata(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.show_metadata and show_obj and self._has_show_metadata(show_obj):
|
|
logger.debug(u'Metadata provider %s updating show indexer metadata file for %s' % (
|
|
self.name, show_obj.unique_name))
|
|
|
|
nfo_file_path = self.get_show_file_path(show_obj)
|
|
with io.open(nfo_file_path, 'r', encoding='utf8') as xmlFileObj:
|
|
show_xml = etree.ElementTree(file=xmlFileObj)
|
|
|
|
tvid = show_xml.find('indexer')
|
|
prodid = show_xml.find('id')
|
|
|
|
root = show_xml.getroot()
|
|
show_tvid = str(show_obj.tvid)
|
|
if None is not tvid:
|
|
tvid.text = '%s' % show_tvid
|
|
else:
|
|
etree.SubElement(root, 'indexer').text = '%s' % show_tvid
|
|
|
|
show_prodid = str(show_obj.prodid)
|
|
if None is not prodid:
|
|
prodid.text = '%s' % show_prodid
|
|
else:
|
|
etree.SubElement(root, 'id').text = '%s' % show_prodid
|
|
|
|
# Make it purdy
|
|
sg_helpers.indent_xml(root)
|
|
|
|
sg_helpers.write_file(nfo_file_path, show_xml, xmltree=True, utf8=True)
|
|
|
|
return True
|
|
|
|
def create_fanart(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.fanart and show_obj and not self._has_fanart(show_obj):
|
|
logger.debug(u'Metadata provider %s creating fanart for %s' % (self.name, show_obj.unique_name))
|
|
return self.save_fanart(show_obj)
|
|
return False
|
|
|
|
def create_poster(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.poster and show_obj and not self._has_poster(show_obj):
|
|
logger.debug(u'Metadata provider %s creating poster for %s' % (self.name, show_obj.unique_name))
|
|
return self.save_poster(show_obj)
|
|
return False
|
|
|
|
def create_banner(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.banner and show_obj and not self._has_banner(show_obj):
|
|
logger.debug(u'Metadata provider %s creating banner for %s' % (self.name, show_obj.unique_name))
|
|
return self.save_banner(show_obj)
|
|
return False
|
|
|
|
def create_episode_thumb(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> bool
|
|
if self.episode_thumbnails and ep_obj and not self.has_episode_thumb(ep_obj):
|
|
logger.log(u"Metadata provider " + self.name + " creating episode thumbnail for " + ep_obj.pretty_name(),
|
|
logger.DEBUG)
|
|
return self.save_thumbnail(ep_obj)
|
|
return False
|
|
|
|
def create_season_posters(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.season_posters and show_obj:
|
|
result = []
|
|
for season, _ in iteritems(show_obj.sxe_ep_obj):
|
|
if not self._has_season_poster(show_obj, season):
|
|
logger.debug(u'Metadata provider %s creating season posters for %s' % (
|
|
self.name, show_obj.unique_name))
|
|
result = result + [self.save_season_posters(show_obj, season)]
|
|
return all(result)
|
|
return False
|
|
|
|
def create_season_banners(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.season_banners and show_obj:
|
|
result = []
|
|
for season, _ in iteritems(show_obj.sxe_ep_obj):
|
|
if not self._has_season_banner(show_obj, season):
|
|
logger.debug(u'Metadata provider %s creating season banners for %s' % (
|
|
self.name, show_obj.unique_name))
|
|
result = result + [self.save_season_banners(show_obj, season)]
|
|
return all(result)
|
|
return False
|
|
|
|
def create_season_all_poster(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.season_all_poster and show_obj and not self._has_season_all_poster(show_obj):
|
|
logger.debug(u'Metadata provider %s creating season all posters for %s' % (
|
|
self.name, show_obj.unique_name))
|
|
return self.save_season_all_poster(show_obj)
|
|
return False
|
|
|
|
def create_season_all_banner(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
if self.season_all_banner and show_obj and not self._has_season_all_banner(show_obj):
|
|
logger.debug(u'Metadata provider %s creating season all banner for %s' % (
|
|
self.name, show_obj.unique_name))
|
|
return self.save_season_all_banner(show_obj)
|
|
return False
|
|
|
|
@staticmethod
|
|
def _get_episode_thumb_url(ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> Optional[AnyStr]
|
|
"""
|
|
Returns the URL to use for downloading an episode's thumbnail. Uses
|
|
theTVDB.com and TVRage.com data.
|
|
|
|
:param ep_obj: a TVEpisode object for which to grab the thumb URL
|
|
:return: URL to thumb
|
|
"""
|
|
|
|
ep_obj_list = [ep_obj] + ep_obj.related_ep_obj
|
|
|
|
# validate show
|
|
from .. import helpers
|
|
if not helpers.validate_show(ep_obj.show_obj):
|
|
return None
|
|
|
|
# try all included episodes in case some have thumbs and others don't
|
|
for cur_ep_obj in ep_obj_list:
|
|
if TVINFO_TVDB == cur_ep_obj.show_obj.tvid:
|
|
show_lang = cur_ep_obj.show_obj.lang
|
|
|
|
try:
|
|
tvinfo_config = sickgear.TVInfoAPI(TVINFO_TVDB).api_params.copy()
|
|
tvinfo_config['dvdorder'] = 0 != cur_ep_obj.show_obj.dvdorder
|
|
tvinfo_config['no_dummy'] = True
|
|
|
|
if show_lang and not 'en' == show_lang:
|
|
tvinfo_config['language'] = show_lang
|
|
|
|
t = sickgear.TVInfoAPI(TVINFO_TVDB).setup(**tvinfo_config)
|
|
|
|
ep_info = t[cur_ep_obj.show_obj.prodid][cur_ep_obj.season][cur_ep_obj.episode]
|
|
except (BaseTVinfoEpisodenotfound, BaseTVinfoSeasonnotfound, TypeError):
|
|
ep_info = None
|
|
else:
|
|
ep_info = helpers.validate_show(cur_ep_obj.show_obj, cur_ep_obj.season, cur_ep_obj.episode)
|
|
|
|
if not ep_info:
|
|
continue
|
|
|
|
thumb_url = getattr(ep_info, 'filename', None) \
|
|
or (isinstance(ep_info, dict) and ep_info.get('filename', None))
|
|
if thumb_url not in (None, False, ''):
|
|
return thumb_url
|
|
|
|
return None
|
|
|
|
def write_show_file(self, show_obj):
|
|
# type: (sickgear.tv.TVShow) -> bool
|
|
"""
|
|
Generates and writes show_obj's metadata under the given path to the
|
|
filename given by get_show_file_path()
|
|
|
|
show_obj: TVShow object for which to create the metadata
|
|
|
|
path: An absolute or relative path where we should put the file. Note that
|
|
the file name will be the default show_file_name.
|
|
|
|
Note that this method expects that _show_data will return an ElementTree
|
|
object. If your _show_data returns data in another format you'll need to
|
|
override this method.
|
|
"""
|
|
|
|
data = self._show_data(show_obj)
|
|
|
|
if not data:
|
|
return False
|
|
|
|
nfo_file_path = self.get_show_file_path(show_obj)
|
|
|
|
logger.log(u'Writing show metadata file: %s' % nfo_file_path, logger.DEBUG)
|
|
|
|
return sg_helpers.write_file(nfo_file_path, data, xmltree=True, utf8=True)
|
|
|
|
def write_ep_file(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> bool
|
|
"""
|
|
Generates and writes ep_obj's metadata under the given path with the
|
|
given filename root. Uses the episode's name with the extension in
|
|
_ep_nfo_extension.
|
|
|
|
ep_obj: TVEpisode object for which to create the metadata
|
|
|
|
file_name_path: The file name to use for this metadata. Note that the extension
|
|
will be automatically added based on _ep_nfo_extension. This should
|
|
include an absolute path.
|
|
|
|
Note that this method expects that _ep_data will return an ElementTree
|
|
object. If your _ep_data returns data in another format you'll need to
|
|
override this method.
|
|
"""
|
|
|
|
data = self._ep_data(ep_obj)
|
|
|
|
if not data:
|
|
return False
|
|
|
|
nfo_file_path = self.get_episode_file_path(ep_obj)
|
|
|
|
logger.log(u'Writing episode metadata file: %s' % nfo_file_path, logger.DEBUG)
|
|
|
|
return sg_helpers.write_file(nfo_file_path, data, xmltree=True, utf8=True)
|
|
|
|
def save_thumbnail(self, ep_obj):
|
|
# type: (sickgear.tv.TVEpisode) -> bool
|
|
"""
|
|
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.get_show_image(thumb_url, show_name=ep_obj.show_obj.name)
|
|
|
|
result = self._write_image(thumb_data, file_path)
|
|
|
|
if not result:
|
|
return False
|
|
|
|
for cur_ep_obj in [ep_obj] + ep_obj.related_ep_obj:
|
|
cur_ep_obj.hastbn = True
|
|
|
|
return True
|
|
|
|
def save_fanart(self, show_obj, which=None):
|
|
# type: (sickgear.tv.TVShow, Optional[AnyStr]) -> bool
|
|
"""
|
|
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)
|
|
|
|
fanart_data = self._retrieve_show_image('fanart', show_obj, which,
|
|
img_cache_type=sickgear.image_cache.ImageCache.FANART)
|
|
|
|
if not fanart_data:
|
|
logger.log(u"No fanart image was retrieved, unable to write fanart", logger.DEBUG)
|
|
return False
|
|
|
|
return self._write_image(fanart_data, fanart_path)
|
|
|
|
def save_poster(self, show_obj, which=None):
|
|
# type: (sickgear.tv.TVShow, Optional[AnyStr]) -> bool
|
|
"""
|
|
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)
|
|
|
|
poster_data = self._retrieve_show_image('poster', show_obj, which,
|
|
img_cache_type=sickgear.image_cache.ImageCache.POSTER)
|
|
|
|
if not poster_data:
|
|
logger.log(u"No show poster image was retrieved, unable to write poster", logger.DEBUG)
|
|
return False
|
|
|
|
return self._write_image(poster_data, poster_path)
|
|
|
|
def save_banner(self, show_obj, which=None):
|
|
# type: (sickgear.tv.TVShow, Optional[AnyStr]) -> bool
|
|
"""
|
|
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
|
|
"""
|
|
|
|
# use the default banner name
|
|
banner_path = self.get_banner_path(show_obj)
|
|
|
|
banner_data = self._retrieve_show_image('banner', show_obj, which,
|
|
img_cache_type=sickgear.image_cache.ImageCache.BANNER)
|
|
|
|
if not banner_data:
|
|
logger.log(u"No show banner image was retrieved, unable to write banner", logger.DEBUG)
|
|
return False
|
|
|
|
return self._write_image(banner_data, banner_path)
|
|
|
|
def save_season_posters(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow, int) -> bool
|
|
"""
|
|
Saves all season posters 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 posters if possible.
|
|
"""
|
|
|
|
season_dict = self._season_image_dict(show_obj, season, 'seasons')
|
|
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 is there for now
|
|
art_id, season_url = cur_season_art.popitem()
|
|
|
|
season_poster_file_path = self.get_season_poster_path(show_obj, cur_season)
|
|
|
|
if not season_poster_file_path:
|
|
logger.log(u'Path for season ' + str(cur_season) + ' came back blank, skipping this season',
|
|
logger.DEBUG)
|
|
continue
|
|
|
|
season_data = metadata_helpers.get_show_image(season_url, show_name=show_obj.name)
|
|
|
|
if not season_data:
|
|
logger.log(u'No season poster data available, skipping this season', logger.DEBUG)
|
|
continue
|
|
|
|
result = result + [self._write_image(season_data, season_poster_file_path)]
|
|
|
|
if result:
|
|
return all(result)
|
|
return False
|
|
|
|
def save_season_banners(self, show_obj, season):
|
|
# type: (sickgear.tv.TVShow, int) -> bool
|
|
"""
|
|
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 is there for now
|
|
art_id, season_url = cur_season_art.popitem()
|
|
|
|
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.get_show_image(season_url, show_name=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, which=None):
|
|
# type: (sickgear.tv.TVShow, Optional[AnyStr]) -> bool
|
|
# use the default season all poster name
|
|
poster_path = self.get_season_all_poster_path(show_obj)
|
|
|
|
poster_data = self._retrieve_show_image('poster', show_obj, which,
|
|
img_cache_type=sickgear.image_cache.ImageCache.POSTER)
|
|
|
|
if not poster_data:
|
|
logger.log(u"No show poster image was retrieved, unable to write season all poster", logger.DEBUG)
|
|
return False
|
|
|
|
return self._write_image(poster_data, poster_path)
|
|
|
|
def save_season_all_banner(self, show_obj, which=None):
|
|
# type: (sickgear.tv.TVShow, Optional[AnyStr]) -> bool
|
|
# use the default season all banner name
|
|
banner_path = self.get_season_all_banner_path(show_obj)
|
|
|
|
banner_data = self._retrieve_show_image('banner', show_obj, which,
|
|
img_cache_type=sickgear.image_cache.ImageCache.BANNER)
|
|
|
|
if not banner_data:
|
|
logger.log(u"No show banner image was retrieved, unable to write season all banner", logger.DEBUG)
|
|
return False
|
|
|
|
return self._write_image(banner_data, banner_path)
|
|
|
|
@staticmethod
|
|
def _write_image(image_data, image_path, force=False):
|
|
# type: (bytes, AnyStr, bool) -> bool
|
|
"""
|
|
Saves the data in image_data to the location image_path. Returns True/False
|
|
to represent success or failure.
|
|
|
|
image_data: binary image data to write to file
|
|
image_path: file location to save the image to
|
|
"""
|
|
|
|
# don't bother overwriting it
|
|
if not force and os.path.isfile(image_path):
|
|
logger.log(u"Image already exists, not downloading", logger.DEBUG)
|
|
return False
|
|
|
|
if not image_data:
|
|
logger.log(u"Unable to retrieve image, skipping", logger.WARNING)
|
|
return False
|
|
|
|
image_dir = os.path.dirname(image_path)
|
|
|
|
try:
|
|
if not os.path.isdir(image_dir):
|
|
logger.log(u"Metadata dir didn't exist, creating it at " + image_dir, logger.DEBUG)
|
|
os.makedirs(image_dir)
|
|
sg_helpers.chmod_as_parent(image_dir)
|
|
|
|
out_file = open(image_path, 'wb')
|
|
out_file.write(image_data)
|
|
out_file.close()
|
|
sg_helpers.chmod_as_parent(image_path)
|
|
except IOError as e:
|
|
logger.log(
|
|
u"Unable to write image to " + image_path + " - are you sure the show folder is writable? " + ex(e),
|
|
logger.ERROR)
|
|
return False
|
|
|
|
return True
|
|
|
|
@staticmethod
|
|
def gen_show_infos_dict(show_obj):
|
|
# type: (TVShow) -> ShowInfosDict
|
|
show_infos = ShowInfosDict()
|
|
|
|
def _get_show_info(tv_id):
|
|
try:
|
|
show_lang = show_obj.lang
|
|
# There's gotta be a better way of doing this, but we don't want to
|
|
# change the language value elsewhere
|
|
tvinfo_config = sickgear.TVInfoAPI(tv_id).api_params.copy()
|
|
tvinfo_config['fanart'] = True
|
|
tvinfo_config['posters'] = True
|
|
tvinfo_config['banners'] = True
|
|
tvinfo_config['dvdorder'] = 0 != show_obj.dvdorder
|
|
|
|
if show_lang and not 'en' == show_lang:
|
|
tvinfo_config['language'] = show_lang
|
|
|
|
t = sickgear.TVInfoAPI(tv_id).setup(**tvinfo_config)
|
|
return t.get_show((show_obj.ids[tv_id]['id'], show_obj.prodid)[tv_src == show_obj.tvid],
|
|
load_episodes=False, banners=True, posters=True, fanart=True, language=show_obj.lang)
|
|
except (BaseTVinfoError, IOError) as e:
|
|
logger.log(u"Unable to look up show on " + sickgear.TVInfoAPI(
|
|
tv_id).name + ", not downloading images: " + ex(e), logger.WARNING)
|
|
|
|
# todo: when tmdb is added as tv source remove the hardcoded TVINFO_TMDB
|
|
for tv_src in list(OrderedDict.fromkeys([show_obj.tvid] + list(sickgear.TVInfoAPI().search_sources) +
|
|
[TVINFO_TMDB])):
|
|
if tv_src != show_obj.tvid and not show_obj.ids.get(tv_src, {}).get('id'):
|
|
continue
|
|
if tv_src == show_obj.tvid:
|
|
show_infos[tv_src] = _get_show_info(tv_src)
|
|
else:
|
|
show_infos[tv_src] = _get_show_info
|
|
|
|
return show_infos
|
|
|
|
def _retrieve_image_urls(self, show_obj, image_type, show_infos):
|
|
# type: (TVShow, AnyStr, TVInfoShow) -> Generator
|
|
image_urls, alt_tvdb_urls, fanart_fetched, de_dupe, show_lang = [], [], False, set(), show_obj.lang
|
|
|
|
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
|
|
|
|
def _get_fanart_tv():
|
|
return [_de_dupe((f_item[2], (f_item[2], f_item[2]))[image_type in ('poster', 'banner')])
|
|
for f_item in self._fanart_urls_from_show(show_obj, image_type, show_lang) or []]
|
|
|
|
def _de_dupe(images_list):
|
|
# type:(Union[List[AnyStr], AnyStr]) -> Optional[Union[List[AnyStr], AnyStr]]
|
|
if not isinstance(images_list, list):
|
|
return_list = False
|
|
temp_list = [images_list]
|
|
else:
|
|
return_list = True
|
|
temp_list = images_list
|
|
images_list = [i for i in temp_list if i not in de_dupe]
|
|
[de_dupe.add(_i) for _i in images_list]
|
|
if not return_list:
|
|
if images_list:
|
|
return images_list[0]
|
|
return None
|
|
return images_list
|
|
|
|
if image_type.startswith('fanart'):
|
|
for r in _get_fanart_tv():
|
|
yield r
|
|
|
|
for tv_src in show_infos:
|
|
if not self._valid_show(show_infos[tv_src], show_obj):
|
|
continue
|
|
|
|
if 'poster_thumb' == image_type:
|
|
if None is not getattr(show_infos[tv_src], image_type, None):
|
|
image_urls, alt_tvdb_urls = build_url(show_infos[tv_src], image_type)
|
|
elif None is not getattr(show_infos[tv_src], 'poster', None):
|
|
image_urls, alt_tvdb_urls = build_url(show_infos[tv_src], 'poster')
|
|
|
|
image_urls, alt_tvdb_urls = _de_dupe(image_urls), _de_dupe(alt_tvdb_urls)
|
|
for item in image_urls + alt_tvdb_urls:
|
|
yield item
|
|
|
|
elif 'banner_thumb' == image_type:
|
|
if None is not getattr(show_infos[tv_src], image_type, None):
|
|
image_urls, alt_tvdb_urls = build_url(show_infos[tv_src], image_type)
|
|
elif None is not getattr(show_infos[tv_src], 'banner', None):
|
|
image_urls, alt_tvdb_urls = build_url(show_infos[tv_src], 'banner')
|
|
|
|
image_urls, alt_tvdb_urls = _de_dupe(image_urls), _de_dupe(alt_tvdb_urls)
|
|
for item in image_urls + alt_tvdb_urls:
|
|
yield item
|
|
|
|
else:
|
|
if None is not getattr(show_infos[tv_src], image_type, None):
|
|
image_url = show_infos[tv_src][image_type]
|
|
if image_type in ('poster', 'banner'):
|
|
if None is not getattr(show_infos[tv_src], '%s_thumb' % image_type, None):
|
|
thumb_url = show_infos[tv_src]['%s_thumb' % image_type]
|
|
else:
|
|
thumb_url = image_url
|
|
else:
|
|
thumb_url = None
|
|
if image_url:
|
|
r = _de_dupe(((image_url, thumb_url), image_url)[None is thumb_url])
|
|
if r:
|
|
yield r
|
|
|
|
# check extra provided images in '_banners' key
|
|
if None is not getattr(show_infos[tv_src], '_banners', None) and \
|
|
isinstance(show_infos[tv_src]['_banners'].get(image_type, None), (list, dict)):
|
|
for res, value in iteritems(show_infos[tv_src]['_banners'][image_type]):
|
|
for item in itervalues(value):
|
|
thumb = item['thumbnailpath']
|
|
if not thumb:
|
|
thumb = item['bannerpath']
|
|
|
|
r = _de_dupe((item['bannerpath'], (item['bannerpath'], thumb))[
|
|
image_type in ('poster', 'banner')])
|
|
if r:
|
|
yield r
|
|
|
|
# extra images via images property
|
|
tvinfo_type = map_image_types.get(image_type)
|
|
tvinfo_size = (TVInfoImageSize.original, TVInfoImageSize.medium)['_thumb' in image_type]
|
|
if tvinfo_type and getattr(show_infos[tv_src], 'images', None) and \
|
|
show_infos[tv_src].images.get(tvinfo_type):
|
|
for img in show_infos[tv_src].images[tvinfo_type]: # type: TVInfoImage
|
|
for img_size, img_url in iteritems(img.sizes):
|
|
if tvinfo_size == img_size:
|
|
img_url = _de_dupe(img_url)
|
|
if not img_url:
|
|
continue
|
|
if image_type in ('poster', 'banner'):
|
|
thumb_url = img.sizes.get(TVInfoImageSize.medium, img_url)
|
|
if thumb_url:
|
|
thumb_url = _de_dupe(thumb_url)
|
|
if not thumb_url:
|
|
thumb_url = img_url
|
|
yield img_url, thumb_url
|
|
elif img_url:
|
|
yield img_url
|
|
|
|
if not image_type.startswith('fanart'):
|
|
for r in _get_fanart_tv():
|
|
yield r
|
|
|
|
def _retrieve_show_image(self,
|
|
image_type, # type: AnyStr
|
|
show_obj, # type: sickgear.tv.TVShow
|
|
which=None, # type: int
|
|
return_links=False, # type: bool
|
|
show_infos=None, # type: ShowInfosDict
|
|
img_cache_type=None # type: int
|
|
):
|
|
# type: (...) -> Optional[bytes, List[AnyStr]]
|
|
"""
|
|
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
|
|
"""
|
|
if not show_infos:
|
|
show_infos = self.gen_show_infos_dict(show_obj)
|
|
|
|
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 " + sickgear.TVInfoAPI(
|
|
show_obj.tvid).name + " object", logger.ERROR)
|
|
return
|
|
|
|
image_urls = self._retrieve_image_urls(show_obj, image_type, show_infos)
|
|
|
|
if image_urls:
|
|
if return_links:
|
|
return image_urls
|
|
else:
|
|
img_data = None
|
|
image_cache = sickgear.image_cache.ImageCache()
|
|
for image_url in image_urls or []:
|
|
if image_type in ('poster', 'banner'):
|
|
if isinstance(image_url, tuple):
|
|
image_url = image_url[0]
|
|
img_data = metadata_helpers.get_show_image(image_url, which, show_obj.name)
|
|
if img_cache_type and img_cache_type != image_cache.which_type(img_data, is_binary=True):
|
|
img_data = None
|
|
continue
|
|
if None is not img_data:
|
|
break
|
|
|
|
if None is not img_data:
|
|
return img_data
|
|
|
|
def _season_image_dict(self, show_obj, season, image_type):
|
|
# type: (sickgear.tv.TVShow, int, AnyStr) -> Dict[int, Dict[int, AnyStr]]
|
|
"""
|
|
image_type : Type of image to fetch, 'seasons' or 'seasonwides'
|
|
image_type type : String
|
|
|
|
Should return a dict like:
|
|
|
|
result = {<season number>:
|
|
{1: '<url 1>', 2: <url 2>, ...},}
|
|
"""
|
|
result = {}
|
|
|
|
try:
|
|
# There's gotta be a better way of doing this, but we don't want to
|
|
# change the language value elsewhere
|
|
tvinfo_config = sickgear.TVInfoAPI(show_obj.tvid).api_params.copy()
|
|
tvinfo_config[image_type] = True
|
|
tvinfo_config['dvdorder'] = 0 != show_obj.dvdorder
|
|
|
|
if 'en' != getattr(show_obj, 'lang', None):
|
|
tvinfo_config['language'] = show_obj.lang
|
|
|
|
t = sickgear.TVInfoAPI(show_obj.tvid).setup(**tvinfo_config)
|
|
tvinfo_obj_show = t[show_obj.prodid]
|
|
except (BaseTVinfoError, IOError) as e:
|
|
logger.log(u'Unable to look up show on ' + sickgear.TVInfoAPI(
|
|
show_obj.tvid).name + ', not downloading images: ' + ex(e), logger.WARNING)
|
|
return result
|
|
|
|
if not self._valid_show(tvinfo_obj_show, show_obj):
|
|
return result
|
|
|
|
season_images = getattr(tvinfo_obj_show, 'banners', {}).get(
|
|
('season', 'seasonwide')['seasonwides' == image_type], {}).get(season, {})
|
|
for image_id in season_images:
|
|
if season not in result:
|
|
result[season] = {}
|
|
result[season][image_id] = season_images[image_id]['bannerpath']
|
|
|
|
return result
|
|
|
|
def retrieve_show_metadata(self, folder):
|
|
# type: (AnyStr) -> Union[Tuple[int, int, AnyStr], Tuple[None, None, None]]
|
|
"""
|
|
Used only when mass adding Existing Shows,
|
|
using previously generated Show metadata to reduce the need to query TVDB.
|
|
"""
|
|
|
|
from sickgear.indexers.indexer_config import TVINFO_TVDB
|
|
|
|
empty_return = (None, None, None)
|
|
|
|
metadata_path = os.path.join(folder, self._show_metadata_filename)
|
|
|
|
if not os.path.isdir(folder) or not os.path.isfile(metadata_path):
|
|
logger.log(u"Can't load the metadata file from " + repr(metadata_path) + ", it doesn't exist", logger.DEBUG)
|
|
return empty_return
|
|
|
|
logger.log(u"Loading show info from metadata file in " + folder, logger.DEBUG)
|
|
|
|
try:
|
|
with io.open(metadata_path, 'r', encoding='utf8') as xmlFileObj:
|
|
show_xml = etree.ElementTree(file=xmlFileObj)
|
|
|
|
if None is show_xml.findtext('title') \
|
|
or all(None is _f for _f in (show_xml.find('//uniqueid[@type]'),
|
|
show_xml.findtext('tvdbid'),
|
|
show_xml.findtext('id'),
|
|
show_xml.findtext('indexer'))):
|
|
logger.log(u"Invalid info in tvshow.nfo (missing name or id):"
|
|
+ str(show_xml.findtext('title')) + ' '
|
|
+ str(show_xml.findtext('indexer')) + ' '
|
|
+ str(show_xml.findtext('tvdbid')) + ' '
|
|
+ str(show_xml.findtext('id')))
|
|
return empty_return
|
|
|
|
name = show_xml.findtext('title')
|
|
|
|
try:
|
|
tvid = int(show_xml.findtext('indexer'))
|
|
except (BaseException, Exception):
|
|
tvid = None
|
|
|
|
# handle v2 format of .nfo file
|
|
default_source = show_xml.find('//uniqueid[@default="true"]')
|
|
if None is not default_source:
|
|
use_tvid = default_source.attrib.get('type') or tvid
|
|
if isinstance(use_tvid, string_types):
|
|
use_tvid = {sickgear.TVInfoAPI(x).config['slug']: x
|
|
for x, _ in iteritems(sickgear.TVInfoAPI().all_sources)}.get(use_tvid)
|
|
prodid = sg_helpers.try_int(default_source.text, None)
|
|
if use_tvid and None is not prodid:
|
|
return use_tvid, prodid, name
|
|
|
|
prodid = show_xml.find('//uniqueid[@type="tvdb"]')
|
|
if None is not prodid:
|
|
prodid = int(prodid.text)
|
|
tvid = TVINFO_TVDB
|
|
elif None is not show_xml.findtext('tvdbid'):
|
|
prodid = int(show_xml.findtext('tvdbid'))
|
|
tvid = TVINFO_TVDB
|
|
elif None is not show_xml.findtext('id'):
|
|
prodid = int(show_xml.findtext('id'))
|
|
try:
|
|
tvid = TVINFO_TVDB if [s for s in show_xml.findall('.//*')
|
|
if s.text and -1 != s.text.find('thetvdb.com')] else tvid
|
|
except (BaseException, Exception):
|
|
pass
|
|
else:
|
|
logger.log(u"Empty <id> or <tvdbid> field in NFO, unable to find a ID", logger.WARNING)
|
|
return empty_return
|
|
|
|
if None is prodid:
|
|
logger.log(u"Invalid Show ID (%s), not using metadata file" % prodid, logger.WARNING)
|
|
return empty_return
|
|
|
|
except (BaseException, Exception) as e:
|
|
logger.log(
|
|
u"There was an error parsing your existing metadata file: '" + metadata_path + "' error: " + ex(e),
|
|
logger.WARNING)
|
|
return empty_return
|
|
|
|
return tvid, prodid, name
|
|
|
|
def _fanart_urls_from_show(self, show_obj, image_type='banner', lang='en', thumb=False):
|
|
# type: (sickgear.tv.TVShow, AnyStr, AnyStr, bool) -> Optional[List[int, int, AnyStr]]
|
|
try:
|
|
tvdb_id = show_obj.ids.get(indexer_config.TVINFO_TVDB, {}).get('id', None)
|
|
if tvdb_id:
|
|
return self._fanart_urls(tvdb_id, image_type, lang, thumb)
|
|
except (BaseException, Exception):
|
|
pass
|
|
|
|
logger.log(u'Could not find any %s images on Fanart.tv for %s' % (image_type, show_obj.name), logger.DEBUG)
|
|
|
|
@staticmethod
|
|
def _fanart_urls(tvdb_id, image_type='banner', lang='en', thumb=False):
|
|
# type: (int, AnyStr, AnyStr, bool) -> Optional[List[int, int, AnyStr]]
|
|
types = {'poster': fanart.TYPE.TV.POSTER,
|
|
'banner': fanart.TYPE.TV.BANNER,
|
|
'fanart': fanart.TYPE.TV.BACKGROUND,
|
|
'poster_thumb': fanart.TYPE.TV.POSTER,
|
|
'banner_thumb': fanart.TYPE.TV.BANNER}
|
|
|
|
try:
|
|
if tvdb_id:
|
|
request = fanartRequest(apikey=sickgear.FANART_API_KEY, tvdb_id=tvdb_id, types=types[image_type])
|
|
resp = request.response()
|
|
itemlist = []
|
|
dedupe = []
|
|
for art in filter(lambda i: 10 < len(i.get('url', '')) and (lang == i.get('lang', '')[0:2]),
|
|
# remove "[0:2]" ... to strictly use only data where "en" is at source
|
|
resp[types[image_type]]): # type: dict
|
|
try:
|
|
url = (art['url'], art['url'].replace('/fanart/', '/preview/'))[thumb]
|
|
if url not in dedupe:
|
|
dedupe += [url]
|
|
itemlist += [
|
|
[int(art['id']), int(art['likes']), url]
|
|
]
|
|
except (BaseException, Exception):
|
|
continue
|
|
|
|
itemlist.sort(key=lambda a: (a[1], a[0]), reverse=True)
|
|
return itemlist
|
|
|
|
except (BaseException, Exception):
|
|
raise
|
|
|
|
def retrieve_show_image(self, image_type, show_obj, which=None, return_links=False, show_infos=None):
|
|
# type: (AnyStr, sickgear.tv.TVShow, bool, bool, ShowInfosDict) -> Optional[bytes]
|
|
return self._retrieve_show_image(image_type=image_type, show_obj=show_obj, which=which,
|
|
return_links=return_links, show_infos=show_infos)
|
|
|
|
def write_image(self, image_data, image_path, force=False):
|
|
# type: (bytes, AnyStr, bool) -> bool
|
|
return self._write_image(image_data=image_data, image_path=image_path, force=force)
|