Module: deluge Branch: master Commit: 510c81776f13c803aff8f32d434efefd4d3d6b5e
Author: Nick Lanham <[email protected]> Date: Thu Feb 24 13:24:35 2011 +0100 add status command --- deluge/ui/console/commands/status.py | 126 ++++++++++++++++++++++++++++++++++ 1 files changed, 126 insertions(+), 0 deletions(-) diff --git a/deluge/ui/console/commands/status.py b/deluge/ui/console/commands/status.py new file mode 100644 index 0000000..9dc6b29 --- /dev/null +++ b/deluge/ui/console/commands/status.py @@ -0,0 +1,126 @@ +# +# status.py +# +# Copyright (C) 2011 Nick Lanham <[email protected]> +# +# 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 optparse import make_option +from twisted.internet import defer +from deluge.ui.console.main import BaseCommand +from deluge.ui.client import client +import deluge.common +import deluge.component as component + +class Command(BaseCommand): + """Shows a various status information from the daemon.""" + option_list = BaseCommand.option_list + ( + make_option('-r', '--raw', action='store_true', default=False, dest='raw', + help='Don\'t format upload/download rates in KiB/s (useful for scripts that want to do their own parsing)'), + make_option('-n', '--no-torrents', action='store_false', default=True, dest='show_torrents', + help='Don\'t show torrent status (this will make the command a bit faster)'), + ) + + + usage = "Usage: status [-r] [-n]" + def handle(self, *args, **options): + self.console = component.get("ConsoleUI") + self.status = None + self.connections = None + if options["show_torrents"]: + self.torrents = None + else: + self.torrents = -2 + + self.raw = options["raw"] + + def on_session_status(status): + self.status = status + if self.status != None and self.connections != None and self.torrents != None: + self.print_status() + + def on_num_connections(conns): + self.connections = conns + if self.status != None and self.connections != None and self.torrents != None: + self.print_status() + + def on_torrents_status(status): + self.torrents = status + if self.status != None and self.connections != None and self.torrents != None: + self.print_status() + + def on_torrents_status_fail(reason): + self.torrents = -1 + if self.status != None and self.connections != None and self.torrents != None: + self.print_status() + + deferreds = [] + + ds = client.core.get_session_status(["payload_upload_rate","payload_download_rate","dht_nodes"]) + ds.addCallback(on_session_status) + deferreds.append(ds) + + dc = client.core.get_num_connections() + dc.addCallback(on_num_connections) + deferreds.append(dc) + + if options["show_torrents"]: + dt = client.core.get_torrents_status({}, ["state"]) + dt.addCallback(on_torrents_status) + dt.addErrback(on_torrents_status_fail) + deferreds.append(dt) + + return defer.DeferredList(deferreds) + + def print_status(self): + self.console.set_batch_write(True) + if self.raw: + self.console.write("{!info!}Total upload: %f"%self.status["payload_upload_rate"]) + self.console.write("{!info!}Total download: %f"%self.status["payload_download_rate"]) + else: + self.console.write("{!info!}Total upload: %s"%deluge.common.fspeed(self.status["payload_upload_rate"])) + self.console.write("{!info!}Total download: %s"%deluge.common.fspeed(self.status["payload_download_rate"])) + self.console.write("{!info!}DHT Nodes: %i"%self.status["dht_nodes"]) + self.console.write("{!info!}Total connections: %i"%self.connections) + if self.torrents == -1: + self.console.write("{!error!}Error getting torrent info") + elif self.torrents != -2: + self.console.write("{!info!}Total torrents: %i"%len(self.torrents)) + states = ["Downloading","Seeding","Paused","Checking","Error","Queued"] + state_counts = {} + for state in states: + state_counts[state] = 0 + for t in self.torrents: + s = self.torrents[t] + state_counts[s["state"]] += 1 + for state in states: + self.console.write("{!info!} %s: %i"%(state,state_counts[state])) + + self.console.set_batch_write(False) -- You received this message because you are subscribed to the Google Groups "deluge-commit" group. To post to this group, send email to [email protected]. To unsubscribe from this group, send email to [email protected]. For more options, visit this group at http://groups.google.com/group/deluge-commit?hl=en.
