From 983826fa2b494957f6bbdf4bb2b7b83e64a3a0d1 Mon Sep 17 00:00:00 2001 From: Eirik Byrkjeflot Anonsen Date: Thu, 19 May 2011 21:19:13 +0200 Subject: [PATCH] Implement console command for displaying and setting per-torrent options. --- deluge/ui/console/commands/manage.py | 155 ++++++++++++++++++++++++++++++++++ 1 files changed, 155 insertions(+), 0 deletions(-) create mode 100644 deluge/ui/console/commands/manage.py diff --git a/deluge/ui/console/commands/manage.py b/deluge/ui/console/commands/manage.py new file mode 100644 index 0000000..d0c94b9 --- /dev/null +++ b/deluge/ui/console/commands/manage.py @@ -0,0 +1,155 @@ +# +# manage.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: manage [ [ ...]] [ [ ...]]\n'\ + ' manage [ ...] --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('\n') + 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 = [] + request_options = [] + + for arg in args: + if arg in torrent_options: + request_options.append(arg) + else: + ids = self.console.match_torrent(arg) + if not ids: + self.console.write("{!error!}The argument '" + arg + "' is not a recognized option nor did it match any torrents") + return + torrent_ids.extend(ids) + + if not torrent_ids: + torrent_ids.extend(self.console.match_torrent("")) + + 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 = [] + for arg in args: + ids = self.console.match_torrent(arg) + if not ids: + self.console.write("{!error!}The argument '" + arg + "' did not match any torrents") + return + torrent_ids.extend(ids) + if not torrent_ids: + self.console.write('{!error!}No torrents named') + return + key = options['set'][0] + val = options['set'][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 + + def complete(self, text): + torrents = component.get('ConsoleUI').tab_complete_torrent(text) + options = [ x for x in torrent_options if x.startswith(text) ] + # This should probably only return options immediately after --set. + return torrents + options -- 1.7.2.5