mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-27 05:03:37 +00:00
ae4173e8c0
Change use specific logger functions for debug, warning, error.
171 lines
6.8 KiB
Python
171 lines
6.8 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 re
|
|
|
|
import sickgear
|
|
from . import logger
|
|
from .classes import NZBDataSearchResult, NZBSearchResult
|
|
from .common import Quality
|
|
from .helpers import try_int
|
|
from .providers.generic import GenericProvider
|
|
|
|
from lib.xmlrpclib_to import ServerProxy
|
|
|
|
from _23 import b64encodestring
|
|
from six import moves
|
|
|
|
|
|
def test_nzbget(host, use_https, username, password, timeout=300):
|
|
|
|
result = False
|
|
if not host:
|
|
msg = 'No NZBGet host found. Please configure it'
|
|
logger.error(msg)
|
|
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 = ServerProxy(url, timeout=timeout)
|
|
try:
|
|
msg = 'Success. Connected'
|
|
if rpc_client.writelog('INFO', 'SickGear connected as a test'):
|
|
logger.debug(msg)
|
|
else:
|
|
msg += ', but unable to send a message'
|
|
logger.error(msg)
|
|
result = True
|
|
logger.debug(f'NZBGet URL: {url}')
|
|
|
|
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.error(msg)
|
|
|
|
except moves.xmlrpc_client.ProtocolError as e:
|
|
if 'Unauthorized' == e.errmsg:
|
|
msg = 'NZBGet username or password is incorrect'
|
|
logger.error(msg)
|
|
else:
|
|
msg = 'Protocol Error: %s' % e.errmsg
|
|
logger.error(msg)
|
|
|
|
return result, msg, rpc_client
|
|
|
|
|
|
def send_nzb(search_result):
|
|
"""
|
|
|
|
:param search_result: search result
|
|
:type search_result: NZBSearchResult or NZBDataSearchResult
|
|
:return:
|
|
"""
|
|
result = False
|
|
add_to_top = False
|
|
nzbget_prio = 0
|
|
|
|
authed, auth_msg, rpc_client = test_nzbget(
|
|
sickgear.NZBGET_HOST, sickgear.NZBGET_USE_HTTPS, sickgear.NZBGET_USERNAME, sickgear.NZBGET_PASSWORD)
|
|
|
|
if not authed:
|
|
return authed
|
|
|
|
dupekey = ''
|
|
dupescore = 0
|
|
# if it aired recently make it high priority and generate DupeKey/Score
|
|
for cur_ep_obj in search_result.ep_obj_list:
|
|
if '' == dupekey:
|
|
dupekey = 'SickGear-%s%s' % (
|
|
sickgear.TVInfoAPI(cur_ep_obj.show_obj.tvid).config.get('dupekey', ''), cur_ep_obj.show_obj.prodid)
|
|
dupekey += '-%s.%s' % (cur_ep_obj.season, cur_ep_obj.episode)
|
|
|
|
if 1 == search_result.priority:
|
|
add_to_top = True
|
|
nzbget_prio = sickgear.NZBGET_PRIORITY
|
|
|
|
if Quality.UNKNOWN != search_result.quality:
|
|
dupescore = search_result.quality * 100
|
|
|
|
dupescore += (0, 9 + search_result.properlevel)[0 < search_result.properlevel]
|
|
|
|
nzbcontent64 = None
|
|
if 'nzbdata' == search_result.resultType:
|
|
data = search_result.get_data()
|
|
if not data:
|
|
return False
|
|
nzbcontent64 = b64encodestring(data, keep_eol=True)
|
|
elif 'Anizb' == search_result.provider.name and 'nzb' == search_result.resultType:
|
|
gen_provider = GenericProvider('')
|
|
data = gen_provider.get_url(search_result.url)
|
|
if None is data:
|
|
return result
|
|
nzbcontent64 = b64encodestring(data, keep_eol=True)
|
|
|
|
logger.log(f'Sending NZB to NZBGet: {search_result.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 = try_int(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 = 0 < rpc_client.append(
|
|
'%s.nzb' % search_result.name, (nzbcontent64, search_result.url)[None is nzbcontent64],
|
|
sickgear.NZBGET_CATEGORY, nzbget_prio, False, False, dupekey, dupescore, 'score')
|
|
|
|
elif 12 == nzbget_version:
|
|
if None is not nzbcontent64:
|
|
nzbget_result = rpc_client.append(
|
|
'%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzbcontent64, False, dupekey, dupescore, 'score')
|
|
else:
|
|
nzbget_result = rpc_client.appendurl(
|
|
'%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
nzbget_prio, False, search_result.url, False, dupekey, dupescore, 'score')
|
|
elif 0 == nzbget_version:
|
|
if None is not nzbcontent64:
|
|
nzbget_result = rpc_client.append('%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
add_to_top, nzbcontent64)
|
|
else:
|
|
if 'nzb' == search_result.resultType:
|
|
gen_provider = GenericProvider('')
|
|
data = gen_provider.get_url(search_result.url)
|
|
if None is data:
|
|
return result
|
|
|
|
nzbcontent64 = b64encodestring(data, keep_eol=True)
|
|
nzbget_result = rpc_client.append('%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
add_to_top, nzbcontent64)
|
|
else:
|
|
if None is not nzbcontent64:
|
|
nzbget_result = rpc_client.append('%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
nzbget_prio, False, nzbcontent64)
|
|
else:
|
|
nzbget_result = rpc_client.appendurl('%s.nzb' % search_result.name, sickgear.NZBGET_CATEGORY,
|
|
nzbget_prio, False, search_result.url)
|
|
|
|
if nzbget_result:
|
|
logger.debug('NZB sent to NZBGet successfully')
|
|
result = True
|
|
else:
|
|
logger.error(f'NZBGet could not add {search_result.name}.nzb to the queue')
|
|
except (BaseException, Exception):
|
|
logger.error(f'Connect Error to NZBGet: could not add {search_result.name}.nzb to the queue')
|
|
|
|
return result
|