SickGear/sickgear/tv.py

5324 lines
235 KiB
Python
Raw Permalink Normal View History

#
# 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
import random
import weakref
from collections import Counter, OrderedDict
from functools import reduce
from itertools import chain
from weakref import ReferenceType
import datetime
import glob
import inspect
import os.path
import re
import requests
import stat
import threading
import traceback
from imdbpie import ImdbAPIError
from lxml_etree import etree
# noinspection PyPep8Naming
import encodingKludge as ek
import exceptions_helper
from exceptions_helper import ex
import sickgear
from . import db, helpers, history, image_cache, indexermapper, logger, \
name_cache, network_timezones, notifiers, postProcessor, subtitles
from .anime import AniGroupList
from .classes import weakList
from .common import Quality, statusStrings, \
ARCHIVED, DOWNLOADED, FAILED, IGNORED, SKIPPED, SNATCHED, SNATCHED_ANY, SNATCHED_PROPER, UNAIRED, UNKNOWN, WANTED, \
NAMING_DUPLICATE, NAMING_EXTEND, NAMING_LIMITED_EXTEND, NAMING_LIMITED_EXTEND_E_PREFIXED, NAMING_SEPARATED_REPEAT
from .generic_queue import QueuePriorities
from .helpers import try_float, try_int
from .indexermapper import del_mapping, MapStatus, save_mapping
from .indexers.indexer_config import TVINFO_IMDB, TVINFO_TMDB, TVINFO_TRAKT, TVINFO_TVDB, TVINFO_TVMAZE, TVINFO_TVRAGE
from .name_parser.parser import InvalidNameException, InvalidShowException, NameParser
from .sgdatetime import SGDatetime, timestamp_near
from .tv_base import TVEpisodeBase, TVShowBase
from lib import imdbpie, subliminal
from lib.dateutil import tz
from lib.dateutil.parser import parser as du_parser
from lib.fuzzywuzzy import fuzz
from lib.tvinfo_base import RoleTypes, TVINFO_FACEBOOK, TVINFO_INSTAGRAM, TVINFO_SLUG, TVINFO_TWITTER, TVINFO_WIKIPEDIA
from lib.tvinfo_base.exceptions import *
from sg_helpers import calc_age, int_to_time, remove_file_perm, time_to_int
from _23 import filter_iter, filter_list, list_keys
from six import integer_types, iteritems, itervalues, moves, PY2, string_types
# noinspection PyUnreachableCode
if False:
from typing import Any, AnyStr, Dict, List, Optional, Set, Text, Tuple, Union
from sqlite3 import Row
from lib.tvinfo_base import CastList, Character as TVINFO_Character, Person as TVINFO_Person, \
TVInfoEpisode, TVInfoShow
coreid_warnings = False
if coreid_warnings:
import warnings
warnings.simplefilter('always', DeprecationWarning)
tz_p = du_parser()
invalid_date_limit = datetime.date(1900, 1, 1)
# status codes for switching tv show source
TVSWITCH_DUPLICATE_SHOW = 0
TVSWITCH_ID_CONFLICT = 1
TVSWITCH_NO_NEW_ID = 2
TVSWITCH_NORMAL = 3
TVSWITCH_NOT_FOUND_ERROR = 4
TVSWITCH_SAME_ID = 5
TVSWITCH_SOURCE_NOT_FOUND_ERROR = 6
TVSWITCH_VERIFY_ERROR = 7
tvswitch_names = {
TVSWITCH_DUPLICATE_SHOW: 'Duplicate Show',
TVSWITCH_ID_CONFLICT: 'new id conflicts with existing show',
TVSWITCH_NO_NEW_ID: 'no new id found',
TVSWITCH_NORMAL: 'Normal',
TVSWITCH_NOT_FOUND_ERROR: 'Not found on new tv source',
TVSWITCH_SAME_ID: 'unchanged ids given',
TVSWITCH_SOURCE_NOT_FOUND_ERROR: 'source show not found',
TVSWITCH_VERIFY_ERROR: 'Verification error',
}
TVSWITCH_EP_DELETED = 1
TVSWITCH_EP_RENAMED = 2
tvswitch_ep_names = {
TVSWITCH_EP_DELETED: 'deleted',
TVSWITCH_EP_RENAMED: 'renamed'
}
concurrent_show_not_found_days = 7
show_not_found_retry_days = 7
prodid_bitshift = 4
tvid_bitmask = (1 << prodid_bitshift) - 1
class TVidProdid(object):
"""
Helper class to standardise the use of a TV info source id with its associated TV show id
Examples of setting and using on the one action, tvid and prodid are numbers of type int or string;
TVidProdid({tvid: prodid})
TVidProdid({tvid: prodid})([])
TVidProdid({tvid: prodid})(list)
TVidProdid({tvid: prodid})('list')
TVidProdid({tvid: prodid}).list
Whitespace may exist between the number value and <self.glue> in these strings
TVidProdid('tvid <self.glue> prodid')({})
TVidProdid('tvid <self.glue> prodid')(dict)
TVidProdid('tvid <self.glue> prodid')('dict')
TVidProdid('tvid <self.glue> prodid').dict
"""
glue = ':'
def __init__(self, tvid_prodid=None):
"""
:param tvid_prodid: TV info data source ID, and TV show ID
:type tvid_prodid: Dict or String
"""
self.tvid = None
self.prodid = None
self.sid_int = None
if isinstance(tvid_prodid, dict) and 1 == len(tvid_prodid):
try:
for (tvid, prodid) in iteritems(tvid_prodid):
self.tvid, self.prodid = int(tvid), int(prodid)
except ValueError:
pass
elif isinstance(tvid_prodid, string_types):
if self.glue in tvid_prodid:
try:
for (tvid, prodid) in [re.findall(r'(\d+)\s*%s\s*(\d+)' % self.glue, tvid_prodid)[0]]:
self.tvid, self.prodid = int(tvid), int(prodid)
except IndexError:
pass
else:
try:
legacy_showid = int(re.findall(r'(?i)\d+', tvid_prodid)[0])
show_obj = (helpers.find_show_by_id({TVINFO_TVDB: legacy_showid})
or helpers.find_show_by_id({TVINFO_TVRAGE: legacy_showid}))
pre_msg = 'Todo: Deprecate show_id used without a tvid'
if None is show_obj:
pre_msg += ' and show_obj not found'
else:
self.tvid, self.prodid = show_obj.tvid, legacy_showid
if coreid_warnings:
logger.log('%s\n' % pre_msg +
'|>%s^-- Note: Bootstrap & Tornado startup functions stripped from traceback log.' %
'|>'.join(filter_iter(lambda text: not re.search(r'(?i)bootstrap|traceback\.'
r'format_stack|pydevd|tornado'
r'|webserveinit', text),
traceback.format_stack(inspect.currentframe()))))
except IndexError:
pass
elif isinstance(tvid_prodid, integer_types):
self.tvid = tvid_prodid & tvid_bitmask
self.prodid = tvid_prodid >> prodid_bitshift
self.sid_int = tvid_prodid
return
if None not in (self.prodid, self.tvid):
self.sid_int = self.prodid << prodid_bitshift | self.tvid
def __call__(self, *args, **kwargs):
return self._get(*args, **kwargs)
def __repr__(self):
return self._get()
def __str__(self):
return self._get()
def __tuple__(self):
return self._get(tuple)
def __int__(self):
return self._get(int)
@property
def list(self):
return self._get(list)
@property
def dict(self):
return self._get(dict)
@property
def tuple(self):
return self._get(tuple)
@property
def int(self):
return self._get(int)
@staticmethod
def _checktype(value, t):
if isinstance(value, string_types) and not isinstance(value, type) and value:
if value == getattr(t, '__name__', None):
return True
elif (isinstance(value, type) and value == t) or isinstance(value, t):
return True
return False
def _get(self, kind=None):
if None is not kind:
if self._checktype(kind, int):
return self.sid_int
elif self._checktype(kind, dict):
return {self.tvid: self.prodid}
elif self._checktype(kind, tuple):
return self.tvid, self.prodid
if None is kind or self._checktype(kind, string_types):
return '%s%s%s' % (self.tvid, self.glue, self.prodid)
return [self.tvid, self.prodid]
@property
def _tvid(self):
return self.tvid
@property
def _prodid(self):
return self.prodid
class PersonGenders(object):
UNKNOWN = 0
male = 1
female = 2
possible_values = [0, 1, 2]
names = {UNKNOWN: 'Unknown', male: 'Male', female: 'Female'}
tmdb_map = {0: UNKNOWN, 1: female, 2: male}
def _make_date(dt):
# type: (datetime.date) -> integer_types
if not dt:
return None
return dt.toordinal()
def usable_id(value):
# type: (Union[AnyStr, int]) -> Optional[AnyStr, int]
"""
return value as int if value is numerical,
or value as string if value is valid id:format
otherwise None
"""
value_id = try_int(value, None)
if None is not value_id:
return value_id
return usable_rid(value)
def usable_rid(value):
# type: (Union[AnyStr]) -> Optional[AnyStr]
"""
return value if is a id:format is valid
otherwise None if value fails basic id format validation
"""
if isinstance(value, string_types) and ':' in value:
temp = value.split(':')
if 2 == len(temp) and None not in [try_int(_x, None) for _x in temp]:
return value
class Referential(object):
"""
This class superimposes handling for a string based ID named `ref_id`.
__init__ will pass back control untouched to the object deriving
from this one if an integer is passed so that integer ID can perform
"""
def __init__(self, sid=None):
self.id = None # type: integer_types
self.ids = {} # type: Dict[int, integer_types]
self._rid = usable_rid(sid)
def has_ref_id(self, item):
if isinstance(item, integer_types):
return self.id == item
tvid, src_id = self.ref_id(rid=item, string=False)
return self.ids.get(tvid) == src_id
def ref_id(self,
rid=None, # type: Optional[AnyStr, int]
string=True # type: Optional[bool]
): # type: (...) -> Union[tuple[Optional[AnyStr, int], Optional[AnyStr, int]], AnyStr]
"""
return either,
1) a preferred external unique id for a consistent reliable `reference`, or the internal row id as fallback
2) convert a consistent `reference` to src and src_id to use for matching to an internal row id
uses param rid or self._rid as reference id to convert if either is not Nonetype
for example,
1) if string is False, preferred static tuple[tvid, id], or tuple[None, id] if ids not set
2) otherwise, preferred static 'tvid:id', or 'id' if ids not set
3) if string is False and self._rid contains ':', list[src, src_id]
4) otherwise if self._rid contains ':', list['src', 'src_id']
reason, a row id is highly volatile, but a preferred ID is more reliable.
use cases,
show removed and readded, or src switched, then refreshing any character/person views will be consistent.
can share/save a character/person link/bookmark as it will be consistent across instances.
cast glide startAt is unreliable using glide index or row id.
"""
# if self._rid is a temporary type string containing ':' (use case during init)
# return it as a tuple[int, int] -> src, src_id
rid = rid or self._rid
if usable_rid(rid):
self._rid = None # consume once
parts = [try_int(_x, None) for _x in rid.split(':')]
# if string: no usage for this case
# ? return rid
# ? return '%s' % parts[0], '%s' % parts[1]
return parts[0], parts[1]
# get an available id from an order of preferred ids for use
if self.ids:
for cur_tvid in [TVINFO_IMDB, TVINFO_TVMAZE, TVINFO_TMDB, TVINFO_TRAKT]:
if self.ids.get(cur_tvid):
if string:
return '%s:%s' % (cur_tvid, self.ids.get(cur_tvid))
return cur_tvid, self.ids.get(cur_tvid)
if string:
return '%s' % self.id
return None, self.id # default internal id has no slug
class Person(Referential):
def __init__(
self,
name=None, # type: AnyStr
gender=PersonGenders.UNKNOWN, # type: int
birthday=None, # type: datetime.date
deathday=None, # type: datetime.date
bio=None, # type: AnyStr
ids=None, # type: Dict[int, integer_types]
sid=None, # type: integer_types
birthplace=None, # type: AnyStr
homepage=None, # type: AnyStr
image_url=None, # type: AnyStr
thumb_url=None, # type: AnyStr
show_obj=None, # type: TVShow
updated=1, # type: integer_types
deathplace=None, # type: AnyStr
height=None, # type: Union[integer_types, float]
real_name=None, # type: AnyStr
nicknames=None, # type: Set[AnyStr]
akas=None, # type: Set[AnyStr]
character_obj=None, # type: Character
tmp_character_obj=None # type: Character
): # type: (...) -> Person
super(Person, self).__init__(sid)
self.updated = updated # type: integer_types
self._data_failure = False # type: bool
self._data_fetched = False # type: bool
self.name = name # type: AnyStr
self.character_obj = character_obj # type: Optional[Character]
self._tmp_character_obj = tmp_character_obj # type: Optional[Character]
self.gender = (PersonGenders.UNKNOWN, gender)[gender in PersonGenders.possible_values] # type: int
self.birthday = birthday # type: Optional[datetime.date]
self.deathday = deathday # type: Optional[datetime.date]
self.biography = bio # type: Optional[AnyStr]
self.birthplace = birthplace # type: Optional[AnyStr]
self.homepage = homepage # type: Optional[AnyStr]
self.image_url = image_url # type: Optional[AnyStr]
self.thumb_url = thumb_url # type: Optional[AnyStr]
self.deathplace = deathplace # type: Optional[AnyStr]
self.height = height # type: Optional[Union[integer_types, float]]
self.real_name = real_name # type: Optional[AnyStr]
self.nicknames = nicknames or set() # type: Set[AnyStr]
self.akas = akas or set() # type: Set[AnyStr]
self.ids = ids or {} # type: Dict[int, integer_types]
if not self._rid:
self.id = sid or self._get_sid(show_obj=show_obj) # type: integer_types
new_dirty = not sid
self.dirty_main = new_dirty # type: bool
self.dirty_ids = new_dirty # type: bool
if not sid and self.id:
fetched = self._data_fetched
cur_data = self._remember_properties()
self.load_from_db()
cur_data['ids'] = dict(chain.from_iterable(iteritems(d) for d in (self.ids, ids or {})))
self.update_properties(**cur_data)
self._data_fetched = fetched
elif not self.name:
self.load_from_db()
@staticmethod
def _order_names(name_set):
# type: (Set[AnyStr]) -> List[AnyStr]
rc_aka = re.compile(r'[\x00-\x7f]+')
aka_all_ascii = []
aka_non_ascii = []
for cur_aka in name_set or []:
if rc_aka.match(cur_aka):
aka_all_ascii += [cur_aka]
else:
aka_non_ascii += [cur_aka]
return aka_all_ascii + aka_non_ascii
@property
def lang_ordered_akas(self):
# type: (...) -> List[AnyStr]
return self._order_names(self.akas)
@property
def lang_ordered_nicknames(self):
# type: (...) -> List[AnyStr]
return self._order_names(self.nicknames)
def _remember_properties(self):
# type: (...) -> Dict
return {k: self.__dict__[k] for k in
['akas', 'biography', 'birthday', 'birthplace', 'deathday', 'deathplace', 'gender', 'height',
'homepage', 'ids', 'image_url', 'name', 'nicknames', 'real_name', 'thumb_url']}
def reset(self, person_obj=None):
# type: (TVINFO_Person) -> None
"""
reset all properties with the exception of: name, id, ids
:param person_obj: TVInfo Person object to reset to
"""
self._data_failure = False
self._data_fetched = False
self.akas = person_obj.akas or set()
self.biography = person_obj.bio or None
self.birthday = person_obj.birthdate or None
self.birthplace = person_obj.birthplace or None
self.deathday = person_obj.deathdate or None
self.deathplace = person_obj.deathplace or None
self.dirty_main = True
self.gender = person_obj.gender or PersonGenders.UNKNOWN
self.height = person_obj.height or None
self.homepage = person_obj.homepage or None
self.image_url = person_obj.image or None
self.nicknames = person_obj.nicknames or set()
self.real_name = person_obj.real_name or None
self.thumb_url = person_obj.thumb_url or None
self.updated = 1
def _get_sid(self, mass_action_result=None, show_obj=None):
# type: (List, TVShow) -> Optional[integer_types]
if getattr(self, 'id', None):
return self.id
def _get_from_ids():
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT person_id
FROM person_ids
WHERE %s
""" % ' OR '.join(['( src = ? AND src_id = ? )'] * len(self.ids)),
list(reduce(lambda a, b: a + b, iteritems(self.ids))))
for cur_id in sql_result or []:
return cur_id['person_id']
if mass_action_result:
try:
return mass_action_result[-1:][0][0]['last_rowid']
except (BaseException, Exception):
return
elif self.ids:
i = _get_from_ids()
if i:
return i
self.get_missing_data(show_obj=show_obj)
if self.ids:
return _get_from_ids()
def _set_updated(self):
# type: (...) -> integer_types
"""
set and return new update date
"""
self.updated = datetime.date.today().toordinal() - (0, 120)[self._data_failure]
return self.updated
def remove_img(self):
"""
removes image of person
"""
try:
remove_file_perm(image_cache.ImageCache().person_path(self))
except (BaseException, Exception):
pass
def remove_thumb(self):
"""
removes thumb image of person
"""
try:
remove_file_perm(image_cache.ImageCache().person_thumb_path(self))
except (BaseException, Exception):
pass
def remove_all_img(self):
"""
remove all images of person
"""
try:
for r in image_cache.ImageCache().person_both_paths(self):
try:
remove_file_perm(r)
except (BaseException, Exception):
pass
except (BaseException, Exception):
pass
def update_properties(self, **kwargs):
"""
updates properties of class, if changed sets dirty
:param kwargs: properties of class
"""
if (not self.image_url and not self.thumb_url) or (not self.image_url and kwargs.get('image_url')):
if kwargs.get('image_url'):
self.dirty_main = True
self.remove_img()
self.image_url = kwargs.get('image_url')
# thumb changed, remove old image
if kwargs.get('thumb_url'):
self.remove_thumb()
self.thumb_url = kwargs.get('thumb_url')
else:
self.remove_thumb()
self.thumb_url = None
elif kwargs.get('thumb_url'):
self.dirty_main = True
self.thumb_url = kwargs.get('thumb_url')
# image changed remove old image
self.remove_img()
if kwargs.get('image_url'):
self.remove_img()
self.image_url = kwargs.get('image_url')
else:
self.remove_img()
self.image_url = None
for k, v in iteritems(kwargs):
if k in ('image_url', 'thumb_url'):
continue
if k not in self.__dict__:
raise Exception('Person has no property [%s]' % k)
if None is not v:
if 'akas' == k:
v.update(self.akas)
elif 'nicknames' == k:
v.update(self.nicknames)
if v != self.__dict__[k]:
if 'biography' == k:
if len(v) > len(self.biography or ''):
self.biography = v
self.dirty_main = True
elif 'gender' == k and v not in (PersonGenders.male, PersonGenders.female):
continue
else:
self.__dict__[k] = v
if 'ids' == k:
self.dirty_ids = True
else:
self.dirty_main = True
def load_from_db(self):
if self.id or usable_rid(self._rid):
my_db = db.DBConnection()
src, src_id = self.ref_id(string=False)
sql_result = my_db.select(
"""
SELECT persons.*,
(SELECT group_concat(person_ids.src || ':' || person_ids.src_id, ';;;')
FROM person_ids
WHERE person_ids.person_id = persons.id) AS p_ids
FROM persons
LEFT JOIN person_ids pi ON persons.id = pi.person_id
WHERE %s
""" % ('pi.src = ? and pi.src_id = ?', 'persons.id = ?')[not src],
([src, src_id], [self.id])[not src])
for cur_person in sql_result or []:
birthdate = try_int(cur_person['birthdate'], None)
birthdate = birthdate and datetime.date.fromordinal(cur_person['birthdate'])
deathdate = try_int(cur_person['deathdate'], None)
deathdate = deathdate and datetime.date.fromordinal(cur_person['deathdate'])
p_ids = {}
for cur_ids in (cur_person['p_ids'] and cur_person['p_ids'].split(';;;')) or []:
k, v = cur_ids.split(':')
k = try_int(k, None)
if v and None is not k:
p_ids[k] = v if k in (TVINFO_FACEBOOK, TVINFO_INSTAGRAM, TVINFO_TWITTER, TVINFO_WIKIPEDIA) \
else try_int(v, None)
(self._data_failure, self._data_fetched,
self.akas, self.biography,
self.birthday, self.birthplace, self.deathday, self.deathplace,
self.gender, self.height, self.homepage,
self.id, self.ids, self.image_url, self.name,
self.nicknames,
self.real_name, self.thumb_url, self.updated) = \
(False, False,
set((cur_person['akas'] and cur_person['akas'].split(';;;')) or []), cur_person['bio'],
birthdate, cur_person['birthplace'], deathdate, cur_person['deathplace'],
cur_person['gender'], cur_person['height'], cur_person['homepage'],
cur_person['id'], p_ids, cur_person['image_url'], cur_person['name'],
set((cur_person['nicknames'] and cur_person['nicknames'].split(';;;')) or []),
cur_person['realname'], cur_person['thumb_url'], cur_person['updated'])
break
def update_prop_from_tvinfo_person(self, person_obj):
# type: (TVINFO_Person) -> None
"""
update person with tvinfo person object info
Note: doesn't change: name, id, image_url, thumb_url
:param person_obj: TVInfo Person object
"""
if person_obj:
char_obj = self.character_obj or self._tmp_character_obj
if person_obj.characters and char_obj and char_obj.show_obj \
and char_obj.show_obj.ids.get(TVINFO_IMDB, {}).get('id'):
p_char = [pc for pc in person_obj.characters
if pc.show.ids.imdb == char_obj.show_obj.ids.get(TVINFO_IMDB, {}).get('id')]
p_count = len(p_char)
if 1 == p_count:
if (p_char[0].start_year or p_char[0].end_year) and getattr(self, 'id', None):
char_obj.combine_start_end_years(
{self.id: {'start': p_char[0].start_year, 'end': p_char[0].end_year}})
elif 1 < p_count:
start, end, found, p_p = None, None, False, None
for c_p in p_char:
if not start or start > (c_p.start_year or 3000):
start = c_p.start_year
if not end or end < (c_p.end_year or 0):
end = c_p.end_year
if c_p.name == char_obj.name:
start = c_p.start_year
end = c_p.end_year
found = True
break
if not found:
try:
p_p = sorted([
(fuzz.UWRatio(char_obj.name, p.name),
p.end_year and p.start_year and p.end_year - p.start_year, p) for p in p_char],
key=lambda a: (a[0], a[1]), reverse=True)[0][2]
except (BaseException, Exception):
p_p = None
if None is not p_p and (p_p.start_year or p_p.end_year):
start = p_p.start_year
end = p_p.end_year
if (start or end) and getattr(self, 'id', None):
char_obj.combine_start_end_years({self.id: {'start': start, 'end': end}})
self.update_properties(
gender=person_obj.gender, homepage=person_obj.homepage,
ids=dict(chain.from_iterable(
iteritems(d) for d in (self.ids, {k: v for k, v in iteritems(person_obj.ids)
if v and TVINFO_SLUG != k} or {}))),
birthday=person_obj.birthdate, deathday=person_obj.deathdate, biography=person_obj.bio,
birthplace=person_obj.birthplace, deathplace=person_obj.deathplace, height=person_obj.height,
real_name=person_obj.real_name, nicknames=person_obj.nicknames, akas=person_obj.akas
)
def get_missing_data(self, show_obj=None, character_obj=None, stop_event=None, force_id=False):
# type: (TVShow, Character, threading.Event, bool) -> None
"""
load missing data for person from trakt + tmdb
:param show_obj:
:param character_obj:
:param stop_event:
:param force_id: refetch all external ids, normally only source ids and missing are fetched
"""
if character_obj and not self.character_obj:
self.character_obj = character_obj
if self._data_failure:
self.dirty_main = True
self._data_failure = False
self._data_fetched = True
tvsrc_result, found_persons, found_on_src, search_sources, found_ids, ids_to_check, tv_loop, imdb_confirmed, \
source_confirmed = \
None, {}, set(), [TVINFO_TRAKT, TVINFO_TMDB, TVINFO_IMDB], \
set([k for k, v in iteritems(self.ids) if v] + ['text']), {}, 0, False, {}
# confirmed_character = False
max_search_src = len(search_sources)
logger.debug('Getting extra data for: %s' % self.name)
for tv_loop in moves.xrange(0, max_search_src):
search_sources = [s for s in search_sources if s not in found_on_src]
for cur_tv_info_src in search_sources:
if stop_event and stop_event.is_set():
return
rp, confirmed_on_src = None, False
tvinfo_config = sickgear.TVInfoAPI(cur_tv_info_src).api_params.copy()
t = sickgear.TVInfoAPI(cur_tv_info_src).setup(**tvinfo_config)
if 0 == tv_loop or cur_tv_info_src not in ids_to_check:
ids_to_check[cur_tv_info_src] = t.supported_person_id_searches + ['text']
found_persons[cur_tv_info_src] = set()
for tv_src in ids_to_check[cur_tv_info_src]:
if 'text' != tv_src and not self.ids.get(tv_src):
continue
try:
if 'text' != tv_src:
kw = {'ids': {tv_src: self.ids[tv_src]}}
else:
kw = {'name': self.name}
tvsrc_result = t.search_person(**kw)
except (BaseException, Exception):
self._data_failure = True
continue
if tvsrc_result:
# verify we have the correct person
for cur_person in tvsrc_result: # type: TVINFO_Person
if None is not rp:
break
if not (imdb_confirmed and TVINFO_IMDB == tv_src) \
and cur_person.id in found_persons[cur_tv_info_src]:
continue
found_persons[cur_tv_info_src].add(cur_person.id)
try:
pd = t.get_person(cur_person.ids.get(cur_tv_info_src),
get_show_credits=True, get_images=True)
except (BaseException, Exception) as e:
self._data_failure = True
logger.warning('Error searching extra info for person: %s - %s' % (self.name, ex(e)))
continue
if None is not pd and imdb_confirmed and TVINFO_IMDB == tv_src:
rp = pd
break
# noinspection PyUnresolvedReferences
if show_obj and None is not pd and pd.characters:
clean_show_name = indexermapper.clean_show_name(show_obj.name.lower())
for ch in pd.characters or []: # type: TVINFO_Character
if clean_show_name == indexermapper.clean_show_name(ch.show.seriesname.lower()):
rp = pd
confirmed_on_src = True
# confirmed_character = True
break
elif any(t_src == s_src and bool(t_id) and t_id == s_id['id']
for t_src, t_id in iteritems(ch.show.ids)
for s_src, s_id in iteritems(show_obj.ids)):
rp = pd
confirmed_on_src = True
# confirmed_character = True
break
if rp:
if TVINFO_IMDB == cur_tv_info_src and confirmed_on_src:
imdb_confirmed = True
found_on_src.add(cur_tv_info_src)
break
ids_to_check[cur_tv_info_src] = [i for i in ids_to_check[cur_tv_info_src] if i not in found_ids]
if None is not rp:
if confirmed_on_src:
for i in (TVINFO_TRAKT, TVINFO_IMDB, TVINFO_TMDB, TVINFO_TVMAZE, TVINFO_TVDB):
# in case it's the current source use it's id and lock if from being changed
if cur_tv_info_src == i and rp.ids.get(i):
source_confirmed[i] = True
if rp.ids.get(i) != self.ids.get(i):
self.ids[i] = rp.ids[i]
self.dirty_ids = True
if not source_confirmed.get(i) and \
(rp.ids.get(i) and not self.ids.get(i) or
(force_id and rp.ids.get(i) and rp.ids.get(i) != self.ids.get(i))):
self.ids[i] = rp.ids[i]
found_ids.add(i)
self.dirty_ids = True
for i in (TVINFO_INSTAGRAM, TVINFO_TWITTER, TVINFO_FACEBOOK, TVINFO_WIKIPEDIA):
if rp.social_ids.get(i) and not self.ids.get(i) or \
(rp.social_ids.get(i) and rp.social_ids.get(i) != self.ids.get(i)):
self.ids[i] = rp.social_ids[i]
found_ids.add(i)
self.dirty_ids = True
self.update_prop_from_tvinfo_person(rp)
if not self.image_url and rp.image:
self.image_url = rp.image
if rp.thumb_url:
self.thumb_url = rp.thumb_url
else:
self.remove_thumb()
self.thumb_url = None
self.dirty_main = True
elif not self.thumb_url and rp.thumb_url:
self.thumb_url = rp.thumb_url
if not rp.image:
self.remove_img()
self.image_url = None
self.dirty_main = True
def save_to_db(self, show_obj=None, character_obj=None, force=False, stop_event=None):
# type: (TVShow, Character, bool, threading.Event) -> None
if not self._data_fetched:
self.get_missing_data(show_obj=show_obj, character_obj=character_obj, stop_event=stop_event)
self._data_fetched = False
if not any(d for d in (self.dirty_main, self.dirty_ids)):
return
my_db = db.DBConnection()
if not self.id:
cl = [[
"""
INSERT INTO persons (akas, bio, birthdate, birthplace,
deathdate, deathplace, gender, height, homepage,
image_url, name, nicknames, realname, thumb_url,
updated)
VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?);
""", [';;;'.join(self.akas or []), self.biography, _make_date(self.birthday), self.birthplace,
_make_date(self.deathday), self.deathplace, self.gender, self.height, self.homepage,
self.image_url, self.name, ';;;'.join(self.nicknames or []), self.real_name, self.thumb_url,
self._set_updated()]
]]
c_ids = len(self.ids)
if c_ids:
f_d = next(iter(self.ids))
cl.extend([[
"""
INSERT INTO person_ids (src, src_id, person_id)
VALUES (?,?,last_insert_rowid())
""", [f_d, self.ids[f_d]]]
])
if 1 < c_ids:
cl.extend([[
"""
INSERT INTO person_ids (src, src_id, person_id)
VALUES (?,?,
(SELECT person_ids.person_id FROM person_ids
WHERE person_ids.id = last_insert_rowid())
);
""", [s, self.ids[s]]] for s in list(self.ids)[1:]
])
cl.extend([[
"""
SELECT person_ids.person_id AS last_rowid
FROM person_ids
WHERE person_ids.id = last_insert_rowid()
"""]
])
else:
cl.extend([['SELECT last_insert_rowid() as last_rowid;']])
else:
cl = []
if force or self.dirty_main:
cl = [[
"""
UPDATE persons
SET name = ?, gender = ?, birthdate = ?, deathdate = ?, bio = ?,
birthplace = ?, homepage = ?, image_url = ?, thumb_url = ?, updated = ?,
deathplace = ?, height = ?, realname = ?, nicknames = ?, akas = ?
WHERE id = ?
""",
[self.name, self.gender, _make_date(self.birthday), _make_date(self.deathday), self.biography,
self.birthplace, self.homepage, self.image_url, self.thumb_url,
self._set_updated(), self.deathplace, self.height, self.real_name,
';;;'.join(self.nicknames or []), ';;;'.join(self.akas or []), self.id]],
]
if force or self.dirty_ids:
for s, v in iteritems(self.ids):
cl.extend([
['UPDATE person_ids SET src_id = ? WHERE person_id = ? AND src = ?', [v, self.id, s]],
["INSERT INTO person_ids (src, src_id, person_id) SELECT %s, '%s', %s WHERE changes() == 0"
% (s, v, self.id)]
])
if cl:
r_id = my_db.mass_action(cl)
if r_id and r_id[-1:][0]:
self.id = self._get_sid(r_id)
self.dirty_main, self.dirty_ids = False, False
def calc_age(self, date=None):
# type: (Optional[datetime.date]) -> Optional[int]
"""
returns age based on current date or given date
:param date:
"""
return calc_age(self.birthday, self.deathday, date)
@property
def age(self):
# type: (...) -> Optional[int]
"""
:return: age of person if birthdate is known, in case of deathdate is known return age of death
"""
return self.calc_age()
def __bool__(self):
return bool(self.name)
def __eq__(self, other):
return (self.id not in [None, 0] and other.id == self.id) \
and any(self.ids[o] == v for o, v in iteritems(other.ids))
def __str__(self):
lived, id_str, id_list = '', '', []
if self.birthday:
lived += ' %s' % self.birthday
if self.deathday:
lived += ' - %s' % self.deathday
for i, d in iteritems(self.ids):
id_list.append('%s: %s' % (sickgear.TVInfoAPI(i).name, d))
if id_list:
id_str = ' (%s)' % ', '.join(id_list)
return '<Person %s%s%s>' % (self.name, lived, id_str)
def __repr__(self):
return self.__str__()
__nonzero__ = __bool__
class Character(Referential):
def __init__(
self,
name=None, # type: AnyStr
person=None, # type: List[Person]
ids=None, # type: Dict[int, integer_types]
bio=None, # type: AnyStr
sid=None, # type: integer_types
image_url=None, # type: AnyStr
thumb_url=None, # type: AnyStr
show_obj=None, # type: TVShow
updated=1, # type: integer_types
persons_years=None, # type: Dict[integer_types, Dict[AnyStr, int]]
tmp=False # type:bool
): # type: (...) -> Character
super(Character, self).__init__(sid)
self.updated = updated # type: integer_types
self.name = name # type: AnyStr
self.persons_years = persons_years or {} # type: Dict[integer_types, Dict[AnyStr, int]]
self.person = person or [] # type: List[Person]
self.biography = bio # type: Optional[AnyStr]
self.ids = ids or {} # type: Dict[int, integer_types]
self.image_url = image_url # type: Optional[AnyStr]
self.thumb_url = thumb_url # type: Optional[AnyStr]
self.show_obj = show_obj # type: TVShow
if not tmp:
if not self._rid:
self.id = sid or self._get_sid() # type: integer_types
new_dirty = not sid
self.dirty_main = new_dirty # type: bool
self.dirty_ids = new_dirty # type: bool
self.dirty_person = new_dirty # type: bool
self.dirty_years = new_dirty # type: bool
if not sid and self.id:
self.load_from_db()
self.update_properties(name=name, person=person, biography=bio,
ids=dict(chain.from_iterable(iteritems(d) for d in (self.ids, ids or {}))),
image_url=image_url, thumb_url=thumb_url)
elif not self.name:
self.load_from_db()
def _get_sid(self, mass_action_result=None):
# type: (List) -> Optional[integer_types]
if getattr(self, 'id', None):
return self.id
if mass_action_result:
try:
return mass_action_result[-1:][0][0]['last_rowid']
except (BaseException, Exception):
return
elif self.ids:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT character_id
FROM character_ids
WHERE %s
""" % ' OR '.join(['( src = ? AND src_id = ? )'] * len(self.ids)),
list(reduce(lambda a, b: a+b, iteritems(self.ids))))
if sql_result:
return sql_result[0]['character_id']
elif self.person:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT characters.id AS c_id
FROM characters
LEFT JOIN character_person_map cpm ON characters.id = cpm.character_id
WHERE name = ? AND cpm.person_id IN (%s)
""" % ','.join(['?' * len(self.person)]),
[self.name] + [p.id for p in self.person])
if sql_result:
return sql_result[0]['c_id']
def _set_updated(self):
# type: (...) -> integer_types
"""
set and return new update date
"""
self.updated = datetime.date.today().toordinal()
return self.updated
def remove_img(self):
"""
removes image of person
"""
try:
remove_file_perm(image_cache.ImageCache().character_path(self, self.show_obj))
except (BaseException, Exception):
pass
def remove_thumb(self):
"""
removes thumb image of person
"""
try:
remove_file_perm(image_cache.ImageCache().character_thumb_path(self, self.show_obj))
except (BaseException, Exception):
pass
def remove_all_img(self, include_person=False, tvid=None, proid=None):
# type: (bool, integer_types, integer_types) -> None
"""
remove all images of person
"""
try:
for r in image_cache.ImageCache().character_both_path(self, show_obj=self.show_obj, tvid=tvid, proid=proid):
try:
remove_file_perm(r)
except (BaseException, Exception):
pass
except (BaseException, Exception):
pass
if include_person:
for p in self.person or []:
try:
p.remove_all_img()
except (BaseException, Exception):
pass
def update_properties(self, **kwargs):
for k, v in iteritems(kwargs):
if k not in self.__dict__:
raise Exception('Character has no property [%s]' % k)
if None is not v and v != self.__dict__[k]:
if 'image_url' == k:
self.dirty_main = True
self.remove_img()
if not kwargs.get('thumb_url'):
self.remove_thumb()
self.thumb_url = None
elif 'thumb_url' == k:
self.dirty_main = True
self.remove_thumb()
if not kwargs.get('image_url'):
self.remove_img()
self.image_url = None
self.__dict__[k] = v
if 'ids' == k:
self.dirty_ids = True
elif 'person' == k:
self.dirty_person = True
else:
self.dirty_main = True
def combine_start_end_years(self, new_years):
# type: (Dict[integer_types, Dict[AnyStr, int]]) -> None
new_dict = dict(chain.from_iterable(
iteritems(d) for d in (self.persons_years, {k: v for k, v in iteritems(new_years) if v} or {})))
if new_dict != self.persons_years:
self.persons_years = new_dict
self.dirty_years = True
def load_from_db(self):
if self.id or usable_rid(self._rid):
my_db = db.DBConnection()
src, src_id = self.ref_id(string=False)
sql_result = my_db.select(
"""
SELECT characters.name AS name, characters.bio AS bio,
characters.thumb_url AS thumb_url, characters.image_url AS image_url,
characters.updated AS c_updated, characters.id AS c_id,
(SELECT group_concat(character_ids.src || ':' || character_ids.src_id, ';;;')
FROM character_ids WHERE character_ids.character_id = characters.id) AS c_ids,
(SELECT group_concat(character_person_map.person_id, ';;;')
FROM character_person_map
WHERE character_person_map.character_id = characters.id) AS person_ids,
(SELECT group_concat(character_person_years.person_id || ':' ||
character_person_years.start_year || '-' || character_person_years.end_year, ';;;')
FROM character_person_years WHERE character_person_years.character_id = characters.id)
AS p_years
FROM characters
LEFT JOIN character_ids ci ON characters.id = ci.character_id
WHERE %s
""" % ('ci.src = ? and ci.src_id = ?', 'characters.id = ?')[not src],
([src, src_id], [self.id])[not src])
for cur_row in (sql_result or []):
c_ids = {}
for cur_ids in (cur_row['c_ids'] and cur_row['c_ids'].split(';;;')) or []:
k, v = cur_ids.split(':')
v = try_int(v, None)
if v:
c_ids[int(k)] = try_int(v, None)
p_years = {}
for cur_years in (cur_row['p_years'] and cur_row['p_years'].split(';;;')) or []:
p_id, py = cur_years.split(':')
start, end = py.split('-')
p_years[int(p_id)] = {'start': try_int(start, None), 'end': try_int(end, None)}
(self.biography, self.id, self.ids, self.image_url, self.name,
self.person,
self.persons_years, self.thumb_url, self.updated) = \
(cur_row['bio'], cur_row['c_id'], c_ids, cur_row['image_url'], cur_row['name'],
[Person(sid=int(p), character_obj=self)
for p in (cur_row['person_ids'] and cur_row['person_ids'].split(';;;')) or []],
p_years, cur_row['thumb_url'], cur_row['c_updated'])
self.dirty_main, self.dirty_ids, self.dirty_person = False, False, False
break
def _get_start_end_year(self):
if self.person:
tvinfo_config = sickgear.TVInfoAPI(TVINFO_IMDB).api_params.copy()
t = sickgear.TVInfoAPI(TVINFO_IMDB).setup(**tvinfo_config)
for cur_per in self.person:
if not cur_per.id:
continue
if cur_per.ids.get(TVINFO_IMDB):
try:
res = t.get_person(cur_per.ids.get(TVINFO_IMDB), get_show_credits=True)
except (BaseException, Exception):
continue
if res and res.characters:
cur_per.character_obj = self
cur_per.update_prop_from_tvinfo_person(res)
def save_to_db(self, show_obj=None, force=False, stop_event=None):
# type: (TVShow, bool, threading.Event) -> None
if self.person:
for cur_per in self.person:
if stop_event and stop_event.is_set():
return
if force or self.dirty_person or not cur_per.id or cur_per.dirty_main or cur_per.dirty_ids or \
14 < (datetime.date.today().toordinal() - cur_per.updated):
cur_per.save_to_db(show_obj=show_obj, character_obj=self, force=force, stop_event=stop_event)
self.dirty_person = True
if ((None is self.id and not self.persons_years)
or self.dirty_main or 14 < datetime.date.today().toordinal() - self.updated) and self.person:
self._get_start_end_year()
my_db = db.DBConnection()
if not self.id:
cl = [[
"""
INSERT INTO characters (bio, image_url, name, thumb_url, updated)
VALUES (?,?,?,?,?)
""", [self.biography, self.image_url, self.name, self.thumb_url, self._set_updated()]],
]
c_ids = len(self.ids)
if c_ids:
f_d = next(iter(self.ids))
cl.extend([[
"""
INSERT INTO character_ids (src, src_id, character_id)
VALUES (?,?,last_insert_rowid())
""", [f_d, self.ids[f_d]]]
])
if 1 < c_ids:
cl.extend([[
"""
INSERT INTO character_ids (src, src_id, character_id)
VALUES (?,?,
(SELECT character_ids.character_id FROM character_ids
WHERE character_ids.id = last_insert_rowid())
);
""", [s, self.ids[s]]] for s in list(self.ids)[1:]
])
cl.extend([[
"""
SELECT character_ids.character_id AS last_rowid
FROM character_ids
WHERE character_ids.id = last_insert_rowid()
"""]
])
else:
cl.extend([['SELECT last_insert_rowid() AS last_rowid;']])
else:
cl = []
if force or self.dirty_main:
cl = [[
"""
UPDATE characters SET bio = ?, image_url = ?, name = ?, thumb_url = ?, updated = ?
WHERE id = ?
""", [self.biography, self.image_url, self.name, self.thumb_url, self._set_updated(), self.id]],
]
if force or self.dirty_ids:
for cur_tvid, cur_src_id in iteritems(self.ids):
cl.extend([[
"""
UPDATE character_ids SET src_id = ?
WHERE src = ? AND character_id = ?
""", [cur_src_id, cur_tvid, self.id]
], [
"""
INSERT INTO character_ids (src, src_id, character_id)
SELECT %s, %s, %s WHERE changes() == 0
""" % (cur_tvid, cur_src_id, self.id)]
])
# in case we don't have a character id yet, we need to fetch it for the next step
if not self.id and cl:
r_id = my_db.mass_action(cl)
cl = []
if r_id and r_id[-1:][0]:
# r_id = list(itertools.chain.from_iterable(r_id))
if r_id:
self.id = self._get_sid(r_id)
if (force or self.dirty_person) and self.person:
cl.append([
"""
DELETE FROM character_person_map
WHERE character_id = ? AND person_id NOT IN (%s)
""" % ','.join(['?'] * len(self.person)),
[self.id] + [p.id for p in self.person]])
for cur_per in self.person:
cl.extend([[
"""
INSERT OR IGNORE INTO character_person_map (character_id, person_id)
VALUES (?,?)
""", [self.id, cur_per.id]]
])
if (force or self.dirty_years) and self.person:
cl.append([
"""
DELETE FROM character_person_years
WHERE character_id = ? AND person_id NOT IN (%s)
""" % ','.join(['?'] * len(self.person)),
[self.id] + [p.id for p in self.person]])
for cur_per in self.person:
if cur_per.id and any(1 for v in itervalues(self.persons_years.get(cur_per.id, {})) if v):
p_years = self.persons_years.get(cur_per.id, {})
cl.append([
"""
REPLACE INTO character_person_years (character_id, person_id, start_year, end_year)
VALUES (?,?,?,?)
""", [self.id, cur_per.id, p_years.get('start'), p_years.get('end')]]
)
if cl:
my_db.mass_action(cl)
self.dirty_ids, self.dirty_main, self.dirty_person, self.dirty_years = False, False, False, False
def __bool__(self):
return bool(self.name) or bool(self.person) or bool(self.ids)
def __eq__(self, other):
return other.person == self.person and ((self.id not in [None, 0] and other.id == self.id)
or any(self.ids[o] == v for o, v in iteritems(other.ids))
or (not other.ids and other.name == self.name))
def __str__(self):
id_str, id_list = '', []
for i, d in iteritems(self.ids):
id_list.append('%s: %s' % (sickgear.TVInfoAPI(i).name, d))
if id_list:
id_str = ' (%s)' % ', '.join(id_list)
return '<Character %s (%s)%s>' % (self.name, ', '.join(p.name for p in self.person), id_str)
def __repr__(self):
return self.__str__()
__nonzero__ = __bool__
class TVShow(TVShowBase):
__slots__ = (
'path',
'unique_name',
)
def __init__(self, tvid, prodid, lang='', show_result=None, imdb_info_result=None):
# type: (int, int, Text, Optional[Row], Optional[Union[Row, Dict]]) -> None
super(TVShow, self).__init__(tvid, prodid, lang)
self.unique_name = ''
self.tvid = int(tvid)
self.prodid = int(prodid)
self.sid_int = self.create_sid(self.tvid, self.prodid)
if None is not helpers.find_show_by_id(self.sid_int, check_multishow=True):
raise exceptions_helper.MultipleShowObjectsException('Can\'t create a show if it already exists')
self._airtime = None # type: Optional[datetime.time]
self._cast_list = None # type: Optional[ReferenceType[List[Character]]]
self._last_found_on_indexer = -1 # type: int
self._location = self.path = '' # type: AnyStr
# self._is_location_good = None
self._network_country = None # type: Optional[AnyStr]
self._network_country_code = None # type: Optional[AnyStr]
self._network_id = None # type: Optional[int]
self._network_is_stream = None # type: Optional[bool]
self._not_found_count = None # type: None or int
self._paused = 0
self._src_update_time = None # type: Optional[integer_types]
self.internal_ids = {} # type: Dict
self.internal_timezone = None # type: Optional[AnyStr]
self.lock = threading.RLock()
self.nextaired = '' # type: AnyStr
# noinspection added so that None _can_ be excluded from type annotation
# so that this property evaluates directly to the class on ctrl+hover instead of "multiple implementations"
# noinspection PyTypeChecker
self.release_groups = None # type: AniGroupList
self.sxe_ep_obj = {} # type: Dict
self.load_from_db(show_result=show_result, imdb_info_result=imdb_info_result)
def _get_end_episode(self, last=False, exclude_specials=False):
# type: (bool, bool) -> Optional[TVEpisode]
eps = self.get_all_episodes()
if eps:
ed = datetime.date(1900, 1, 1)
return next(iter(sorted((ep for ep in eps if ep.airdate > ed and (not exclude_specials or ep.season != 0)),
key=lambda a: a.airdate, reverse=last)), None)
@property
def first_aired_episode(self):
# type: (...) -> Optional[TVEpisode]
"""
returns first aired episode
"""
return self._get_end_episode()
@property
def first_aired_regular_episode(self):
# type: (...) -> Optional[TVEpisode]
"""
returns first aired regular episode
"""
return self._get_end_episode(exclude_specials=True)
@property
def latest_aired_episode(self):
# type: (...) -> Optional[TVEpisode]
"""
returns latest aired episode
"""
return self._get_end_episode(last=True)
@property
def latest_aired_regular_episode(self):
# type: (...) -> Optional[TVEpisode]
"""
returns latest aired regular episode
"""
return self._get_end_episode(last=True, exclude_specials=True)
@property
def cast_list(self):
# type: (...) -> List[Character]
if None is self._cast_list or (isinstance(self._cast_list, ReferenceType) and None is self._cast_list()):
v = self._load_cast_from_db()
self._cast_list = weakref.ref(v)
return self._cast_list()
return self._cast_list()
@cast_list.setter
def cast_list(self, value):
# type: (weakList[Character]) -> None
self._cast_list = None if not isinstance(value, weakList) else weakref.ref(value)
@property
def network_id(self):
return self._network_id
@network_id.setter
def network_id(self, *args):
self.dirty_setter('_network_id')(self, *args)
@property
def network_country(self):
return self._network_country
@network_country.setter
def network_country(self, *args):
self.dirty_setter('_network_country')(self, *args)
@property
def network_country_code(self):
return self._network_country_code
@network_country_code.setter
def network_country_code(self, *args):
self.dirty_setter('_network_country_code')(self, *args)
@property
def network_is_stream(self):
return self._network_is_stream
@network_is_stream.setter
def network_is_stream(self, *args):
self.dirty_setter('_network_is_stream')(self, *args)
@property
def airtime(self):
return self._airtime
@airtime.setter
def airtime(self, *args):
self.dirty_setter('_airtime')(self, *args)
@property
def timezone(self):
return self.internal_timezone
@timezone.setter
def timezone(self, *args):
self.dirty_setter('internal_timezone')(self, *args)
@staticmethod
def create_sid(tvid, prodid):
# type: (int, int) -> int
return int(prodid) << prodid_bitshift | int(tvid)
@property
def _tvid(self):
return self.tvid
@_tvid.setter
def _tvid(self, val):
self.dirty_setter('tvid')(self, int(val))
self.tvid_prodid = self.create_sid(val, self.prodid)
# TODO: remove the following when indexer is gone
# in deprecation transition, tvid also sets indexer so that existing uses continue to work normally
self.dirty_setter('_indexer')(self, int(val))
@property
def _prodid(self):
return self.prodid
@_prodid.setter
def _prodid(self, val):
# type: (int) -> None
self.dirty_setter('prodid')(self, int(val))
self.tvid_prodid = self.create_sid(self.tvid, val)
# TODO: remove the following when indexerid is gone
# in deprecation transition, prodid also sets indexerid so that existing usages continue as normal
self.dirty_setter('_indexerid')(self, int(val))
@property
def tvid_prodid(self):
# type: (...) -> AnyStr
return TVidProdid({self.tvid: self.prodid})()
@tvid_prodid.setter
def tvid_prodid(self, val):
tvid_prodid_obj = TVidProdid(val)
if getattr(self, 'tvid_prodid', None) != tvid_prodid_obj():
self.tvid, self.prodid = tvid_prodid_obj.list
self.dirty = True
tvid_prodid_int = int(tvid_prodid_obj)
if getattr(self, 'sid_int', None) != tvid_prodid_int:
self.sid_int = tvid_prodid_int
def helper_load_failed_db(self, sql_result=None):
# type: (Optional[Row, Dict]) -> None
if None is self._not_found_count or -1 == self._last_found_on_indexer:
if sql_result and self.prodid == sql_result['indexer_id'] and self.tvid == sql_result['indexer']:
sql_result = [sql_result]
else:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT fail_count, last_success
FROM tv_shows_not_found
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
if sql_result:
self._not_found_count = helpers.try_int(sql_result[0]['fail_count'])
self._last_found_on_indexer = helpers.try_int(sql_result[0]['last_success'])
else:
self._not_found_count = 0
self._last_found_on_indexer = 0
@property
def not_found_count(self):
self.helper_load_failed_db()
return self._not_found_count
@not_found_count.setter
def not_found_count(self, v):
if isinstance(v, integer_types) and v != self._not_found_count:
self._last_found_on_indexer = self.last_found_on_indexer
my_db = db.DBConnection()
# noinspection PyUnresolvedReferences
last_check = int(timestamp_near(datetime.datetime.now()))
# in case of flag change (+/-) don't change last_check date
if abs(v) == abs(self._not_found_count):
sql_result = my_db.select(
"""
SELECT last_check
FROM tv_shows_not_found
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
if sql_result:
last_check = helpers.try_int(sql_result[0]['last_check'])
my_db.upsert('tv_shows_not_found',
dict(fail_count=v, last_check=last_check, last_success=self._last_found_on_indexer),
dict(indexer=self.tvid, indexer_id=self.prodid))
self._not_found_count = v
@property
def last_found_on_indexer(self):
self.helper_load_failed_db()
return (self._last_found_on_indexer, self.last_update_indexer)[0 >= self._last_found_on_indexer]
def inc_not_found_count(self):
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT last_check
FROM tv_shows_not_found
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
days = (show_not_found_retry_days - 1, 0)[abs(self.not_found_count) <= concurrent_show_not_found_days]
if not sql_result or datetime.datetime.fromtimestamp(helpers.try_int(sql_result[0]['last_check'])) + \
datetime.timedelta(days=days, hours=18) < datetime.datetime.now():
self.not_found_count += (-1, 1)[0 <= self.not_found_count]
def reset_not_found_count(self):
if 0 != self.not_found_count:
self._not_found_count = 0
self._last_found_on_indexer = 0
my_db = db.DBConnection()
my_db.action(
"""
DELETE FROM tv_shows_not_found
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
@property
def paused(self):
"""
:rtype: bool
"""
return self._paused
@paused.setter
def paused(self, value):
if value != self._paused:
if isinstance(value, bool) or (isinstance(value, integer_types) and value in [0, 1]):
self._paused = int(value)
self.dirty = True
else:
logger.log('tried to set paused property to invalid value: %s of type: %s' % (value, type(value)),
logger.ERROR)
@property
def ids(self):
if not self.internal_ids:
acquired_lock = self.lock.acquire(False)
if acquired_lock:
try:
indexermapper.map_indexers_to_show(self)
finally:
self.lock.release()
return self.internal_ids
@ids.setter
def ids(self, value):
if isinstance(value, dict):
for k, v in iteritems(value):
if k not in indexermapper.indexer_list or \
not isinstance(v, dict) or \
not isinstance(v.get('id'), integer_types) or \
not isinstance(v.get('status'), integer_types) or \
v.get('status') not in indexermapper.MapStatus.allstatus or \
not isinstance(v.get('date'), datetime.date):
return
self.internal_ids = value
@property
def is_anime(self):
# type: (...) -> bool
return 0 < int(self.anime)
@property
def is_sports(self):
# type: (...) -> bool
return 0 < int(self.sports)
@property
def is_scene(self):
# type: (...) -> bool
return 0 < int(self.scene)
def _get_location(self):
# type: (...) -> AnyStr
# no dir check needed if missing show dirs are created during post-processing
if sickgear.CREATE_MISSING_SHOW_DIRS:
return self._location
if ek.ek(os.path.isdir, self._location):
return self._location
raise exceptions_helper.ShowDirNotFoundException('Show folder does not exist: \'%s\'' % self._location)
def _set_location(self, new_location):
# type: (AnyStr) -> None
logger.log('Setter sets location to %s' % new_location, logger.DEBUG)
# Don't validate dir if user wants to add shows without creating a dir
if sickgear.ADD_SHOWS_WO_DIR or ek.ek(os.path.isdir, new_location):
self.dirty_setter('_location')(self, new_location)
self.path = new_location
# self._is_location_good = True
else:
raise exceptions_helper.NoNFOException('Invalid folder for the show!')
location = property(_get_location, _set_location)
# delete references to anything that's not in the internal lists
def flush_episodes(self):
for cur_season_number in self.sxe_ep_obj:
for cur_ep_number in self.sxe_ep_obj[cur_season_number]:
# noinspection PyUnusedLocal
ep_obj = self.sxe_ep_obj[cur_season_number][cur_ep_number]
self.sxe_ep_obj[cur_season_number][cur_ep_number] = None
del ep_obj
def get_all_episodes(self, season=None, has_location=False, check_related_eps=True):
# type: (Optional[integer_types], bool, bool) -> List[TVEpisode]
"""
:param season: None or season number
:param has_location: return only with location
:param check_related_eps: get related episodes
:return: List of TVEpisode objects
"""
sql_selection = 'SELECT *'
if check_related_eps:
# subselection to detect multi-episodes early, share_location > 0
sql_selection += """
, (SELECT COUNT (*)
FROM tv_episodes
WHERE showid = tve.showid AND indexer = tve.indexer AND season = tve.season AND location != ''
AND location = tve.location AND episode != tve.episode
) AS share_location
"""
sql_selection += ' FROM tv_episodes tve WHERE indexer = ? AND showid = ?'
sql_parameter = [self.tvid, self.prodid]
if None is not season:
sql_selection += ' AND season = ?'
sql_parameter += [season]
if has_location:
sql_selection += ' AND location != "" '
# need ORDER episode ASC to rename multi-episodes in order S01E01-02
sql_selection += ' ORDER BY season ASC, episode ASC'
my_db = db.DBConnection()
sql_result = my_db.select(sql_selection, sql_parameter)
ep_obj_list = []
for cur_row in sql_result:
ep_obj = self.get_episode(int(cur_row['season']), int(cur_row['episode']), ep_result=[cur_row])
if ep_obj:
ep_obj.related_ep_obj = []
if check_related_eps and ep_obj.location:
# if there is a location, check if it's a multi-episode (share_location > 0)
# and put into related_ep_obj
if 0 < cur_row['share_location']:
# noinspection SqlRedundantOrderingDirection
related_ep_sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND location = ? AND episode != ?
ORDER BY episode ASC
""", [self.tvid, self.prodid, ep_obj.season, ep_obj.location, ep_obj.episode])
for cur_ep_row in related_ep_sql_result:
related_ep_obj = self.get_episode(int(cur_ep_row['season']),
int(cur_ep_row['episode']),
ep_result=[cur_ep_row])
if related_ep_obj not in ep_obj.related_ep_obj:
ep_obj.related_ep_obj.append(related_ep_obj)
ep_obj_list.append(ep_obj)
return ep_obj_list
def get_episode(self,
season=None, # type: Optional[int]
episode=None, # type: Optional[int]
path=None, # type: Optional[AnyStr]
no_create=False, # type: bool
absolute_number=None, # type: Optional[int]
ep_result=None, # type: Optional[List[Row]]
existing_only=False # type: bool
): # type: (...) -> Optional[TVEpisode]
"""
Initialise sxe_ep_obj with db fetched season keys, and then fill the TVShow episode property
and return an TVEpisode object if no_create is False
:param season: Season number
:param episode: Episode number
:param path: path to file episode
:param no_create: return None instead of an instantiated TVEpisode object
:param absolute_number: absolute number
:param ep_result:
:param existing_only: only return existing episodes
:return: TVEpisode object
"""
# if we get an anime get the real season and episode
if self.is_anime and absolute_number and not season and not episode:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT season, episode
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND absolute_number = ? AND season != 0
LIMIT 2
""", [self.tvid, self.prodid, absolute_number])
if 1 != len(sql_result):
msg = 'found for absolute number: %s in show: %s' % (absolute_number, self._name)
if not len(sql_result):
logger.debug('No entries %s' % msg)
else:
logger.error('Multiple entries %s' % msg)
return
season = int(sql_result[0]['season'])
episode = int(sql_result[0]['episode'])
logger.debug('Found episode by absolute_number: %s which is %sx%s' % (absolute_number, season, episode))
if season not in self.sxe_ep_obj:
self.sxe_ep_obj[season] = {}
if episode not in self.sxe_ep_obj[season] or None is self.sxe_ep_obj[season][episode]:
if no_create:
return
# logger.log('%s: An object for episode %sx%s did not exist in the cache, trying to create it' %
# (self.tvid_prodid, season, episode), logger.DEBUG)
if path and not existing_only:
ep_obj = TVEpisode(self, season, episode, path, show_result=ep_result)
else:
ep_obj = TVEpisode(self, season, episode, show_result=ep_result, existing_only=existing_only)
if None is not ep_obj:
self.sxe_ep_obj[season][episode] = ep_obj
return self.sxe_ep_obj[season][episode]
def _load_cast_from_db(self):
# type: (...) -> List[Character]
# cast_list = []
old_cast = [] if None is self._cast_list else self._cast_list()
old_list = {c.id for c in old_cast or []}
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT castlist.sort_order AS sort_order, characters.name AS name,
characters.bio AS c_bio, characters.id AS c_id,
characters.image_url AS image_url, characters.thumb_url AS thumb_url,
characters.updated AS c_updated, castlist.updated AS cast_updated,
persons.name AS p_name, persons.gender AS gender, persons.bio AS p_bio,
persons.birthdate AS birthdate, persons.thumb_url AS p_thumb,
persons.image_url AS p_image, persons.deathdate AS deathdate, persons.id AS p_id,
persons.birthplace AS birthplace, persons.updated AS p_updated,
persons.deathplace AS deathplace, persons.height AS height,
persons.realname AS realname, persons.nicknames AS nicknames,
persons.akas AS akas,
(SELECT group_concat(person_ids.src || ':' || person_ids.src_id, ';;;')
FROM person_ids WHERE person_ids.person_id = persons.id) AS p_ids,
(SELECT group_concat(character_ids.src || ':' || character_ids.src_id, ';;;')
FROM character_ids WHERE character_ids.character_id = characters.id) AS c_ids,
(SELECT group_concat(character_person_years.person_id || ':' ||
character_person_years.start_year || '-' || character_person_years.end_year, ';;;')
FROM character_person_years WHERE
character_person_years.character_id = characters.id) AS p_years
FROM castlist
LEFT JOIN character_person_map
ON castlist.character_id = character_person_map.character_id
LEFT JOIN characters ON character_person_map.character_id = characters.id
LEFT JOIN persons ON character_person_map.person_id = persons.id
WHERE castlist.indexer = ? AND castlist.indexer_id = ?
ORDER BY castlist.sort_order
""", [self.tvid, self.prodid])
old_cast = old_cast or []
for cur_row in sql_result:
existing_character = next(
(c for c in old_cast or [] if None is not c.id and c.id == cur_row['c_id']),
None) # type: Optional[Character]
birthdate = try_int(cur_row['birthdate'], None)
birthdate = birthdate and datetime.date.fromordinal(cur_row['birthdate'])
deathdate = try_int(cur_row['deathdate'], None)
deathdate = deathdate and datetime.date.fromordinal(cur_row['deathdate'])
p_years = {}
for p in (cur_row['p_years'] and cur_row['p_years'].split(';;;')) or []:
p_id, py = p.split(':')
start, end = py.split('-')
p_years[int(p_id)] = {'start': try_int(start, None), 'end': try_int(end, None)}
p_ids, c_ids = {}, {}
for i in (cur_row['p_ids'] and cur_row['p_ids'].split(';;;')) or []:
k, v = i.split(':')
k = try_int(k, None)
if v:
if k in (TVINFO_INSTAGRAM, TVINFO_TWITTER, TVINFO_FACEBOOK, TVINFO_WIKIPEDIA):
p_ids[k] = v
else:
p_ids[k] = try_int(v, None)
for i in (cur_row['c_ids'] and cur_row['c_ids'].split(';;;')) or []:
k, v = i.split(':')
v = try_int(v, None)
if v:
c_ids[int(k)] = try_int(v, None)
person = Person(cur_row['p_name'], cur_row['gender'],
akas=set((cur_row['akas'] and cur_row['akas'].split(';;;')) or []),
bio=cur_row['p_bio'],
birthday=birthdate, birthplace=cur_row['birthplace'],
deathday=deathdate, deathplace=cur_row['deathplace'],
height=cur_row['height'],
ids=p_ids,
image_url=cur_row['p_image'],
nicknames=set((cur_row['nicknames'] and cur_row['nicknames'].split(';;;')) or []),
real_name=cur_row['realname'], show_obj=self, sid=cur_row['p_id'],
thumb_url=cur_row['p_thumb'], updated=cur_row['p_updated'])
if existing_character:
existing_character.updated = cur_row['cast_updated']
try:
old_list.remove(existing_character.id)
except (BaseException, Exception):
pass
existing_person = next((p for p in existing_character.person if (None is not p.id
and p.ids.get(self.tvid) == person.ids.get(self.tvid))
or p.name == person.name),
None) # type: Optional[Person]
existing_character.combine_start_end_years(p_years)
if existing_person:
existing_person.update_properties(
akas=set((cur_row['akas'] and cur_row['akas'].split(';;;')) or []),
biography=person.biography,
birthday=person.birthday, birthplace=person.birthplace,
deathday=person.deathday, deathplace=cur_row['deathplace'],
gender=person.gender, height=cur_row['height'],
ids=dict(chain.from_iterable(iteritems(d) for d in (existing_person.ids, person.ids or {}))),
image_url=person.image_url, name=person.name,
nicknames=set((cur_row['nicknames'] and cur_row['nicknames'].split(';;;')) or []),
real_name=cur_row['realname'], thumb_url=person.thumb_url
)
else:
existing_character.person.append(person)
else:
old_cast.append(Character(
cur_row['name'],
bio=cur_row['c_bio'], ids=c_ids, image_url=cur_row['image_url'], person=[person],
persons_years=p_years, show_obj=self, sid=cur_row['c_id'],
thumb_url=cur_row['thumb_url'], updated=cur_row['cast_updated']))
cast_list = weakList(c for c in old_cast or [] if c.id not in old_list)
self.cast_list = cast_list
return cast_list
def cast_list_id(self):
# type: (...) -> Set
"""
return an identifier that represents the current state of the show cast list
used to tell if a change has occurred in a cast list after time/process
"""
return set((c.name, c.image_url or '', c.thumb_url or '',
hash(*([', '.join(p.name for p in c.person or [] if p.name)])))
for c in self.cast_list or [] if c.name)
@staticmethod
def orphaned_cast_sql():
# type: (...) -> List
"""
returns list of sql that removes all character and person entries that are not linked to any show
"""
return [
['DELETE FROM castlist WHERE castlist.id IN (SELECT c.id FROM castlist c'
' LEFT JOIN tv_shows ON tv_shows.indexer = c.indexer AND tv_shows.indexer_id = c.indexer_id'
' WHERE tv_shows.indexer_id is NULL);'],
['DELETE FROM characters WHERE characters.id IN (SELECT c.id FROM characters c'
' LEFT JOIN castlist on castlist.character_id = c.id WHERE castlist.id is NULL);'],
['DELETE FROM character_ids WHERE character_ids.character_id IN'
' (SELECT c.character_id FROM character_ids c'
' LEFT JOIN characters ON c.character_id = characters.id WHERE characters.id is NULL);'],
['DELETE FROM character_person_map WHERE character_id IN (SELECT c.character_id FROM character_person_map c'
' LEFT JOIN characters ON c.character_id = characters.id WHERE characters.id is NULL)'],
['DELETE FROM persons WHERE persons.id IN (SELECT p.id FROM persons p'
' LEFT JOIN character_person_map c on p.id = c.person_id WHERE c.id is NULL);'],
['DELETE FROM person_ids WHERE person_ids.person_id IN (SELECT p.person_id FROM person_ids p'
' LEFT JOIN persons ON p.person_id = persons.id WHERE persons.id is NULL);'],
['DELETE FROM castlist WHERE castlist.id IN (SELECT c.id FROM castlist c'
' LEFT JOIN character_person_map cpm on c.character_id = cpm.character_id WHERE cpm.id IS NULL);'],
['DELETE FROM character_person_years WHERE character_person_years.ROWID IN (SELECT c.ROWID FROM'
' character_person_years c'
' LEFT JOIN characters ch on c.character_id = ch.id WHERE ch.id IS NULL);'],
['DELETE FROM character_person_years WHERE character_person_years.ROWID IN (SELECT c.ROWID FROM'
' character_person_years c'
' LEFT JOIN persons p on c.person_id = p.id WHERE p.id IS NULL);'],
]
def _save_cast_list(self, removed_char_ids=None, force=False, stop_event=None, cast_list=None):
# type: (Union[List[integer_types], Set[integer_types]], bool, threading.Event, List[Character]) -> None
if cast_list:
my_db = db.DBConnection()
cl = []
for cur_id in removed_char_ids or []:
cl.extend([[
"""
DELETE FROM castlist
WHERE indexer = ? AND indexer_id = ? AND character_id = ?;
""", [self.tvid, self.prodid, cur_id]]
])
update_date = datetime.date.today().toordinal()
for cur_enum, cur_cast in enumerate(cast_list, 1):
if stop_event and stop_event.is_set():
return
cur_cast.save_to_db(show_obj=self, force=force, stop_event=stop_event)
cl.extend([[
"""
UPDATE castlist SET sort_order = ?, updated = ?
WHERE indexer = ? AND indexer_id = ?
AND character_id = ?;
""", [cur_enum, update_date, self.tvid, self.prodid, cur_cast.id]
], [
"""
INSERT INTO castlist (indexer, indexer_id, character_id, sort_order, updated)
SELECT %s, %s, %s, %s, %s WHERE changes() == 0;
""" % (self.tvid, self.prodid, cur_cast.id, cur_enum, update_date)]
])
if removed_char_ids:
# remove orphaned entries
cl.extend(self.orphaned_cast_sql())
if cl:
my_db.mass_action(cl)
def should_update(self, update_date=datetime.date.today(), last_indexer_change=None):
# type: (datetime.date, integer_types) -> bool
if last_indexer_change and self._src_update_time:
# update if show was updated or not updated within the last 30 days
if self._src_update_time < last_indexer_change or \
(self._last_update_indexer and datetime.date.today().toordinal() - self._last_update_indexer > 30):
return True
return False
# In some situations self.status = None.. need to figure out where that is!
if not self._status:
self.status = ''
logger.log('Status missing for show: [%s] with status: [%s]' %
(self.tvid_prodid, self._status), logger.DEBUG)
last_update_indexer = datetime.date.fromordinal(self._last_update_indexer)
# if show was not found for 1 week, only retry to update once a week
if (concurrent_show_not_found_days < abs(self.not_found_count)) \
and (update_date - last_update_indexer) < datetime.timedelta(days=show_not_found_retry_days):
return False
my_db = db.DBConnection()
sql_result = my_db.mass_action([[
"""
SELECT airdate
FROM [tv_episodes]
WHERE indexer = ? AND showid = ? AND season > 0
ORDER BY season DESC, episode DESC
LIMIT 1
""", [self.tvid, self.prodid]
], [
"""
SELECT airdate
FROM [tv_episodes]
WHERE indexer = ? AND showid = ? AND season > 0 AND airdate > 1
ORDER BY airdate DESC
LIMIT 1
""", [self.tvid, self.prodid]]])
last_airdate_unknown = 1 >= int(sql_result[0][0]['airdate']) if sql_result and sql_result[0] else True
last_airdate = datetime.date.fromordinal(sql_result[1][0]['airdate']) \
if sql_result and sql_result[1] else datetime.date.fromordinal(1)
# if show is not 'Ended' and last episode aired less then 460 days ago
# or don't have an airdate for the last episode always update (status 'Continuing' or '')
update_days_limit = 2013
ended_limit = datetime.timedelta(days=update_days_limit)
if 'Ended' not in self._status and (last_airdate == datetime.date.fromordinal(1)
or last_airdate_unknown
or (update_date - last_airdate) <= ended_limit
or (update_date - last_update_indexer) > ended_limit):
return True
# in the first 460 days (last airdate), update regularly
airdate_diff = update_date - last_airdate
last_update_diff = update_date - last_update_indexer
update_step_list = [[60, 1], [120, 3], [180, 7], [1281, 15], [update_days_limit, 30]]
for date_diff, interval in update_step_list:
if airdate_diff <= datetime.timedelta(days=date_diff) \
and last_update_diff >= datetime.timedelta(days=interval):
return True
# update shows without an airdate for the last episode for update_days_limit days every 7 days
return last_airdate_unknown and airdate_diff <= ended_limit and last_update_diff >= datetime.timedelta(days=7)
def write_show_nfo(self, force=False):
# type: (bool) -> bool
result = False
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, skipping NFO generation' % self.tvid_prodid)
return False
logger.log('%s: Writing NFOs for show' % self.tvid_prodid)
for cur_provider in itervalues(sickgear.metadata_provider_dict):
result = cur_provider.create_show_metadata(self, force) or result
return result
def write_metadata(self, show_only=False, force=False):
# type:(bool, bool) -> None
"""
:param show_only: only for show
:param force:
"""
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, skipping NFO generation' % self.tvid_prodid)
return
self.get_images()
force_nfo = force or not db.DBConnection().has_flag('kodi_nfo_uid')
self.write_show_nfo(force_nfo)
if not show_only:
self.write_episode_nfo(force_nfo)
def write_episode_nfo(self, force=False):
# type: (bool) -> None
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, skipping NFO generation' % self.tvid_prodid)
return
logger.log('%s: Writing NFOs for all episodes' % self.tvid_prodid)
my_db = db.DBConnection()
# noinspection SqlResolve
sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND location != ''
""", [self.tvid, self.prodid])
processed = []
for cur_row in sql_result:
if (cur_row['season'], cur_row['episode']) in processed:
continue
logger.log('%s: Retrieving/creating episode %sx%s'
% (self.tvid_prodid, cur_row['season'], cur_row['episode']), logger.DEBUG)
ep_obj = self.get_episode(cur_row['season'], cur_row['episode'], ep_result=[cur_row])
if not ep_obj.related_ep_obj:
processed += [(cur_row['season'], cur_row['episode'])]
else:
logger.log('%s: Found related to %sx%s episode(s)... %s'
% (self.tvid_prodid, cur_row['season'], cur_row['episode'],
', '.join(['%sx%s' % (x.season, x.episode) for x in ep_obj.related_ep_obj])),
logger.DEBUG)
processed += list(set([(cur_row['season'], cur_row['episode'])] +
[(x.season, x.episode) for x in ep_obj.related_ep_obj]))
ep_obj.create_meta_files(force)
def update_metadata(self):
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, skipping NFO generation' % self.tvid_prodid)
return
self.update_show_nfo()
def update_show_nfo(self):
result = False
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, skipping NFO generation' % self.tvid_prodid)
return False
logger.log('%s: Updating NFOs for show with new TV info' % self.tvid_prodid)
for cur_provider in itervalues(sickgear.metadata_provider_dict):
result = cur_provider.update_show_indexer_metadata(self) or result
return result
# find all media files in the show folder and create episodes for as many as possible
def load_episodes_from_dir(self):
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, not loading episodes from disk' % self.tvid_prodid)
return
logger.log('%s: Loading all episodes from the show directory %s' % (self.tvid_prodid, self._location))
# get file list
file_list = helpers.list_media_files(self._location)
# create TVEpisodes from each media file (if possible)
sql_l = []
for cur_media_file in file_list:
parse_result = None
ep_obj = None
logger.log('%s: Creating episode from %s' % (self.tvid_prodid, cur_media_file), logger.DEBUG)
try:
ep_obj = self.ep_obj_from_file(ek.ek(os.path.join, self._location, cur_media_file))
except (exceptions_helper.ShowNotFoundException, exceptions_helper.EpisodeNotFoundException) as e:
logger.log('Episode %s returned an exception: %s' % (cur_media_file, ex(e)), logger.ERROR)
continue
except exceptions_helper.EpisodeDeletedException:
logger.log('The episode deleted itself when I tried making an object for it', logger.DEBUG)
if None is ep_obj:
continue
# see if we should save the release name in the db
ep_file_name = ek.ek(os.path.basename, ep_obj.location)
ep_file_name = ek.ek(os.path.splitext, ep_file_name)[0]
try:
parse_result = None
np = NameParser(False, show_obj=self)
parse_result = np.parse(ep_file_name)
except (InvalidNameException, InvalidShowException):
pass
if ep_file_name and parse_result and None is not parse_result.release_group and not ep_obj.release_name:
logger.log(
'Name %s gave release group of %s, seems valid' % (ep_file_name, parse_result.release_group),
logger.DEBUG)
ep_obj.release_name = ep_file_name
# store the reference in the show
if None is not ep_obj:
if sickgear.USE_SUBTITLES and self.subtitles:
try:
ep_obj.refresh_subtitles()
except (BaseException, Exception):
logger.log('%s: Could not refresh subtitles' % self.tvid_prodid, logger.ERROR)
logger.log(traceback.format_exc(), logger.ERROR)
result = ep_obj.get_sql()
if None is not result:
sql_l.append(result)
if 0 < len(sql_l):
my_db = db.DBConnection()
my_db.mass_action(sql_l)
def load_episodes_from_db(self, update=False):
# type: (bool) -> Dict[int, Dict[int, TVEpisode]]
"""
:param update:
:return:
"""
logger.log('Loading all episodes for [%s] from the DB' % self._name)
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ?
""", [self.tvid, self.prodid])
scanned_eps = {}
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
if self._lang:
tvinfo_config['language'] = self._lang
if 0 != self._dvdorder:
tvinfo_config['dvdorder'] = True
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
cached_show = None
try:
cached_show = t.get_show(self.prodid, language=self._lang)
except BaseTVinfoError as e:
logger.log('Unable to find cached seasons from %s: %s' % (
sickgear.TVInfoAPI(self.tvid).name, ex(e)), logger.WARNING)
if None is cached_show:
return scanned_eps
scene_sql_result = my_db.select(
"""
SELECT *
FROM scene_numbering
WHERE indexer == ? AND indexer_id = ?
""", [self.tvid, self.prodid])
cached_seasons = {}
cl = []
for cur_row in sql_result:
delete_ep = False
season = int(cur_row['season'])
episode = int(cur_row['episode'])
if season not in cached_seasons:
try:
cached_seasons[season] = cached_show[season]
except BaseTVinfoSeasonnotfound as e:
logger.log('Error when trying to load the episode for [%s] from %s: %s' %
(self._name, sickgear.TVInfoAPI(self.tvid).name, ex(e)), logger.WARNING)
delete_ep = True
if season not in scanned_eps:
scanned_eps[season] = {}
logger.log('Loading episode %sx%s for [%s] from the DB' % (season, episode, self.name), logger.DEBUG)
try:
ep_obj = self.get_episode(season, episode, ep_result=[cur_row]) # type: TVEpisode
# if we found out that the ep is no longer on TVDB then delete it from our database too
if delete_ep and helpers.should_delete_episode(ep_obj.status):
cl.extend(ep_obj.delete_episode(return_sql=True))
else:
ep_obj.load_from_db(season, episode, show_result=[cur_row], scene_result=scene_sql_result)
ep_obj.load_from_tvinfo(tvapi=t, update=update, cached_show=cached_show)
scanned_eps[season][episode] = True
except exceptions_helper.EpisodeDeletedException:
logger.log('Tried loading an episode that should have been deleted from the DB [%s], skipping it'
% self._name, logger.DEBUG)
continue
if cl:
my_db.mass_action(cl)
return scanned_eps
def switch_ep_change_sql(self, old_tvid, old_prodid, season, episode, reason):
# type: (int, integer_types, int, int, int) -> List[AnyStr]
return [
"""
REPLACE INTO switch_ep_result
(old_indexer, old_indexer_id, new_indexer, new_indexer_id, season, episode, reason)
VALUES (?,?,?,?,?,?,?)
""", [old_tvid, old_prodid, self.tvid, self.prodid, season, episode, reason]]
def load_episodes_from_tvinfo(self, cache=True, update=False, tvinfo_data=None, switch=False, old_tvid=None,
old_prodid=None):
# type: (bool, bool, TVInfoShow, bool, int, integer_types) -> Optional[Dict[int, Dict[int, TVEpisode]]]
"""
:param cache:
:param update:
:param tvinfo_data:
:param switch:
:param old_tvid:
:param old_prodid:
:return:
"""
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
if not cache:
tvinfo_config['cache'] = False
if self._lang:
tvinfo_config['language'] = self._lang
if 0 != self._dvdorder:
tvinfo_config['dvdorder'] = True
logger.log('%s: Loading all episodes for [%s] from %s..'
% (self.tvid_prodid, self._name, sickgear.TVInfoAPI(self.tvid).name))
if getattr(tvinfo_data, 'id', None) == self.prodid:
show_obj = tvinfo_data
t = None
else:
try:
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
show_obj = t.get_show(self.prodid, language=self._lang)
except BaseTVinfoError:
logger.log('%s timed out, unable to update episodes for [%s] from %s' %
(sickgear.TVInfoAPI(self.tvid).name, self._name, sickgear.TVInfoAPI(self.tvid).name),
logger.ERROR)
return None
scanned_eps = {}
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ?
""", [self.tvid, self.prodid])
sql_l = []
for season in show_obj:
scanned_eps[season] = {}
for episode in show_obj[season]:
# need some examples of wtf episode 0 means to decide if we want it or not
if 0 == episode:
continue
try:
ep_obj = self.get_episode(season, episode, ep_result=sql_result) # type: TVEpisode
except exceptions_helper.EpisodeNotFoundException:
logger.log('%s: %s object for %sx%s from [%s] is incomplete, skipping this episode' %
(self.tvid_prodid, sickgear.TVInfoAPI(self.tvid).name, season, episode, self._name))
continue
else:
try:
ep_obj.load_from_tvinfo(tvapi=t, update=update, cached_show=show_obj, switch=switch,
old_tvid=old_tvid, old_prodid=old_prodid, switch_list=sql_l)
except exceptions_helper.EpisodeDeletedException:
logger.log('The episode from [%s] was deleted, skipping the rest of the load' % self._name)
continue
with ep_obj.lock:
logger.log('%s: Loading info from %s for episode %sx%s from [%s]' %
(self.tvid_prodid, sickgear.TVInfoAPI(self.tvid).name, season, episode, self._name),
logger.DEBUG)
ep_obj.load_from_tvinfo(season, episode, tvapi=t, update=update, cached_show=show_obj,
switch=switch, old_tvid=old_tvid, old_prodid=old_prodid,
switch_list=sql_l)
result = ep_obj.get_sql()
if None is not result:
sql_l.append(result)
scanned_eps[season][episode] = True
if 0 < len(sql_l):
my_db = db.DBConnection()
my_db.mass_action(sql_l)
# Done updating save last update date
self.last_update_indexer = datetime.date.today().toordinal()
self.save_to_db()
return scanned_eps
def get_images(self):
fanart_result = poster_result = banner_result = False
season_posters_result = season_banners_result = season_all_poster_result = season_all_banner_result = False
for cur_provider in itervalues(sickgear.metadata_provider_dict):
# FIXME: Needs to not show this message if the option is not enabled?
logger.log('Running metadata routines for %s' % cur_provider.name, logger.DEBUG)
fanart_result = cur_provider.create_fanart(self) or fanart_result
poster_result = cur_provider.create_poster(self) or poster_result
banner_result = cur_provider.create_banner(self) or banner_result
season_posters_result = cur_provider.create_season_posters(self) or season_posters_result
season_banners_result = cur_provider.create_season_banners(self) or season_banners_result
season_all_poster_result = cur_provider.create_season_all_poster(self) or season_all_poster_result
season_all_banner_result = cur_provider.create_season_all_banner(self) or season_all_banner_result
return fanart_result or poster_result or banner_result or season_posters_result or season_banners_result \
or season_all_poster_result or season_all_banner_result
# make a TVEpisode object from a media file
def ep_obj_from_file(self, path):
# type: (AnyStr) -> Optional[TVEpisode]
"""
:param path:
:return:
"""
if not ek.ek(os.path.isfile, path):
logger.log('%s: Not a real file... %s' % (self.tvid_prodid, path))
return None
logger.log('%s: Creating episode object from %s' % (self.tvid_prodid, path), logger.DEBUG)
try:
my_parser = NameParser(show_obj=self)
parse_result = my_parser.parse(path)
except InvalidNameException:
logger.log('Unable to parse the filename %s into a valid episode' % path, logger.DEBUG)
return None
except InvalidShowException:
logger.log('Unable to parse the filename %s into a valid show' % path, logger.DEBUG)
return None
if not len(parse_result.episode_numbers):
logger.log('parse_result: %s' % parse_result)
logger.log('No episode number found in %s, ignoring it' % path, logger.ERROR)
return None
# for now lets assume that any episode in the show dir belongs to that show
season_number = parse_result.season_number if None is not parse_result.season_number else 1
episode_numbers = parse_result.episode_numbers
root_ep_obj = None
sql_l = []
for cur_ep_num in episode_numbers:
cur_ep_num = int(cur_ep_num)
logger.log('%s: %s parsed to %s %sx%s' % (self.tvid_prodid, path, self._name, season_number, cur_ep_num),
logger.DEBUG)
check_quality_again = False
same_file = False
ep_obj = self.get_episode(season_number, cur_ep_num)
if None is ep_obj:
try:
ep_obj = self.get_episode(season_number, cur_ep_num, path)
except exceptions_helper.EpisodeNotFoundException:
logger.log('%s: Unable to figure out what this file is, skipping' % self.tvid_prodid, logger.ERROR)
continue
else:
# if there is a new file associated with this ep then re-check the quality
status, quality = sickgear.common.Quality.splitCompositeStatus(ep_obj.status)
if IGNORED == status:
continue
if (ep_obj.location and ek.ek(os.path.normpath, ep_obj.location) != ek.ek(os.path.normpath, path)) or \
(not ep_obj.location and path) or \
(SKIPPED == status):
logger.log('The old episode had a different file associated with it, re-checking the quality ' +
'based on the new filename %s' % path, logger.DEBUG)
check_quality_again = True
with ep_obj.lock:
old_size = ep_obj.file_size if ep_obj.location and status != SKIPPED else 0
ep_obj.location = path
# if the sizes are the same then it's probably the same file
if old_size and ep_obj.file_size == old_size:
same_file = True
else:
same_file = False
ep_obj.check_for_meta_files()
if None is root_ep_obj:
root_ep_obj = ep_obj
else:
if ep_obj not in root_ep_obj.related_ep_obj:
root_ep_obj.related_ep_obj.append(ep_obj)
# if it's a new file then
if not same_file:
ep_obj.release_name = ''
# if user replaces a file, attempt to recheck the quality unless it's know to be the same file
if check_quality_again and not same_file:
new_quality = Quality.nameQuality(path, self.is_anime)
if Quality.UNKNOWN == new_quality:
new_quality = Quality.fileQuality(path)
logger.log('Since this file was renamed, file %s was checked and quality "%s" found'
% (path, Quality.qualityStrings[new_quality]), logger.DEBUG)
status, quality = sickgear.common.Quality.splitCompositeStatus(ep_obj.status)
if Quality.UNKNOWN != new_quality or status in (SKIPPED, UNAIRED):
ep_obj.status = Quality.compositeStatus(DOWNLOADED, new_quality)
# check for status/quality changes as long as it's a new file
elif not same_file and sickgear.helpers.has_media_ext(path)\
and ep_obj.status not in Quality.DOWNLOADED + Quality.ARCHIVED + [IGNORED]:
old_status, old_quality = Quality.splitCompositeStatus(ep_obj.status)
new_quality = Quality.nameQuality(path, self.is_anime)
if Quality.UNKNOWN == new_quality:
new_quality = Quality.fileQuality(path)
if Quality.UNKNOWN == new_quality:
new_quality = Quality.assumeQuality(path)
new_status = None
# if it was snatched and now exists then set the status correctly
if SNATCHED == old_status and old_quality <= new_quality:
logger.log('STATUS: this episode used to be snatched with quality %s but'
' a file exists with quality %s so setting the status to DOWNLOADED'
% (Quality.qualityStrings[old_quality], Quality.qualityStrings[new_quality]),
logger.DEBUG)
new_status = DOWNLOADED
# if it was snatched proper and we found a higher quality one then allow the status change
elif SNATCHED_PROPER == old_status and old_quality < new_quality:
logger.log('STATUS: this episode used to be snatched proper with quality %s but'
' a file exists with quality %s so setting the status to DOWNLOADED'
% (Quality.qualityStrings[old_quality], Quality.qualityStrings[new_quality]),
logger.DEBUG)
new_status = DOWNLOADED
elif old_status not in SNATCHED_ANY:
new_status = DOWNLOADED
if None is not new_status:
with ep_obj.lock:
logger.log('STATUS: we have an associated file, so setting the status from %s to DOWNLOADED/%s'
% (ep_obj.status, Quality.compositeStatus(new_status, new_quality)), logger.DEBUG)
ep_obj.status = Quality.compositeStatus(new_status, new_quality)
elif same_file:
status, quality = Quality.splitCompositeStatus(ep_obj.status)
if status in (SKIPPED, UNAIRED):
new_quality = Quality.nameQuality(path, self.is_anime)
if Quality.UNKNOWN == new_quality:
new_quality = Quality.fileQuality(path)
logger.log('Since this file has status: "%s", file %s was checked and quality "%s" found'
% (statusStrings[status], path, Quality.qualityStrings[new_quality]), logger.DEBUG)
ep_obj.status = Quality.compositeStatus(DOWNLOADED, new_quality)
with ep_obj.lock:
result = ep_obj.get_sql()
if None is not result:
sql_l.append(result)
if 0 < len(sql_l):
my_db = db.DBConnection()
my_db.mass_action(sql_l)
# creating metafiles on the root should be good enough
if sickgear.USE_FAILED_DOWNLOADS and None is not root_ep_obj:
with root_ep_obj.lock:
root_ep_obj.create_meta_files()
return root_ep_obj
def load_from_db(self, show_result=None, imdb_info_result=None):
# type: (Optional[Row], Optional[Union[Row, Dict]]) -> Optional[bool]
"""
:return:
"""
if not show_result or self.tvid != show_result['indexer'] or self.prodid != show_result['indexer_id']:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT *
FROM tv_shows
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
if 1 != len(sql_result):
if len(sql_result):
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
if self._lang:
tvinfo_config['language'] = self._lang
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
cached_show = t.get_show(self.prodid, load_episodes=False, language=self._lang)
vals = (self.prodid, '' if not cached_show else ' [%s]' % cached_show['seriesname'].strip())
if len(sql_result):
logger.log('%s: Loading show info%s from database' % vals)
raise exceptions_helper.MultipleDBShowsException()
logger.log('%s-%s: Unable to find the show%s in the database' % (self.tvid, self.prodid, self.name))
return
show_result = next(iter(sql_result))
if not self.tvid:
self.tvid, self.prodid = int(show_result['indexer']), int(show_result['indexer_id'])
self._air_by_date = show_result['air_by_date'] or 0
self._airs = show_result['airs'] or ''
self._airtime = self._make_airtime(show_result['airtime'])
self._anime = show_result['anime'] or 0
self._classification = None is self._classification and show_result['classification'] or ''
self._dvdorder = show_result['dvdorder'] or 0
self._flatten_folders = int(show_result['flatten_folders'])
self._genre = '|'.join([_g for _g in (self._genre or show_result['genre'] or '').split('|') if _g])
self._lang = self._lang or show_result['lang'] or ''
self._last_update_indexer = show_result['last_update_indexer']
self._name = self._name or show_result['show_name'] or ''
self._network_country = show_result['network_country']
self._network_country_code = show_result['network_country_code']
self._network_id = show_result['network_id']
self._network_is_stream = bool(show_result['network_is_stream'])
self._overview = self._overview or show_result['overview'] or ''
self._paused = int(show_result['paused'])
self._prune = show_result['prune'] or 0
self._quality = int(show_result['quality'])
self._runtime = show_result['runtime']
self._scene = show_result['scene'] or 0
self._sports = show_result['sports'] or 0
self._src_update_time = try_int(show_result['src_update_timestamp'], None)
self._startyear = show_result['startyear'] or 0
self._status = show_result['status'] or ''
self._subtitles = show_result['subtitles'] and 1 or 0
self._tag = show_result['tag'] or 'Show List'
self._upgrade_once = show_result['archive_firstmatch'] or 0
if not self._imdbid:
imdbid = show_result['imdb_id'] or ''
self._imdbid = ('', imdbid)[2 < len(imdbid)]
self._rls_global_exclude_ignore = helpers.split_word_str(
show_result['rls_global_exclude_ignore'])[0]
self._rls_global_exclude_require = helpers.split_word_str(
show_result['rls_global_exclude_require'])[0]
self._rls_ignore_words, self._rls_ignore_words_regex = helpers.split_word_str(
show_result['rls_ignore_words'])
self._rls_require_words, self._rls_require_words_regex = helpers.split_word_str(
show_result['rls_require_words'])
self.internal_network = self.internal_network or show_result['network'] or ''
self.internal_timezone = show_result['timezone']
if not self.internal_timezone and self.internal_network:
_, self.internal_timezone = network_timezones.get_network_timezone(self.internal_network,
return_name=True)
try:
self.location = show_result['location']
except (BaseException, Exception):
self.dirty_setter('_location')(self, show_result['location'])
# self._is_location_good = False
self.release_groups = self._anime and AniGroupList(self.tvid, self.prodid, self.tvid_prodid) or None
logger.log(u'Loaded.. {: <9} {: <8} {}'.format(
sickgear.TVInfoAPI(self.tvid).config.get('name') + ',', '%s,' % self.prodid, self.name))
# Get IMDb_info from database
if not imdb_info_result or \
self.tvid != imdb_info_result['indexer'] or self.prodid != imdb_info_result['indexer_id']:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT *
FROM imdb_info
WHERE indexer = ? AND indexer_id = ?
""", [self.tvid, self.prodid])
else:
sql_result = [imdb_info_result]
if 0 < len(sql_result):
# this keys() is not a dict
if isinstance(sql_result[0], dict):
self._imdb_info = sql_result[0]
else:
self._imdb_info = dict(zip(sql_result[0].keys(), [(r, '')[None is r] for r in sql_result[0]]))
if 'is_mini_series' in self._imdb_info:
self._imdb_info['is_mini_series'] = bool(self._imdb_info['is_mini_series'])
elif sickgear.USE_IMDB_INFO:
logger.log('%s: The next show update will attempt to find IMDb info for [%s]' %
(self.tvid_prodid, self.name), logger.DEBUG)
return
self.dirty = False
return True
def _get_tz_info(self):
if self.internal_timezone:
return tz.gettz(self.internal_timezone, zoneinfo_priority=True)
elif self.internal_network:
return network_timezones.get_network_timezone(self.internal_network)
def _make_airtime(self, airtime=None):
# type: (Optional[integer_types]) -> Optional[datetime.time]
if isinstance(airtime, integer_types):
return int_to_time(airtime)
if self._airs:
hour, minute = network_timezones.parse_time(self._airs)
return datetime.time(hour=hour, minute=minute)
return None
def _get_airtime(self):
# type: (...) -> Tuple[Optional[datetime.time], Optional[string_types], Optional[string_types]]
"""
try to get airtime from trakt
"""
try:
tvinfo_config = sickgear.TVInfoAPI(TVINFO_TRAKT).api_params.copy()
t = sickgear.TVInfoAPI(TVINFO_TRAKT).setup(**tvinfo_config)
results = t.search_show(ids={self._tvid: self._prodid})
for _r in results:
if 'show' != _r['type']:
continue
if _r['ids'][TVINFO_TMDB] == self._prodid:
try:
h, m = _r['airs']['time'].split(':')
h, m = try_int(h, None), try_int(m, None)
if None is not h and None is not m:
return datetime.time(hour=h, minute=m), _r['airs'].get('time'), _r['airs'].get('day')
except (BaseException, Exception):
continue
except (BaseException, Exception):
return None, None, None
return None, None, None
def _should_cast_update(self, show_info_cast):
# type: (CastList) -> bool
"""
return if cast list has changed and should be updated
"""
cast_list = self.cast_list
existing_cast = set(hash(*([', '.join(p.name for p in c.person or [] if p.name)]))
for c in cast_list or [] if c.name)
new_cast = set(hash(*([', '.join(p.name for p in c.person or [] if p.name)]))
for c_t, c_l in iteritems(show_info_cast or {}) for c in c_l or [] if c.name
and c_t in (RoleTypes.ActorMain, RoleTypes.Host, RoleTypes.Interviewer, RoleTypes.Presenter))
now = datetime.date.today().toordinal()
max_age = random.randint(30, 60)
no_image_cast = any(not (c.image_url or c.thumb_url) and (random.randint(4, 14) < now - c.updated)
for c in cast_list or [] if c.name)
too_old_char = any(max_age < now - c.updated for c in cast_list or [] if c.name)
# too_old_person = any(any(max_age < now - p.updated for p in c.person or [] if p.name)
# for c in cast_list or [] if c.name)
return existing_cast != new_cast or no_image_cast or too_old_char # or too_old_person
def load_from_tvinfo(self, cache=True, tvapi=None, tvinfo_data=None, scheduled_update=False, switch=False):
# type: (bool, bool, TVInfoShow, bool, bool) -> Optional[Union[bool, TVInfoShow]]
"""
:param cache:
:param tvapi:
:param tvinfo_data:
:param scheduled_update:
:param switch:
"""
# There's gotta be a better way of doing this but we don't wanna
# change the cache value elsewhere
if None is tvapi:
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
if not cache:
tvinfo_config['cache'] = False
if self._lang:
tvinfo_config['language'] = self._lang
if 0 != self._dvdorder:
tvinfo_config['dvdorder'] = True
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
else:
t = tvapi
if getattr(tvinfo_data, 'id', None) == self.prodid:
show_info = tvinfo_data
else:
show_info = t.get_show(self.prodid, actors=True, language=self._lang) # type: Optional[TVInfoShow]
if None is show_info or getattr(t, 'show_not_found', False):
if getattr(t, 'show_not_found', False):
self.inc_not_found_count()
logger.log('Show [%s] not found (maybe even removed?)' % self._name, logger.WARNING)
else:
logger.log('Show data [%s] not found' % self._name, logger.WARNING)
return False
self.reset_not_found_count()
try:
self.name = show_info['seriesname'].strip()
except AttributeError:
raise BaseTVinfoAttributenotfound(
"Found %s, but attribute 'seriesname' was empty." % self.tvid_prodid)
if show_info:
logger.log('%s: Loading show info [%s] from %s' % (
self.tvid_prodid, self._name, sickgear.TVInfoAPI(self.tvid).name))
self.classification = self.dict_prevent_nonetype(show_info, 'classification', 'Scripted')
self.genre = '|'.join([_g for _g in (self.dict_prevent_nonetype(show_info, 'genre') or '').split('|') if _g])
self.network = self.dict_prevent_nonetype(show_info, 'network')
self.runtime = self.dict_prevent_nonetype(show_info, 'runtime')
self.dirty_setter('_src_update_time')(self, show_info.updated_timestamp)
old_imdb = self.imdbid
if show_info.ids.imdb:
self.imdbid = 'tt%07d' % show_info.ids.imdb
else:
self.imdbid = self.dict_prevent_nonetype(show_info, 'imdb_id')
if old_imdb != self.imdbid:
try:
imdb_id = try_int(self.imdbid.replace('tt', ''), None)
mapped_imdb = self.ids.get(TVINFO_IMDB, {'id': 0})['id']
if imdb_id != mapped_imdb:
indexermapper.map_indexers_to_show(self, recheck=True)
except (BaseException, Exception):
pass
# tmdb doesn't provide air time, try to get it from other source
if TVINFO_TMDB == self._tvid and None is show_info.time:
show_info.time, show_info.airs_time, show_info.airs_dayofweek = self._get_airtime()
if None is not getattr(show_info, 'airs_dayofweek', None) and None is not getattr(show_info, 'airs_time', None):
self.airs = ('%s %s' % (show_info['airs_dayofweek'], show_info['airs_time'])).strip()
if None is not getattr(show_info, 'firstaired', None):
self.startyear = int(str(show_info["firstaired"]).split('-')[0])
self.status = self.dict_prevent_nonetype(show_info, 'status')
self.overview = self.dict_prevent_nonetype(show_info, 'overview')
self.network_id = show_info.network_id
self.network_country = show_info.network_country
self.network_country_code = show_info.network_country_code
self.network_is_stream = show_info.network_is_stream
self.timezone = show_info.network_timezone
if not self.internal_timezone and self.internal_network:
_, self.timezone = network_timezones.get_network_timezone(self.internal_network, return_name=True)
self.airtime = self._make_airtime(show_info.time)
if show_info.cast and self._should_cast_update(show_info.cast):
sickgear.people_queue_scheduler.action.add_cast_update(show_obj=self, show_info_cast=show_info.cast,
scheduled_update=scheduled_update, switch=switch)
else:
logger.log('Not updating cast for show because data is unchanged.')
return show_info
@staticmethod
def _update_person_properties_helper(person_obj, src_person, p_ids):
# type: (Person, TVINFO_Person, Dict) -> None
person_obj.update_properties(
name=src_person.name, gender=src_person.gender,
birthday=src_person.birthdate, deathday=src_person.deathdate,
biography=src_person.bio,
ids=dict(chain.from_iterable(iteritems(d) for d in
(person_obj.ids, p_ids))),
deathplace=src_person.deathplace, akas=src_person.akas,
nicknames=src_person.nicknames, real_name=src_person.real_name,
height=src_person.height)
def load_cast_from_tvinfo(self, show_info_cast, force=False, stop_event=None):
# type: (CastList, bool, threading.Event) -> None
"""
add and fetch cast list info for show
Note: currently not multithread safe
:param show_info_cast: tvinfo castlist
:param force: force reload of cast info
:param stop_event:
"""
if stop_event and stop_event.is_set():
return
if not show_info_cast:
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
show_info = t.get_show(self.prodid, load_episodes=False, actors=True,
language=self._lang) # type: Optional[TVInfoShow]
if None is show_info:
return
show_info_cast = show_info.cast
cast_list = self._load_cast_from_db()
remove_char_ids = {c.id for c in cast_list or []}
cast_ordered = weakList()
for ct, c_l in iteritems(show_info_cast): # type: (integer_types, List[TVINFO_Character])
if ct not in (RoleTypes.ActorMain, RoleTypes.Host, RoleTypes.Interviewer, RoleTypes.Presenter):
continue
for c in c_l:
if stop_event and stop_event.is_set():
return
unique_name = 1 == len([cu for cu in c_l if (None is not c.id and cu.id == c.id)
or cu.name == c.name])
mc = next((cl for cl in cast_list or []
if (None is not c.id and cl.ids.get(self.tvid) == c.id)
or (unique_name and c.name and cl.name == c.name)
or any(cl.ids.get(src) == c.ids.get(src) for src in c.ids or {})),
None) # type: Optional[Character]
if not mc:
unique_person = not any(1 for _cp in
Counter([_p.name for _cha in c_l for _p in _cha.person
if any(_p.name == _pp.name for _pp in c.person)]).values()
if 1 != _cp)
if unique_person:
pc = [cl for cl in cast_list or [] if cl.person
and any(1 for p in cl.person if c.person
and ((None is not p.ids.get(self.tvid) and any(p.ids.get(self.tvid) == cp.id
for cp in c.person))
or any(p.name == cp.name for cp in c.person)))]
if 1 == len(pc):
mc = pc[0]
if mc:
try:
remove_char_ids.remove(mc.id)
except KeyError:
# can happen with duplicate characters on source
logger.debug('%s - error character: %s (%s)' % (self.name, mc.id, mc.name))
old_person_ids = {op.id for op in mc.person if None is not op.id}
for src_person in c.person:
if src_person.id:
person_ids = {self.tvid: src_person.id}
else:
person_ids = {}
existing_person = next(
(cur_person for cur_person in mc.person
if (None is not src_person.id and cur_person.ids.get(self.tvid) == src_person.id)
or (cur_person.name and cur_person.name == src_person.name)),
None) # type: Optional[Person]
new_person = None
if not existing_person:
new_person = Person(src_person.name, src_person.gender, src_person.birthdate,
src_person.deathdate, src_person.bio, akas=src_person.akas,
character_obj=mc, deathplace=src_person.deathplace,
height=src_person.height, ids=person_ids, image_url=src_person.image,
nicknames=src_person.nicknames, real_name=src_person.real_name,
show_obj=self, thumb_url=src_person.thumb_url)
if new_person and new_person.id and any(1 for pi in mc.person if pi.id == new_person.id):
existing_person = next((pi for pi in mc.person if pi.id == new_person.id), None)
if existing_person:
try:
old_person_ids.remove(existing_person.id)
except KeyError:
logger.log('%s - Person error: %s (%s)' %
(self.name, existing_person.name, existing_person.id), logger.ERROR)
pass
if force:
existing_person.reset(src_person)
self._update_person_properties_helper(existing_person, src_person, person_ids)
elif None is not new_person:
mc.person.append(new_person)
if old_person_ids:
mc.person = [cp for cp in mc.person if cp.id not in old_person_ids]
mc.update_properties(
name=c.name, image_url=c.image, thumb_url=c.thumb_url,
ids=dict(chain.from_iterable(
iteritems(d) for d in (mc.ids, ({}, {self.tvid: c.id})[None is not c.id]))))
else:
persons = []
for s_pers in c.person:
existing_person = next((_p for _c in cast_list for _p in _c.person
if (s_pers.id and _p.ids.get(self.tvid) == s_pers.id)
or (not s_pers.id and s_pers.name == _p.name)),
None) # type: Optional[Person]
if s_pers.id:
person_ids = {self.tvid: s_pers.id}
else:
person_ids = {}
if existing_person:
if force:
existing_person.reset(s_pers)
self._update_person_properties_helper(existing_person, s_pers, person_ids)
persons.append(existing_person)
else:
tmp_char = Character(c.name, ids=({}, {self.tvid: c.id})[None is not c.id],
image_url=c.image, show_obj=self, thumb_url=c.thumb_url, tmp=True)
new_person = Person(
s_pers.name, s_pers.gender, s_pers.birthdate, s_pers.deathdate, s_pers.bio,
akas=s_pers.akas, deathplace=s_pers.deathplace, height=s_pers.height,
ids=({}, {self.tvid: s_pers.id})[None is not s_pers.id],
image_url=s_pers.image, nicknames=s_pers.nicknames, real_name=s_pers.real_name,
show_obj=self, thumb_url=s_pers.thumb_url, tmp_character_obj=tmp_char
)
if force:
new_person.reset(s_pers)
self._update_person_properties_helper(new_person, s_pers, person_ids)
persons.append(new_person)
mc = Character(c.name, ids=({}, {self.tvid: c.id})[None is not c.id], image_url=c.image,
person=persons, show_obj=self, thumb_url=c.thumb_url)
cast_list.append(mc)
cast_ordered.append(mc)
if stop_event and stop_event.is_set():
return
if remove_char_ids:
[c.remove_all_img(include_person=True) for c in cast_list or [] if c.id in remove_char_ids]
self.cast_list = cast_ordered
self._save_cast_list(force=force, removed_char_ids=remove_char_ids, stop_event=stop_event,
cast_list=cast_ordered)
def load_imdb_info(self):
if not sickgear.USE_IMDB_INFO:
return
logger.log('Retrieving show info [%s] from IMDb' % self._name, logger.DEBUG)
try:
self._get_imdb_info()
except (BaseException, Exception) as e:
logger.log('Error loading IMDb info: %s' % ex(e), logger.ERROR)
logger.log('%s' % traceback.format_exc(), logger.ERROR)
@staticmethod
def check_imdb_redirect(imdb_id):
# type: (Union[AnyStr, integer_types]) -> Optional[AnyStr]
"""
:param imdb_id: imdb id
"""
page_url = 'https://www.imdb.com/title/{0}/'.format(imdb_id)
try:
response = requests.head(page_url, allow_redirects=True)
if response.history and any([h for h in response.history if 301 == h.status_code]):
return helpers.parse_imdb_id(response.url)
except (BaseException, Exception):
pass
def _get_imdb_info(self, retry=False):
# type: (bool) -> None
if not self._imdbid and 0 >= self.ids.get(indexermapper.TVINFO_IMDB, {'id': 0}).get('id', 0):
return
imdb_info = {'imdb_id': self._imdbid or 'tt%07d' % self.ids[indexermapper.TVINFO_IMDB]['id'],
'title': '',
'year': '',
'akas': '',
'runtimes': try_int(self._runtime, None),
'is_mini_series': False,
'episode_count': None,
'genres': '',
'countries': '',
'country_codes': '',
'certificates': '',
'rating': '',
'votes': '',
'last_update': ''}
imdb_id = None
imdb_certificates = None
try:
imdb_id = str(self._imdbid or 'tt%07d' % self.ids[indexermapper.TVINFO_IMDB]['id'])
redirect_check = self.check_imdb_redirect(imdb_id)
if redirect_check:
self._imdbid = redirect_check
imdb_id = redirect_check
imdb_info['imdb_id'] = self.imdbid
i = imdbpie.Imdb(exclude_episodes=True, cachedir=ek.ek(os.path.join, sickgear.CACHE_DIR, 'imdb-pie'))
if not helpers.parse_imdb_id(imdb_id):
logger.log('Not a valid imdbid: %s for show: %s' % (imdb_id, self._name), logger.WARNING)
return
imdb_ratings = i.get_title_ratings(imdb_id=imdb_id)
imdb_akas = i.get_title_versions(imdb_id=imdb_id)
imdb_tv = i.get_title_auxiliary(imdb_id=imdb_id)
ipie = getattr(imdbpie.__dict__.get('imdbpie'), '_SIMPLE_GET_ENDPOINTS', None)
if ipie:
ipie.update({
u'get_title_certificates': u'/title/{imdb_id}/certificates',
u'get_title_parentalguide': u'/title/{imdb_id}/parentalguide',
})
imdb_certificates = i.get_title_certificates(imdb_id=imdb_id)
except LookupError as e:
if 'Title was an episode' in ex(e) and imdb_id == 'tt%07d' % self.ids[indexermapper.TVINFO_IMDB]['id']:
self.ids[indexermapper.TVINFO_IMDB]['id'] = 0
self.ids[indexermapper.TVINFO_IMDB]['status'] = MapStatus.NOT_FOUND
if datetime.date.today() != self.ids[indexermapper.TVINFO_IMDB]['date']:
indexermapper.map_indexers_to_show(self, force=True)
if not retry and imdb_id != 'tt%07d' % self.ids[indexermapper.TVINFO_IMDB]['id']:
# add retry arg to prevent endless loops
logger.log('imdbid: %s not found. retrying with newly found id: %s' %
(imdb_id, 'tt%07d' % self.ids[indexermapper.TVINFO_IMDB]['id']), logger.DEBUG)
self._get_imdb_info(retry=True)
return
logger.log('imdbid: %s not found. Error: %s' % (imdb_id, ex(e)), logger.WARNING)
return
except ImdbAPIError as e:
logger.log('Imdb API Error: %s' % ex(e), logger.WARNING)
return
except (BaseException, Exception) as e:
logger.log('Error: %s retrieving imdb id: %s' % (ex(e), imdb_id), logger.WARNING)
return
# ratings
if isinstance(imdb_ratings.get('rating'), (int, float)):
imdb_info['rating'] = try_float(imdb_ratings.get('rating'), '')
if isinstance(imdb_ratings.get('ratingCount'), int):
imdb_info['votes'] = try_int(imdb_ratings.get('ratingCount'), '')
en_cc = ['GB', 'US', 'CA', 'AU']
# akas
if isinstance(imdb_akas.get('alternateTitles'), (list, tuple)):
akas_head = OrderedDict([(k, None) for k in en_cc])
akas_tail = []
for t in imdb_akas.get('alternateTitles'):
if isinstance(t, dict) and t.get('title') and t.get('region'):
cc = t.get('region').upper()
cc_aka = '%s::%s' % (cc, t.get('title'))
if cc in akas_head:
akas_head[cc] = cc_aka
else:
akas_tail += [cc_aka]
imdb_info['akas'] = '|'.join([aka for aka in itervalues(akas_head) if aka] + sorted(akas_tail))
# tv
if isinstance(imdb_tv.get('title'), string_types):
imdb_info['title'] = imdb_tv.get('title')
if isinstance(imdb_tv.get('year'), (int, string_types)):
imdb_info['year'] = try_int(imdb_tv.get('year'), '')
if isinstance(imdb_tv.get('runningTimes'), list):
try:
for _t in imdb_tv.get('runningTimes'):
try:
if isinstance(_t.get('attributes'), list) and \
any(1 for _a in _t.get('attributes') if 'entire' in _a):
continue
except (BaseException, Exception):
continue
imdb_info['runtimes'] = try_int(_t.get('timeMinutes'), '')
break
except (BaseException, Exception):
pass
if isinstance(imdb_tv.get('titleType'), string_types):
imdb_info['is_mini_series'] = 'mini' in imdb_tv.get('titleType').lower()
if isinstance(imdb_tv.get('numberOfEpisodes'), (int, string_types)):
imdb_info['episode_count'] = try_int(imdb_tv.get('numberOfEpisodes'), 1)
if isinstance(imdb_tv.get('genres'), (list, tuple)):
imdb_info['genres'] = '|'.join(filter_iter(lambda _v: _v, imdb_tv.get('genres')))
if isinstance(imdb_tv.get('origins'), list):
imdb_info['country_codes'] = '|'.join(filter_iter(lambda _v: _v, imdb_tv.get('origins')))
# certificate
if isinstance(imdb_certificates.get('certificates'), dict):
certs_head = OrderedDict([(k, None) for k in en_cc])
certs_tail = []
for cc, values in iteritems(imdb_certificates.get('certificates')):
if cc and isinstance(values, (list, tuple)):
for cert in values:
if isinstance(cert, dict) and cert.get('certificate'):
extra_info = ''
if isinstance(cert.get('attributes'), list):
extra_info = ' (%s)' % ', '.join(cert.get('attributes'))
cc = cc.upper()
cc_cert = '%s:%s%s' % (cc, cert.get('certificate'), extra_info)
if cc in certs_head:
certs_head[cc] = cc_cert
else:
certs_tail += [cc_cert]
imdb_info['certificates'] = '|'.join([cert for cert in itervalues(certs_head) if cert] + sorted(certs_tail))
if (not imdb_info['certificates'] and isinstance(imdb_tv.get('certificate'), dict)
and isinstance(imdb_tv.get('certificate').get('certificate'), string_types)):
imdb_info['certificates'] = '%s:%s' % (u'US', imdb_tv.get('certificate').get('certificate'))
imdb_info['last_update'] = datetime.date.today().toordinal()
# Rename dict keys without spaces for DB upsert
self.imdb_info = dict(
[(k.replace(' ', '_'), k(v) if hasattr(v, 'keys') else v) for k, v in iteritems(imdb_info)])
logger.log('%s: Obtained info from IMDb -> %s' % (self.tvid_prodid, self._imdb_info), logger.DEBUG)
logger.log('%s: Parsed latest IMDb show info for [%s]' % (self.tvid_prodid, self._name))
def next_episode(self):
logger.log('%s: Finding the episode which airs next for: %s' % (self.tvid_prodid, self._name), logger.DEBUG)
cur_date = datetime.date.today().toordinal()
if not self.nextaired or self.nextaired and cur_date > self.nextaired:
my_db = db.DBConnection()
# noinspection SqlRedundantOrderingDirection
sql_result = my_db.select(
"""
SELECT airdate, season, episode
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND airdate >= ? AND status in (?,?,?)
ORDER BY airdate ASC
LIMIT 1
""", [self.tvid, self.prodid, datetime.date.today().toordinal(), UNAIRED, WANTED, FAILED])
if None is sql_result or 0 == len(sql_result):
logger.log('%s: No episode found... need to implement a show status' % self.tvid_prodid, logger.DEBUG)
self.nextaired = ''
else:
logger.log('%s: Found episode %sx%s' % (
self.tvid_prodid, sql_result[0]['season'], sql_result[0]['episode']), logger.DEBUG)
self.nextaired = sql_result[0]['airdate']
return self.nextaired
def delete_show(self, full=False):
# type: (bool) -> None
"""
:param full:
"""
try:
sickgear.people_queue_scheduler.action.abort_cast_update(show_obj=self)
except (BaseException, Exception):
pass
try:
sickgear.show_queue_scheduler.action.abort_show(show_obj=self)
except (BaseException, Exception):
pass
try:
sickgear.search_queue_scheduler.action.abort_show(show_obj=self)
except (BaseException, Exception):
pass
sql_l = [['DELETE FROM tv_episodes WHERE indexer = ? AND showid = ?', [self.tvid, self.prodid]],
['DELETE FROM tv_shows WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM imdb_info WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM xem_refresh WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM scene_numbering WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM allowlist WHERE indexer = ? AND show_id = ?', [self.tvid, self.prodid]],
['DELETE FROM blocklist WHERE indexer = ? AND show_id = ?', [self.tvid, self.prodid]],
['DELETE FROM indexer_mapping WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM tv_shows_not_found WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]],
['DELETE FROM castlist WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid]]
] + self.orphaned_cast_sql()
my_db = db.DBConnection()
my_db.mass_action(sql_l)
self.remove_character_images()
name_cache.remove_from_namecache(self.tvid, self.prodid)
try:
sickgear.name_parser.parser.name_parser_cache.flush(self)
except (BaseException, Exception):
pass
action = ('delete', 'trash')[sickgear.TRASH_REMOVE_SHOW]
# remove self from show list
sickgear.showList = filter_list(lambda so: so.tvid_prodid != self.tvid_prodid, sickgear.showList)
try:
del sickgear.showDict[self.sid_int]
except (BaseException, Exception):
pass
sickgear.webserve.Home.make_showlist_unique_names()
sickgear.MEMCACHE['history_tab'] = sickgear.webserve.History.menu_tab(sickgear.MEMCACHE['history_tab_limit'])
try:
tvid_prodid = self.tvid_prodid
if tvid_prodid in sickgear.switched_shows:
sickgear.switched_shows.pop(tvid_prodid)
elif tvid_prodid in itervalues(sickgear.switched_shows):
sickgear.switched_shows = {k: v for k, v in iteritems(sickgear.switched_shows) if tvid_prodid != v}
except (BaseException, Exception):
pass
# clear the cache
ic = image_cache.ImageCache()
for cache_obj in ek.ek(glob.glob, ic.fanart_path(self.tvid, self.prodid).replace('fanart.jpg', '*')) \
+ ek.ek(glob.glob, ic.poster_thumb_path(self.tvid, self.prodid).replace('poster.jpg', '*')) \
+ ek.ek(glob.glob, ic.poster_path(self.tvid, self.prodid).replace('poster.jpg', '*')):
cache_dir = ek.ek(os.path.isdir, cache_obj)
result = helpers.remove_file(cache_obj, tree=cache_dir, log_level=logger.WARNING)
if result:
logger.log('%s cache %s %s' % (result, cache_dir and 'dir' or 'file', cache_obj))
if self.tvid_prodid in sickgear.FANART_RATINGS:
del sickgear.FANART_RATINGS[self.tvid_prodid]
# remove entire show folder
if full:
try:
logger.log('Attempt to %s show folder %s' % (action, self._location))
# check first the read-only attribute
file_attribute = ek.ek(os.stat, self.location)[0]
if not file_attribute & stat.S_IWRITE:
# File is read-only, so make it writeable
logger.log('Attempting to make writeable the read only folder %s' % self._location, logger.DEBUG)
try:
ek.ek(os.chmod, self.location, stat.S_IWRITE)
except (BaseException, Exception):
logger.log('Unable to change permissions of %s' % self._location, logger.WARNING)
result = helpers.remove_file(self.location, tree=True)
if result:
logger.log('%s show folder %s' % (result, self._location))
except exceptions_helper.ShowDirNotFoundException:
logger.log('Show folder does not exist, no need to %s %s' % (action, self._location), logger.WARNING)
except OSError as e:
logger.log('Unable to %s %s: %s / %s' % (action, self._location, repr(e), ex(e)), logger.WARNING)
def populate_cache(self, force=False):
# type: (bool) -> None
"""
:param force:
"""
cache_inst = image_cache.ImageCache()
logger.log('Checking & filling cache for show %s' % self._name)
cache_inst.fill_cache(self, force)
def refresh_dir(self):
# make sure the show dir is where we think it is unless dirs are created on the fly
if not ek.ek(os.path.isdir, self._location) and not sickgear.CREATE_MISSING_SHOW_DIRS:
return False
# load from dir
self.load_episodes_from_dir()
# run through all locations from DB, check that they exist
logger.log('%s: Loading all episodes for [%s] with a location from the database'
% (self.tvid_prodid, self._name))
my_db = db.DBConnection()
# noinspection SqlResolve
sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND location != ''
ORDER BY season DESC, episode DESC
""", [self.tvid, self.prodid])
kept = 0
deleted = 0
attempted = []
sql_l = []
for cur_row in sql_result:
season = int(cur_row['season'])
episode = int(cur_row['episode'])
location = ek.ek(os.path.normpath, cur_row['location'])
try:
ep_obj = self.get_episode(season, episode, ep_result=[cur_row])
except exceptions_helper.EpisodeDeletedException:
logger.log('The episode from [%s] was deleted while we were refreshing it, moving on to the next one'
% self._name, logger.DEBUG)
continue
# if the path exist and if it's in our show dir
if (self.prune and season and ep_obj.location not in attempted and 0 < helpers.get_size(ep_obj.location) and
ek.ek(os.path.normpath, location).startswith(ek.ek(os.path.normpath, self.location))):
with ep_obj.lock:
if ep_obj.status in Quality.DOWNLOADED:
# locations repeat but attempt to delete once
attempted += ep_obj.location
if kept >= self.prune:
result = helpers.remove_file(ep_obj.location, prefix_failure=u'%s: ' % self.tvid_prodid)
if result:
logger.log(u'%s: %s file %s' % (self.tvid_prodid, result, ep_obj.location),
logger.DEBUG)
deleted += 1
else:
kept += 1
# if the path doesn't exist or if it's not in our show dir
if not ek.ek(os.path.isfile, location) or not ek.ek(os.path.normpath, location).startswith(
ek.ek(os.path.normpath, self.location)):
# check if downloaded files still exist, update our data if this has changed
if 1 != sickgear.SKIP_REMOVED_FILES:
with ep_obj.lock:
# if it used to have a file associated with it and it doesn't anymore then set it to IGNORED
if ep_obj.location and ep_obj.status in Quality.DOWNLOADED:
if ARCHIVED == sickgear.SKIP_REMOVED_FILES:
ep_obj.status = Quality.compositeStatus(
ARCHIVED, Quality.qualityDownloaded(ep_obj.status))
else:
ep_obj.status = (sickgear.SKIP_REMOVED_FILES, IGNORED)[
not sickgear.SKIP_REMOVED_FILES]
logger.log(
'%s: File no longer at location for s%02de%02d,' % (self.tvid_prodid, season, episode)
+ ' episode removed and status changed to %s' % statusStrings[ep_obj.status],
logger.DEBUG)
ep_obj.subtitles = list()
ep_obj.subtitles_searchcount = 0
ep_obj.subtitles_lastsearch = str(datetime.datetime.min)
ep_obj.location = ''
ep_obj.hasnfo = False
ep_obj.hastbn = False
ep_obj.release_name = ''
result = ep_obj.get_sql()
if None is not result:
sql_l.append(result)
else:
# the file exists, set its modify file stamp
if sickgear.AIRDATE_EPISODES:
ep_obj.airdate_modify_stamp()
if deleted:
logger.log('%s: %s %s media file%s and kept %s most recent downloads' % (
self.tvid_prodid, ('Permanently deleted', 'Trashed')[sickgear.TRASH_REMOVE_SHOW],
deleted, helpers.maybe_plural(deleted), kept))
if 0 < len(sql_l):
my_db = db.DBConnection()
my_db.mass_action(sql_l)
def download_subtitles(self, force=False):
# type: (bool) -> None
"""
:param force:
"""
# TODO: Add support for force option
if not ek.ek(os.path.isdir, self._location):
logger.log('%s: Show directory doesn\'t exist, can\'t download subtitles' % self.tvid_prodid, logger.DEBUG)
return
logger.log('%s: Downloading subtitles' % self.tvid_prodid, logger.DEBUG)
try:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT location
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND LENGTH(location) != 0
ORDER BY season DESC, episode DESC
""", [self.tvid, self.prodid])
for cur_row in sql_result:
ep_obj = self.ep_obj_from_file(cur_row['location'])
_ = ep_obj.download_subtitles(force=force)
except (BaseException, Exception):
logger.log('Error occurred when downloading subtitles: %s' % traceback.format_exc(), logger.ERROR)
return
def remove_character_images(self):
try:
img_obj = image_cache.ImageCache()
delete_list = []
for character in self.cast_list:
for p in character.person:
person_img = img_obj.person_both_paths(p)
for i in person_img:
delete_list.append(i)
character_img = img_obj.character_both_path(character, self)
for i in character_img:
delete_list.append(i)
for d in delete_list:
try:
remove_file_perm(d)
except (BaseException, Exception):
pass
except (BaseException, Exception):
pass
def switch_infosrc(self, old_tvid, old_prodid, pausestatus_after=None, update_show=True):
# type: (integer_types, integer_types, Optional[bool], bool) -> None
"""
:param old_tvid: old tvid
:param old_prodid: old prodid
:param pausestatus_after: pause after switch
:param update_show:
"""
with self.lock:
my_db = db.DBConnection()
my_db.mass_action([
['UPDATE tv_shows SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE tv_episodes SET indexer = ?, showid = ?, indexerid = 0 WHERE indexer = ? AND showid = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE blocklist SET indexer = ?, show_id = ? WHERE indexer = ? AND show_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE history SET indexer = ?, showid = ? WHERE indexer = ? AND showid = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE imdb_info SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE scene_exceptions SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE scene_numbering SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE allowlist SET indexer = ?, show_id = ? WHERE indexer = ? AND show_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['UPDATE xem_refresh SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]],
['DELETE FROM tv_shows_not_found WHERE indexer = ? AND indexer_id = ?',
[old_tvid, old_prodid]],
['UPDATE castlist SET indexer = ?, indexer_id = ? WHERE indexer = ? AND indexer_id = ?',
[self.tvid, self.prodid, old_tvid, old_prodid]]
])
my_failed_db = db.DBConnection('failed.db')
my_failed_db.action('UPDATE history SET indexer = ?, showid = ? WHERE indexer = ? AND showid = ?',
[self.tvid, self.prodid, old_tvid, old_prodid])
del_mapping(old_tvid, old_prodid)
try:
for c in self.cast_list: # type: Character
c.remove_all_img(tvid=old_tvid, proid=old_prodid)
except (BaseException, Exception):
pass
self._cast_list = None
self.sxe_ep_obj = {}
self.ids[old_tvid]['status'] = MapStatus.NONE
self.ids[self.tvid]['status'] = MapStatus.SOURCE
self.ids[self.tvid]['id'] = self.prodid
if isinstance(self.imdb_info, dict):
self.imdb_info['indexer'] = self.tvid
self.imdb_info['indexer_id'] = self.prodid
save_mapping(self)
name_cache.remove_from_namecache(old_tvid, old_prodid)
image_cache_dir = ek.ek(os.path.join, sickgear.CACHE_DIR, 'images', 'shows')
old_dir = ek.ek(os.path.join, image_cache_dir, '%s-%s' % (old_tvid, old_prodid))
new_dir = ek.ek(os.path.join, image_cache_dir, '%s-%s' % (self.tvid, self.prodid))
try:
ek.ek(os.rename, old_dir, new_dir)
except (BaseException, Exception) as e:
logger.log('Unable to rename %s to %s: %s / %s' % (old_dir, new_dir, repr(e), ex(e)),
logger.WARNING)
old_id = TVidProdid({old_tvid: old_prodid})()
rating = sickgear.FANART_RATINGS.get(old_id)
if rating:
del sickgear.FANART_RATINGS[old_id]
sickgear.FANART_RATINGS[self.tvid_prodid] = rating
sickgear.save_config()
name_cache.buildNameCache(self)
self.reset_not_found_count()
old_sid_int = self.create_sid(old_tvid, old_prodid)
if old_sid_int != self.sid_int:
try:
del sickgear.showDict[old_sid_int]
except (BaseException, Exception):
pass
sickgear.showDict[self.sid_int] = self
self.save_to_db()
if update_show:
# force the update
try:
sickgear.show_queue_scheduler.action.updateShow(
self, force=True, web=True, priority=QueuePriorities.VERYHIGH,
pausestatus_after=pausestatus_after, switch_src=True)
except exceptions_helper.CantUpdateException as e:
logger.log('Unable to update this show. %s' % ex(e), logger.ERROR)
def save_to_db(self, force_save=False):
# type: (bool) -> None
"""
:param force_save:
"""
if not self.dirty and not force_save:
logger.log('%s: Not saving show to db - record is not dirty' % self.tvid_prodid, logger.DEBUG)
return
logger.log('%s: Saving show info to database' % self.tvid_prodid, logger.DEBUG)
new_value_dict = dict(
air_by_date=self._air_by_date,
airs=self._airs,
airtime=time_to_int(self._airtime),
anime=self._anime,
archive_firstmatch=self._upgrade_once,
classification=self._classification,
dvdorder=self._dvdorder,
flatten_folders=self._flatten_folders,
genre=self._genre,
indexer=self.tvid,
lang=self._lang, imdb_id=self._imdbid,
last_update_indexer=self._last_update_indexer,
location=self._location,
network=self.internal_network,
network_country=self._network_country,
network_country_code=self._network_country_code,
network_id=self._network_id,
network_is_stream=self._network_is_stream,
overview=self._overview,
paused=self._paused,
prune=self._prune,
quality=self._quality,
rls_global_exclude_ignore=','.join(self._rls_global_exclude_ignore),
rls_global_exclude_require=','.join(self._rls_global_exclude_require),
rls_ignore_words=helpers.generate_word_str(self._rls_ignore_words, self._rls_ignore_words_regex),
rls_require_words=helpers.generate_word_str(self._rls_require_words, self._rls_require_words_regex),
runtime=self._runtime,
scene=self._scene,
show_name=self._name,
sports=self._sports,
startyear=self.startyear,
status=self._status,
subtitles=self._subtitles,
tag=self._tag,
timezone=self.internal_timezone,
)
control_value_dict = dict(indexer=self.tvid, indexer_id=self.prodid)
my_db = db.DBConnection()
my_db.upsert('tv_shows', new_value_dict, control_value_dict)
self.dirty = False
if sickgear.USE_IMDB_INFO and len(self._imdb_info):
new_value_dict = self._imdb_info
my_db = db.DBConnection()
my_db.upsert('imdb_info', new_value_dict, control_value_dict)
def __ne__(self, o):
# type: (TVShow) -> bool
return not self.__eq__(o)
def __eq__(self, o):
# type: (TVShow) -> bool
if not isinstance(o, TVShow):
return False
return o.tvid == self.tvid and o.prodid == self.prodid
def __hash__(self):
return hash((self.tvid, self.prodid))
def __repr__(self):
return 'TVShow(%s)' % self.__str__()
def __str__(self):
return 'prodid: %s\n' % self.prodid \
+ 'tvid: %s\n' % self.tvid \
+ 'name: %s\n' % self.name \
+ 'location: %s\n' % self._location \
+ ('', 'network: %s\n' % self.network)[self.network not in (None, '')] \
+ ('', 'airs: %s\n' % self.airs)[self.airs not in (None, '')] \
+ ('', 'status: %s\n' % self.status)[self.status not in (None, '')] \
+ 'startyear: %s\n' % self.startyear \
+ ('', 'genre: %s\n' % self.genre)[self.genre not in (None, '')] \
+ 'classification: %s\n' % self.classification \
+ 'runtime: %s\n' % self.runtime \
+ 'quality: %s\n' % self.quality \
+ 'scene: %s\n' % self.is_scene \
+ 'sports: %s\n' % self.is_sports \
+ 'anime: %s\n' % self.is_anime \
+ 'prune: %s\n' % self.prune
def want_episode(self, season, episode, quality, manual_search=False, multi_ep=False):
# type: (integer_types, integer_types, integer_types, bool, bool) -> bool
"""
:param season: season number
:param episode: episode number
:param quality: quality
:param manual_search: manual search
:param multi_ep: multiple episodes
:return:
"""
logger.log('Checking if found %sepisode %sx%s is wanted at quality %s' %
(('', 'multi-part ')[multi_ep], season, episode, Quality.qualityStrings[quality]), logger.DEBUG)
if not multi_ep:
try:
wq = getattr(self.sxe_ep_obj.get(season, {}).get(episode, {}), 'wanted_quality', None)
if None is not wq:
if quality in wq:
cur_status, cur_quality = Quality.splitCompositeStatus(self.sxe_ep_obj[season][episode].status)
if cur_status in (WANTED, UNAIRED, SKIPPED, FAILED):
logger.log('Existing episode status is wanted/unaired/skipped/failed,'
' getting found episode', logger.DEBUG)
return True
elif manual_search:
logger.log('Usually ignoring found episode, but forced search allows the quality,'
' getting found episode', logger.DEBUG)
return True
elif quality > cur_quality:
logger.log(
'Episode already exists but the found episode has better quality,'
' getting found episode', logger.DEBUG)
return True
logger.log('None of the conditions were met, ignoring found episode', logger.DEBUG)
return False
except (BaseException, Exception):
pass
# if the quality isn't one we want under any circumstances then just say no
initial_qualities, archive_qualities = Quality.splitQuality(self._quality)
all_qualities = list(set(initial_qualities + archive_qualities))
initial = '= (%s)' % ','.join([Quality.qualityStrings[qual] for qual in initial_qualities])
if 0 < len(archive_qualities):
initial = '+ upgrade to %s + (%s)'\
% (initial, ','.join([Quality.qualityStrings[qual] for qual in archive_qualities]))
logger.log('Want initial %s and found %s' % (initial, Quality.qualityStrings[quality]), logger.DEBUG)
if quality not in all_qualities:
logger.log('Don\'t want this quality, ignoring found episode', logger.DEBUG)
return False
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT status
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?
""", [self.tvid, self.prodid, season, episode])
if not sql_result or not len(sql_result):
logger.log('Unable to find a matching episode in database, ignoring found episode', logger.DEBUG)
return False
cur_status, cur_quality = Quality.splitCompositeStatus(int(sql_result[0]['status']))
ep_status_text = statusStrings[cur_status]
logger.log('Existing episode status: %s (%s)' % (statusStrings[cur_status], ep_status_text), logger.DEBUG)
# if we know we don't want it then just say no
if cur_status in [IGNORED, ARCHIVED] + ([SKIPPED], [])[multi_ep] and not manual_search:
logger.log('Existing episode status is %signored/archived, ignoring found episode' %
('skipped/', '')[multi_ep], logger.DEBUG)
return False
# if it's one of these then we want it as long as it's in our allowed initial qualities
if quality in all_qualities:
if cur_status in [WANTED, UNAIRED, SKIPPED, FAILED] + ([], SNATCHED_ANY)[multi_ep]:
logger.log('Existing episode status is wanted/unaired/skipped/failed, getting found episode',
logger.DEBUG)
return True
elif manual_search:
logger.log(
'Usually ignoring found episode, but forced search allows the quality, getting found episode',
logger.DEBUG)
return True
else:
logger.log('Quality is on wanted list, need to check if it\'s better than existing quality',
logger.DEBUG)
downloaded_status_list = SNATCHED_ANY + [DOWNLOADED]
# special case: already downloaded quality is not in any of the wanted Qualities
if cur_status in downloaded_status_list and cur_quality not in all_qualities:
wanted_qualities = all_qualities
else:
wanted_qualities = archive_qualities
# if re-downloading then only keep items in the archiveQualities list and better than what we have
if cur_status in downloaded_status_list and quality in wanted_qualities and quality > cur_quality:
logger.log('Episode already exists but the found episode has better quality, getting found episode',
logger.DEBUG)
return True
else:
logger.log('Episode already exists and the found episode has same/lower quality, ignoring found episode',
logger.DEBUG)
logger.log('None of the conditions were met, ignoring found episode', logger.DEBUG)
return False
def get_overview(self, ep_status, split_snatch=False):
# type: (integer_types, bool) -> integer_types
"""
:param ep_status: episode status
:param split_snatch:
:return:
:rtype: int
"""
return helpers.get_overview(ep_status, self.quality, self.upgrade_once, split_snatch=split_snatch)
def __getstate__(self):
d = dict(self.__dict__)
del d['lock']
return d
def __setstate__(self, d):
d['lock'] = threading.RLock()
self.__dict__.update(d)
def __bool__(self):
return bool(self.tvid) and bool(self.prodid)
__nonzero__ = __bool__
class TVEpisode(TVEpisodeBase):
def __init__(self, show_obj, season, episode, path='', existing_only=False, show_result=None):
# type: (TVShow, integer_types, integer_types, AnyStr, bool, List) -> None
super(TVEpisode, self).__init__(season, episode, int(show_obj.tvid))
self._airtime = None # type: Optional[datetime.time]
self._epid = 0 # type: int
self._location = path # type: AnyStr
self._network = None # type: Optional[AnyStr]
self._network_country = None # type: Optional[AnyStr]
self._network_country_code = None # type: Optional[AnyStr]
self._network_id = None # type: Optional[int]
self._network_is_stream = None # type: Optional[bool]
self._runtime = 0 # type: integer_types
self._show_obj = show_obj # type: TVShow
self._timestamp = None # type: Optional[int]
self._timezone = None # type: Optional[AnyStr]
self._tvid = int(show_obj.tvid) # type: int
self.check_for_meta_files()
self.lock = threading.RLock()
self.related_ep_obj = [] # type: List
self.scene_absolute_number = 0 # type: int
self.scene_episode = 0 # type: int
self.scene_season = 0 # type: int
self.specify_episode(self._season, self._episode, existing_only=existing_only, show_result=show_result)
self.wanted_quality = [] # type: List
@property
def network(self):
return self._network
@network.setter
def network(self, *args):
self.dirty_setter('_network')(self, *args)
@property
def network_id(self):
return self._network_id
@network_id.setter
def network_id(self, *args):
self.dirty_setter('_network_id')(self, *args)
@property
def network_country(self):
return self._network_country
@network_country.setter
def network_country(self, *args):
self.dirty_setter('_network_country')(self, *args)
@property
def network_country_code(self):
return self._network_country_code
@network_country_code.setter
def network_country_code(self, *args):
self.dirty_setter('_network_country_code')(self, *args)
@property
def network_is_stream(self):
return self._network_is_stream
@network_is_stream.setter
def network_is_stream(self, *args):
self.dirty_setter('_network_is_stream')(self, *args)
@property
def airtime(self):
# type: (...) -> Optional[datetime.time]
return self._airtime
@airtime.setter
def airtime(self, *args):
# type: (Optional[datetime.time, AnyStr]) -> None
self.dirty_setter('_airtime')(self, *args)
@property
def runtime(self):
return self._runtime
@runtime.setter
def runtime(self, *args):
self.dirty_setter('_runtime')(self, *args)
@property
def timezone(self):
return self._timezone
@timezone.setter
def timezone(self, *args):
self.dirty_setter('_timezone')(self, *args)
@property
def timestamp(self):
return self._timestamp
@timestamp.setter
def timestamp(self, *args):
self.dirty_setter('_timestamp')(self, *args)
@property
def show_obj(self):
"""
:return: TVShow object
:rtype: TVShow
"""
return self._show_obj
@show_obj.setter
def show_obj(self, val):
self._show_obj = val
@property
def tvid(self):
"""
:rtype: int
"""
return self._tvid
@tvid.setter
def tvid(self, val):
self.dirty_setter('_tvid')(self, int(val))
# TODO: remove the following when indexerid is gone
# in deprecation transition, tvid will also set indexer so that existing uses continue to work normally
self.dirty_setter('_indexer')(self, int(val))
@property
def epid(self):
"""
:rtype: int or long
"""
return self._epid
@epid.setter
def epid(self, val):
self.dirty_setter('_epid')(self, int(val))
# TODO: remove the following when indexerid is gone
# in deprecation transition, epid will also set indexerid so that existing uses continue as normal
self.dirty_setter('_indexerid')(self, int(val))
def _set_location(self, val):
log_vals = (('clears', ''), ('sets', ' to ' + val))[any(val)]
# noinspection PyStringFormat
logger.log(u'Setter %s location%s' % log_vals, logger.DEBUG)
# self._location = newLocation
self.dirty_setter('_location')(self, val)
if val and ek.ek(os.path.isfile, val):
self.file_size = ek.ek(os.path.getsize, val)
else:
self.file_size = 0
location = property(lambda self: self._location, _set_location)
def refresh_subtitles(self):
"""Look for subtitles files and refresh the subtitles property"""
if sickgear.USE_SUBTITLES:
self.subtitles = subtitles.subtitles_languages(self.location)
def download_subtitles(self, force=False):
"""
:param force:
:type force: bool
:return:
:rtype:
"""
if not sickgear.USE_SUBTITLES:
return
# TODO: Add support for force option
if not ek.ek(os.path.isfile, self.location):
logger.log('%s: Episode file doesn\'t exist, can\'t download subtitles for episode %sx%s' %
(self.show_obj.tvid_prodid, self.season, self.episode), logger.DEBUG)
return
logger.log('%s: Downloading subtitles for episode %sx%s'
% (self.show_obj.tvid_prodid, self.season, self.episode), logger.DEBUG)
previous_subtitles = self.subtitles
try:
need_languages = list(set(sickgear.SUBTITLES_LANGUAGES) - set(self.subtitles))
subs = subliminal.download_subtitles([self.location], languages=need_languages,
services=sickgear.subtitles.get_enabled_service_list(),
force=force, multi=True, cache_dir=sickgear.CACHE_DIR,
os_auth=sickgear.SUBTITLES_SERVICES_AUTH[0],
os_hash=sickgear.SUBTITLES_OS_HASH)
if sickgear.SUBTITLES_DIR:
for video in subs:
subs_new_path = ek.ek(os.path.join, ek.ek(os.path.dirname, video.path), sickgear.SUBTITLES_DIR)
dir_exists = helpers.make_dir(subs_new_path)
if not dir_exists:
logger.log('Unable to create subtitles folder %s' % subs_new_path, logger.ERROR)
else:
helpers.chmod_as_parent(subs_new_path)
for subtitle in subs.get(video):
new_file_path = ek.ek(os.path.join, subs_new_path, ek.ek(os.path.basename, subtitle.path))
helpers.move_file(subtitle.path, new_file_path)
helpers.chmod_as_parent(new_file_path)
else:
for video in subs:
for subtitle in subs.get(video):
helpers.chmod_as_parent(subtitle.path)
except (BaseException, Exception):
logger.log('Error occurred when downloading subtitles: %s' % traceback.format_exc(), logger.ERROR)
return
self.refresh_subtitles()
# added the if because sometime it raises an error
self.subtitles_searchcount = self.subtitles_searchcount + 1 if self.subtitles_searchcount else 1
self.subtitles_lastsearch = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
self.save_to_db()
newsubtitles = set(self.subtitles).difference(set(previous_subtitles))
if newsubtitles:
try:
subtitle_list = ", ".join([subliminal.language.Language(x).name for x in newsubtitles])
except (BaseException, Exception):
logger.log('Could not parse a language to use to fetch subtitles for episode %sx%s' %
(self.season, self.episode), logger.DEBUG)
return
logger.log('%s: Downloaded %s subtitles for episode %sx%s' %
(self.show_obj.tvid_prodid, subtitle_list, self.season, self.episode), logger.DEBUG)
notifiers.notify_subtitle_download(self, subtitle_list)
else:
logger.log('%s: No subtitles downloaded for episode %sx%s'
% (self.show_obj.tvid_prodid, self.season, self.episode), logger.DEBUG)
if sickgear.SUBTITLES_HISTORY:
for video in subs:
for subtitle in subs.get(video):
history.log_subtitle(self.show_obj.tvid, self.show_obj.prodid,
self.season, self.episode, self.status, subtitle)
return subs
def check_for_meta_files(self):
"""
:return:
:rtype: bool
"""
oldhasnfo = self._hasnfo
oldhastbn = self._hastbn
hasnfo = False
hastbn = False
# check for nfo and tbn
if ek.ek(os.path.isfile, self.location):
for cur_provider in itervalues(sickgear.metadata_provider_dict):
if cur_provider.episode_metadata:
new_result = cur_provider.has_episode_metadata(self)
else:
new_result = False
hasnfo = new_result or hasnfo
if cur_provider.episode_thumbnails:
new_result = cur_provider.has_episode_thumb(self)
else:
new_result = False
hastbn = new_result or hastbn
self.hasnfo = hasnfo
self.hastbn = hastbn
# if either setting has changed return true, if not return false
return oldhasnfo != self._hasnfo or oldhastbn != self._hastbn
def specify_episode(self, season, episode, existing_only=False, **kwargs):
"""
kwargs['show_result']: type: Optional[List[Row]] passed thru
:param season: season number
:type season: int
:param episode: episode number
:type episode: int
:param existing_only:
:type existing_only: bool
"""
if not self.load_from_db(season, episode, **kwargs):
# only load from NFO if we didn't load from DB
if ek.ek(os.path.isfile, self.location):
try:
self.load_from_nfo(self.location)
except exceptions_helper.NoNFOException:
logger.log('%s: There was an error loading the NFO for episode %sx%s' %
(self.show_obj.tvid_prodid, season, episode), logger.ERROR)
pass
# if we tried loading it from NFO and didn't find the NFO, try the Indexers
if not self._hasnfo:
try:
self.load_from_tvinfo(season, episode)
except exceptions_helper.EpisodeDeletedException:
# if we failed SQL *and* NFO, Indexers then fail
raise exceptions_helper.EpisodeNotFoundException(
'Couldn\'t find episode %sx%s' % (season, episode))
elif existing_only:
raise exceptions_helper.EpisodeNotFoundException(
'Couldn\'t find episode %sx%s' % (season, episode))
def load_from_db(self, season, episode, show_result=None, **kwargs):
# type: (int, int, Optional[List[Row]], Any) -> bool
"""
kwargs['scene_result']: type: Optional[List[Row]] passed thru
:param season: season number
:param episode: episode number
:param show_result:
"""
logger.log('%s: Loading episode details from DB for episode %sx%s'
% (self._show_obj.tvid_prodid, season, episode), logger.DEBUG)
show_result = show_result and next(iter(show_result), None)
if not show_result or episode != show_result['episode'] or season != show_result['season']:
my_db = db.DBConnection()
sql_result = my_db.select(
"""
SELECT *
FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?
LIMIT 2
""", [self._show_obj.tvid, self._show_obj.prodid, season, episode])
if 1 != len(sql_result):
if len(sql_result):
raise exceptions_helper.MultipleDBEpisodesException('DB has multiple records for the same show')
logger.log('%s: Episode %sx%s not found in the database'
% (self._show_obj.tvid_prodid, self._season, self._episode), logger.DEBUG)
return False
show_result = next(iter(sql_result))
# todo: change to _tvid , _epid after removing indexer, indexerid
self._tvid = int(show_result['indexer'])
self._epid = int(show_result['indexerid'])
self._absolute_number = show_result['absolute_number']
self._airdate = datetime.date.fromordinal(int(show_result['airdate']))
self._airtime = int_to_time(try_int(show_result['airtime'], None))
self._description = self._description if not show_result['description'] else show_result['description']
self._episode = episode
self._file_size = try_int(show_result['file_size'])
self._is_proper = self._is_proper if not show_result['is_proper'] else try_int(show_result['is_proper'])
self._name = self._name if not show_result['name'] else show_result['name']
self._network = show_result['network'] or self._show_obj.internal_network
self._network_country = show_result['network_country']
self._network_country_code = show_result['network_country_code']
self._network_id = show_result['network_id']
self._network_is_stream = bool(show_result['network_is_stream'])
self._runtime = show_result['runtime']
self._season = season
self._status = self._status if None is show_result['status'] else int(show_result['status'])
subt_value = show_result['subtitles']
if ',,' in subt_value: # a defensive fix for a data issue
subt_value = re.sub(r',+', '', subt_value)
subt_value = ('', subt_value)[2 == len(subt_value)]
self._subtitles = subt_value and subt_value.split(',') or []
self._subtitles_lastsearch = show_result['subtitles_lastsearch']
self._subtitles_searchcount = show_result['subtitles_searchcount']
self._timestamp = show_result['timestamp'] or self._make_timestamp()
self._version = self._version if not show_result['version'] else int(show_result['version'])
self.location = show_result['location'] and ek.ek(os.path.normpath, show_result['location']) or self.location
if None is not show_result['release_group']:
self._release_group = show_result['release_group']
if None is not show_result['release_name']:
self._release_name = show_result['release_name']
sickgear.scene_numbering.xem_refresh(self._show_obj.tvid, self._show_obj.prodid)
self.scene_absolute_number = try_int(show_result['scene_absolute_number'])
if 0 == self.scene_absolute_number:
self.scene_absolute_number = sickgear.scene_numbering.get_scene_absolute_numbering(
self._show_obj.tvid, self._show_obj.prodid,
absolute_number=self._absolute_number, season=self._season, episode=episode,
show_result=show_result, show_obj=self._show_obj, **kwargs)
self.scene_season = try_int(show_result['scene_season'])
self.scene_episode = try_int(show_result['scene_episode'])
if 0 == self.scene_season or 0 == self.scene_episode:
self.scene_season, self.scene_episode = sickgear.scene_numbering.get_scene_numbering(
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode, show_obj=self._show_obj,
show_result=show_result, **kwargs)
self._timezone = show_result['timezone']
if not self._timezone and (self._network or self._show_obj.internal_timezone):
if self._show_obj.internal_timezone:
self._timezone = self._show_obj.internal_timezone
else:
_, self.timezone = network_timezones.get_network_timezone(self._network, return_name=True)
self.dirty = False
return True
# noinspection PyProtectedMember
def _make_timestamp(self):
# type: (...) -> Optional[integer_types]
if self.airdate:
if isinstance(self._airtime, datetime.time):
ep_time = self._airtime
elif isinstance(self._show_obj._airtime, datetime.time):
ep_time = self._show_obj._airtime
else:
ep_time = datetime.time(hour=0, minute=0)
tzinfo = None
if isinstance(self._timezone, string_types) and self._timezone:
tzinfo = tz.gettz(self._timezone, zoneinfo_priority=True)
elif isinstance(self._network, string_types) and self._network:
tzinfo = tz.gettz(self._network, zoneinfo_priority=True)
elif isinstance(self._show_obj.timezone, string_types) and self._show_obj.timezone:
tzinfo = self._show_obj.timezone
elif isinstance(self._show_obj.network, string_types) and self._show_obj.network:
tzinfo = network_timezones.get_network_timezone(self._show_obj.network)
if PY2:
return SGDatetime.combine(self.airdate, ep_time).replace(tzinfo=tzinfo).timestamp_far()
return SGDatetime.combine(self.airdate, ep_time, tzinfo=tzinfo).timestamp_far()
return None
def load_from_tvinfo(
self,
season=None, # type: integer_types
episode=None, # type: integer_types
cache=True, # type: bool
tvapi=None, # type: Any
cached_season=None, # type: Dict
update=False, # type: bool
cached_show=None, # type: Dict
switch=False, # type: bool
old_tvid=None, # type: int
old_prodid=None, # type: integer_types
switch_list=None # type: List
): # type: (...) -> Optional[bool]
"""
:param season: season number
:param episode: episode number
:param cache:
:param tvapi:
:param cached_season:
:param update:
:param cached_show:
:param switch:
:param old_tvid:
:param old_prodid:
:param switch_list:
"""
if None is season:
season = self._season
if None is episode:
episode = self._episode
logger.log('%s: Loading episode details from %s for episode %sx%s' %
(self._show_obj.tvid_prodid, sickgear.TVInfoAPI(self._show_obj.tvid).name, season, episode),
logger.DEBUG)
try:
if cached_show:
ep_info = cached_show[season][episode]
elif None is cached_season:
if None is tvapi:
tvinfo_config = sickgear.TVInfoAPI(self.tvid).api_params.copy()
if not cache:
tvinfo_config['cache'] = False
show_lang = self._show_obj.lang
if show_lang:
tvinfo_config['language'] = show_lang
if 0 != self._show_obj.dvdorder:
tvinfo_config['dvdorder'] = True
t = sickgear.TVInfoAPI(self.tvid).setup(**tvinfo_config)
else:
t = tvapi
ep_info = t.get_show(self._show_obj.prodid,
language=self.show_obj.lang)[season][episode] # type: TVInfoEpisode
else:
ep_info = cached_season[episode] # type: TVInfoEpisode
except (BaseTVinfoEpisodenotfound, BaseTVinfoSeasonnotfound):
logger.log('Unable to find the episode on %s... has it been removed? Should I delete from db?' %
sickgear.TVInfoAPI(self.tvid).name, logger.DEBUG)
# if I'm no longer on the Indexers but I once was then delete myself from the DB
if -1 != self._epid and helpers.should_delete_episode(self._status):
self.delete_episode()
elif UNKNOWN == self._status:
self.status = SKIPPED
return
except (BaseTVinfoError, IOError) as e:
logger.log('%s threw up an error: %s' % (sickgear.TVInfoAPI(self.tvid).name, ex(e)), logger.DEBUG)
# if the episode is already valid just log it, if not throw it up
if UNKNOWN == self._status:
self.status = SKIPPED
if self._name:
logger.log('%s timed out but there is enough info from other sources, allowing the error' %
sickgear.TVInfoAPI(self.tvid).name, logger.DEBUG)
return
logger.log('%s timed out, unable to create the episode' % sickgear.TVInfoAPI(self.tvid).name,
logger.ERROR)
return False
if getattr(ep_info, 'absolute_number', None) in (None, ''):
logger.debug('This episode (%s - %sx%s) has no absolute number on %s' %
(self.show_obj.unique_name, season, episode, sickgear.TVInfoAPI(self.tvid).name))
else:
logger.log('%s: The absolute_number for %sx%s is : %s' %
(self._show_obj.tvid_prodid, season, episode, ep_info['absolute_number']), logger.DEBUG)
self.absolute_number = int(ep_info['absolute_number'])
if switch and None is not switch_list:
if self._name != self.dict_prevent_nonetype(ep_info, 'episodename'):
switch_list.append(self.show_obj.switch_ep_change_sql(old_tvid, old_prodid, episode, season,
TVSWITCH_EP_RENAMED))
self.name = self.dict_prevent_nonetype(ep_info, 'episodename')
self.season = season
self.episode = episode
sickgear.scene_numbering.xem_refresh(self._show_obj.tvid, self._show_obj.prodid)
self.scene_absolute_number = sickgear.scene_numbering.get_scene_absolute_numbering(
self._show_obj.tvid, self._show_obj.prodid,
absolute_number=self._absolute_number,
season=self._season, episode=self._episode, show_obj=self._show_obj)
self.scene_season, self.scene_episode = sickgear.scene_numbering.get_scene_numbering(
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode, show_obj=self._show_obj)
self.description = self.dict_prevent_nonetype(ep_info, 'overview')
firstaired = getattr(ep_info, 'firstaired', None)
if None is firstaired or firstaired in '0000-00-00':
firstaired = str(datetime.date.fromordinal(1))
raw_airdate = [int(x) for x in firstaired.split('-')]
old_airdate_future = self._airdate == datetime.date.fromordinal(1) or self._airdate >= datetime.date.today()
try:
self.airdate = datetime.date(raw_airdate[0], raw_airdate[1], raw_airdate[2])
except (ValueError, IndexError):
logger.error('Malformed air date retrieved from %s (%s - %sx%s)' %
(sickgear.TVInfoAPI(self.tvid).name, self.show_obj.unique_name, season, episode))
# if I'm incomplete on TVDB but I once was complete then just delete myself from the DB for now
if -1 != self._epid and helpers.should_delete_episode(self._status):
self.delete_episode()
elif UNKNOWN == self._status:
self.status = SKIPPED
return False
self.network = ep_info.network
if not self._network:
self.network = self._show_obj.internal_network
self.network_id = ep_info.network_id
self.network_country = ep_info.network_country
self.network_country_code = ep_info.network_country_code
self.timezone = ep_info.network_timezone
if not self._timezone and (self._network or self._show_obj.internal_timezone):
if self._show_obj.internal_timezone:
self._timezone = self._show_obj.internal_timezone
else:
_, self.timezone = network_timezones.get_network_timezone(self._network, return_name=True)
self.network_is_stream = ep_info.network_is_stream
self.airtime = ep_info.airtime
self.runtime = ep_info.runtime
invalid_date = False
if not (self._airdate and self._airdate < invalid_date_limit):
self.timestamp = ep_info.timestamp or self._make_timestamp()
else:
self.timestamp = None
invalid_date = True
today = datetime.date.today()
delta = datetime.timedelta(days=1)
if invalid_date:
show_time = None
elif self._timestamp and self._timezone:
try:
show_time = SGDatetime.from_timestamp(self._timestamp, tz_aware=True, local_time=False,
tzinfo=tz.gettz(self._timezone))
except OverflowError:
logger.debug('Invalid timestamp: %s, using fallback' % self._timestamp)
show_time = network_timezones.parse_date_time(self._airdate.toordinal(),
self._airtime or self._show_obj.airs,
self._network or self._show_obj.network)
else:
show_time = network_timezones.parse_date_time(self._airdate.toordinal(),
self._airtime or self._show_obj.airs,
self._network or self._show_obj.network)
tz_now = datetime.datetime.now(network_timezones.SG_TIMEZONE)
show_length = datetime.timedelta(minutes=helpers.try_int(self._runtime or self._show_obj.runtime, 60))
future_airtime = not invalid_date and (
(self._airdate > (today + delta) or
(not self._airdate < (today - delta) and ((show_time + show_length) > tz_now))))
# early conversion to int so that episode doesn't get marked dirty
self.epid = getattr(ep_info, 'id', None)
if None is self._epid:
logger.log('Failed to retrieve ID from %s' % sickgear.TVInfoAPI(self.tvid).name, logger.ERROR)
if helpers.should_delete_episode(self._status):
self.delete_episode()
elif UNKNOWN == self._status:
self.status = (SKIPPED, UNAIRED)[future_airtime]
return False
# don't update show status if show dir is missing, unless it's missing on purpose
# noinspection PyProtectedMember
if not ek.ek(os.path.isdir, self._show_obj._location) \
and not sickgear.CREATE_MISSING_SHOW_DIRS and not sickgear.ADD_SHOWS_WO_DIR:
if UNKNOWN == self._status:
self.status = (SKIPPED, UNAIRED)[future_airtime]
logger.log('The show directory is missing but an episode status found at Unknown is set Skipped')
else:
logger.log('The show directory is missing,'
' not bothering to change the episode statuses since it\'d probably be invalid')
return
if self._location:
logger.log('%s: Setting status for %sx%s based on status %s and existence of %s' %
(self._show_obj.tvid_prodid, season, episode, statusStrings[self._status], self._location),
logger.DEBUG)
# if we don't have the file
if not ek.ek(os.path.isfile, self._location):
if self._status in [SKIPPED, UNAIRED, UNKNOWN, WANTED]:
very_old_delta = datetime.timedelta(days=90)
very_old_airdate = datetime.date.fromordinal(1) < self._airdate < (today - very_old_delta)
# if this episode hasn't aired yet set the status to UNAIRED
if future_airtime:
msg = 'Episode airs in the future, marking it %s'
self.status = UNAIRED
# if there's no airdate then set it to unaired (and respect ignored)
elif self._airdate == datetime.date.fromordinal(1):
if IGNORED == self._status:
msg = 'Episode has no air date and marked %s, no change'
else:
msg = 'Episode has no air date, marking it %s'
self.status = UNAIRED
# if the airdate is in the past
elif UNAIRED == self._status:
msg = ('Episode status %s%s, with air date in the past, marking it ' % (
statusStrings[self._status], ','.join([(' is a special', '')[0 < self._season],
('', ' is paused')[self._show_obj.paused]])) + '%s')
self.status = (SKIPPED, WANTED)[0 < self._season
and not self._show_obj.paused and not very_old_airdate]
# if still UNKNOWN or SKIPPED with the deprecated future airdate method
elif UNKNOWN == self._status or (SKIPPED == self._status and old_airdate_future):
msg = ('Episode status %s%s, with air date in the past, marking it ' % (
statusStrings[self._status], ','.join([
('', ' has old future date format')[SKIPPED == self._status and old_airdate_future],
('', ' is being updated')[bool(update)], (' is a special', '')[0 < self._season]])) + '%s')
self.status = (SKIPPED, WANTED)[update and not self._show_obj.paused and 0 < self._season
and not very_old_airdate]
else:
msg = 'Not touching episode status %s, with air date in the past, because there is no file'
else:
msg = 'Not touching episode status %s, because there is no file'
logger.log(msg % statusStrings[self._status], logger.DEBUG)
# if we have a media file then it's downloaded
elif sickgear.helpers.has_media_ext(self._location):
if IGNORED == self._status:
logger.log('File exists for %sx%s, ignoring because of status %s' %
(self._season, self._episode, statusStrings[self._status]), logger.DEBUG)
# leave propers alone, you have to either post-process them or manually change them back
elif self._status not in Quality.SNATCHED_ANY + Quality.DOWNLOADED + Quality.ARCHIVED:
msg = '(1) Status changes from %s to ' % statusStrings[self._status]
self.status = Quality.statusFromNameOrFile(self._location, anime=self._show_obj.is_anime)
logger.log('%s%s' % (msg, statusStrings[self._status]), logger.DEBUG)
# shouldn't get here probably
else:
msg = '(2) Status changes from %s to ' % statusStrings[self._status]
self.status = UNKNOWN
logger.log('%s%s' % (msg, statusStrings[self._status]), logger.DEBUG)
def load_from_nfo(self, location):
"""
:param location:
:type location: AnyStr
"""
# noinspection PyProtectedMember
if not ek.ek(os.path.isdir, self._show_obj._location):
logger.log('%s: The show directory is missing, not bothering to try loading the episode NFO'
% self._show_obj.tvid_prodid)
return
logger.log('%s: Loading episode details from the NFO file associated with %s'
% (self.show_obj.tvid_prodid, location), logger.DEBUG)
self.location = location
if '' != self.location:
if UNKNOWN == self._status and sickgear.helpers.has_media_ext(self.location):
status_quality = Quality.statusFromNameOrFile(self.location, anime=self._show_obj.is_anime)
logger.log('(3) Status changes from %s to %s' % (self._status, status_quality), logger.DEBUG)
self.status = status_quality
nfo_file = sickgear.helpers.replace_extension(self.location, 'nfo')
logger.log('%s: Using NFO name %s' % (self._show_obj.tvid_prodid, nfo_file), logger.DEBUG)
if ek.ek(os.path.isfile, nfo_file):
try:
show_xml = etree.ElementTree(file=nfo_file)
except (SyntaxError, ValueError) as e:
logger.log('Error loading the NFO, backing up the NFO and skipping for now: %s' % ex(e),
logger.ERROR) # TODO: figure out what's wrong and fix it
try:
ek.ek(os.rename, nfo_file, '%s.old' % nfo_file)
except (BaseException, Exception) as e:
logger.log(
'Failed to rename your episode\'s NFO file - you need to delete it or fix it: %s' % ex(e),
logger.ERROR)
raise exceptions_helper.NoNFOException('Error in NFO format')
# TODO: deprecated function getiterator needs to be replaced
# for epDetails in showXML.getiterator('episodedetails'):
for epDetails in list(show_xml.iter('episodedetails')):
if None is epDetails.findtext('season') or int(epDetails.findtext('season')) != self._season or \
None is epDetails.findtext('episode') or int(
epDetails.findtext('episode')) != self._episode:
logger.log('%s: NFO has an <episodedetails> block for a different episode - wanted %sx%s'
' but got %sx%s' %
(self._show_obj.tvid_prodid, self._season, self._episode,
epDetails.findtext('season'), epDetails.findtext('episode')), logger.DEBUG)
continue
if None is epDetails.findtext('title') or None is epDetails.findtext('aired'):
raise exceptions_helper.NoNFOException('Error in NFO format (missing episode title or airdate)')
self.name = epDetails.findtext('title')
self.episode = int(epDetails.findtext('episode'))
self.season = int(epDetails.findtext('season'))
sickgear.scene_numbering.xem_refresh(self._show_obj.tvid, self._show_obj.prodid)
self.scene_absolute_number = sickgear.scene_numbering.get_scene_absolute_numbering(
self._show_obj.tvid, self._show_obj.prodid,
absolute_number=self._absolute_number,
season=self._season, episode=self._episode, show_obj=self._show_obj)
self.scene_season, self.scene_episode = sickgear.scene_numbering.get_scene_numbering(
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode,
show_obj=self._show_obj)
self.description = epDetails.findtext('plot')
if None is self._description:
self.description = ''
if epDetails.findtext('aired'):
raw_airdate = [int(x) for x in epDetails.findtext('aired').split("-")]
self.airdate = datetime.date(raw_airdate[0], raw_airdate[1], raw_airdate[2])
else:
self.airdate = datetime.date.fromordinal(1)
self.hasnfo = True
else:
self.hasnfo = False
if ek.ek(os.path.isfile, sickgear.helpers.replace_extension(nfo_file, 'tbn')):
self.hastbn = True
else:
self.hastbn = False
def __ne__(self, o):
# type: (TVEpisode) -> bool
return not self.__eq__(o)
def __eq__(self, o):
# type: (TVEpisode) -> bool
if not isinstance(o, TVEpisode):
return False
return o._show_obj == self._show_obj and o._epid == self._epid
def __hash__(self):
return hash((self._show_obj, self._epid))
def __repr__(self):
return 'TVEpisode(%s)' % self.__str__()
def __str__(self):
return '%s - %sx%s - %s\n' % (self.show_obj.unique_name, self.season, self.episode, self.name) \
+ 'location: %s\n' % self.location \
+ 'description: %s\n' % self.description \
+ 'subtitles: %s\n' % ','.join(self.subtitles) \
+ 'subtitles_searchcount: %s\n' % self.subtitles_searchcount \
+ 'subtitles_lastsearch: %s\n' % self.subtitles_lastsearch \
+ 'airdate: %s (%s)\n' % (self.airdate.toordinal(), self.airdate) \
+ 'hasnfo: %s\n' % self.hasnfo \
+ 'hastbn: %s\n' % self.hastbn \
+ 'status: %s\n' % self.status
def create_meta_files(self, force=False):
# noinspection PyProtectedMember
if not ek.ek(os.path.isdir, self.show_obj._location):
logger.log('%s: The show directory is missing, not bothering to try to create metadata'
% self.show_obj.tvid_prodid)
return
self.create_nfo(force)
self.create_thumbnail()
if self.check_for_meta_files():
self.save_to_db()
def create_nfo(self, force=False):
"""
:return:
:rtype: bool
"""
result = False
for cur_provider in itervalues(sickgear.metadata_provider_dict):
result = cur_provider.create_episode_metadata(self, force) or result
return result
def create_thumbnail(self):
"""
:return:
:rtype: bool
"""
result = False
for cur_provider in itervalues(sickgear.metadata_provider_dict):
result = cur_provider.create_episode_thumb(self) or result
return result
def delete_episode(self, return_sql=False):
# type: (bool) -> Optional[List[List]]
"""
deletes episode from db, alternatively returns sql to remove episode from db
:param return_sql: only return sql to delete episode
"""
logger.debug('Deleting %s %sx%s from the DB' % (self._show_obj.unique_name, self._season, self._episode))
# remove myself from the show dictionary
if self.show_obj.get_episode(self._season, self._episode, no_create=True) == self:
logger.log('Removing myself from my show\'s list', logger.DEBUG)
del self.show_obj.sxe_ep_obj[self._season][self._episode]
# delete myself from the DB
logger.log('Deleting myself from the database', logger.DEBUG)
sql = [['DELETE FROM tv_episodes WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?',
[self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode]]]
if return_sql:
return sql
my_db = db.DBConnection()
my_db.mass_action(sql)
raise exceptions_helper.EpisodeDeletedException()
def get_sql(self, force_save=False):
# type: (bool) -> Optional[List[AnyStr, List]]
"""
Creates SQL queue for this episode if any of its data has been changed since the last save.
:param force_save: If True it will create SQL queue even if no data has been changed since the last save
(aka if the record is not dirty).
"""
if not self.dirty and not force_save:
logger.log('%s: Not creating SQL queue - record is not dirty' % self._show_obj.tvid_prodid, logger.DEBUG)
return
self.dirty = False
return [
"""
INSERT OR REPLACE INTO tv_episodes
(episode_id,
indexerid, indexer, name, description,
subtitles, subtitles_searchcount, subtitles_lastsearch,
airdate, hasnfo, hastbn, status, location, file_size,
release_name, is_proper, showid, season, episode, absolute_number,
version, release_group,
network, network_id, network_country, network_country_code, network_is_stream,
airtime, runtime, timestamp, timezone,
scene_absolute_number, scene_season, scene_episode)
VALUES
((SELECT episode_id FROM tv_episodes
WHERE indexer = ? AND showid = ?
AND season = ? AND episode = ?)
,?,?
,?,?
,?,?,?
,?,?,?,?,?,?
,?,?
,?,?,?,?
,?,?
,?,?,?,?,?
,?,?,?,?,
(SELECT scene_absolute_number FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?),
(SELECT scene_season FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?),
(SELECT scene_episode FROM tv_episodes
WHERE indexer = ? AND showid = ? AND season = ? AND episode = ?));
""", [self._show_obj.tvid, self._show_obj.prodid,
self._season, self._episode,
self._epid, self._tvid,
self._name, self._description,
','.join([sub for sub in self._subtitles]), self._subtitles_searchcount, self._subtitles_lastsearch,
self._airdate.toordinal(), self._hasnfo, self._hastbn, self._status, self._location, self._file_size,
self._release_name, self._is_proper,
self._show_obj.prodid, self._season, self._episode, self._absolute_number,
self._version, self._release_group,
self._network, self._network_id,
self._network_country, self._network_country_code, self._network_is_stream,
time_to_int(self._airtime), self._runtime, self._timestamp, self._timezone,
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode,
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode,
self._show_obj.tvid, self._show_obj.prodid, self._season, self._episode]]
def save_to_db(self, force_save=False):
"""
Saves this episode to the database if any of its data has been changed since the last save.
:param force_save: If True it will save to the database even if no data has been changed since the
last save (aka if the record is not dirty).
"""
if not self.dirty and not force_save:
logger.log('%s: Not saving episode to db - record is not dirty' % self._show_obj.tvid_prodid, logger.DEBUG)
return
logger.log('%s: Saving episode details to database' % self._show_obj.tvid_prodid, logger.DEBUG)
logger.log('STATUS IS %s' % statusStrings[self._status], logger.DEBUG)
new_value_dict = dict(
absolute_number=self._absolute_number,
airdate=self._airdate.toordinal(),
airtime=time_to_int(self._airtime),
description=self._description,
file_size=self._file_size,
hasnfo=self._hasnfo,
hastbn=self._hastbn,
indexer=self._tvid,
indexerid=self._epid,
is_proper=self._is_proper,
location=self._location,
name=self._name,
network=self._network,
network_country=self._network_country,
network_country_code=self._network_country_code,
network_id=self._network_id,
network_is_stream=self._network_is_stream,
release_group=self._release_group,
release_name=self._release_name,
runtime=self._runtime,
status=self._status,
subtitles=','.join([sub for sub in self._subtitles]),
subtitles_lastsearch=self._subtitles_lastsearch,
subtitles_searchcount=self._subtitles_searchcount,
timestamp=self._timestamp,
timezone=self._timezone,
version=self._version,
)
control_value_dict = dict(
indexer=self.show_obj.tvid, showid=self.show_obj.prodid, season=self.season, episode=self.episode)
# use a custom update/insert method to get the data into the DB
my_db = db.DBConnection()
my_db.upsert('tv_episodes', new_value_dict, control_value_dict)
self.dirty = False
# # TODO: remove if unused
# def full_location(self):
# if self.location in (None, ''):
# return None
# return ek.ek(os.path.join, self.show_obj.location, self.location)
#
# # TODO: remove if unused
# def create_strings(self, pattern=None):
# patterns = [
# '%S.N.S%SE%0E',
# '%S.N.S%0SE%E',
# '%S.N.S%SE%E',
# '%S.N.S%0SE%0E',
# '%SN S%SE%0E',
# '%SN S%0SE%E',
# '%SN S%SE%E',
# '%SN S%0SE%0E'
# ]
#
# strings = []
# if not pattern:
# for p in patterns:
# strings += [self._format_pattern(p)]
# return strings
# return self._format_pattern(pattern)
def pretty_name(self):
"""
Returns the name of this episode in a "pretty" human-readable format. Used for logging
and notifications and such.
:return: A string representing the episode's name and season/ep numbers
:rtype: AnyStr
"""
if self._show_obj.anime and not self._show_obj.scene:
return self._format_pattern('%SN - %AB - %EN')
if self._show_obj.air_by_date:
return self._format_pattern('%SN - %AD - %EN')
return self._format_pattern('%SN - %Sx%0E - %EN')
def _ep_name(self):
"""
:return: the name of the episode to use during renaming. Combines the names of related episodes.
Eg. "Ep Name (1)" and "Ep Name (2)" becomes "Ep Name"
"Ep Name" and "Other Ep Name" becomes "Ep Name & Other Ep Name"
:rtype: AnyStr
"""
multi_name_regex = r'(.*) \(\d{1,2}\)'
self.related_ep_obj = sorted(self.related_ep_obj, key=lambda se: se.episode)
if 0 == len(self.related_ep_obj):
good_name = self._name
else:
single_name = True
known_good_name = None
for cur_name in [self._name] + [x.name for x in self.related_ep_obj]:
match = re.match(multi_name_regex, cur_name)
if not match:
single_name = False
break
if None is known_good_name:
known_good_name = match.group(1)
elif known_good_name != match.group(1):
single_name = False
break
if single_name:
good_name = known_good_name
else:
good_name = self._name
for ep_obj in self.related_ep_obj:
good_name += ' & ' + ep_obj.name
return good_name or 'tba'
def _replace_map(self):
"""
Generates a replacement map for this episode which maps all possible custom naming patterns to the correct
value for this episode.
Returns: A dict with patterns as the keys and their replacement values as the values.
"""
ep_name = self._ep_name()
def dot(name):
return helpers.sanitize_scene_name(name)
def us(name):
return re.sub('[ -]', '_', name)
def release_name(name, is_anime=False):
if name:
name = helpers.remove_non_release_groups(name, is_anime)
return name
def release_group(show_obj, name):
if name:
name = helpers.remove_non_release_groups(name, show_obj.is_anime)
else:
return ''
try:
np = NameParser(name, show_obj=show_obj, naming_pattern=True)
parse_result = np.parse(name)
except (InvalidNameException, InvalidShowException) as e:
logger.log('Unable to get parse release_group: %s' % ex(e), logger.DEBUG)
return ''
if not parse_result.release_group:
return ''
return parse_result.release_group
ep_status, ep_qual = Quality.splitCompositeStatus(self._status)
if sickgear.NAMING_STRIP_YEAR:
show_name = re.sub(r'\(\d+\)$', '', self._show_obj.name).rstrip()
else:
show_name = self._show_obj.name
return {
'%SN': show_name,
'%S.N': dot(show_name),
'%S_N': us(show_name),
'%EN': ep_name,
'%E.N': dot(ep_name),
'%E_N': us(ep_name),
'%QN': Quality.qualityStrings[ep_qual],
'%Q.N': dot(Quality.qualityStrings[ep_qual]),
'%Q_N': us(Quality.qualityStrings[ep_qual]),
'%S': str(self._season),
'%0S': '%02d' % self._season,
'%E': str(self._episode),
'%0E': '%02d' % self._episode,
'%XS': str(self.scene_season),
'%0XS': '%02d' % self.scene_season,
'%XE': str(self.scene_episode),
'%0XE': '%02d' % self.scene_episode,
'%AB': '%(#)03d' % {'#': self._absolute_number},
'%XAB': '%(#)03d' % {'#': self.scene_absolute_number},
'%RN': release_name(self._release_name, self._show_obj.is_anime),
'%RG': release_group(self._show_obj, self._release_name),
'%AD': str(self._airdate).replace('-', ' '),
'%A.D': str(self._airdate).replace('-', '.'),
'%A_D': us(str(self._airdate)),
'%A-D': str(self._airdate),
'%Y': str(self._airdate.year),
'%M': str(self._airdate.month),
'%D': str(self._airdate.day),
'%0M': '%02d' % self._airdate.month,
'%0D': '%02d' % self._airdate.day,
'%RT': "PROPER" if self.is_proper else "",
'%V': 'v%s' % self._version if self._show_obj.is_anime and 1 < self._version else '',
}
@staticmethod
def _format_string(pattern, replace_map):
"""
Replaces all template strings with the correct value
"""
result_name = pattern
# do the replacements
for cur_replacement in sorted(list_keys(replace_map), reverse=True):
result_name = result_name.replace(cur_replacement, helpers.sanitize_filename(replace_map[cur_replacement]))
result_name = result_name.replace(cur_replacement.lower(),
helpers.sanitize_filename(replace_map[cur_replacement].lower()))
return result_name
def _format_pattern(self, pattern=None, multi=None, anime_type=None):
"""
Manipulates an episode naming pattern and then fills the template in
"""
if None is pattern:
pattern = sickgear.NAMING_PATTERN
if None is multi:
multi = sickgear.NAMING_MULTI_EP
if None is anime_type:
anime_type = sickgear.NAMING_ANIME
replace_map = self._replace_map()
result_name = pattern
# if there's no release group then replace it with a reasonable facsimile
if not replace_map['%RN']:
if self._show_obj.air_by_date or self._show_obj.sports:
result_name = result_name.replace('%RN', '%S.N.%A.D.%E.N-SickGear')
result_name = result_name.replace('%rn', '%s.n.%A.D.%e.n-SickGear')
elif 3 != anime_type:
result_name = result_name.replace('%RN', '%S.N.%AB.%E.N-SickGear')
result_name = result_name.replace('%rn', '%s.n.%ab.%e.n-SickGear')
else:
result_name = result_name.replace('%RN', '%S.N.S%0SE%0E.%E.N-SickGear')
result_name = result_name.replace('%rn', '%s.n.s%0se%0e.%e.n-SickGear')
result_name = result_name.replace('%RG', 'SickGear')
result_name = result_name.replace('%rg', 'SickGear')
logger.log('Episode has no release name, replacing it with a generic one: %s' % result_name, logger.DEBUG)
if not replace_map['%RT']:
result_name = re.sub('([ _.-]*)%RT([ _.-]*)', r'\2', result_name)
# split off ep name part only
name_groups = re.split(r'[\\/]', result_name)
# figure out the double-ep numbering style for each group, if applicable
for cur_name_group in name_groups:
season_ep_regex = r'''
(?P<pre_sep>[ _.-]*)
((?:s(?:eason|eries)?\s*)?%0?S(?![._]?N))
(.*?)
(%0?E(?![._]?N))
(?P<post_sep>[ _.-]*)
'''
ep_only_regex = '(E?%0?E(?![._]?N))'
# try the normal way
season_ep_match = re.search(season_ep_regex, cur_name_group, re.I | re.X)
ep_only_match = re.search(ep_only_regex, cur_name_group, re.I | re.X)
# if we have a season and episode then collect the necessary data
if season_ep_match:
season_format = season_ep_match.group(2)
ep_sep = season_ep_match.group(3)
ep_format = season_ep_match.group(4)
sep = season_ep_match.group('pre_sep')
if not sep:
sep = season_ep_match.group('post_sep')
if not sep:
sep = ' '
# force 2-3-4 format if they chose to extend
if multi in (NAMING_EXTEND, NAMING_LIMITED_EXTEND, NAMING_LIMITED_EXTEND_E_PREFIXED):
ep_sep = '-'
regex_used = season_ep_regex
# if there's no season then there's not much choice so we'll just force them to use 03-04-05 style
elif ep_only_match:
season_format = ''
ep_sep = '-'
ep_format = ep_only_match.group(1)
sep = ''
regex_used = ep_only_regex
else:
continue
# we need at least this much info to continue
if not ep_sep or not ep_format:
continue
# start with the ep string, eg. E03
ep_string = self._format_string(ep_format.upper(), replace_map)
for cur_ep_obj in self.related_ep_obj:
# for limited extend we only append the last ep
if multi in (NAMING_LIMITED_EXTEND, NAMING_LIMITED_EXTEND_E_PREFIXED) \
and cur_ep_obj != self.related_ep_obj[-1]:
continue
elif multi == NAMING_DUPLICATE:
# add " - S01"
ep_string += sep + season_format
elif multi == NAMING_SEPARATED_REPEAT:
ep_string += sep
# add "E04"
ep_string += ep_sep
if multi == NAMING_LIMITED_EXTEND_E_PREFIXED:
ep_string += 'E'
# noinspection PyProtectedMember
ep_string += cur_ep_obj._format_string(ep_format.upper(), cur_ep_obj._replace_map())
if 3 != anime_type:
absolute_number = (self._absolute_number, self._episode)[0 == self._absolute_number]
if 0 != self._season: # dont set absolute numbers if we are on specials !
if 1 == anime_type: # this crazy person wants both ! (note: +=)
ep_string += sep + '%(#)03d' % {'#': absolute_number}
elif 2 == anime_type: # total anime freak only need the absolute number ! (note: =)
ep_string = '%(#)03d' % {'#': absolute_number}
for cur_ep_obj in self.related_ep_obj:
if 0 != cur_ep_obj.absolute_number:
ep_string += '-' + '%(#)03d' % {'#': cur_ep_obj.absolute_number}
else:
ep_string += '-' + '%(#)03d' % {'#': cur_ep_obj.episode}
regex_replacement = None
if 2 == anime_type:
regex_replacement = r'\g<pre_sep>' + ep_string + r'\g<post_sep>'
elif season_ep_match:
regex_replacement = r'\g<pre_sep>\g<2>\g<3>' + ep_string + r'\g<post_sep>'
elif ep_only_match:
regex_replacement = ep_string
if regex_replacement:
# fill out the template for this piece and then insert this piece into the actual pattern
cur_name_group_result = re.sub('(?i)(?x)' + regex_used, regex_replacement, cur_name_group)
# cur_name_group_result = cur_name_group.replace(ep_format, ep_string)
# logger.log(u"found "+ep_format+" as the ep pattern using "+regex_used+"
# and replaced it with "+regex_replacement+" to result in "+cur_name_group_result+"
# from "+cur_name_group, logger.DEBUG)
result_name = result_name.replace(cur_name_group, cur_name_group_result)
result_name = self._format_string(result_name, replace_map)
logger.log('formatting pattern: %s -> %s' % (pattern, result_name), logger.DEBUG)
return result_name
def proper_path(self):
"""
Figures out the path where this episode SHOULD live according to the renaming rules, relative from the show dir
:return:
:rtype: AnyStr
"""
anime_type = sickgear.NAMING_ANIME
if not self._show_obj.is_anime:
anime_type = 3
result = self.formatted_filename(anime_type=anime_type)
# if they want us to flatten it and we're allowed to flatten it then we will
if self._show_obj.flatten_folders and not sickgear.NAMING_FORCE_FOLDERS:
return result
# if not we append the folder on and use that
return ek.ek(os.path.join, self.formatted_dir(), result)
def formatted_dir(self, pattern=None, multi=None):
"""
Just the folder name of the episode
"""
if None is pattern:
# we only use ABD if it's enabled, this is an ABD show, AND this is not a multi-ep
if self._show_obj.air_by_date and sickgear.NAMING_CUSTOM_ABD and not self.related_ep_obj:
pattern = sickgear.NAMING_ABD_PATTERN
elif self._show_obj.sports and sickgear.NAMING_CUSTOM_SPORTS and not self.related_ep_obj:
pattern = sickgear.NAMING_SPORTS_PATTERN
elif self._show_obj.anime and sickgear.NAMING_CUSTOM_ANIME:
pattern = sickgear.NAMING_ANIME_PATTERN
else:
pattern = sickgear.NAMING_PATTERN
# split off the dirs only, if they exist
name_groups = re.split(r'[\\/]', pattern)
if 1 == len(name_groups):
logger.debug('No Season Folder set in Naming pattern: %s' % pattern)
return ''
return self._format_pattern(ek.ek(os.sep.join, name_groups[:-1]), multi)
def formatted_filename(self, pattern=None, multi=None, anime_type=None):
"""
Just the filename of the episode, formatted based on the naming settings
"""
if None is pattern:
# we only use ABD if it's enabled, this is an ABD show, AND this is not a multi-ep
if self._show_obj.air_by_date and sickgear.NAMING_CUSTOM_ABD and not self.related_ep_obj:
pattern = sickgear.NAMING_ABD_PATTERN
elif self._show_obj.sports and sickgear.NAMING_CUSTOM_SPORTS and not self.related_ep_obj:
pattern = sickgear.NAMING_SPORTS_PATTERN
elif self._show_obj.anime and sickgear.NAMING_CUSTOM_ANIME:
pattern = sickgear.NAMING_ANIME_PATTERN
else:
pattern = sickgear.NAMING_PATTERN
# split off the dirs only, if they exist
name_groups = re.split(r'[\\/]', pattern)
return self._format_pattern(name_groups[-1], multi, anime_type)
def rename(self):
"""
Renames an episode file and all related files to the location and filename as specified
in the naming settings.
"""
if not ek.ek(os.path.isfile, self.location):
logger.log('Can\'t perform rename on %s when it doesn\'t exist, skipping' % self.location, logger.WARNING)
return
proper_path = self.proper_path()
absolute_proper_path = ek.ek(os.path.join, self._show_obj.location, proper_path)
absolute_current_path_no_ext, file_ext = ek.ek(os.path.splitext, self.location)
absolute_current_path_no_ext_length = len(absolute_current_path_no_ext)
related_subs = []
current_path = absolute_current_path_no_ext
if absolute_current_path_no_ext.startswith(self._show_obj.location):
current_path = absolute_current_path_no_ext[len(self._show_obj.location):]
logger.log('Renaming/moving episode from the base path %s to %s' % (self.location, absolute_proper_path),
logger.DEBUG)
# if it's already named correctly then don't do anything
if proper_path == current_path:
logger.log('%s: File %s is already named correctly, skipping' % (self._epid, self.location),
logger.DEBUG)
return
related_files = postProcessor.PostProcessor(self.location).list_associated_files(
self.location, base_name_only=True)
if self.show_obj.subtitles and '' != sickgear.SUBTITLES_DIR:
related_subs = postProcessor.PostProcessor(self.location).list_associated_files(sickgear.SUBTITLES_DIR,
subtitles_only=True)
# absolute_proper_subs_path = ek.ek(os.path.join, sickgear.SUBTITLES_DIR, self.formatted_filename())
logger.log('Files associated to %s: %s' % (self.location, related_files), logger.DEBUG)
# move the ep file
result = helpers.rename_ep_file(self.location, absolute_proper_path, absolute_current_path_no_ext_length)
# move related files
for cur_related_file in related_files:
renamed = helpers.rename_ep_file(cur_related_file, absolute_proper_path,
absolute_current_path_no_ext_length)
if not renamed:
logger.log('%s: Unable to rename file %s' % (self._epid, cur_related_file), logger.ERROR)
for cur_related_sub in related_subs:
absolute_proper_subs_path = ek.ek(os.path.join, sickgear.SUBTITLES_DIR, self.formatted_filename())
renamed = helpers.rename_ep_file(cur_related_sub, absolute_proper_subs_path,
absolute_current_path_no_ext_length)
if not renamed:
logger.log('%s: Unable to rename file %s' % (self._epid, cur_related_sub), logger.ERROR)
# save the ep
with self.lock:
if result:
self.location = absolute_proper_path + file_ext
for ep_obj in self.related_ep_obj:
ep_obj.location = absolute_proper_path + file_ext
# in case something changed with the metadata just do a quick check
for cur_ep_obj in [self] + self.related_ep_obj:
cur_ep_obj.check_for_meta_files()
# save any changes to the database
sql_l = []
with self.lock:
for ep_obj in [self] + self.related_ep_obj:
result = ep_obj.get_sql()
if None is not result:
sql_l.append(result)
if 0 < len(sql_l):
my_db = db.DBConnection()
my_db.mass_action(sql_l)
def airdate_modify_stamp(self):
"""
Make the modify date and time of a file reflect the show air date and time.
Note: Also called from postProcessor
"""
has_timestamp = isinstance(self._timestamp, int) and 1 < self._timestamp
if not has_timestamp and (not isinstance(self._airdate, datetime.date) or 1 == self._airdate.year):
logger.log('%s: Did not change modify date of %s because episode date is never aired or invalid'
% (self._show_obj.tvid_prodid, ek.ek(os.path.basename, self.location)), logger.DEBUG)
return
aired_dt = None
if not has_timestamp:
if isinstance(self._airtime, datetime.time):
airtime = self._airtime
else:
hr, m = network_timezones.parse_time(self._show_obj.airs)
airtime = datetime.time(hr, m)
aired_dt = SGDatetime.combine(self.airdate, airtime)
try:
aired_epoch = SGDatetime.to_file_timestamp(aired_dt)
filemtime = int(ek.ek(os.path.getmtime, self.location))
except (BaseException, Exception):
return
if filemtime != aired_epoch:
result, loglevel = 'Changed', logger.MESSAGE
if not helpers.touch_file(self.location, aired_epoch):
result, loglevel = 'Error changing', logger.WARNING
logger.log('%s: %s modify date of %s to show air date %s'
% (self._show_obj.tvid_prodid, result, ek.ek(os.path.basename, self.location),
'n/a' if not aired_dt else aired_dt.strftime('%b %d,%Y (%H:%M)')), loglevel)
def __getstate__(self):
d = dict(self.__dict__)
del d['lock']
return d
def __setstate__(self, d):
d['lock'] = threading.RLock()
self.__dict__.update(d)
def __bool__(self):
return bool(self._tvid) and bool(self._epid)
__nonzero__ = __bool__