def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.app.config.allow_high_priority:
# if it aired recently make it high priority
for curEp in result.episodes:
if date.today() - curEp.airdate <= timedelta(days=7):
result.priority = 1
if re.search(r'(^|[. _-])(proper|repack)([. _-]|$)', result.name,
re.I) is not None:
endStatus = SNATCHED_PROPER
# get result content
result.content = result.provider.get_content(result.url)
dlResult = False
if result.resultType in ("nzb", "nzbdata"):
if sickrage.app.config.nzb_method == "blackhole":
dlResult = result.provider.download_result(result)
elif sickrage.app.config.nzb_method == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.app.config.nzb_method == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.app.log.error("Unknown NZB action specified in config: " +
sickrage.app.config.nzb_method)
elif result.resultType in ("torrent", "torznab"):
# add public trackers to torrent result
if not result.provider.private:
result = result.provider.add_trackers(result)
if sickrage.app.config.torrent_method == "blackhole":
dlResult = result.provider.download_result(result)
else:
if any([result.content, result.url.startswith('magnet:')]):
client = getClientIstance(sickrage.app.config.torrent_method)()
dlResult = client.send_torrent(result)
else:
sickrage.app.log.warning("Torrent file content is empty")
else:
sickrage.app.log.error(
"Unknown result type, unable to download it (%r)" %
result.resultType)
# no download results found
if not dlResult:
return False
FailedHistory.logSnatch(result)
sickrage.app.alerts.message(_('Episode snatched'), result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(
SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(
endStatus, result.quality)
# save episode to DB
curEpObj.save_to_db()
if curEpObj.status not in Quality.DOWNLOADED:
try:
Notifiers.mass_notify_snatch(
curEpObj._format_pattern('%SN - %Sx%0E - %EN - %QN') +
" from " + result.provider.name)
except:
sickrage.app.log.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.app.notifier_providers[
'trakt'].trakt_episode_data_generate(trakt_data)
if sickrage.app.config.use_trakt and sickrage.app.config.trakt_sync_watchlist:
sickrage.app.log.debug("Add episodes, showid: indexerid " +
str(result.show.indexerid) + ", Title " +
str(result.show.name) + " to Traktv Watchlist")
if data:
sickrage.app.notifier_providers['trakt'].update_watchlist(
result.show, data_episode=data, update="add")
return True
def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.srCore.srConfig.ALLOW_HIGH_PRIORITY:
# if it aired recently make it high priority
for curEp in result.episodes:
if date.today() - curEp.airdate <= timedelta(days=7):
result.priority = 1
if re.search(r'(^|[\. _-])(proper|repack)([\. _-]|$)', result.name, re.I) is not None:
endStatus = SNATCHED_PROPER
if result.url.startswith('magnet') or result.url.endswith('torrent'):
result.resultType = 'torrent'
dlResult = False
if result.resultType in ("nzb", "nzbdata"):
if sickrage.srCore.srConfig.NZB_METHOD == "blackhole":
dlResult = _downloadResult(result)
elif sickrage.srCore.srConfig.NZB_METHOD == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.srCore.srConfig.NZB_METHOD == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.srCore.srLogger.error(
"Unknown NZB action specified in config: " + sickrage.srCore.srConfig.NZB_METHOD)
elif result.resultType == "torrent":
if sickrage.srCore.srConfig.TORRENT_METHOD == "blackhole":
dlResult = _downloadResult(result)
else:
if all([not result.content, not result.url.startswith('magnet:')]):
result.content = sickrage.srCore.srWebSession.get(result.url).content
if any([result.content, result.url.startswith('magnet:')]):
# add public trackers to magnet url for non-private torrent providers
if not result.provider.private and result.url.startswith('magnet:'):
result.url += '&tr='.join(
[x.strip() for x in sickrage.srCore.srConfig.TORRENT_TRACKERS.split(',') if x.strip()])
client = getClientIstance(sickrage.srCore.srConfig.TORRENT_METHOD)()
dlResult = client.sendTORRENT(result)
else:
sickrage.srCore.srLogger.warning("Torrent file content is empty")
else:
sickrage.srCore.srLogger.error("Unknown result type, unable to download it (%r)" % result.resultType)
# no download results found
if not dlResult:
return False
if sickrage.srCore.srConfig.USE_FAILED_DOWNLOADS:
FailedHistory.logSnatch(result)
sickrage.srCore.srNotifications.message('Episode snatched', result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(endStatus, result.quality)
# save episode to DB
curEpObj.saveToDB()
if curEpObj.status not in Quality.DOWNLOADED:
try:
srNotifiers.notify_snatch(
curEpObj._format_pattern('%SN - %Sx%0E - %EN - %QN') + " from " + result.provider.name)
except:
sickrage.srCore.srLogger.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.srCore.notifiersDict['trakt'].trakt_episode_data_generate(trakt_data)
if sickrage.srCore.srConfig.USE_TRAKT and sickrage.srCore.srConfig.TRAKT_SYNC_WATCHLIST:
sickrage.srCore.srLogger.debug(
"Add episodes, showid: indexerid " + str(result.show.indexerid) + ", Title " + str(
result.show.name) + " to Traktv Watchlist")
if data:
sickrage.srCore.notifiersDict['trakt'].update_watchlist(result.show, data_episode=data, update="add")
return True
def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.srCore.srConfig.ALLOW_HIGH_PRIORITY:
# if it aired recently make it high priority
for curEp in result.episodes:
if date.today() - curEp.airdate <= timedelta(days=7):
result.priority = 1
if re.search(r'(^|[\. _-])(proper|repack)([\. _-]|$)', result.name, re.I) is not None:
endStatus = SNATCHED_PROPER
if result.url.startswith('magnet') or result.url.endswith('torrent'):
result.resultType = 'torrent'
dlResult = False
if result.resultType in ("nzb", "nzbdata"):
if sickrage.srCore.srConfig.NZB_METHOD == "blackhole":
dlResult = _downloadResult(result)
elif sickrage.srCore.srConfig.NZB_METHOD == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.srCore.srConfig.NZB_METHOD == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.srCore.srLogger.error(
"Unknown NZB action specified in config: " + sickrage.srCore.srConfig.NZB_METHOD)
elif result.resultType == "torrent":
if sickrage.srCore.srConfig.TORRENT_METHOD == "blackhole":
dlResult = _downloadResult(result)
else:
if all([not result.content, not result.url.startswith('magnet:')]):
result.content = sickrage.srCore.srWebSession.get(result.url).content
if any([result.content, result.url.startswith('magnet:')]):
# add public trackers to magnet url for non-private torrent providers
if not result.provider.private and result.url.startswith('magnet:'):
result.url += '&tr='.join(
[x.strip() for x in sickrage.srCore.srConfig.TORRENT_TRACKERS.split(',') if x.strip()])
client = getClientIstance(sickrage.srCore.srConfig.TORRENT_METHOD)()
dlResult = client.sendTORRENT(result)
else:
sickrage.srCore.srLogger.warning("Torrent file content is empty")
else:
sickrage.srCore.srLogger.error("Unknown result type, unable to download it (%r)" % result.resultType)
# no download results found
if not dlResult:
return False
if sickrage.srCore.srConfig.USE_FAILED_DOWNLOADS:
FailedHistory.logSnatch(result)
sickrage.srCore.srNotifications.message('Episode snatched', result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
sql_l = []
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(endStatus, result.quality)
sql_q = curEpObj.saveToDB(False)
if sql_q:
sql_l.append(sql_q)
if curEpObj.status not in Quality.DOWNLOADED:
try:
srNotifiers.notify_snatch(
curEpObj._format_pattern('%SN - %Sx%0E - %EN - %QN') + " from " + result.provider.name)
except:
sickrage.srCore.srLogger.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.srCore.notifiersDict.trakt_notifier.trakt_episode_data_generate(trakt_data)
if sickrage.srCore.srConfig.USE_TRAKT and sickrage.srCore.srConfig.TRAKT_SYNC_WATCHLIST:
sickrage.srCore.srLogger.debug(
"Add episodes, showid: indexerid " + str(result.show.indexerid) + ", Title " + str(
result.show.name) + " to Traktv Watchlist")
if data:
sickrage.srCore.notifiersDict.trakt_notifier.update_watchlist(result.show, data_episode=data, update="add")
if len(sql_l) > 0:
main_db.MainDB().mass_upsert(sql_l)
del sql_l # cleanup
return True
def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.ALLOW_HIGH_PRIORITY:
# if it aired recently make it high priority
for curEp in result.episodes:
if datetime.date.today() - curEp.airdate <= datetime.timedelta(
days=7):
result.priority = 1
if re.search(r'(^|[\. _-])(proper|repack)([\. _-]|$)', result.name,
re.I) is not None:
endStatus = SNATCHED_PROPER
if result.url.startswith('magnet') or result.url.endswith('torrent'):
result.resultType = 'torrent'
# NZBs can be sent straight to SAB or saved to disk
if result.resultType in ("nzb", "nzbdata"):
if sickrage.NZB_METHOD == "blackhole":
dlResult = _downloadResult(result)
elif sickrage.NZB_METHOD == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.NZB_METHOD == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.LOGGER.error("Unknown NZB action specified in config: " +
sickrage.NZB_METHOD)
dlResult = False
# TORRENTs can be sent to clients or saved to disk
elif result.resultType == "torrent":
# torrents are saved to disk when blackhole mode
if sickrage.TORRENT_METHOD == "blackhole":
dlResult = _downloadResult(result)
else:
if not result.content and not result.url.startswith('magnet'):
result.content = result.provider.getURL(result.url,
needBytes=True)
if result.content or result.url.startswith('magnet'):
client = getClientIstance(sickrage.TORRENT_METHOD)()
dlResult = client.sendTORRENT(result)
else:
sickrage.LOGGER.warning("Torrent file content is empty")
dlResult = False
else:
sickrage.LOGGER.error(
"Unknown result type, unable to download it (%r)" %
result.resultType)
dlResult = False
if not dlResult:
return False
if sickrage.USE_FAILED_DOWNLOADS:
FailedHistory.logSnatch(result)
notifications.message('Episode snatched', result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
sql_l = []
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(
SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(
endStatus, result.quality)
sql_l.append(curEpObj.get_sql())
if curEpObj.status not in Quality.DOWNLOADED:
try:
notify_snatch(
curEpObj._format_pattern('%SN - %Sx%0E - %EN - %QN') +
" from " + result.provider.name)
except:
sickrage.LOGGER.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.NOTIFIERS.trakt_notifier.trakt_episode_data_generate(
trakt_data)
if sickrage.USE_TRAKT and sickrage.TRAKT_SYNC_WATCHLIST:
sickrage.LOGGER.debug("Add episodes, showid: indexerid " +
str(result.show.indexerid) + ", Title " +
str(result.show.name) + " to Traktv Watchlist")
if data:
sickrage.NOTIFIERS.trakt_notifier.update_watchlist(
result.show, data_episode=data, update="add")
if len(sql_l) > 0:
main_db.MainDB().mass_action(sql_l)
return True
def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.app.config.allow_high_priority:
# if it aired recently make it high priority
for curEp in result.episodes:
if date.today() - curEp.airdate <= timedelta(days=7):
result.priority = 1
if re.search(r'(^|[. _-])(proper|repack)([. _-]|$)', result.name, re.I) is not None:
endStatus = SNATCHED_PROPER
# get result content
result.content = result.provider.get_content(result.url)
dlResult = False
if result.resultType in ("nzb", "nzbdata"):
if sickrage.app.config.nzb_method == "blackhole":
dlResult = result.provider.download_result(result)
elif sickrage.app.config.nzb_method == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.app.config.nzb_method == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.app.log.error("Unknown NZB action specified in config: " + sickrage.app.config.nzb_method)
elif result.resultType in ("torrent", "torznab"):
# add public trackers to torrent result
if not result.provider.private:
result = result.provider.add_trackers(result)
if sickrage.app.config.torrent_method == "blackhole":
dlResult = result.provider.download_result(result)
else:
if any([result.content, result.url.startswith('magnet:')]):
client = getClientIstance(sickrage.app.config.torrent_method)()
dlResult = client.send_torrent(result)
else:
sickrage.app.log.warning("Torrent file content is empty")
else:
sickrage.app.log.error("Unknown result type, unable to download it (%r)" % result.resultType)
# no download results found
if not dlResult:
return False
FailedHistory.logSnatch(result)
sickrage.app.alerts.message(_('Episode snatched'), result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(endStatus, result.quality)
# save episode to DB
curEpObj.save_to_db()
if curEpObj.status not in Quality.DOWNLOADED:
try:
Notifiers.mass_notify_snatch(
curEpObj._format_pattern('%SN - %Sx%0E - %EN - %QN') + " from " + result.provider.name)
except:
sickrage.app.log.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.app.notifier_providers['trakt'].trakt_episode_data_generate(trakt_data)
if sickrage.app.config.use_trakt and sickrage.app.config.trakt_sync_watchlist:
sickrage.app.log.debug(
"Add episodes, showid: indexerid " + str(result.show.indexerid) + ", Title " + str(
result.show.name) + " to Traktv Watchlist")
if data:
sickrage.app.notifier_providers['trakt'].update_watchlist(result.show, data_episode=data, update="add")
return True
def snatchEpisode(result, endStatus=SNATCHED):
"""
Contains the internal logic necessary to actually "snatch" a result that
has been found.
:param result: SearchResult instance to be snatched.
:param endStatus: the episode status that should be used for the episode object once it's snatched.
:return: boolean, True on success
"""
if result is None:
return False
result.priority = 0 # -1 = low, 0 = normal, 1 = high
if sickrage.ALLOW_HIGH_PRIORITY:
# if it aired recently make it high priority
for curEp in result.episodes:
if datetime.date.today() - curEp.airdate <= datetime.timedelta(days=7):
result.priority = 1
if re.search(r"(^|[\. _-])(proper|repack)([\. _-]|$)", result.name, re.I) is not None:
endStatus = SNATCHED_PROPER
if result.url.startswith("magnet") or result.url.endswith("torrent"):
result.resultType = "torrent"
# NZBs can be sent straight to SAB or saved to disk
if result.resultType in ("nzb", "nzbdata"):
if sickrage.NZB_METHOD == "blackhole":
dlResult = _downloadResult(result)
elif sickrage.NZB_METHOD == "sabnzbd":
dlResult = SabNZBd.sendNZB(result)
elif sickrage.NZB_METHOD == "nzbget":
is_proper = True if endStatus == SNATCHED_PROPER else False
dlResult = NZBGet.sendNZB(result, is_proper)
else:
sickrage.LOGGER.error("Unknown NZB action specified in config: " + sickrage.NZB_METHOD)
dlResult = False
# TORRENTs can be sent to clients or saved to disk
elif result.resultType == "torrent":
# torrents are saved to disk when blackhole mode
if sickrage.TORRENT_METHOD == "blackhole":
dlResult = _downloadResult(result)
else:
if not result.content and not result.url.startswith("magnet"):
result.content = result.provider.getURL(result.url, needBytes=True)
if result.content or result.url.startswith("magnet"):
client = getClientIstance(sickrage.TORRENT_METHOD)()
dlResult = client.sendTORRENT(result)
else:
sickrage.LOGGER.warning("Torrent file content is empty")
dlResult = False
else:
sickrage.LOGGER.error("Unknown result type, unable to download it (%r)" % result.resultType)
dlResult = False
if not dlResult:
return False
if sickrage.USE_FAILED_DOWNLOADS:
FailedHistory.logSnatch(result)
notifications.message("Episode snatched", result.name)
History.logSnatch(result)
# don't notify when we re-download an episode
sql_l = []
trakt_data = []
for curEpObj in result.episodes:
with curEpObj.lock:
if isFirstBestMatch(result):
curEpObj.status = Quality.compositeStatus(SNATCHED_BEST, result.quality)
else:
curEpObj.status = Quality.compositeStatus(endStatus, result.quality)
sql_l.append(curEpObj.get_sql())
if curEpObj.status not in Quality.DOWNLOADED:
try:
notify_snatch(curEpObj._format_pattern("%SN - %Sx%0E - %EN - %QN") + " from " + result.provider.name)
except:
sickrage.LOGGER.debug("Failed to send snatch notification")
trakt_data.append((curEpObj.season, curEpObj.episode))
data = sickrage.NOTIFIERS.trakt_notifier.trakt_episode_data_generate(trakt_data)
if sickrage.USE_TRAKT and sickrage.TRAKT_SYNC_WATCHLIST:
sickrage.LOGGER.debug(
"Add episodes, showid: indexerid "
+ str(result.show.indexerid)
+ ", Title "
+ str(result.show.name)
+ " to Traktv Watchlist"
)
if data:
sickrage.NOTIFIERS.trakt_notifier.update_watchlist(result.show, data_episode=data, update="add")
if len(sql_l) > 0:
main_db.MainDB().mass_action(sql_l)
return True
