You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
5.4 KiB
138 lines
5.4 KiB
from bencode import bencode, bdecode
|
|
from couchpotato.core.downloaders.base import Downloader
|
|
from couchpotato.core.helpers.encoding import isInt, ss
|
|
from couchpotato.core.logger import CPLog
|
|
from hashlib import sha1
|
|
from multipartpost import MultipartPostHandler
|
|
import cookielib
|
|
import httplib
|
|
import re
|
|
import time
|
|
import urllib
|
|
import urllib2
|
|
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class uTorrent(Downloader):
|
|
|
|
type = ['torrent', 'torrent_magnet']
|
|
utorrent_api = None
|
|
|
|
def download(self, data, movie, manual = False, filedata = None):
|
|
|
|
if self.isDisabled(manual) or not self.isCorrectType(data.get('type')):
|
|
return
|
|
|
|
log.debug('Sending "%s" (%s) to uTorrent.', (data.get('name'), data.get('type')))
|
|
|
|
# Load host from config and split out port.
|
|
host = self.conf('host').split(':')
|
|
if not isInt(host[1]):
|
|
log.error('Config properties are not filled in correctly, port is missing.')
|
|
return False
|
|
|
|
torrent_params = {}
|
|
if self.conf('label'):
|
|
torrent_params['label'] = self.conf('label')
|
|
|
|
if not filedata and data.get('type') == 'torrent':
|
|
log.error('Failed sending torrent, no data')
|
|
return False
|
|
if data.get('type') == 'torrent_magnet':
|
|
torrent_hash = re.findall('urn:btih:([\w]{32,40})', data.get('url'))[0].upper()
|
|
torrent_params['trackers'] = '%0D%0A%0D%0A'.join(self.torrent_trackers)
|
|
else:
|
|
info = bdecode(filedata)["info"]
|
|
torrent_hash = sha1(bencode(info)).hexdigest().upper()
|
|
torrent_filename = self.createFileName(data, filedata, movie)
|
|
|
|
# Send request to uTorrent
|
|
try:
|
|
if not self.utorrent_api:
|
|
self.utorrent_api = uTorrentAPI(host[0], port = host[1], username = self.conf('username'), password = self.conf('password'))
|
|
|
|
if data.get('type') == 'torrent_magnet':
|
|
self.utorrent_api.add_torrent_uri(data.get('url'))
|
|
else:
|
|
self.utorrent_api.add_torrent_file(torrent_filename, filedata)
|
|
|
|
# Change settings of added torrents
|
|
self.utorrent_api.set_torrent(torrent_hash, torrent_params)
|
|
if self.conf('paused', default = 0):
|
|
self.utorrent_api.pause_torrent(torrent_hash)
|
|
return True
|
|
except Exception, err:
|
|
log.error('Failed to send torrent to uTorrent: %s', err)
|
|
return False
|
|
|
|
|
|
class uTorrentAPI(object):
|
|
|
|
def __init__(self, host = 'localhost', port = 8000, username = None, password = None):
|
|
|
|
super(uTorrentAPI, self).__init__()
|
|
|
|
self.url = 'http://' + str(host) + ':' + str(port) + '/gui/'
|
|
self.token = ''
|
|
self.last_time = time.time()
|
|
cookies = cookielib.CookieJar()
|
|
self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookies), MultipartPostHandler)
|
|
self.opener.addheaders = [('User-agent', 'couchpotato-utorrent-client/1.0')]
|
|
if username and password:
|
|
password_manager = urllib2.HTTPPasswordMgrWithDefaultRealm()
|
|
password_manager.add_password(realm = None, uri = self.url, user = username, passwd = password)
|
|
self.opener.add_handler(urllib2.HTTPBasicAuthHandler(password_manager))
|
|
self.opener.add_handler(urllib2.HTTPDigestAuthHandler(password_manager))
|
|
elif username or password:
|
|
log.debug('User or password missing, not using authentication.')
|
|
self.token = self.get_token()
|
|
|
|
def _request(self, action, data = None):
|
|
if time.time() > self.last_time + 1800:
|
|
self.last_time = time.time()
|
|
self.token = self.get_token()
|
|
request = urllib2.Request(self.url + "?token=" + self.token + "&" + action, data)
|
|
try:
|
|
open_request = self.opener.open(request)
|
|
response = open_request.read()
|
|
log.debug('response: %s', response)
|
|
if response:
|
|
log.debug('uTorrent action successfull')
|
|
return response
|
|
else:
|
|
log.debug('Unknown failure sending command to uTorrent. Return text is: %s', response)
|
|
except httplib.InvalidURL, err:
|
|
log.error('Invalid uTorrent host, check your config %s', err)
|
|
except urllib2.HTTPError, err:
|
|
if err.code == 401:
|
|
log.error('Invalid uTorrent Username or Password, check your config')
|
|
else:
|
|
log.error('uTorrent HTTPError: %s', err)
|
|
except urllib2.URLError, err:
|
|
log.error('Unable to connect to uTorrent %s', err)
|
|
return False
|
|
|
|
def get_token(self):
|
|
request = self.opener.open(self.url + "token.html")
|
|
token = re.findall("<div.*?>(.*?)</", request.read())[0]
|
|
return token
|
|
|
|
def add_torrent_uri(self, torrent):
|
|
action = "action=add-url&s=%s" % urllib.quote(torrent)
|
|
return self._request(action)
|
|
|
|
def add_torrent_file(self, filename, filedata):
|
|
action = "action=add-file"
|
|
return self._request(action, {"torrent_file": (ss(filename), filedata)})
|
|
|
|
def set_torrent(self, hash, params):
|
|
action = "action=setprops&hash=%s" % hash
|
|
for k, v in params.iteritems():
|
|
action += "&s=%s&v=%s" % (k, v)
|
|
return self._request(action)
|
|
|
|
def pause_torrent(self, hash):
|
|
action = "action=pause&hash=%s" % hash
|
|
return self._request(action)
|
|
|