Browse Source

Merge branch 'develop' of github.com:RuudBurger/CouchPotatoServer into develop

pull/2497/head
Ruud 12 years ago
parent
commit
cec88319fe
  1. 8
      couchpotato/core/downloaders/rtorrent/__init__.py
  2. 4
      couchpotato/core/downloaders/rtorrent/main.py
  3. 5
      couchpotato/core/downloaders/utorrent/__init__.py
  4. 15
      couchpotato/core/downloaders/utorrent/main.py

8
couchpotato/core/downloaders/rtorrent/__init__.py

@ -59,14 +59,6 @@ config = [{
'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',
'type': 'bool',
'advanced': True,

4
couchpotato/core/downloaders/rtorrent/main.py

@ -125,9 +125,7 @@ class rTorrent(Downloader):
if self.conf('label'):
torrent.set_custom(1, self.conf('label'))
if self.conf('directory') and self.conf('append_label'):
torrent.set_directory(os.path.join(self.conf('directory'), self.conf('label')))
elif self.conf('directory'):
if self.conf('directory'):
torrent.set_directory(self.conf('directory'))
# Set Ratio Group

5
couchpotato/core/downloaders/utorrent/__init__.py

@ -37,6 +37,11 @@ config = [{
'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',
'label': 'Remove torrent',
'default': True,

15
couchpotato/core/downloaders/utorrent/main.py

@ -77,6 +77,7 @@ class uTorrent(Downloader):
else:
info = bdecode(filedata)["info"]
torrent_hash = sha1(benc(info)).hexdigest().upper()
torrent_filename = self.createFileName(data, filedata, movie)
if data.get('seed_ratio'):
@ -91,11 +92,17 @@ class uTorrent(Downloader):
if len(torrent_hash) == 32:
torrent_hash = b16encode(b32decode(torrent_hash))
# Set download directory
if self.conf('directory'):
directory = self.conf('directory')
else:
directory = False
# Send request to uTorrent
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:
self.utorrent_api.add_torrent_file(torrent_filename, filedata)
self.utorrent_api.add_torrent_file(torrent_filename, filedata, directory)
# Change settings of added torrent
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):
action = "action=add-url&s=%s" % urllib.quote(torrent)
if add_folder:
action += "&path=%s" % urllib.quote(filename)
action += "&path=%s" % urllib.quote(add_folder)
return self._request(action)
def add_torrent_file(self, filename, filedata, add_folder = False):
action = "action=add-file"
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)})
def set_torrent(self, hash, params):

Loading…
Cancel
Save