mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-11 03:45:01 -05:00
0d9fbc1ad7
This version of SickBeard uses both TVDB and TVRage to search and gather it's series data from allowing you to now have access to and download shows that you couldn't before because of being locked into only what TheTVDB had to offer. Also this edition is based off the code we used in our XEM editon so it does come with scene numbering support as well as all the other features our XEM edition has to offer. Please before using this with your existing database (sickbeard.db) please make a backup copy of it and delete any other database files such as cache.db and failed.db if present, we HIGHLY recommend starting out with no database files at all to make this a fresh start but the choice is at your own risk! Enjoy!
285 lines
12 KiB
Python
285 lines
12 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of Sick Beard.
|
|
#
|
|
# Sick Beard 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.
|
|
#
|
|
# Sick Beard 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 Sick Beard. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os.path
|
|
|
|
import sickbeard
|
|
|
|
from sickbeard import helpers, logger, exceptions
|
|
from sickbeard import encodingKludge as ek
|
|
|
|
from sickbeard.metadata.generic import GenericMetadata
|
|
|
|
from lib.hachoir_parser import createParser
|
|
from lib.hachoir_metadata import extractMetadata
|
|
|
|
class ImageCache:
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def _cache_dir(self):
|
|
"""
|
|
Builds up the full path to the image cache directory
|
|
"""
|
|
return ek.ek(os.path.abspath, ek.ek(os.path.join, sickbeard.CACHE_DIR, 'images'))
|
|
|
|
def _thumbnails_dir(self):
|
|
"""
|
|
Builds up the full path to the thumbnails image cache directory
|
|
"""
|
|
return ek.ek(os.path.abspath, ek.ek(os.path.join, self._cache_dir(), 'thumbnails'))
|
|
|
|
def poster_path(self, indexer_id):
|
|
"""
|
|
Builds up the path to a poster cache for a given indexer id
|
|
|
|
returns: a full path to the cached poster file for the given indexer id
|
|
|
|
indexer_id: ID of the show to use in the file name
|
|
"""
|
|
poster_file_name = str(indexer_id) + '.poster.jpg'
|
|
return ek.ek(os.path.join, self._cache_dir(), poster_file_name)
|
|
|
|
def banner_path(self, indexer_id):
|
|
"""
|
|
Builds up the path to a banner cache for a given indexer id
|
|
|
|
returns: a full path to the cached banner file for the given indexer id
|
|
|
|
indexer_id: ID of the show to use in the file name
|
|
"""
|
|
banner_file_name = str(indexer_id) + '.banner.jpg'
|
|
return ek.ek(os.path.join, self._cache_dir(), banner_file_name)
|
|
|
|
def poster_thumb_path(self, indexer_id):
|
|
"""
|
|
Builds up the path to a poster cache for a given indexer id
|
|
|
|
returns: a full path to the cached poster file for the given indexer id
|
|
|
|
indexer_id: ID of the show to use in the file name
|
|
"""
|
|
posterthumb_file_name = str(indexer_id) + '.poster.jpg'
|
|
return ek.ek(os.path.join, self._thumbnails_dir(), posterthumb_file_name)
|
|
|
|
def banner_thumb_path(self, indexer_id):
|
|
"""
|
|
Builds up the path to a poster cache for a given indexer id
|
|
|
|
returns: a full path to the cached poster file for the given indexer id
|
|
|
|
indexer_id: ID of the show to use in the file name
|
|
"""
|
|
bannerthumb_file_name = str(indexer_id) + '.banner.jpg'
|
|
return ek.ek(os.path.join, self._thumbnails_dir(), bannerthumb_file_name)
|
|
|
|
def has_poster(self, indexer_id):
|
|
"""
|
|
Returns true if a cached poster exists for the given indexer id
|
|
"""
|
|
poster_path = self.poster_path(indexer_id)
|
|
logger.log(u"Checking if file "+str(poster_path)+" exists", logger.DEBUG)
|
|
return ek.ek(os.path.isfile, poster_path)
|
|
|
|
def has_banner(self, indexer_id):
|
|
"""
|
|
Returns true if a cached banner exists for the given indexer id
|
|
"""
|
|
banner_path = self.banner_path(indexer_id)
|
|
logger.log(u"Checking if file "+str(banner_path)+" exists", logger.DEBUG)
|
|
return ek.ek(os.path.isfile, banner_path)
|
|
|
|
def has_poster_thumbnail(self, indexer_id):
|
|
"""
|
|
Returns true if a cached poster thumbnail exists for the given indexer id
|
|
"""
|
|
poster_thumb_path = self.poster_thumb_path(indexer_id)
|
|
logger.log(u"Checking if file "+str(poster_thumb_path)+" exists", logger.DEBUG)
|
|
return ek.ek(os.path.isfile, poster_thumb_path)
|
|
|
|
def has_banner_thumbnail(self, indexer_id):
|
|
"""
|
|
Returns true if a cached banner exists for the given indexer id
|
|
"""
|
|
banner_thumb_path = self.banner_thumb_path(indexer_id)
|
|
logger.log(u"Checking if file "+str(banner_thumb_path)+" exists", logger.DEBUG)
|
|
return ek.ek(os.path.isfile, banner_thumb_path)
|
|
|
|
|
|
BANNER = 1
|
|
POSTER = 2
|
|
BANNER_THUMB = 3
|
|
POSTER_THUMB = 4
|
|
|
|
def which_type(self, path):
|
|
"""
|
|
Analyzes the image provided and attempts to determine whether it is a poster or banner.
|
|
|
|
returns: BANNER, POSTER if it concluded one or the other, or None if the image was neither (or didn't exist)
|
|
|
|
path: full path to the image
|
|
"""
|
|
|
|
if not ek.ek(os.path.isfile, path):
|
|
logger.log(u"Couldn't check the type of "+str(path)+" cause it doesn't exist", logger.WARNING)
|
|
return None
|
|
|
|
# use hachoir to parse the image for us
|
|
img_parser = createParser(path)
|
|
img_metadata = extractMetadata(img_parser)
|
|
|
|
if not img_metadata:
|
|
logger.log(u"Unable to get metadata from "+str(path)+", not using your existing image", logger.DEBUG)
|
|
return None
|
|
|
|
img_ratio = float(img_metadata.get('width'))/float(img_metadata.get('height'))
|
|
|
|
img_parser.stream._input.close()
|
|
|
|
# most posters are around 0.68 width/height ratio (eg. 680/1000)
|
|
if 0.55 < img_ratio < 0.8:
|
|
return self.POSTER
|
|
|
|
# most banners are around 5.4 width/height ratio (eg. 758/140)
|
|
elif 5 < img_ratio < 6:
|
|
return self.BANNER
|
|
else:
|
|
logger.log(u"Image has size ratio of "+str(img_ratio)+", unknown type", logger.WARNING)
|
|
return None
|
|
|
|
def _cache_image_from_file(self, image_path, img_type, indexer_id):
|
|
"""
|
|
Takes the image provided and copies it to the cache folder
|
|
|
|
returns: bool representing success
|
|
|
|
image_path: path to the image we're caching
|
|
img_type: BANNER or POSTER
|
|
indexer_id: id of the show this image belongs to
|
|
"""
|
|
|
|
# generate the path based on the type & indexer_id
|
|
if img_type == self.POSTER:
|
|
dest_path = self.poster_path(indexer_id)
|
|
elif img_type == self.BANNER:
|
|
dest_path = self.banner_path(indexer_id)
|
|
else:
|
|
logger.log(u"Invalid cache image type: "+str(img_type), logger.ERROR)
|
|
return False
|
|
|
|
# make sure the cache folder exists before we try copying to it
|
|
if not ek.ek(os.path.isdir, self._cache_dir()):
|
|
logger.log(u"Image cache dir didn't exist, creating it at "+str(self._cache_dir()))
|
|
ek.ek(os.makedirs, self._cache_dir())
|
|
|
|
if not ek.ek(os.path.isdir, self._thumbnails_dir()):
|
|
logger.log(u"Thumbnails cache dir didn't exist, creating it at "+str(self._thumbnails_dir()))
|
|
ek.ek(os.makedirs, self._thumbnails_dir())
|
|
|
|
logger.log(u"Copying from "+image_path+" to "+dest_path)
|
|
helpers.copyFile(image_path, dest_path)
|
|
|
|
return True
|
|
|
|
def _cache_image_from_tvdb(self, show_obj, img_type):
|
|
"""
|
|
Retrieves an image of the type specified from TVDB and saves it to the cache folder
|
|
|
|
returns: bool representing success
|
|
|
|
show_obj: TVShow object that we want to cache an image for
|
|
img_type: BANNER or POSTER
|
|
"""
|
|
|
|
# generate the path based on the type & indexer_id
|
|
if img_type == self.POSTER:
|
|
img_type_name = 'poster'
|
|
dest_path = self.poster_path(show_obj.indexerid)
|
|
elif img_type == self.BANNER:
|
|
img_type_name = 'banner'
|
|
dest_path = self.banner_path(show_obj.indexerid)
|
|
elif img_type == self.POSTER_THUMB:
|
|
img_type_name = 'poster_thumb'
|
|
dest_path = self.poster_thumb_path(show_obj.indexerid)
|
|
elif img_type == self.BANNER_THUMB:
|
|
img_type_name = 'banner_thumb'
|
|
dest_path = self.banner_thumb_path(show_obj.indexerid)
|
|
else:
|
|
logger.log(u"Invalid cache image type: "+str(img_type), logger.ERROR)
|
|
return False
|
|
|
|
# retrieve the image from TVDB using the generic metadata class
|
|
#TODO: refactor
|
|
metadata_generator = GenericMetadata()
|
|
img_data = metadata_generator._retrieve_show_image(img_type_name, show_obj)
|
|
result = metadata_generator._write_image(img_data, dest_path)
|
|
|
|
return result
|
|
|
|
def fill_cache(self, show_obj):
|
|
"""
|
|
Caches all images for the given show. Copies them from the show dir if possible, or
|
|
downloads them from TVDB if they aren't in the show dir.
|
|
|
|
show_obj: TVShow object to cache images for
|
|
"""
|
|
|
|
logger.log(u"Checking if we need any cache images for show "+str(show_obj.indexerid), logger.DEBUG)
|
|
|
|
# check if the images are already cached or not
|
|
need_images = {self.POSTER: not self.has_poster(show_obj.indexerid),
|
|
self.BANNER: not self.has_banner(show_obj.indexerid),
|
|
self.POSTER_THUMB: not self.has_poster_thumbnail(show_obj.indexerid),
|
|
self.BANNER_THUMB: not self.has_banner_thumbnail(show_obj.indexerid)}
|
|
|
|
if not need_images[self.POSTER] and not need_images[self.BANNER] and not need_images[self.POSTER_THUMB] and not need_images[self.BANNER_THUMB]:
|
|
logger.log(u"No new cache images needed, not retrieving new ones")
|
|
return
|
|
|
|
# check the show dir for poster or banner images and use them
|
|
if need_images[self.POSTER] or need_images[self.BANNER]:
|
|
try:
|
|
for cur_provider in sickbeard.metadata_provider_dict.values():
|
|
logger.log(u"Checking if we can use the show image from the "+cur_provider.name+" metadata", logger.DEBUG)
|
|
if ek.ek(os.path.isfile, cur_provider.get_poster_path(show_obj)):
|
|
cur_file_name = os.path.abspath(cur_provider.get_poster_path(show_obj))
|
|
cur_file_type = self.which_type(cur_file_name)
|
|
|
|
if cur_file_type == None:
|
|
logger.log(u"Unable to retrieve image type, not using the image from "+str(cur_file_name), logger.WARNING)
|
|
continue
|
|
|
|
logger.log(u"Checking if image "+cur_file_name+" (type "+str(cur_file_type)+" needs metadata: "+str(need_images[cur_file_type]), logger.DEBUG)
|
|
|
|
if cur_file_type in need_images and need_images[cur_file_type]:
|
|
logger.log(u"Found an image in the show dir that doesn't exist in the cache, caching it: "+cur_file_name+", type "+str(cur_file_type), logger.DEBUG)
|
|
self._cache_image_from_file(cur_file_name, cur_file_type, show_obj.indexerid)
|
|
need_images[cur_file_type] = False
|
|
except exceptions.ShowDirNotFoundException:
|
|
logger.log(u"Unable to search for images in show dir because it doesn't exist", logger.WARNING)
|
|
|
|
# download from TVDB for missing ones
|
|
for cur_image_type in [self.POSTER, self.BANNER, self.POSTER_THUMB, self.BANNER_THUMB]:
|
|
logger.log(u"Seeing if we still need an image of type "+str(cur_image_type)+": "+str(need_images[cur_image_type]), logger.DEBUG)
|
|
if cur_image_type in need_images and need_images[cur_image_type]:
|
|
self._cache_image_from_tvdb(show_obj, cur_image_type)
|
|
|
|
|
|
logger.log(u"Done cache check")
|