mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-24 03:33:38 +00:00
144 lines
4.9 KiB
Python
144 lines
4.9 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 sickgear
|
|
from . import logger
|
|
from .classes import NZBDataSearchResult, NZBSearchResult
|
|
|
|
|
|
def send_nzb(search_result):
|
|
"""
|
|
Sends an nzb to SABnzbd via the API.
|
|
|
|
:param search_result: The NZBSearchResult object to send to SAB
|
|
:type search_result: NZBSearchResult or NZBDataSearchResult
|
|
:return: success
|
|
:rtype: bool
|
|
"""
|
|
|
|
success, result, nzb_type = True, '', 'nzb'
|
|
if search_result.resultType in ('nzb', 'nzbdata'):
|
|
|
|
# set up a dict with the URL params in it
|
|
params = {'output': 'json'}
|
|
if None is not sickgear.SAB_USERNAME:
|
|
params['ma_username'] = sickgear.SAB_USERNAME
|
|
if None is not sickgear.SAB_PASSWORD:
|
|
params['ma_password'] = sickgear.SAB_PASSWORD
|
|
if None is not sickgear.SAB_APIKEY:
|
|
params['apikey'] = sickgear.SAB_APIKEY
|
|
if None is not sickgear.SAB_CATEGORY:
|
|
params['cat'] = sickgear.SAB_CATEGORY
|
|
|
|
# use high priority if specified (recently aired episode)
|
|
if 1 == search_result.priority:
|
|
params['priority'] = 1
|
|
|
|
params['nzbname'] = '%s.nzb' % search_result.name
|
|
|
|
kwargs = {}
|
|
# if it's a normal result we just pass SAB the URL
|
|
if 'nzb' == search_result.resultType:
|
|
nzb_type = 'nzb url'
|
|
params['mode'] = 'addurl'
|
|
params['name'] = search_result.url
|
|
kwargs['params'] = params
|
|
|
|
# if we get a raw data result we want to upload it to SAB
|
|
else:
|
|
nzb_type = 'file nzb'
|
|
params['mode'] = 'addfile'
|
|
kwargs['post_data'] = params
|
|
nzb_data = search_result.get_data()
|
|
if not nzb_data:
|
|
return False
|
|
kwargs['files'] = {'nzbfile': ('%s.nzb' % search_result.name, nzb_data)}
|
|
|
|
logger.log(f'Sending {nzb_type} to SABnzbd: {search_result.name}')
|
|
|
|
url = '%sapi' % sickgear.SAB_HOST
|
|
logger.debug(f'SABnzbd at {url} sent params: {params}')
|
|
success, result = _get_url(url, **kwargs)
|
|
|
|
if not success:
|
|
return False
|
|
|
|
# do some crude parsing of the result text to determine what SAB said
|
|
if result.get('status'):
|
|
logger.debug(f'Success from SABnzbd using {nzb_type}')
|
|
return True
|
|
elif 'error' in result:
|
|
logger.error(f'Failed using {nzb_type} with SABnzbd, response: {result.get("error", "und")}')
|
|
else:
|
|
logger.error(f'Failure unknown using {nzb_type} with SABnzbd, response: {result}')
|
|
return False
|
|
|
|
|
|
def _check_sab_response(result):
|
|
|
|
if 0 == len(result):
|
|
logger.error('No data returned from SABnzbd, nzb not used')
|
|
return False, 'No data from SABnzbd'
|
|
|
|
if 'error' in result:
|
|
logger.error(result['error'])
|
|
return False, result['error']
|
|
return True, result
|
|
|
|
|
|
def _get_url(url, params=None, **kwargs):
|
|
|
|
result = sickgear.helpers.get_url(url, params=params, parse_json=True, **kwargs)
|
|
if None is result:
|
|
logger.error('Error, no response from SABnzbd')
|
|
return False, 'Error, no response from SABnzbd'
|
|
return True, result
|
|
|
|
|
|
def access_method(host):
|
|
|
|
success, result = _get_url('%sapi' % host, params={'mode': 'auth', 'output': 'json'})
|
|
if not success:
|
|
return False, result
|
|
return True, result['auth']
|
|
|
|
|
|
def test_authentication(host=None, username=None, password=None, apikey=None):
|
|
"""
|
|
Sends a simple API request to SAB to determine if the given connection information is correct
|
|
|
|
Returns: A tuple containing the success boolean and a message
|
|
:param host: The host where SAB is running (incl port)
|
|
:param username: The username to use for the HTTP request
|
|
:param password: The password to use for the HTTP request
|
|
:param apikey: The API key to provide to SAB
|
|
"""
|
|
|
|
# build up the URL parameters
|
|
params = {'mode': 'queue', 'ma_username': username, 'ma_password': password, 'apikey': apikey, 'output': 'json'}
|
|
url = '%sapi' % host
|
|
|
|
# send the test request
|
|
logger.debug(f'SABnzbd test URL: {url} with parameters: {params}')
|
|
success, result = _get_url(url, params=params)
|
|
if not success:
|
|
return False, result
|
|
|
|
# check the result and determine if it's good or not
|
|
success, response = _check_sab_response(result)
|
|
if not success:
|
|
return False, response
|
|
return True, 'Success'
|