class AnimeTests(SiCKRAGETestDBCase):
def setUp(self):
super(AnimeTests, self).setUp()
self.show = TVShow(1, 1, 'en')
self.show.anime = True
self.show.saveToDB()
def _test_names(self, np, section, transform=None, verbose=False):
"""
Performs a test
:param name_parser: to use for test
:param section:
:param transform:
:param verbose:
:return:
"""
if VERBOSE or verbose:
print()
print('Running', section, 'tests')
for cur_test_base in anime_test_cases[section]:
if transform:
cur_test = transform(cur_test_base)
np.file_name = cur_test
else:
cur_test = cur_test_base
if VERBOSE or verbose:
print('Testing', cur_test)
result = anime_test_cases[section][cur_test_base]
np.showObj.name = result.series_name if result else None
if not result:
self.assertRaises(InvalidNameException, np.parse, cur_test)
return
else:
result.which_regex = {section}
test_result = np.parse(cur_test)
if DEBUG or verbose:
print('air_by_date:', test_result.is_air_by_date, 'air_date:',
test_result.air_date)
print('anime:', test_result.is_anime, 'ab_episode_numbers:',
test_result.ab_episode_numbers)
print(test_result)
print(result)
self.assertEqual(
test_result.which_regex, {section},
'{} : {} != {}'.format(cur_test, test_result.which_regex,
{section}))
self.assertEqual(
test_result, result,
'{} : {} != {}'.format(cur_test, test_result, result))
def test_anime_sxxexx_file_names(self):
"""
Test anime SxxExx file names
"""
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'anime_SxxExx', lambda x: x + '.avi')
def _test_allPossibleShowNames(self, name, indexerid=0, expected=[]):
s = TVShow(1, indexerid)
s.name = name
s.saveToDB()
s.loadFromDB(skipNFO=True)
result = show_names.allPossibleShowNames(s)
self.assertTrue(
len(set(expected).intersection(set(result))) == len(expected))
def _test_allPossibleShowNames(self, name, indexerid=0, expected=[]):
s = TVShow(1, indexerid)
s.name = name
s.saveToDB()
s.loadFromDB(skipNFO=True)
result = show_names.allPossibleShowNames(s)
self.assertTrue(len(set(expected).intersection(set(result))) == len(expected))
def _test_allPossibleShowNames(self, name, indexerid=0, expected=None):
if expected is None:
expected = []
s = TVShow(1, indexerid)
s.name = name
s.saveToDB()
result = show_names.allPossibleShowNames(s)
self.assertTrue(len(set(expected).intersection(set(result))) == len(expected))
def test_init_empty_db(self):
show = TVShow(1, 0001, "en")
show.saveToDB()
show.loadFromDB(skipNFO=True)
ep = TVEpisode(show, 1, 1)
ep.name = "asdasdasdajkaj"
ep.saveToDB()
ep.loadFromDB(1, 1)
self.assertEqual(ep.name, "asdasdasdajkaj")
def test_getEpisode(self):
show = TVShow(1, 0001, "en")
show.name = "show name"
show.network = "cbs"
show.genre = "crime"
show.runtime = 40
show.status = "Ended"
show.default_ep_status = "5"
show.airs = "monday"
show.startyear = 1987
show.saveToDB()
sickrage.showList = [show]
def test_process(self):
show = TVShow(1, 3)
show.name = self.SHOWNAME
show.location = self.SHOWDIR
show.saveToDB()
sickrage.app.showlist = [show]
sickrage.app.name_cache.put('show name', 3)
self.post_processor = PostProcessor(self.FILEPATH,
process_method='move')
self.post_processor._log = _log
self.assertTrue(self.post_processor.process)
def test_getEpisode(self):
show = TVShow(1, 0001, "en")
show.name = "show name"
show.network = "cbs"
show.genre = "crime"
show.runtime = 40
show.status = "Ended"
show.default_ep_status = "5"
show.airs = "monday"
show.startyear = 1987
show.saveToDB()
show.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST = [show]
def test_process(self):
show = TVShow(1, 3)
show.name = self.SHOWNAME
show.location = self.SHOWDIR
show.saveToDB()
sickrage.app.showlist = [show]
ep = TVEpisode(show, self.SEASON, self.EPISODE)
ep.name = "some ep name"
ep.saveToDB()
sickrage.app.name_cache.put('show name', 3)
self.pp = PostProcessor(self.FILEPATH, process_method='move')
self.assertTrue(self.pp.process)
def loadFromDB(self):
"""
Populates the showList with shows from the database
"""
for s in [s['doc'] for s in sickrage.srCore.mainDB.db.all('tv_shows', with_doc=True)]:
try:
curShow = TVShow(int(s["indexer"]), int(s["indexer_id"]))
curShow.saveToDB()
curShow.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(curShow)
except Exception as e:
print "There was an error creating the show"
def loadFromDB(self):
"""
Populates the showList with shows from the database
"""
for s in [s['doc'] for s in MainDB().db.all('tv_shows', with_doc=True)]:
try:
curShow = TVShow(int(s["indexer"]), int(s["indexer_id"]))
curShow.saveToDB()
curShow.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(curShow)
except Exception as e:
print "There was an error creating the show"
def test_process(self):
show = TVShow(1, 3)
show.name = SHOWNAME
show.location = SHOWDIR
show.saveToDB()
sickrage.showList = [show]
ep = TVEpisode(show, SEASON, EPISODE)
ep.name = "some ep name"
ep.saveToDB()
sickrage.NAMECACHE.addNameToCache("show name", 3)
self.pp = PostProcessor(FILEPATH, process_method="move")
self.assertTrue(self.pp.process)
def test_process(self):
show = TVShow(1, 3)
show.name = SHOWNAME
show.location = SHOWDIR
show.saveToDB()
show.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST = [show]
ep = TVEpisode(show, SEASON, EPISODE)
ep.name = "some ep name"
ep.saveToDB()
sickrage.srCore.NAMECACHE.addNameToCache('show name', 3)
self.pp = PostProcessor(FILEPATH, process_method='move')
self.assertTrue(self.pp.process)
def test_process(self):
show = TVShow(1, 3)
show.name = SHOWNAME
show.location = SHOWDIR
show.saveToDB()
show.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST = [show]
ep = TVEpisode(show, SEASON, EPISODE)
ep.name = "some ep name"
ep.saveToDB()
sickrage.srCore.NAMECACHE.put('show name', 3)
self.pp = PostProcessor(FILEPATH, process_method='move')
self.assertTrue(self.pp.process)
def test_isGoodName(self):
list_of_cases = [('Show.Name.S01E02.Test-Test', 'Show/Name'),
('Show.Name.S01E02.Test-Test', 'Show. Name'),
('Show.Name.S01E02.Test-Test', 'Show- Name'),
('Show.Name.Part.IV.Test-Test', 'Show Name'),
('Show.Name.S01.Test-Test', 'Show Name'),
('Show.Name.E02.Test-Test', 'Show: Name'),
('Show Name Season 2 Test', 'Show: Name')]
for testCase in list_of_cases:
scene_name, show_name = testCase
s = TVShow(1, 0)
s.name = show_name
s.saveToDB()
self._test_isGoodName(scene_name, s)
def test_change_indexerid(self):
show = TVShow(1, 0001, "en")
show.name = "show name"
show.network = "cbs"
show.genre = "crime"
show.runtime = 40
show.status = "Ended"
show.default_ep_status = "5"
show.airs = "monday"
show.startyear = 1987
show.indexerid = 0002
show.saveToDB()
show.loadFromDB(skipNFO=True)
self.assertEqual(show.indexerid, 0002)
def loadFromDB(self):
"""
Populates the showList with shows from the database
"""
sqlResults = main_db.MainDB().select("SELECT * FROM tv_shows")
for sqlShow in sqlResults:
try:
curShow = TVShow(int(sqlShow["indexer"]), int(sqlShow["indexer_id"]))
curShow.saveToDB()
curShow.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(curShow)
except Exception as e:
print "There was an error creating the show"
def loadFromDB(self):
"""
Populates the showList with shows from the database
"""
sqlResults = main_db.MainDB().select("SELECT * FROM tv_shows")
for sqlShow in sqlResults:
try:
curShow = TVShow(int(sqlShow["indexer"]),
int(sqlShow["indexer_id"]))
curShow.saveToDB()
curShow.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(curShow)
except Exception as e:
print "There was an error creating the show"
def loadShowsFromDB(self):
"""
Populates the showList with shows from the database
"""
for s in [
s['doc']
for s in sickrage.app.main_db.db.all('tv_shows', with_doc=True)
]:
try:
curShow = TVShow(int(s["indexer"]), int(s["indexer_id"]))
curShow.saveToDB()
curShow.loadFromDB(skipNFO=True)
sickrage.app.showlist.append(curShow)
except Exception:
pass
class UnicodeTests(SiCKRAGETestDBCase):
def setUp(self):
super(UnicodeTests, self).setUp()
self.show = TVShow(1, 1, 'en')
self.show.name = "The Big Bang Theory"
self.show.saveToDB()
def _test_unicode(self, name, result):
np = NameParser(True, showObj=self.show, validate_show=False)
parse_result = np.parse(name)
# this shouldn't raise an exception
repr(str(parse_result))
self.assertEqual(parse_result.extra_info, result.extra_info)
def test_unicode(self):
for (name, result) in unicode_test_cases:
self._test_unicode(name, result)
def test_isGoodName(self):
listOfcases = [('Show.Name.S01E02.Test-Test', 'Show/Name'),
('Show.Name.S01E02.Test-Test', 'Show. Name'),
('Show.Name.S01E02.Test-Test', 'Show- Name'),
('Show.Name.Part.IV.Test-Test', 'Show Name'),
('Show.Name.S01.Test-Test', 'Show Name'),
('Show.Name.E02.Test-Test', 'Show: Name'),
('Show Name Season 2 Test', 'Show: Name'),
]
for testCase in listOfcases:
scene_name, show_name = testCase
s = TVShow(1, 0)
s.name = show_name
s.saveToDB()
s.loadFromDB(skipNFO=True)
self._test_isGoodName(scene_name, s)
del s
def test_isGoodName(self):
listOfcases = [('Show.Name.S01E02.Test-Test', 'Show/Name'),
('Show.Name.S01E02.Test-Test', 'Show. Name'),
('Show.Name.S01E02.Test-Test', 'Show- Name'),
('Show.Name.Part.IV.Test-Test', 'Show Name'),
('Show.Name.S01.Test-Test', 'Show Name'),
('Show.Name.E02.Test-Test', 'Show: Name'),
('Show Name Season 2 Test', 'Show: Name'),
]
for testCase in listOfcases:
scene_name, show_name = testCase
s = TVShow(1, 0)
s.name = show_name
s.saveToDB()
s.loadFromDB(skipNFO=True)
self._test_isGoodName(scene_name, s)
del s
def test(self):
global searchItems
searchItems = curData[b"i"]
show = TVShow(1, tvdbdid)
show.name = show_name
show.quality = curData[b"q"]
show.saveToDB()
sickrage.showList.append(show)
episode = None
for epNumber in curData[b"e"]:
episode = TVEpisode(show, curData[b"s"], epNumber)
episode.status = WANTED
episode.saveToDB()
bestResult = searchProviders(show, episode.episode, forceSearch)
if not bestResult:
self.assertEqual(curData[b"b"], bestResult)
self.assertEqual(curData[b"b"], bestResult.name) # first is expected, second is choosen one
def test(self):
global searchItems
searchItems = curData["i"]
show = TVShow(1, tvdbdid)
show.name = show_name
show.quality = curData["q"]
show.saveToDB()
show.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(show)
episode = None
for epNumber in curData["e"]:
episode = TVEpisode(show, curData["s"], epNumber)
episode.status = WANTED
episode.saveToDB()
bestResult = searchProviders(show, episode.episode, forceSearch)
if not bestResult:
self.assertEqual(curData["b"], bestResult)
self.assertEqual(curData["b"], bestResult.name) # first is expected, second is choosen one
class UnicodeTests(SiCKRAGETestDBCase):
def __init__(self, something):
super(UnicodeTests, self).__init__(something)
self.setUp()
self.show = TVShow(1, 1, 'en')
self.show.name = "The Big Bang Theory"
self.show.saveToDB()
self.show.loadFromDB(skipNFO=True)
def _test_unicode(self, name, result):
np = NameParser(True, showObj=self.show)
parse_result = np.parse(name)
# this shouldn't raise an exception
repr(str(parse_result))
self.assertEqual(parse_result.extra_info, result.extra_info)
def test_unicode(self):
for (name, result) in unicode_test_cases:
self._test_unicode(name, result)
class UnicodeTests(SiCKRAGETestDBCase):
def __init__(self, something):
super(UnicodeTests, self).__init__(something)
self.setUp()
self.show = TVShow(1, 1, "en")
self.show.name = "The Big Bang Theory"
self.show.saveToDB()
self.show.loadFromDB(skipNFO=True)
def _test_unicode(self, name, result):
np = NameParser(True, showObj=self.show)
parse_result = np.parse(name)
# this shouldn't raise an exception
repr(str(parse_result))
self.assertEqual(parse_result.extra_info, result.extra_info)
def test_unicode(self):
for (name, result) in unicode_test_cases:
self._test_unicode(name, result)
class QueueItemAdd(ShowQueueItem):
def __init__(self, indexer, indexer_id, showDir, default_status, quality, flatten_folders, lang, subtitles, anime,
scene, paused, blacklist, whitelist, default_status_after, archive):
super(QueueItemAdd, self).__init__(ShowQueueActions.ADD, None)
self.indexer = indexer
self.indexer_id = indexer_id
self.showDir = showDir
self.default_status = default_status
self.quality = quality
self.flatten_folders = flatten_folders
self.lang = lang
self.subtitles = subtitles
self.anime = anime
self.scene = scene
self.paused = paused
self.blacklist = blacklist
self.whitelist = whitelist
self.default_status_after = default_status_after
self.archive = archive
self.priority = QueuePriorities.HIGH
@property
def show_name(self):
"""
Returns the show name if there is a show object created, if not returns
the dir that the show is being added to.
"""
if self.show is None:
return os.path.basename(self.showDir)
return self.show.name
@property
def isLoading(self):
"""
Returns True if we've gotten far enough to have a show object, or False
if we still only know the folder name.
"""
if self.show is None:
return True
def run(self):
super(QueueItemAdd, self).run()
sickrage.srCore.srLogger.info("Started adding show {}".format(self.showDir))
index_name = srIndexerApi(self.indexer).name
# make sure the Indexer IDs are valid
try:
lINDEXER_API_PARMS = srIndexerApi(self.indexer).api_params.copy()
if self.lang:
lINDEXER_API_PARMS['language'] = self.lang
sickrage.srCore.srLogger.info("{}: {}".format(index_name, repr(lINDEXER_API_PARMS)))
t = srIndexerApi(self.indexer).indexer(**lINDEXER_API_PARMS)
s = t[self.indexer_id]
# this usually only happens if they have an NFO in their show dir which gave us a Indexer ID that has no proper english version of the show
if not getattr(s, 'seriesname'):
sickrage.srCore.srLogger.error(
"Show in {} has no name on {}, probably the wrong language used to search with".format(self.showDir,
index_name))
sickrage.srCore.srNotifications.error("Unable to add show",
"Show in {} has no name on {}, probably the wrong language. Delete .nfo and add manually in the correct language".format(
self.showDir, index_name))
return self._finishEarly()
# if the show has no episodes/seasons
if not len(s):
sickrage.srCore.srLogger.error("Show " + str(s['seriesname']) + " is on " + str(
srIndexerApi(self.indexer).name) + " but contains no season/episode data.")
sickrage.srCore.srNotifications.error("Unable to add show",
"Show " + str(s['seriesname']) + " is on " + str(srIndexerApi(
self.indexer).name) + " but contains no season/episode data.")
return self._finishEarly()
except Exception as e:
sickrage.srCore.srLogger.error(
"{}: Error while loading information from indexer {}. Error: {}".format(self.indexer_id, index_name,
e.message))
sickrage.srCore.srNotifications.error(
"Unable to add show",
"Unable to look up the show in {} on {} using ID {}, not using the NFO. Delete .nfo and try adding manually again.".format(
self.showDir, index_name, self.indexer_id)
)
if sickrage.srCore.srConfig.USE_TRAKT:
trakt_id = srIndexerApi(self.indexer).config['trakt_id']
trakt_api = TraktAPI(sickrage.srCore.srConfig.SSL_VERIFY, sickrage.srCore.srConfig.TRAKT_TIMEOUT)
title = self.showDir.split("/")[-1]
data = {
'shows': [
{
'title': title,
'ids': {}
}
]
}
if trakt_id == 'tvdb_id':
data['shows'][0]['ids']['tvdb'] = self.indexer_id
else:
data['shows'][0]['ids']['tvrage'] = self.indexer_id
trakt_api.traktRequest("sync/watchlist/remove", data, method='POST')
return self._finishEarly()
try:
self.show = TVShow(self.indexer, self.indexer_id, self.lang)
self.show.loadFromIndexer()
# set up initial values
self.show.location = self.showDir
self.show.subtitles = self.subtitles or sickrage.srCore.srConfig.SUBTITLES_DEFAULT
self.show.quality = self.quality or sickrage.srCore.srConfig.QUALITY_DEFAULT
self.show.flatten_folders = self.flatten_folders or sickrage.srCore.srConfig.FLATTEN_FOLDERS_DEFAULT
self.show.anime = self.anime or sickrage.srCore.srConfig.ANIME_DEFAULT
self.show.scene = self.scene or sickrage.srCore.srConfig.SCENE_DEFAULT
self.show.archive_firstmatch = self.archive or sickrage.srCore.srConfig.ARCHIVE_DEFAULT
self.show.paused = self.paused or False
# set up default new/missing episode status
sickrage.srCore.srLogger.info(
"Setting all current episodes to the specified default status: " + str(self.default_status))
self.show.default_ep_status = self.default_status
if self.show.anime:
self.show.release_groups = BlackAndWhiteList(self.show.indexerid)
if self.blacklist:
self.show.release_groups.set_black_keywords(self.blacklist)
if self.whitelist:
self.show.release_groups.set_white_keywords(self.whitelist)
# # be smartish about this
# if self.show.genre and "talk show" in self.show.genre.lower():
# self.show.air_by_date = 1
# if self.show.genre and "documentary" in self.show.genre.lower():
# self.show.air_by_date = 0
# if self.show.classification and "sports" in self.show.classification.lower():
# self.show.sports = 1
except indexer_exception as e:
sickrage.srCore.srLogger.error(
"Unable to add show due to an error with " + srIndexerApi(
self.indexer).name + ": {}".format(e.message))
if self.show:
sickrage.srCore.srNotifications.error(
"Unable to add " + str(self.show.name) + " due to an error with " + srIndexerApi(
self.indexer).name + "")
else:
sickrage.srCore.srNotifications.error(
"Unable to add show due to an error with " + srIndexerApi(self.indexer).name + "")
return self._finishEarly()
except MultipleShowObjectsException:
sickrage.srCore.srLogger.warning("The show in " + self.showDir + " is already in your show list, skipping")
sickrage.srCore.srNotifications.error('Show skipped',
"The show in " + self.showDir + " is already in your show list")
return self._finishEarly()
except Exception as e:
sickrage.srCore.srLogger.error("Error trying to add show: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
raise self._finishEarly()
sickrage.srCore.srLogger.debug("Retrieving show info from TMDb")
try:
self.show.loadTMDbInfo()
except Exception as e:
sickrage.srCore.srLogger.error("Error loading TMDb info: {}".format(e.message))
try:
sickrage.srCore.srLogger.debug("Attempting to retrieve show info from IMDb")
self.show.loadIMDbInfo()
except Exception as e:
sickrage.srCore.srLogger.error("Error loading IMDb info: {}".format(e.message))
# Load XEM data to DB for show
xem_refresh(self.show.indexerid, self.show.indexer, force=True)
# check if show has XEM mapping so we can determin if searches should go by scene numbering or indexer numbering.
if not self.scene and get_xem_numbering_for_show(self.show.indexerid, self.show.indexer):
self.show.scene = 1
try:
self.show.saveToDB()
except Exception as e:
sickrage.srCore.srLogger.error("Error saving the show to the database: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
raise self._finishEarly()
# add it to the show list
sickrage.srCore.SHOWLIST.append(self.show)
try:
self.show.loadEpisodesFromIndexer()
except Exception as e:
sickrage.srCore.srLogger.error(
"Error with " + srIndexerApi(
self.show.indexer).name + ", not creating episode list: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
try:
self.show.loadEpisodesFromDir()
except Exception as e:
sickrage.srCore.srLogger.debug("Error searching dir for episodes: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
# if they set default ep status to WANTED then run the backlog to search for episodes
if self.show.default_ep_status == WANTED:
sickrage.srCore.srLogger.info("Launching backlog for this show since its episodes are WANTED")
sickrage.srCore.BACKLOGSEARCHER.searchBacklog([self.show])
self.show.writeMetadata()
self.show.updateMetadata()
self.show.populateCache()
if sickrage.srCore.srConfig.USE_TRAKT:
# if there are specific episodes that need to be added by trakt
sickrage.srCore.TRAKTSEARCHER.manageNewShow(self.show)
# add show to trakt.tv library
if sickrage.srCore.srConfig.TRAKT_SYNC:
sickrage.srCore.TRAKTSEARCHER.addShowToTraktLibrary(self.show)
if sickrage.srCore.srConfig.TRAKT_SYNC_WATCHLIST:
sickrage.srCore.srLogger.info("update watchlist")
sickrage.srCore.notifiersDict.trakt_notifier.update_watchlist(show_obj=self.show)
# After initial add, set to default_status_after.
sickrage.srCore.srLogger.info(
"Setting all future episodes to the specified default status: " + str(self.default_status_after))
self.show.default_ep_status = self.default_status_after
self.show.saveToDB()
sickrage.srCore.NAMECACHE.buildNameCache(self.show)
sickrage.srCore.srLogger.info("Finished adding show {}".format(self.showDir))
def _finishEarly(self):
if self.show:
sickrage.srCore.SHOWQUEUE.removeShow(self.show)
class QueueItemAdd(ShowQueueItem):
def __init__(self, indexer, indexer_id, showDir, default_status, quality, flatten_folders, lang, subtitles,
subtitles_sr_metadata, anime, scene, paused, blacklist, whitelist, default_status_after, archive):
super(QueueItemAdd, self).__init__(None, ShowQueueActions.ADD)
self.indexer = indexer
self.indexer_id = indexer_id
self.showDir = showDir
self.default_status = default_status
self.quality = quality
self.flatten_folders = flatten_folders
self.lang = lang
self.subtitles = subtitles
self.subtitles_sr_metadata = subtitles_sr_metadata
self.anime = anime
self.scene = scene
self.paused = paused
self.blacklist = blacklist
self.whitelist = whitelist
self.default_status_after = default_status_after
self.archive = archive
self.priority = srQueuePriorities.HIGH
@property
def show_name(self):
"""
Returns the show name if there is a show object created, if not returns
the dir that the show is being added to.
"""
if self.show is None:
return os.path.basename(self.showDir)
return self.show.name
@property
def isLoading(self):
"""
Returns True if we've gotten far enough to have a show object, or False
if we still only know the folder name.
"""
if self.show is None:
return True
def run(self):
sickrage.app.log.info("Started adding show {} from show dir: {}".format(self.show_name, self.showDir))
index_name = IndexerApi(self.indexer).name
# make sure the Indexer IDs are valid
try:
lINDEXER_API_PARMS = IndexerApi(self.indexer).api_params.copy()
lINDEXER_API_PARMS['cache'] = False
lINDEXER_API_PARMS['language'] = self.lang or sickrage.app.config.indexer_default_language
sickrage.app.log.info("{}: {}".format(index_name, repr(lINDEXER_API_PARMS)))
t = IndexerApi(self.indexer).indexer(**lINDEXER_API_PARMS)
s = t[self.indexer_id]
if not s: return self._finishEarly()
# this usually only happens if they have an NFO in their show dir which gave us a Indexer ID that has no proper english version of the show
if not getattr(s, 'seriesname'):
sickrage.app.log.error(
"Show in {} has no name on {}, probably the wrong language used to search with".format(self.showDir,
index_name))
sickrage.app.alerts.error(_("Unable to add show"),
_(
"Show in {} has no name on {}, probably the wrong language. Delete .nfo and add manually in the correct language").format(
self.showDir, index_name))
return self._finishEarly()
# if the show has no episodes/seasons
if not len(s):
sickrage.app.log.error("Show " + str(s['seriesname']) + " is on " + str(
IndexerApi(self.indexer).name) + " but contains no season/episode data.")
sickrage.app.alerts.error(_("Unable to add show"),
_("Show ") + str(s['seriesname']) + _(" is on ") + str(
IndexerApi(
self.indexer).name) + _(
" but contains no season/episode data."))
return self._finishEarly()
except Exception as e:
sickrage.app.log.error(
"{}: Error while loading information from indexer {}. Error: {}".format(self.indexer_id, index_name,
e.message))
sickrage.app.alerts.error(
_("Unable to add show"),
_("Unable to look up the show in {} on {} using ID {}, not using the NFO. Delete .nfo and try adding "
"manually again.").format(self.showDir, index_name, self.indexer_id)
)
if sickrage.app.config.use_trakt:
title = self.showDir.split("/")[-1]
data = {
'shows': [
{
'title': title,
'ids': {IndexerApi(self.indexer).trakt_id: self.indexer_id}
}
]
}
srTraktAPI()["sync/watchlist"].remove(data)
return self._finishEarly()
try:
self.show = TVShow(self.indexer, self.indexer_id, self.lang)
self.show.loadFromIndexer()
# set up initial values
self.show.location = self.showDir
self.show.subtitles = self.subtitles or sickrage.app.config.subtitles_default
self.show.subtitles_sr_metadata = self.subtitles_sr_metadata
self.show.quality = self.quality or sickrage.app.config.quality_default
self.show.flatten_folders = self.flatten_folders or sickrage.app.config.flatten_folders_default
self.show.anime = self.anime or sickrage.app.config.anime_default
self.show.scene = self.scene or sickrage.app.config.scene_default
self.show.archive_firstmatch = self.archive or sickrage.app.config.archive_default
self.show.paused = self.paused or False
# set up default new/missing episode status
sickrage.app.log.info(
"Setting all current episodes to the specified default status: " + str(self.default_status))
self.show.default_ep_status = self.default_status
if self.show.anime:
self.show.release_groups = BlackAndWhiteList(self.show.indexerid)
if self.blacklist:
self.show.release_groups.set_black_keywords(self.blacklist)
if self.whitelist:
self.show.release_groups.set_white_keywords(self.whitelist)
# # be smartish about this
# if self.show.genre and "talk show" in self.show.genre.lower():
# self.show.air_by_date = 1
# if self.show.genre and "documentary" in self.show.genre.lower():
# self.show.air_by_date = 0
# if self.show.classification and "sports" in self.show.classification.lower():
# self.show.sports = 1
except indexer_exception as e:
sickrage.app.log.error(
"Unable to add show due to an error with " + IndexerApi(
self.indexer).name + ": {}".format(e.message))
if self.show:
sickrage.app.alerts.error(
_("Unable to add ") + str(self.show.name) + _(" due to an error with ") + IndexerApi(
self.indexer).name + "")
else:
sickrage.app.alerts.error(
_("Unable to add show due to an error with ") + IndexerApi(self.indexer).name + "")
return self._finishEarly()
except MultipleShowObjectsException:
sickrage.app.log.warning("The show in " + self.showDir + " is already in your show list, skipping")
sickrage.app.alerts.error(_('Show skipped'),
_("The show in ") + self.showDir + _(" is already in your show list"))
return self._finishEarly()
except Exception as e:
sickrage.app.log.error("Error trying to add show: {}".format(e.message))
sickrage.app.log.debug(traceback.format_exc())
raise self._finishEarly()
try:
sickrage.app.log.debug("Attempting to retrieve show info from IMDb")
self.show.loadIMDbInfo()
except Exception as e:
sickrage.app.log.error("Error loading IMDb info: {}".format(e.message))
try:
self.show.saveToDB()
except Exception as e:
sickrage.app.log.error("Error saving the show to the database: {}".format(e.message))
sickrage.app.log.debug(traceback.format_exc())
raise self._finishEarly()
# add it to the show list
if not findCertainShow(sickrage.app.showlist, self.indexer_id): sickrage.app.showlist.append(self.show)
try:
self.show.loadEpisodesFromIndexer()
except Exception as e:
sickrage.app.log.error(
"Error with " + IndexerApi(
self.show.indexer).name + ", not creating episode list: {}".format(e.message))
sickrage.app.log.debug(traceback.format_exc())
try:
self.show.loadEpisodesFromDir()
except Exception as e:
sickrage.app.log.debug("Error searching dir for episodes: {}".format(e.message))
sickrage.app.log.debug(traceback.format_exc())
# if they set default ep status to WANTED then run the backlog to search for episodes
if self.show.default_ep_status == WANTED:
sickrage.app.log.info("Launching backlog for this show since its episodes are WANTED")
sickrage.app.backlog_searcher.searchBacklog([self.show])
self.show.writeMetadata()
self.show.updateMetadata()
self.show.populateCache()
if sickrage.app.config.use_trakt:
# if there are specific episodes that need to be added by trakt
sickrage.app.trakt_searcher.manageNewShow(self.show)
# add show to trakt.tv library
if sickrage.app.config.trakt_sync:
sickrage.app.trakt_searcher.addShowToTraktLibrary(self.show)
if sickrage.app.config.trakt_sync_watchlist:
sickrage.app.log.info("update watchlist")
sickrage.app.notifier_providers['trakt'].update_watchlist(show_obj=self.show)
# Load XEM data to DB for show
xem_refresh(self.show.indexerid, self.show.indexer, force=True)
# check if show has XEM mapping so we can determin if searches should go by scene numbering or indexer numbering.
if not self.scene and get_xem_numbering_for_show(self.show.indexerid, self.show.indexer):
self.show.scene = 1
self.show.default_ep_status = self.default_status_after
self.show.saveToDB()
sickrage.app.name_cache.build(self.show)
self.finish()
sickrage.app.log.info(
"Finished adding show {} in {}s from show dir: {}".format(self.show_name,
round(time.time() - self.startTime, 2),
self.showDir))
def _finishEarly(self):
if self.show: sickrage.app.show_queue.removeShow(self.show)
class BasicTests(SiCKRAGETestDBCase):
def setUp(self):
super(BasicTests, self).setUp()
self.show = TVShow(1, 1, 'en')
self.show.saveToDB()
def _test_names(self, np, section, transform=None, verbose=False):
if VERBOSE or verbose:
print('Running', section, 'tests')
for cur_test_base in simple_test_cases[section]:
if transform:
cur_test = transform(cur_test_base)
np.file_name = cur_test
else:
cur_test = cur_test_base
if VERBOSE or verbose:
print('Testing', cur_test)
result = simple_test_cases[section][cur_test_base]
np.showObj.name = result.series_name if result else None
if not result:
self.assertRaises(InvalidNameException, np.parse, cur_test)
return
else:
result.which_regex = {section}
test_result = np.parse(cur_test)
if DEBUG or verbose:
print('air_by_date:', test_result.is_air_by_date, 'air_date:',
test_result.air_date)
print('anime:', test_result.is_anime, 'ab_episode_numbers:',
test_result.ab_episode_numbers)
print(test_result)
print(result)
self.assertEqual(test_result.which_regex, {section})
self.assertEqual(str(test_result), str(result))
def test_standard_names(self):
np = NameParser(True, showObj=self.show, validate_show=False)
self._test_names(np, 'standard')
def test_standard_repeat_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'standard_repeat')
def test_fov_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'fov')
def test_fov_repeat_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'fov_repeat')
def test_stupid_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'stupid')
def test_no_season_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'no_season')
def test_no_season_general_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'no_season_general')
def test_no_season_multi_ep_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'no_season_multi_ep')
def test_season_only_names(self):
np = NameParser(False, showObj=self.show, validate_show=False)
self._test_names(np, 'season_only')
def test_standard_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'standard', lambda x: x + '.avi')
def test_standard_repeat_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'standard_repeat', lambda x: x + '.avi')
def test_fov_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'fov', lambda x: x + '.avi')
def test_fov_repeat_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'fov_repeat', lambda x: x + '.avi')
def test_stupid_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'stupid', lambda x: x + '.avi')
def test_no_season_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'no_season', lambda x: x + '.avi')
def test_no_season_general_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'no_season_general', lambda x: x + '.avi')
def test_no_season_multi_ep_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'no_season_multi_ep', lambda x: x + '.avi')
def test_season_only_file_names(self):
np = NameParser(showObj=self.show, validate_show=False)
self._test_names(np, 'season_only', lambda x: x + '.avi')
def test_combination_names(self):
pass
class QueueItemAdd(ShowQueueItem):
def __init__(self, indexer, indexer_id, showDir, default_status, quality,
flatten_folders, lang, subtitles, anime, scene, paused,
blacklist, whitelist, default_status_after, archive):
super(QueueItemAdd, self).__init__(None, ShowQueueActions.ADD)
self.indexer = indexer
self.indexer_id = indexer_id
self.showDir = showDir
self.default_status = default_status
self.quality = quality
self.flatten_folders = flatten_folders
self.lang = lang
self.subtitles = subtitles
self.anime = anime
self.scene = scene
self.paused = paused
self.blacklist = blacklist
self.whitelist = whitelist
self.default_status_after = default_status_after
self.archive = archive
self.priority = srQueuePriorities.HIGH
@property
def show_name(self):
"""
Returns the show name if there is a show object created, if not returns
the dir that the show is being added to.
"""
if self.show is None:
return os.path.basename(self.showDir)
return self.show.name
@property
def isLoading(self):
"""
Returns True if we've gotten far enough to have a show object, or False
if we still only know the folder name.
"""
if self.show is None:
return True
def run(self):
sickrage.srCore.srLogger.info("Started adding show {}".format(
self.show_name))
index_name = srIndexerApi(self.indexer).name
# make sure the Indexer IDs are valid
try:
lINDEXER_API_PARMS = srIndexerApi(self.indexer).api_params.copy()
if self.lang:
lINDEXER_API_PARMS['language'] = self.lang
sickrage.srCore.srLogger.info("{}: {}".format(
index_name, repr(lINDEXER_API_PARMS)))
t = srIndexerApi(self.indexer).indexer(**lINDEXER_API_PARMS)
s = t[self.indexer_id]
if not s: return self._finishEarly()
# this usually only happens if they have an NFO in their show dir which gave us a Indexer ID that has no proper english version of the show
if not getattr(s, 'seriesname'):
sickrage.srCore.srLogger.error(
"Show in {} has no name on {}, probably the wrong language used to search with"
.format(self.showDir, index_name))
sickrage.srCore.srNotifications.error(
"Unable to add show",
"Show in {} has no name on {}, probably the wrong language. Delete .nfo and add manually in the correct language"
.format(self.showDir, index_name))
return self._finishEarly()
# if the show has no episodes/seasons
if not len(s):
sickrage.srCore.srLogger.error(
"Show " + str(s['seriesname']) + " is on " +
str(srIndexerApi(self.indexer).name) +
" but contains no season/episode data.")
sickrage.srCore.srNotifications.error(
"Unable to add show", "Show " + str(s['seriesname']) +
" is on " + str(srIndexerApi(self.indexer).name) +
" but contains no season/episode data.")
return self._finishEarly()
except Exception as e:
sickrage.srCore.srLogger.error(
"{}: Error while loading information from indexer {}. Error: {}"
.format(self.indexer_id, index_name, e.message))
sickrage.srCore.srNotifications.error(
"Unable to add show",
"Unable to look up the show in {} on {} using ID {}, not using the NFO. Delete .nfo and try adding manually again."
.format(self.showDir, index_name, self.indexer_id))
if sickrage.srCore.srConfig.USE_TRAKT:
trakt_id = srIndexerApi(self.indexer).config['trakt_id']
trakt_api = TraktAPI(sickrage.srCore.srConfig.SSL_VERIFY,
sickrage.srCore.srConfig.TRAKT_TIMEOUT)
title = self.showDir.split("/")[-1]
data = {'shows': [{'title': title, 'ids': {}}]}
if trakt_id == 'tvdb_id':
data['shows'][0]['ids']['tvdb'] = self.indexer_id
else:
data['shows'][0]['ids']['tvrage'] = self.indexer_id
trakt_api.traktRequest("sync/watchlist/remove",
data,
method='POST')
return self._finishEarly()
try:
self.show = TVShow(self.indexer, self.indexer_id, self.lang)
self.show.loadFromIndexer()
# set up initial values
self.show.location = self.showDir
self.show.subtitles = self.subtitles or sickrage.srCore.srConfig.SUBTITLES_DEFAULT
self.show.quality = self.quality or sickrage.srCore.srConfig.QUALITY_DEFAULT
self.show.flatten_folders = self.flatten_folders or sickrage.srCore.srConfig.FLATTEN_FOLDERS_DEFAULT
self.show.anime = self.anime or sickrage.srCore.srConfig.ANIME_DEFAULT
self.show.scene = self.scene or sickrage.srCore.srConfig.SCENE_DEFAULT
self.show.archive_firstmatch = self.archive or sickrage.srCore.srConfig.ARCHIVE_DEFAULT
self.show.paused = self.paused or False
# set up default new/missing episode status
sickrage.srCore.srLogger.info(
"Setting all current episodes to the specified default status: "
+ str(self.default_status))
self.show.default_ep_status = self.default_status
if self.show.anime:
self.show.release_groups = BlackAndWhiteList(
self.show.indexerid)
if self.blacklist:
self.show.release_groups.set_black_keywords(self.blacklist)
if self.whitelist:
self.show.release_groups.set_white_keywords(self.whitelist)
# # be smartish about this
# if self.show.genre and "talk show" in self.show.genre.lower():
# self.show.air_by_date = 1
# if self.show.genre and "documentary" in self.show.genre.lower():
# self.show.air_by_date = 0
# if self.show.classification and "sports" in self.show.classification.lower():
# self.show.sports = 1
except indexer_exception as e:
sickrage.srCore.srLogger.error(
"Unable to add show due to an error with " +
srIndexerApi(self.indexer).name + ": {}".format(e.message))
if self.show:
sickrage.srCore.srNotifications.error(
"Unable to add " + str(self.show.name) +
" due to an error with " +
srIndexerApi(self.indexer).name + "")
else:
sickrage.srCore.srNotifications.error(
"Unable to add show due to an error with " +
srIndexerApi(self.indexer).name + "")
return self._finishEarly()
except MultipleShowObjectsException:
sickrage.srCore.srLogger.warning(
"The show in " + self.showDir +
" is already in your show list, skipping")
sickrage.srCore.srNotifications.error(
'Show skipped', "The show in " + self.showDir +
" is already in your show list")
return self._finishEarly()
except Exception as e:
sickrage.srCore.srLogger.error(
"Error trying to add show: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
raise self._finishEarly()
try:
sickrage.srCore.srLogger.debug(
"Attempting to retrieve show info from IMDb")
self.show.loadIMDbInfo()
except Exception as e:
sickrage.srCore.srLogger.error(
"Error loading IMDb info: {}".format(e.message))
# Load XEM data to DB for show
xem_refresh(self.show.indexerid, self.show.indexer, force=True)
# check if show has XEM mapping so we can determin if searches should go by scene numbering or indexer numbering.
if not self.scene and get_xem_numbering_for_show(
self.show.indexerid, self.show.indexer):
self.show.scene = 1
try:
self.show.saveToDB()
except Exception as e:
sickrage.srCore.srLogger.error(
"Error saving the show to the database: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
raise self._finishEarly()
# add it to the show list
sickrage.srCore.SHOWLIST.append(self.show)
try:
self.show.loadEpisodesFromIndexer()
except Exception as e:
sickrage.srCore.srLogger.error(
"Error with " + srIndexerApi(self.show.indexer).name +
", not creating episode list: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
try:
self.show.loadEpisodesFromDir()
except Exception as e:
sickrage.srCore.srLogger.debug(
"Error searching dir for episodes: {}".format(e.message))
sickrage.srCore.srLogger.debug(traceback.format_exc())
# if they set default ep status to WANTED then run the backlog to search for episodes
if self.show.default_ep_status == WANTED:
sickrage.srCore.srLogger.info(
"Launching backlog for this show since its episodes are WANTED"
)
sickrage.srCore.BACKLOGSEARCHER.searchBacklog([self.show])
self.show.writeMetadata()
self.show.updateMetadata()
self.show.populateCache()
if sickrage.srCore.srConfig.USE_TRAKT:
# if there are specific episodes that need to be added by trakt
sickrage.srCore.TRAKTSEARCHER.manageNewShow(self.show)
# add show to trakt.tv library
if sickrage.srCore.srConfig.TRAKT_SYNC:
sickrage.srCore.TRAKTSEARCHER.addShowToTraktLibrary(self.show)
if sickrage.srCore.srConfig.TRAKT_SYNC_WATCHLIST:
sickrage.srCore.srLogger.info("update watchlist")
sickrage.srCore.notifiersDict.trakt_notifier.update_watchlist(
show_obj=self.show)
# After initial add, set to default_status_after.
sickrage.srCore.srLogger.info(
"Setting all future episodes to the specified default status: " +
str(self.default_status_after))
self.show.default_ep_status = self.default_status_after
self.show.saveToDB()
sickrage.srCore.NAMECACHE.build(self.show)
self.finish()
sickrage.srCore.srLogger.info("Finished adding show {}".format(
self.showDir))
def _finishEarly(self):
if self.show: sickrage.srCore.SHOWQUEUE.removeShow(self.show)
def test(self):
show = TVShow(1, int(curData["tvdbid"]))
show.name = name
show.quality = ANY | Quality.UNKNOWN | Quality.RAWHDTV
show.saveToDB()
show.loadFromDB(skipNFO=True)
sickrage.srCore.SHOWLIST.append(show)
for epNumber in curData["e"]:
episode = TVEpisode(show, curData["s"], epNumber)
episode.status = WANTED
# We arent updating scene numbers, so fake it here
episode.scene_season = curData["s"]
episode.scene_episode = epNumber
episode.saveToDB()
provider.show = show
season_strings = provider._get_season_search_strings(episode)
episode_strings = provider._get_episode_search_strings(episode)
fail = False
for cur_string in season_strings, episode_strings:
if not all([isinstance(cur_string, list), isinstance(cur_string[0], dict)]):
print(" %s is using a wrong string format!" % provider.name)
print(cur_string)
fail = True
continue
if fail:
continue
try:
assert (season_strings == curData["s_strings"])
assert (episode_strings == curData["e_strings"])
except AssertionError:
continue
search_strings = episode_strings[0]
# search_strings.update(season_strings[0])
# search_strings.update({"RSS":['']})
# print search_strings
if not provider.public:
continue
items = provider.search(search_strings)
if not items:
print("No results from provider?")
continue
title, url = provider._get_title_and_url(items[0])
for word in show.name.split(" "):
if not word.lower() in title.lower():
print("Show name not in title: %s. URL: %s" % (title, url))
continue
if not url:
print("url is empty")
continue
quality = provider.getQuality(items[0])
size = provider._get_size(items[0])
if not show.quality & quality:
print("Quality not in ANY, %r" % quality)
continue
def test_set_name(self):
show = TVShow(1, 0001, "en")
show.name = "newName"
show.saveToDB()
show.loadFromDB(skipNFO=True)
self.assertEqual(show.name, "newName")
class BasicTests(SiCKRAGETestDBCase):
def __init__(self, something):
super(BasicTests, self).__init__(something)
super(BasicTests, self).setUp()
self.show = TVShow(1, 1, "en")
self.show.saveToDB()
def _test_names(self, np, section, transform=None, verbose=False):
if VERBOSE or verbose:
print("Running", section, "tests")
for cur_test_base in simple_test_cases[section]:
if transform:
cur_test = transform(cur_test_base)
np.file_name = cur_test
else:
cur_test = cur_test_base
if VERBOSE or verbose:
print("Testing", cur_test)
result = simple_test_cases[section][cur_test_base]
self.show.name = result.series_name if result else None
np.showObj = self.show
if not result:
self.assertRaises(InvalidNameException, np.parse, cur_test)
return
else:
result.which_regex = [section]
test_result = np.parse(cur_test)
if DEBUG or verbose:
print("air_by_date:", test_result.is_air_by_date, "air_date:", test_result.air_date)
print("anime:", test_result.is_anime, "ab_episode_numbers:", test_result.ab_episode_numbers)
print(test_result)
print(result)
self.assertEqual(test_result.which_regex, [section])
self.assertEqual(str(test_result), str(result))
def test_standard_names(self):
np = NameParser(True)
self._test_names(np, "standard")
def test_standard_repeat_names(self):
np = NameParser(False)
self._test_names(np, "standard_repeat")
def test_fov_names(self):
np = NameParser(False)
self._test_names(np, "fov")
def test_fov_repeat_names(self):
np = NameParser(False)
self._test_names(np, "fov_repeat")
# def test_bare_names(self):
# np = parser.NameParser(False)
# self._test_names(np, 'bare')
def test_stupid_names(self):
np = NameParser(False)
self._test_names(np, "stupid")
# def test_no_season_names(self):
# np = parser.NameParser(False)
# self._test_names(np, 'no_season')
def test_no_season_general_names(self):
np = NameParser(False)
self._test_names(np, "no_season_general")
def test_no_season_multi_ep_names(self):
np = NameParser(False)
self._test_names(np, "no_season_multi_ep")
def test_season_only_names(self):
np = NameParser(False)
self._test_names(np, "season_only")
# def test_scene_date_format_names(self):
# np = parser.NameParser(False)
# self._test_names(np, 'scene_date_format')
def test_standard_file_names(self):
np = NameParser()
self._test_names(np, "standard", lambda x: x + ".avi")
def test_standard_repeat_file_names(self):
np = NameParser()
self._test_names(np, "standard_repeat", lambda x: x + ".avi")
def test_fov_file_names(self):
np = NameParser()
self._test_names(np, "fov", lambda x: x + ".avi")
def test_fov_repeat_file_names(self):
np = NameParser()
self._test_names(np, "fov_repeat", lambda x: x + ".avi")
# def test_bare_file_names(self):
# np = parser.NameParser()
# self._test_names(np, 'bare', lambda x: x + '.avi')
def test_stupid_file_names(self):
np = NameParser()
self._test_names(np, "stupid", lambda x: x + ".avi")
# def test_no_season_file_names(self):
# np = parser.NameParser()
# self._test_names(np, 'no_season', lambda x: x + '.avi')
def test_no_season_general_file_names(self):
np = NameParser()
self._test_names(np, "no_season_general", lambda x: x + ".avi")
def test_no_season_multi_ep_file_names(self):
np = NameParser()
self._test_names(np, "no_season_multi_ep", lambda x: x + ".avi")
def test_season_only_file_names(self):
np = NameParser()
self._test_names(np, "season_only", lambda x: x + ".avi")
# def test_scene_date_format_file_names(self):
# np = parser.NameParser()
# self._test_names(np, 'scene_date_format', lambda x: x + '.avi')
def test_combination_names(self):
pass
def test_init_indexerid(self):
show = TVShow(1, 0001, "en")
show.saveToDB()
show.loadFromDB(skipNFO=True)
self.assertEqual(show.indexerid, 0001)
class QueueItemAdd(ShowQueueItem):
def __init__(self, indexer, indexer_id, showDir, default_status, quality,
flatten_folders, lang, subtitles, subtitles_sr_metadata,
anime, scene, paused, blacklist, whitelist,
default_status_after, skip_downloaded):
super(QueueItemAdd, self).__init__(None, ShowQueueActions.ADD)
self.indexer = indexer
self.indexer_id = indexer_id
self.showDir = showDir
self.default_status = default_status
self.quality = quality
self.flatten_folders = flatten_folders
self.lang = lang
self.subtitles = subtitles
self.subtitles_sr_metadata = subtitles_sr_metadata
self.anime = anime
self.scene = scene
self.paused = paused
self.blacklist = blacklist
self.whitelist = whitelist
self.default_status_after = default_status_after
self.skip_downloaded = skip_downloaded
self.priority = srQueuePriorities.HIGH
@property
def show_name(self):
"""
Returns the show name if there is a show object created, if not returns
the dir that the show is being added to.
"""
return self.show.name if self.show else os.path.basename(self.showDir)
@property
def is_loading(self):
"""
Returns True if we've gotten far enough to have a show object, or False
if we still only know the folder name.
"""
if self.show is None:
return True
def run(self):
start_time = time.time()
sickrage.app.log.info(
"Started adding show {} from show dir: {}".format(
self.show_name, self.showDir))
index_name = IndexerApi(self.indexer).name
# make sure the Indexer IDs are valid
try:
lINDEXER_API_PARMS = IndexerApi(self.indexer).api_params.copy()
lINDEXER_API_PARMS['cache'] = False
lINDEXER_API_PARMS[
'language'] = self.lang or sickrage.app.config.indexer_default_language
sickrage.app.log.info("{}: {}".format(index_name,
repr(lINDEXER_API_PARMS)))
t = IndexerApi(self.indexer).indexer(**lINDEXER_API_PARMS)
try:
s = t[self.indexer_id]
except indexer_error:
s = None
if not s:
return self._finish_early()
# this usually only happens if they have an NFO in their show dir which gave us a Indexer ID that has no proper english version of the show
if not getattr(s, 'seriesname'):
sickrage.app.log.warning(
"Show in {} has no name on {}, probably the wrong language used to search with"
.format(self.showDir, index_name))
sickrage.app.alerts.error(
_("Unable to add show"),
_("Show in {} has no name on {}, probably the wrong language. Delete .nfo "
"and add manually in the correct language").format(
self.showDir, index_name))
return self._finish_early()
# if the show has no episodes/seasons
if not len(s):
sickrage.app.log.warning(
"Show " + str(s['seriesname']) + " is on " +
str(IndexerApi(self.indexer).name) +
" but contains no season/episode data.")
sickrage.app.alerts.error(
_("Unable to add show"),
_("Show ") + str(s['seriesname']) + _(" is on ") +
str(IndexerApi(self.indexer).name) +
_(" but contains no season/episode data."))
return self._finish_early()
except Exception as e:
sickrage.app.log.error(
"{}: Error while loading information from indexer {}. Error: {}"
.format(self.indexer_id, index_name, e))
sickrage.app.alerts.error(
_("Unable to add show"),
_("Unable to look up the show in {} on {} using ID {}, not using the NFO. Delete .nfo and try adding "
"manually again.").format(self.showDir, index_name,
self.indexer_id))
if sickrage.app.config.use_trakt:
title = self.showDir.split("/")[-1]
data = {
'shows': [{
'title': title,
'ids': {
IndexerApi(self.indexer).trakt_id: self.indexer_id
}
}]
}
srTraktAPI()["sync/watchlist"].remove(data)
return self._finish_early()
try:
self.show = TVShow(self.indexer, self.indexer_id, self.lang)
self.show.loadFromIndexer()
# set up initial values
self.show.location = self.showDir
self.show.subtitles = self.subtitles or sickrage.app.config.subtitles_default
self.show.subtitles_sr_metadata = self.subtitles_sr_metadata
self.show.quality = self.quality or sickrage.app.config.quality_default
self.show.flatten_folders = self.flatten_folders or sickrage.app.config.flatten_folders_default
self.show.anime = self.anime or sickrage.app.config.anime_default
self.show.scene = self.scene or sickrage.app.config.scene_default
self.show.skip_downloaded = self.skip_downloaded or sickrage.app.config.skip_downloaded_default
self.show.paused = self.paused or False
# set up default new/missing episode status
sickrage.app.log.info(
"Setting all current episodes to the specified default status: "
+ str(self.default_status))
self.show.default_ep_status = self.default_status
if self.show.anime:
self.show.release_groups = BlackAndWhiteList(
self.show.indexerid)
if self.blacklist:
self.show.release_groups.set_black_keywords(self.blacklist)
if self.whitelist:
self.show.release_groups.set_white_keywords(self.whitelist)
# # be smartish about this
# if self.show.genre and "talk show" in self.show.genre.lower():
# self.show.air_by_date = 1
# if self.show.genre and "documentary" in self.show.genre.lower():
# self.show.air_by_date = 0
# if self.show.classification and "sports" in self.show.classification.lower():
# self.show.sports = 1
except indexer_exception as e:
sickrage.app.log.warning(
"Unable to add show due to an error with " +
IndexerApi(self.indexer).name + ": {}".format(e))
if self.show:
sickrage.app.alerts.error(
_("Unable to add ") + str(self.show.name) +
_(" due to an error with ") +
IndexerApi(self.indexer).name + "")
else:
sickrage.app.alerts.error(
_("Unable to add show due to an error with ") +
IndexerApi(self.indexer).name + "")
return self._finish_early()
except MultipleShowObjectsException:
sickrage.app.log.warning("The show in " + self.showDir +
" is already in your show list, skipping")
sickrage.app.alerts.error(
_('Show skipped'),
_("The show in ") + self.showDir +
_(" is already in your show list"))
return self._finish_early()
except Exception as e:
sickrage.app.log.error("Error trying to add show: {}".format(e))
sickrage.app.log.debug(traceback.format_exc())
raise self._finish_early()
try:
sickrage.app.log.debug(
"Attempting to retrieve show info from IMDb")
self.show.load_imdb_info()
except Exception as e:
sickrage.app.log.error("Error loading IMDb info: {}".format(e))
try:
self.show.saveToDB()
except Exception as e:
sickrage.app.log.error(
"Error saving the show to the database: {}".format(e))
sickrage.app.log.debug(traceback.format_exc())
raise self._finish_early()
# add it to the show list
if not findCertainShow(self.indexer_id):
sickrage.app.showlist.append(self.show)
try:
self.show.loadEpisodesFromIndexer()
except Exception as e:
sickrage.app.log.error("Error with " +
IndexerApi(self.show.indexer).name +
", not creating episode list: {}".format(e))
sickrage.app.log.debug(traceback.format_exc())
try:
self.show.loadEpisodesFromDir()
except Exception as e:
sickrage.app.log.debug(
"Error searching dir for episodes: {}".format(e))
sickrage.app.log.debug(traceback.format_exc())
# if they set default ep status to WANTED then run the backlog to search for episodes
if self.show.default_ep_status == WANTED:
sickrage.app.log.info(
"Launching backlog for this show since its episodes are WANTED"
)
sickrage.app.backlog_searcher.search_backlog([self.show])
self.show.writeMetadata(force=True)
self.show.populateCache()
if sickrage.app.config.use_trakt:
# if there are specific episodes that need to be added by trakt
sickrage.app.trakt_searcher.manageNewShow(self.show)
# add show to trakt.tv library
if sickrage.app.config.trakt_sync:
sickrage.app.trakt_searcher.addShowToTraktLibrary(self.show)
if sickrage.app.config.trakt_sync_watchlist:
sickrage.app.log.info("update watchlist")
sickrage.app.notifier_providers['trakt'].update_watchlist(
show_obj=self.show)
# Load XEM data to DB for show
xem_refresh(self.show.indexerid, self.show.indexer, force=True)
# check if show has XEM mapping so we can determin if searches should go by scene numbering or indexer numbering.
if not self.scene and get_xem_numbering_for_show(
self.show.indexerid, self.show.indexer):
self.show.scene = 1
self.show.default_ep_status = self.default_status_after
self.show.saveToDB()
sickrage.app.name_cache.build(self.show)
sickrage.app.quicksearch_cache.add_show(self.show.indexerid)
self.finish()
sickrage.app.log.info(
"Finished adding show {} in {}s from show dir: {}".format(
self.show_name, round(time.time() - start_time, 2),
self.showDir))
def _finish_early(self):
if self.show: sickrage.app.show_queue.removeShow(self.show)
def test(self):
show = TVShow(1, int(curData[b"tvdbid"]))
show.name = name
show.quality = ANY | Quality.UNKNOWN | Quality.RAWHDTV
show.saveToDB()
sickrage.showList.append(show)
for epNumber in curData[b"e"]:
episode = TVEpisode(show, curData[b"s"], epNumber)
episode.status = WANTED
# We arent updating scene numbers, so fake it here
episode.scene_season = curData[b"s"]
episode.scene_episode = epNumber
episode.saveToDB()
provider.show = show
season_strings = provider._get_season_search_strings(episode)
episode_strings = provider._get_episode_search_strings(episode)
fail = False
for cur_string in season_strings, episode_strings:
if not all([
isinstance(cur_string, list),
isinstance(cur_string[0], dict)
]):
print(" %s is using a wrong string format!" %
provider.name)
print(cur_string)
fail = True
continue
if fail:
continue
try:
assert (season_strings == curData[b"s_strings"])
assert (episode_strings == curData[b"e_strings"])
except AssertionError:
continue
search_strings = episode_strings[0]
# search_strings.update(season_strings[0])
# search_strings.update({"RSS":['']})
# print search_strings
if not provider.public:
continue
items = provider._doSearch(search_strings)
if not items:
print("No results from provider?")
continue
title, url = provider._get_title_and_url(items[0])
for word in show.name.split(" "):
if not word.lower() in title.lower():
print("Show name not in title: %s. URL: %s" % (title, url))
continue
if not url:
print("url is empty")
continue
quality = provider.getQuality(items[0])
size = provider._get_size(items[0])
if not show.quality & quality:
print("Quality not in ANY, %r" % quality)
continue
