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):
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.show = None
# this will initialize self.show to None
ShowQueueItem.__init__(self, ShowQueueActions.ADD, self.show)
# Process add show in priority
self.priority = QueuePriorities.HIGH
def _getName(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 self.showDir
return self.show.name
show_name = property(_getName)
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
return False
isLoading = property(_isLoading)
def run(self):
ShowQueueItem.run(self)
sickrage.srLogger.info("Started adding show {}".format(self.showDir))
index_name = sickrage.srCore.INDEXER_API(self.indexer).name
# update internal name cache
sickrage.srCore.NAMECACHE.buildNameCache()
# make sure the Indexer IDs are valid
try:
lINDEXER_API_PARMS = sickrage.srCore.INDEXER_API(self.indexer).api_params.copy()
if self.lang:
lINDEXER_API_PARMS[b'language'] = self.lang
sickrage.srLogger.info("{}: {}".format(index_name, repr(lINDEXER_API_PARMS)))
t = sickrage.srCore.INDEXER_API(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 getattr(s, 'seriesname', None) is None:
sickrage.srLogger.error(
"Show in {} has no name on {}, probably the wrong language used to search with".format(self.showDir,
index_name))
notifications.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 s:
sickrage.srLogger.error("Show " + str(s[b'seriesname']) + " is on " + str(
sickrage.srCore.INDEXER_API(self.indexer).name) + " but contains no season/episode data.")
notifications.error("Unable to add show",
"Show " + str(s[b'seriesname']) + " is on " + str(sickrage.srCore.INDEXER_API(
self.indexer).name) + " but contains no season/episode data.")
return self._finishEarly()
except Exception as e:
sickrage.srLogger.error("{}: Error while loading information from indexer {}. Error: {}".format(self.indexer_id, index_name, e.message))
notifications.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.srConfig.USE_TRAKT:
trakt_id = sickrage.srCore.INDEXER_API(self.indexer).config[b'trakt_id']
trakt_api = TraktAPI(sickrage.srConfig.SSL_VERIFY, sickrage.srConfig.TRAKT_TIMEOUT)
title = self.showDir.split("/")[-1]
data = {
'shows': [
{
'title': title,
'ids': {}
}
]
}
if trakt_id == 'tvdb_id':
data[b'shows'][0][b'ids'][b'tvdb'] = self.indexer_id
else:
data[b'shows'][0][b'ids'][b'tvrage'] = self.indexer_id
trakt_api.traktRequest("sync/watchlist/remove", data, method='POST')
self._finishEarly()
return
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.srConfig.SUBTITLES_DEFAULT
self.show.quality = self.quality or sickrage.srConfig.QUALITY_DEFAULT
self.show.flatten_folders = self.flatten_folders or sickrage.srConfig.FLATTEN_FOLDERS_DEFAULT
self.show.anime = self.anime or sickrage.srConfig.ANIME_DEFAULT
self.show.scene = self.scene or sickrage.srConfig.SCENE_DEFAULT
self.show.archive_firstmatch = self.archive or sickrage.srConfig.ARCHIVE_DEFAULT
self.show.paused = self.paused or False
# set up default new/missing episode status
sickrage.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.srLogger.error(
"Unable to add show due to an error with " + sickrage.srCore.INDEXER_API(
self.indexer).name + ": {}".format(e.message))
if self.show:
notifications.error(
"Unable to add " + str(self.show.name) + " due to an error with " + sickrage.srCore.INDEXER_API(
self.indexer).name + "")
else:
notifications.error(
"Unable to add show due to an error with " + sickrage.srCore.INDEXER_API(self.indexer).name + "")
self._finishEarly()
return
except MultipleShowObjectsException:
sickrage.srLogger.warning("The show in " + self.showDir + " is already in your show list, skipping")
notifications.error('Show skipped', "The show in " + self.showDir + " is already in your show list")
self._finishEarly()
return
except Exception as e:
sickrage.srLogger.error("Error trying to add show: {}".format(e.message))
sickrage.srLogger.debug(traceback.format_exc())
self._finishEarly()
raise
sickrage.srLogger.debug("Retrieving show info from TMDb")
try:
self.show.loadTMDbInfo()
except Exception as e:
sickrage.srLogger.error("Error loading TMDb info: {}".format(e.message))
try:
sickrage.srLogger.debug("Attempting to retrieve show info from IMDb")
self.show.loadIMDbInfo()
except Exception as e:
sickrage.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.srLogger.error("Error saving the show to the database: {}".format(e.message))
sickrage.srLogger.debug(traceback.format_exc())
self._finishEarly()
raise
# add it to the show list
sickrage.srCore.SHOWLIST.append(self.show)
try:
self.show.loadEpisodesFromIndexer()
except Exception as e:
sickrage.srLogger.error(
"Error with " + sickrage.srCore.INDEXER_API(
self.show.indexer).name + ", not creating episode list: {}".format(e.message))
sickrage.srLogger.debug(traceback.format_exc())
try:
self.show.loadEpisodesFromDir()
except Exception as e:
sickrage.srLogger.error("Error searching dir for episodes: {}".format(e.message))
sickrage.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.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()
self.show.flushEpisodes()
if sickrage.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.srConfig.TRAKT_SYNC:
sickrage.srCore.TRAKTSEARCHER.addShowToTraktLibrary(self.show)
if sickrage.srConfig.TRAKT_SYNC_WATCHLIST:
sickrage.srLogger.info("update watchlist")
sickrage.srCore.NOTIFIERS.trakt_notifier.update_watchlist(show_obj=self.show)
# After initial add, set to default_status_after.
sickrage.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.srLogger.info("Finished adding show {}".format(self.showDir))
self.finish()
def _finishEarly(self):
if self.show is not None:
sickrage.srCore.SHOWQUEUE.removeShow(self.show)
self.finish()
