From 117d29ae72582f801f8649c0ac17afc3a33ef3eb Mon Sep 17 00:00:00 2001 From: NoGare Date: Mon, 19 Mar 2012 21:51:25 -0400 Subject: [PATCH] Author: eirikba Found this in the trac http://dev.deluge-torrent.org/ticket/1858 description: This patch adds a command to deluge-console to show and set per-torrent options. I have called the command "torrent_option" and mostly modeled it on the "config" command. A better name for the command would be nice, but I have no better suggestion at this time. --- deluge/ui/console/commands/torrent_option.py | 134 +++++++++++++++++++ 1 file changed, 134 insertions(+) create mode 100644 deluge/ui/console/commands/torrent_option.py diff --git a/deluge/ui/console/commands/torrent_option.py b/deluge/ui/console/commands/torrent_option.py new file mode 100644 index 000000000..beb00935e --- /dev/null +++ b/deluge/ui/console/commands/torrent_option.py @@ -0,0 +1,134 @@ +# +# config.py +# +# Copyright (C) 2008-2009 Ido Abramovich +# Copyright (C) 2009 Andrew Resch +# +# Deluge is free software. +# +# You may 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. +# +# deluge 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 deluge. If not, write to: +# The Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor +# Boston, MA 02110-1301, USA. +# +# In addition, as a special exception, the copyright holders give +# permission to link the code of portions of this program with the OpenSSL +# library. +# You must obey the GNU General Public License in all respects for all of +# the code used other than OpenSSL. If you modify file(s) with this +# exception, you may extend this exception to your version of the file(s), +# but you are not obligated to do so. If you do not wish to do so, delete +# this exception statement from your version. If you delete this exception +# statement from all source files in the program, then also delete it here. +# +# + +from twisted.internet import defer + +from deluge.ui.console.main import BaseCommand +import deluge.ui.console.colors as colors +from deluge.ui.client import client +import deluge.component as component +from deluge.log import LOG as log + +from optparse import make_option + +torrent_options = { + "max_download_speed": float, + "max_upload_speed": float, + "max_connections": int, + "max_upload_slots": int, + "private": bool, + "prioritize_first_last": bool, + "is_auto_managed": bool, + "stop_at_ratio": bool, + "stop_ratio": float, + "remove_at_ratio": bool, + "move_on_completed": bool, + "move_on_completed_path": str + } + + +class Command(BaseCommand): + """Show and set per-torrent options""" + + option_list = BaseCommand.option_list + ( + make_option('-s', '--set', action='store', nargs=2, dest='set', + help='set value for key'), + ) + usage = "Usage: torrent_option [ [ ...]]\n"\ + " torrent_option --set " + + def handle(self, *args, **options): + self.console = component.get("ConsoleUI") + if options['set']: + return self._set_option(*args, **options) + else: + return self._get_option(*args, **options) + + + def _get_option(self, *args, **options): + + def on_torrents_status(status): + for torrentid, data in status.items(): + self.console.write('') + if 'name' in data: + self.console.write('{!info!}Name: {!input!}%s' % data.get('name')) + self.console.write('{!info!}ID: {!input!}%s' % torrentid) + for k, v in data.items(): + if k != 'name': + self.console.write('{!info!}%s: {!input!}%s' % (k, v)) + + def on_torrents_status_fail(reason): + self.console.write('{!error!}Failed to get torrent data.') + + torrent_ids = [] + torrent_ids.extend(self.console.match_torrent(args[0])) + + request_options = [] + for opt in args[1:]: + if opt not in torrent_options: + self.console.write('{!error!}Unknown torrent option: %s' % opt) + return + request_options.append(opt) + if not request_options: + request_options = [ opt for opt in torrent_options.keys() ] + request_options.append('name') + + d = client.core.get_torrents_status({"id": torrent_ids}, request_options) + d.addCallback(on_torrents_status) + d.addErrback(on_torrents_status_fail) + return d + + + def _set_option(self, *args, **options): + deferred = defer.Deferred() + torrent_ids = [] + torrent_ids.extend(self.console.match_torrent(args[0])) + key = options["set"][0] + val = options["set"][1] + " " .join(args[1:]) + + if key not in torrent_options: + self.console.write("{!error!}The key '%s' is invalid!" % key) + return + + val = torrent_options[key](val) + + def on_set_config(result): + self.console.write("{!success!}Torrent option successfully updated.") + deferred.callback(True) + + self.console.write("Setting %s to %s for torrents %s.." % (key, val, torrent_ids)) + client.core.set_torrent_options(torrent_ids, {key: val}).addCallback(on_set_config) + return deferred