mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-05 02:43:37 +00:00
deda4bc65a
Simplify request responses within torrent clients. There is no need to pass the response through the parent class when dealing with requests. In many places this makes the code more readable and simplifies logic.
226 lines
8.1 KiB
Python
226 lines
8.1 KiB
Python
# Author: Mr_Orange <mr_orange@hotmail.it>
|
|
# 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 json
|
|
from base64 import b64encode
|
|
|
|
import sickbeard
|
|
from sickbeard import logger
|
|
from sickbeard.clients.generic import GenericClient
|
|
from lib.requests.exceptions import RequestException
|
|
|
|
|
|
class DelugeAPI(GenericClient):
|
|
def __init__(self, host=None, username=None, password=None):
|
|
|
|
super(DelugeAPI, self).__init__('Deluge', host, username, password)
|
|
|
|
self.url = self.host + 'json'
|
|
|
|
def _get_auth(self):
|
|
|
|
post_data = json.dumps({'method': 'auth.login',
|
|
'params': [self.password],
|
|
'id': 1})
|
|
|
|
try:
|
|
self.auth = self.session.post(
|
|
self.url,
|
|
data=post_data.encode('utf-8'),
|
|
verify=sickbeard.TORRENT_VERIFY_CERT
|
|
).json()['result']
|
|
|
|
post_data = json.dumps({'method': 'web.connected',
|
|
'params': [],
|
|
'id': 10})
|
|
|
|
connected = self.session.post(
|
|
self.url,
|
|
data=post_data.encode('utf-8'),
|
|
verify=sickbeard.TORRENT_VERIFY_CERT
|
|
).json()['result']
|
|
|
|
if not connected:
|
|
post_data = json.dumps({'method': 'web.get_hosts',
|
|
'params': [],
|
|
'id': 11})
|
|
|
|
hosts = self.session.post(
|
|
self.url,
|
|
data=post_data.encode('utf-8'),
|
|
verify=sickbeard.TORRENT_VERIFY_CERT
|
|
).json()['result']
|
|
|
|
if len(hosts) == 0:
|
|
logger.log(self.name + u': WebUI does not contain daemons',
|
|
logger.ERROR)
|
|
return None
|
|
|
|
post_data = json.dumps({'method': 'web.connect',
|
|
'params': [hosts[0][0]],
|
|
'id': 11})
|
|
self.session.post(self.url, data=post_data.encode('utf-8'),
|
|
verify=sickbeard.TORRENT_VERIFY_CERT)
|
|
|
|
post_data = json.dumps({'method': 'web.connected',
|
|
'params': [],
|
|
'id': 10})
|
|
connected = self.session.post(
|
|
self.url,
|
|
data=post_data.encode('utf-8'),
|
|
verify=sickbeard.TORRENT_VERIFY_CERT
|
|
).json()['result']
|
|
|
|
if not connected:
|
|
logger.log(self.name + u': WebUI could not connect to daemon',
|
|
logger.ERROR)
|
|
return None
|
|
except RequestException:
|
|
return None
|
|
|
|
return self.auth
|
|
|
|
def _add_torrent_uri(self, result):
|
|
|
|
post_data = json.dumps({
|
|
'method': 'core.add_torrent_magnet',
|
|
'params': [result.url, {
|
|
'move_completed': 'true',
|
|
'move_completed_path': sickbeard.TV_DOWNLOAD_DIR
|
|
}],
|
|
'id': 2
|
|
})
|
|
result.hash = self._request(method='post',
|
|
data=post_data).json()['result']
|
|
|
|
return result.hash
|
|
|
|
def _add_torrent_file(self, result):
|
|
|
|
post_data = json.dumps({'method':
|
|
'core.add_torrent_file',
|
|
'params': [result.name + '.torrent',
|
|
b64encode(result.content),
|
|
{'move_completed': 'true',
|
|
'move_completed_path':
|
|
sickbeard.TV_DOWNLOAD_DIR}],
|
|
'id': 2})
|
|
result.hash = self._request(method='post',
|
|
data=post_data).json()['result']
|
|
|
|
return result.hash
|
|
|
|
def _set_torrent_label(self, result):
|
|
|
|
label = sickbeard.TORRENT_LABEL
|
|
if ' ' in label:
|
|
logger.log(self.name +
|
|
u': Invalid label. Label must not contain a space',
|
|
logger.ERROR)
|
|
return False
|
|
|
|
if label:
|
|
# check if label already exists and create it if not
|
|
post_data = json.dumps({
|
|
'method': 'label.get_labels',
|
|
'params': [],
|
|
'id': 3
|
|
})
|
|
labels = self._request(method='post',
|
|
data=post_data).json()['result']
|
|
|
|
if labels is not None:
|
|
if label not in labels:
|
|
logger.log(self.name + ': ' + label +
|
|
u' label does not exist in ' +
|
|
u'Deluge we must add it',
|
|
logger.DEBUG)
|
|
post_data = json.dumps({
|
|
'method': 'label.add',
|
|
'params': [label],
|
|
'id': 4
|
|
})
|
|
self._request(method='post', data=post_data)
|
|
logger.log(self.name + ': ' + label +
|
|
u' label added to Deluge', logger.DEBUG)
|
|
|
|
# add label to torrent
|
|
post_data = json.dumps({
|
|
'method': 'label.set_torrent',
|
|
'params': [result.hash, label],
|
|
'id': 5
|
|
})
|
|
self._request(method='post', data=post_data)
|
|
logger.log(self.name + ': ' + label +
|
|
u' label added to torrent',
|
|
logger.DEBUG)
|
|
else:
|
|
logger.log(self.name + ': ' +
|
|
u'label plugin not detected',
|
|
logger.DEBUG)
|
|
return False
|
|
|
|
return True
|
|
|
|
def _set_torrent_ratio(self, result):
|
|
|
|
ratio = None
|
|
if result.ratio:
|
|
ratio = result.ratio
|
|
|
|
if ratio:
|
|
post_data = json.dumps({'method': 'core.set_torrent_stop_at_ratio',
|
|
'params': [result.hash, True],
|
|
'id': 5})
|
|
self._request(method='post', data=post_data)
|
|
|
|
post_data = json.dumps({'method': 'core.set_torrent_stop_ratio',
|
|
'params': [result.hash, float(ratio)],
|
|
'id': 6})
|
|
self._request(method='post', data=post_data)
|
|
return True
|
|
|
|
def _set_torrent_path(self, result):
|
|
|
|
if sickbeard.TORRENT_PATH:
|
|
post_data = json.dumps({
|
|
'method': 'core.set_torrent_move_completed',
|
|
'params': [result.hash, True],
|
|
'id': 7
|
|
})
|
|
self._request(method='post', data=post_data)
|
|
|
|
post_data = json.dumps({
|
|
'method': 'core.set_torrent_move_completed_path',
|
|
'params': [result.hash, sickbeard.TORRENT_PATH],
|
|
'id': 8
|
|
})
|
|
self._request(method='post', data=post_data)
|
|
return True
|
|
|
|
def _set_torrent_pause(self, result):
|
|
|
|
if sickbeard.TORRENT_PAUSED:
|
|
post_data = json.dumps({'method': 'core.pause_torrent',
|
|
'params': [[result.hash]],
|
|
'id': 9})
|
|
self._request(method='post', data=post_data)
|
|
return True
|
|
|
|
|
|
api = DelugeAPI()
|