mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-01 00:43:37 +00:00
137 lines
4.7 KiB
Python
137 lines
4.7 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# 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 sickbeard
|
|
from sickbeard import logger
|
|
|
|
|
|
def send_nzb(nzb):
|
|
"""
|
|
Sends an nzb to SABnzbd via the API.
|
|
|
|
:param nzb: The NZBSearchResult object to send to SAB
|
|
"""
|
|
|
|
success, result, nzb_type = True, '', 'nzb'
|
|
if nzb.resultType in ('nzb', 'nzbdata'):
|
|
|
|
# set up a dict with the URL params in it
|
|
params = {'output': 'json'}
|
|
if sickbeard.SAB_USERNAME is not None:
|
|
params['ma_username'] = sickbeard.SAB_USERNAME
|
|
if sickbeard.SAB_PASSWORD is not None:
|
|
params['ma_password'] = sickbeard.SAB_PASSWORD
|
|
if sickbeard.SAB_APIKEY is not None:
|
|
params['apikey'] = sickbeard.SAB_APIKEY
|
|
if sickbeard.SAB_CATEGORY is not None:
|
|
params['cat'] = sickbeard.SAB_CATEGORY
|
|
|
|
# use high priority if specified (recently aired episode)
|
|
if 1 == nzb.priority:
|
|
params['priority'] = 1
|
|
|
|
kwargs = {}
|
|
# if it's a normal result we just pass SAB the URL
|
|
if 'nzb' == nzb.resultType:
|
|
nzb_type = 'nzb url'
|
|
params['mode'] = 'addurl'
|
|
params['name'] = nzb.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
|
|
kwargs['files'] = {'nzbfile': ('%s.nzb' % nzb.name, nzb.extraInfo[0])}
|
|
|
|
logger.log(u'Sending %s to SABnzbd: %s' % (nzb_type, nzb.name))
|
|
|
|
url = '%sapi' % sickbeard.SAB_HOST
|
|
logger.log(u'SABnzbd at %s sent params: %s' % (url, params), logger.DEBUG)
|
|
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.log(u'Success from SABnzbd using %s' % nzb_type, logger.DEBUG)
|
|
return True
|
|
elif 'error' in result:
|
|
logger.log(u'Failed using %s with SABnzbd, response: %s' % (nzb_type, result.get('error', 'und')), logger.ERROR)
|
|
else:
|
|
logger.log(u'Failure unknown using %s with SABnzbd, response: %s' % (nzb_type, result), logger.ERROR)
|
|
return False
|
|
|
|
|
|
def _check_sab_response(result):
|
|
|
|
if 0 == len(result):
|
|
logger.log('No data returned from SABnzbd, nzb not used', logger.ERROR)
|
|
return False, 'No data from SABnzbd'
|
|
|
|
if 'error' in result:
|
|
logger.log(result['error'], logger.ERROR)
|
|
return False, result['error']
|
|
return True, result
|
|
|
|
|
|
def _get_url(url, params=None, **kwargs):
|
|
|
|
result = sickbeard.helpers.getURL(url, params=params, json=True, **kwargs)
|
|
if None is result:
|
|
logger.log('Error, no response from SABnzbd', logger.ERROR)
|
|
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 connect
|
|
|
|
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.log(u'SABnzbd test URL: %s with parameters: %s' % (url, params), logger.DEBUG)
|
|
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'
|