mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-12 14:13:38 +00:00
b56ddd2eec
Change download picked .nzb file on demand and not before.
161 lines
6.5 KiB
Python
161 lines
6.5 KiB
Python
#
|
|
# This file is part of SickGear.
|
|
#
|
|
# SickGear is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickGear is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
import re
|
|
|
|
import sickbeard
|
|
import config
|
|
from lib.six import moves
|
|
from base64 import standard_b64encode
|
|
from common import Quality
|
|
from sickbeard import logger
|
|
from sickbeard.helpers import tryInt
|
|
from sickbeard.providers.generic import GenericProvider
|
|
|
|
|
|
def test_nzbget(host, use_https, username, password):
|
|
|
|
result = False
|
|
if not host:
|
|
msg = 'No NZBget host found. Please configure it'
|
|
logger.log(msg, logger.ERROR)
|
|
return result, msg, None
|
|
|
|
url = 'http%(scheme)s://%(username)s:%(password)s@%(host)s/xmlrpc' % {
|
|
'scheme': ('', 's')[use_https], 'host': re.sub('(?:https?://)?(.*?)/?$', r'\1', host),
|
|
'username': username, 'password': password}
|
|
rpc_client = moves.xmlrpc_client.ServerProxy(url)
|
|
try:
|
|
msg = 'Success. Connected'
|
|
if rpc_client.writelog('INFO', 'SickGear connected as a test'):
|
|
logger.log(msg, logger.DEBUG)
|
|
else:
|
|
msg += ', but unable to send a message'
|
|
logger.log(msg, logger.ERROR)
|
|
result = True
|
|
logger.log(u'NZBGet URL: %s' % url, logger.DEBUG)
|
|
|
|
except moves.http_client.socket.error:
|
|
msg = 'Please check NZBget host and port (if it is running). NZBget is not responding to these values'
|
|
logger.log(msg, logger.ERROR)
|
|
|
|
except moves.xmlrpc_client.ProtocolError as e:
|
|
if 'Unauthorized' == e.errmsg:
|
|
msg = 'NZBget username or password is incorrect'
|
|
logger.log(msg, logger.ERROR)
|
|
else:
|
|
msg = 'Protocol Error: %s' % e.errmsg
|
|
logger.log(msg, logger.ERROR)
|
|
|
|
return result, msg, rpc_client
|
|
|
|
|
|
def send_nzb(nzb, proper=False):
|
|
result = False
|
|
add_to_top = False
|
|
nzbget_prio = 0
|
|
|
|
authed, auth_msg, rpc_client = test_nzbget(
|
|
sickbeard.NZBGET_HOST, sickbeard.NZBGET_USE_HTTPS, sickbeard.NZBGET_USERNAME, sickbeard.NZBGET_PASSWORD)
|
|
|
|
if not authed:
|
|
return authed
|
|
|
|
dupekey = ''
|
|
dupescore = 0
|
|
# if it aired recently make it high priority and generate DupeKey/Score
|
|
for curEp in nzb.episodes:
|
|
if '' == dupekey:
|
|
dupekey = "SickGear-%s%s" % (
|
|
sickbeard.indexerApi(curEp.show.indexer).config.get('dupekey', ''), curEp.show.indexerid)
|
|
dupekey += '-%s.%s' % (curEp.season, curEp.episode)
|
|
|
|
if datetime.date.today() - curEp.airdate <= datetime.timedelta(days=7):
|
|
add_to_top = True
|
|
nzbget_prio = sickbeard.NZBGET_PRIORITY
|
|
|
|
if Quality.UNKNOWN != nzb.quality:
|
|
dupescore = nzb.quality * 100
|
|
if proper:
|
|
dupescore += 10
|
|
|
|
nzbcontent64 = None
|
|
if 'nzbdata' == nzb.resultType:
|
|
data = nzb.get_data()
|
|
if not data:
|
|
return False
|
|
nzbcontent64 = standard_b64encode(data)
|
|
elif 'Anizb' == nzb.provider.name and 'nzb' == nzb.resultType:
|
|
gen_provider = GenericProvider('')
|
|
data = gen_provider.get_url(nzb.url)
|
|
if None is data:
|
|
return result
|
|
nzbcontent64 = standard_b64encode(data)
|
|
|
|
logger.log(u'Sending NZB to NZBGet: %s' % nzb.name)
|
|
|
|
try:
|
|
# Find out if nzbget supports priority (Version 9.0+), old versions beginning with a 0.x will use the old cmd
|
|
nzbget_version_str = rpc_client.version()
|
|
nzbget_version = tryInt(nzbget_version_str[:nzbget_version_str.find('.')])
|
|
|
|
# v13+ has a 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.)
|
|
if 13 <= nzbget_version:
|
|
nzbget_result = rpc_client.append(
|
|
'%s.nzb' % nzb.name, (nzbcontent64, nzb.url)[None is nzbcontent64], sickbeard.NZBGET_CATEGORY,
|
|
nzbget_prio, False, False, dupekey, dupescore, 'score') > 0
|
|
|
|
elif 12 == nzbget_version:
|
|
if nzbcontent64 is not None:
|
|
nzbget_result = rpc_client.append('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzbcontent64, False, dupekey, dupescore, 'score')
|
|
else:
|
|
nzbget_result = rpc_client.appendurl('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzb.url, False, dupekey, dupescore, 'score')
|
|
elif 0 == nzbget_version:
|
|
if nzbcontent64 is not None:
|
|
nzbget_result = rpc_client.append('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
add_to_top, nzbcontent64)
|
|
else:
|
|
if 'nzb' == nzb.resultType:
|
|
gen_provider = GenericProvider('')
|
|
data = gen_provider.get_url(nzb.url)
|
|
if None is data:
|
|
return result
|
|
|
|
nzbcontent64 = standard_b64encode(data)
|
|
nzbget_result = rpc_client.append('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
add_to_top, nzbcontent64)
|
|
else:
|
|
if nzbcontent64 is not None:
|
|
nzbget_result = rpc_client.append('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzbcontent64)
|
|
else:
|
|
nzbget_result = rpc_client.appendurl('%s.nzb' % nzb.name, sickbeard.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzb.url)
|
|
|
|
if nzbget_result:
|
|
logger.log(u'NZB sent to NZBget successfully', logger.DEBUG)
|
|
result = True
|
|
else:
|
|
logger.log(u'NZBget could not add %s to the queue' % ('%s.nzb' % nzb.name), logger.ERROR)
|
|
except:
|
|
logger.log(u'Connect Error to NZBget: could not add %s to the queue' % ('%s.nzb' % nzb.name), logger.ERROR)
|
|
|
|
return result
|