Merge branch 'develop' of github.com:RuudBurger/CouchPotatoServer into develop
This commit is contained in:
@@ -58,14 +58,6 @@ config = [{
|
|||||||
'advanced': True,
|
'advanced': True,
|
||||||
'description': 'Also remove the leftover files.',
|
'description': 'Also remove the leftover files.',
|
||||||
},
|
},
|
||||||
{
|
|
||||||
'name': 'append_label',
|
|
||||||
'label': 'Append Label',
|
|
||||||
'default': False,
|
|
||||||
'advanced': True,
|
|
||||||
'type': 'bool',
|
|
||||||
'description': 'Append label to download location. Requires you to set the download location above.',
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
'name': 'paused',
|
'name': 'paused',
|
||||||
'type': 'bool',
|
'type': 'bool',
|
||||||
|
|||||||
@@ -125,9 +125,7 @@ class rTorrent(Downloader):
|
|||||||
if self.conf('label'):
|
if self.conf('label'):
|
||||||
torrent.set_custom(1, self.conf('label'))
|
torrent.set_custom(1, self.conf('label'))
|
||||||
|
|
||||||
if self.conf('directory') and self.conf('append_label'):
|
if self.conf('directory'):
|
||||||
torrent.set_directory(os.path.join(self.conf('directory'), self.conf('label')))
|
|
||||||
elif self.conf('directory'):
|
|
||||||
torrent.set_directory(self.conf('directory'))
|
torrent.set_directory(self.conf('directory'))
|
||||||
|
|
||||||
# Set Ratio Group
|
# Set Ratio Group
|
||||||
|
|||||||
@@ -36,6 +36,11 @@ config = [{
|
|||||||
'name': 'label',
|
'name': 'label',
|
||||||
'description': 'Label to add torrent as.',
|
'description': 'Label to add torrent as.',
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
'name': 'directory',
|
||||||
|
'type': 'directory',
|
||||||
|
'description': 'Download to this directory. Keep empty for default uTorrent download directory.',
|
||||||
|
},
|
||||||
{
|
{
|
||||||
'name': 'remove_complete',
|
'name': 'remove_complete',
|
||||||
'label': 'Remove torrent',
|
'label': 'Remove torrent',
|
||||||
|
|||||||
@@ -77,6 +77,7 @@ class uTorrent(Downloader):
|
|||||||
else:
|
else:
|
||||||
info = bdecode(filedata)["info"]
|
info = bdecode(filedata)["info"]
|
||||||
torrent_hash = sha1(benc(info)).hexdigest().upper()
|
torrent_hash = sha1(benc(info)).hexdigest().upper()
|
||||||
|
|
||||||
torrent_filename = self.createFileName(data, filedata, movie)
|
torrent_filename = self.createFileName(data, filedata, movie)
|
||||||
|
|
||||||
if data.get('seed_ratio'):
|
if data.get('seed_ratio'):
|
||||||
@@ -91,11 +92,17 @@ class uTorrent(Downloader):
|
|||||||
if len(torrent_hash) == 32:
|
if len(torrent_hash) == 32:
|
||||||
torrent_hash = b16encode(b32decode(torrent_hash))
|
torrent_hash = b16encode(b32decode(torrent_hash))
|
||||||
|
|
||||||
|
# Set download directory
|
||||||
|
if self.conf('directory'):
|
||||||
|
directory = self.conf('directory')
|
||||||
|
else:
|
||||||
|
directory = False
|
||||||
|
|
||||||
# Send request to uTorrent
|
# Send request to uTorrent
|
||||||
if data.get('protocol') == 'torrent_magnet':
|
if data.get('protocol') == 'torrent_magnet':
|
||||||
self.utorrent_api.add_torrent_uri(torrent_filename, data.get('url'))
|
self.utorrent_api.add_torrent_uri(torrent_filename, data.get('url'), directory)
|
||||||
else:
|
else:
|
||||||
self.utorrent_api.add_torrent_file(torrent_filename, filedata)
|
self.utorrent_api.add_torrent_file(torrent_filename, filedata, directory)
|
||||||
|
|
||||||
# Change settings of added torrent
|
# Change settings of added torrent
|
||||||
self.utorrent_api.set_torrent(torrent_hash, torrent_params)
|
self.utorrent_api.set_torrent(torrent_hash, torrent_params)
|
||||||
@@ -249,13 +256,13 @@ class uTorrentAPI(object):
|
|||||||
def add_torrent_uri(self, filename, torrent, add_folder = False):
|
def add_torrent_uri(self, filename, torrent, add_folder = False):
|
||||||
action = "action=add-url&s=%s" % urllib.quote(torrent)
|
action = "action=add-url&s=%s" % urllib.quote(torrent)
|
||||||
if add_folder:
|
if add_folder:
|
||||||
action += "&path=%s" % urllib.quote(filename)
|
action += "&path=%s" % urllib.quote(add_folder)
|
||||||
return self._request(action)
|
return self._request(action)
|
||||||
|
|
||||||
def add_torrent_file(self, filename, filedata, add_folder = False):
|
def add_torrent_file(self, filename, filedata, add_folder = False):
|
||||||
action = "action=add-file"
|
action = "action=add-file"
|
||||||
if add_folder:
|
if add_folder:
|
||||||
action += "&path=%s" % urllib.quote(filename)
|
action += "&path=%s" % urllib.quote(add_folder)
|
||||||
return self._request(action, {"torrent_file": (ss(filename), filedata)})
|
return self._request(action, {"torrent_file": (ss(filename), filedata)})
|
||||||
|
|
||||||
def set_torrent(self, hash, params):
|
def set_torrent(self, hash, params):
|
||||||
|
|||||||
Reference in New Issue
Block a user