def sendNZB(nzb, proper=False): # pylint: disable=too-many-locals, too-many-statements, too-many-branches, too-many-return-statements
"""
Sends NZB to NZBGet client
:param nzb: nzb object
:param proper: True if this is a Proper download, False if not. Defaults to False
"""
if sickbeard.NZBGET_HOST is None:
logger.log('No NZBget host found in configuration. Please configure it.', logger.WARNING)
return False
addToTop = False
nzbgetprio = 0
category = sickbeard.NZBGET_CATEGORY
if nzb.show.is_anime:
category = sickbeard.NZBGET_CATEGORY_ANIME
url = 'http{}://{}:{}@{}/xmlrpc'.format(
's' if sickbeard.NZBGET_USE_HTTPS else '',
sickbeard.NZBGET_USERNAME,
sickbeard.NZBGET_PASSWORD,
sickbeard.NZBGET_HOST)
nzbGetRPC = ServerProxy(url)
try:
if nzbGetRPC.writelog('INFO', 'SickRage connected to drop off {} any moment now.'.format(nzb.name + '.nzb')):
logger.log('Successful connected to NZBget', logger.DEBUG)
else:
logger.log('Successful connected to NZBget, but unable to send a message', logger.WARNING)
except socket.error:
logger.log(
'Please check your NZBget host and port (if it is running). NZBget is not responding to this combination',
logger.WARNING)
return False
except ProtocolError as e:
if e.errmsg == 'Unauthorized':
logger.log('NZBget username or password is incorrect.', logger.WARNING)
else:
logger.log('Protocol Error: ' + e.errmsg, logger.ERROR)
return False
dupekey = ''
dupescore = 0
# if it aired recently make it high priority and generate DupeKey/Score
for curEp in nzb.episodes:
if dupekey == '':
if curEp.show.indexer == 1:
dupekey = 'SickRage-' + str(curEp.show.indexerid)
elif curEp.show.indexer == 2:
dupekey = 'SickRage-tvr' + str(curEp.show.indexerid)
dupekey += '-' + str(curEp.season) + '.' + str(curEp.episode)
if datetime.date.today() - curEp.airdate <= datetime.timedelta(days=7):
addToTop = True
nzbgetprio = sickbeard.NZBGET_PRIORITY
else:
category = sickbeard.NZBGET_CATEGORY_BACKLOG
if nzb.show.is_anime:
category = sickbeard.NZBGET_CATEGORY_ANIME_BACKLOG
if nzb.quality != Quality.UNKNOWN:
dupescore = nzb.quality * 100
if proper:
dupescore += 10
nzbcontent64 = None
if nzb.resultType == 'nzbdata':
data = nzb.extraInfo[0]
nzbcontent64 = standard_b64encode(data)
logger.log('Sending NZB to NZBget')
logger.log('URL: ' + url, logger.DEBUG)
try:
# Find out if nzbget supports priority (Version 9.0+),
# old versions beginning with a 0.x will use the old command
nzbget_version_str = nzbGetRPC.version()
nzbget_version = try_int(nzbget_version_str[:nzbget_version_str.find('.')])
if nzbget_version == 0:
if nzbcontent64:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category, addToTop, nzbcontent64)
else:
if nzb.resultType == 'nzb':
if not nzb.provider.login():
return False
data = nzb.provider.get_url(nzb.url, returns='content')
if data is None:
return False
nzbcontent64 = standard_b64encode(data)
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category, addToTop, nzbcontent64)
elif nzbget_version == 12:
if nzbcontent64 is not None:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category, nzbgetprio, False,
nzbcontent64, False, dupekey, dupescore, 'score')
else:
nzbget_result = nzbGetRPC.appendurl(nzb.name + '.nzb', category, nzbgetprio, False,
nzb.url, False, dupekey, dupescore, 'score')
# v13+ has a new combined append method that accepts both (url and content)
# also the return value has changed from boolean to integer
# (Positive number representing NZBID of the queue item. 0 and negative numbers represent error codes.)
elif nzbget_version >= 13:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', nzbcontent64 if nzbcontent64 is not None else nzb.url,
category, nzbgetprio, False, False, dupekey, dupescore,
'score') > 0
else:
if nzbcontent64 is not None:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category, nzbgetprio, False,
nzbcontent64)
else:
nzbget_result = nzbGetRPC.appendurl(nzb.name + '.nzb', category, nzbgetprio, False,
nzb.url)
if nzbget_result:
logger.log('NZB sent to NZBget successfully', logger.DEBUG)
return True
else:
logger.log('NZBget could not add {} to the queue'.format(nzb.name + '.nzb'), logger.WARNING)
return False
except Exception:
logger.log('Connect Error to NZBget: could not add {} to the queue'.format(nzb.name + '.nzb'), logger.WARNING)
return False
def sendNZB(nzb, proper=False):
"""
Sends NZB to NZBGet client
:param nzb: nzb object
:param proper: True if a Proper download, False if not.
"""
if app.NZBGET_HOST is None:
log.warning('No NZBget host found in configuration.'
' Please configure it.')
return False
addToTop = False
nzbgetprio = 0
category = app.NZBGET_CATEGORY
if nzb.show.is_anime:
category = app.NZBGET_CATEGORY_ANIME
url = 'http{}://{}:{}@{}/xmlrpc'.format(
's' if app.NZBGET_USE_HTTPS else '', app.NZBGET_USERNAME,
app.NZBGET_PASSWORD, app.NZBGET_HOST)
if not NZBConnection(url):
return False
nzbGetRPC = ServerProxy(url)
dupekey = ''
dupescore = 0
# if it aired recently make it high priority and generate DupeKey/Score
for cur_ep in nzb.episodes:
if dupekey == '':
if cur_ep.series.indexer == 1:
dupekey = 'Medusa-' + str(cur_ep.series.indexerid)
elif cur_ep.series.indexer == 2:
dupekey = 'Medusa-tvr' + str(cur_ep.series.indexerid)
dupekey += '-' + str(cur_ep.season) + '.' + str(cur_ep.episode)
if datetime.date.today() - cur_ep.airdate <= datetime.timedelta(
days=7):
addToTop = True
nzbgetprio = app.NZBGET_PRIORITY
else:
category = app.NZBGET_CATEGORY_BACKLOG
if nzb.show.is_anime:
category = app.NZBGET_CATEGORY_ANIME_BACKLOG
if nzb.quality != Quality.UNKNOWN:
dupescore = nzb.quality * 100
if proper:
dupescore += 10
nzbcontent64 = None
if nzb.result_type == 'nzbdata':
data = nzb.extra_info[0]
nzbcontent64 = standard_b64encode(data)
log.info('Sending NZB to NZBget')
log.debug('URL: {}', url)
try:
# Find out if nzbget supports priority (Version 9.0+),
# old versions beginning with a 0.x will use the old command
nzbget_version_str = nzbGetRPC.version()
nzbget_version = try_int(
nzbget_version_str[:nzbget_version_str.find('.')])
if nzbget_version == 0:
if nzbcontent64:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category,
addToTop, nzbcontent64)
else:
if nzb.result_type == 'nzb':
if not nzb.provider.login():
return False
# TODO: Check if this needs exception handling
data = nzb.provider.session(nzb.url).content
if data is None:
return False
nzbcontent64 = standard_b64encode(data)
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category,
addToTop, nzbcontent64)
elif nzbget_version == 12:
if nzbcontent64 is not None:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category,
nzbgetprio, False,
nzbcontent64, False, dupekey,
dupescore, 'score')
else:
nzbget_result = nzbGetRPC.appendurl(nzb.name + '.nzb',
category, nzbgetprio,
False, nzb.url, False,
dupekey, dupescore,
'score')
# v13+ has a new combined append method that accepts both (url and
# content) also the return value has changed from boolean to integer
# (Positive number representing NZBID of the queue item. 0 and negative
# numbers represent error codes.)
elif nzbget_version >= 13:
nzbget_result = nzbGetRPC.append(
nzb.name + '.nzb', nzbcontent64
if nzbcontent64 is not None else nzb.url, category, nzbgetprio,
False, False, dupekey, dupescore, 'score') > 0
else:
if nzbcontent64 is not None:
nzbget_result = nzbGetRPC.append(nzb.name + '.nzb', category,
nzbgetprio, False,
nzbcontent64)
else:
nzbget_result = nzbGetRPC.appendurl(nzb.name + '.nzb',
category, nzbgetprio,
False, nzb.url)
if nzbget_result:
log.debug('NZB sent to NZBget successfully')
return True
else:
log.warning('NZBget could not add {name}.nzb to the queue',
{'name': nzb.name})
return False
except Exception:
log.warning(
'Connect Error to NZBget: could not add {file}.nzb to the'
' queue', {'name': nzb.name})
return False
