1 files changed, 124 insertions, 0 deletions
diff --git a/deluge/ui/console/widgets/statusbars.py b/deluge/ui/console/widgets/statusbars.py
new file mode 100644
index 0000000..1b91737
--- /dev/null
+++ b/deluge/ui/console/widgets/statusbars.py
@@ -0,0 +1,124 @@
+#
+# Copyright (C) 2009 Andrew Resch <andrewresch@gmail.com>
+#
+# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with
+# the additional special exception to link portions of this program with the OpenSSL library.
+# See LICENSE for more details.
+#
+
+import deluge.common
+import deluge.component as component
+from deluge.ui.client import client
+
+DEFAULT_DAEMON_PORT = 58846
+
+
+class StatusBars(component.Component):
+    def __init__(self):
+        component.Component.__init__(self, 'StatusBars', 2, depend=['CoreConfig'])
+        self.config = component.get('CoreConfig')
+
+        # Hold some values we get from the core
+        self.connections = 0
+        self.download = ''
+        self.upload = ''
+        self.dht = 0
+        self.external_ip = ''
+
+        # Default values
+        self.topbar = '{!status!}Deluge %s Console - ' % deluge.common.get_version()
+        self.bottombar = '{!status!}C: %s' % self.connections
+
+    def start(self):
+        self.update()
+
+    def update(self):
+        def on_get_session_status(status):
+            self.upload = deluge.common.fsize(status['payload_upload_rate'])
+            self.download = deluge.common.fsize(status['payload_download_rate'])
+            self.connections = status['peer.num_peers_connected']
+            if 'dht_nodes' in status:
+                self.dht = status['dht.dht_nodes']
+
+            self.update_statusbars()
+
+        def on_get_external_ip(external_ip):
+            self.external_ip = external_ip
+
+        keys = [
+            'peer.num_peers_connected',
+            'payload_upload_rate',
+            'payload_download_rate',
+        ]
+
+        if self.config['dht']:
+            keys.append('dht.dht_nodes')
+
+        client.core.get_session_status(keys).addCallback(on_get_session_status)
+        client.core.get_external_ip().addCallback(on_get_external_ip)
+
+    def update_statusbars(self):
+        # Update the topbar string
+        self.topbar = '{!status!}Deluge %s Console - ' % deluge.common.get_version()
+
+        if client.connected():
+            info = client.connection_info()
+            connection_info = ''
+
+            # Client name
+            if info[2] == 'localclient':
+                connection_info += '{!white,blue!}%s'
+            else:
+                connection_info += '{!green,blue,bold!}%s'
+
+            # Hostname
+            if info[0] == '127.0.0.1':
+                connection_info += '{!white,blue,bold!}@{!white,blue!}%s'
+            else:
+                connection_info += '{!white,blue,bold!}@{!red,blue,bold!}%s'
+
+            # Port
+            if info[1] == DEFAULT_DAEMON_PORT:
+                connection_info += '{!white,blue!}:%s'
+            else:
+                connection_info += '{!status!}:%s'
+
+            # Change color back to normal, just in case
+            connection_info += '{!status!}'
+
+            self.topbar += connection_info % (info[2], info[0], info[1])
+        else:
+            self.topbar += 'Not Connected'
+
+        # Update the bottombar string
+        self.bottombar = '{!status!}C: {!white,blue!}%s{!status!}' % self.connections
+
+        if self.config['max_connections_global'] > -1:
+            self.bottombar += ' (%s)' % self.config['max_connections_global']
+
+        if self.download != '0.0 KiB':
+            self.bottombar += ' D: {!magenta,blue,bold!}%s{!status!}' % self.download
+        else:
+            self.bottombar += ' D: {!white,blue!}%s{!status!}' % self.download
+
+        if self.config['max_download_speed'] > -1:
+            self.bottombar += (
+                ' (%s ' % self.config['max_download_speed'] + _('KiB/s') + ')'
+            )
+
+        if self.upload != '0.0 KiB':
+            self.bottombar += ' U: {!green,blue,bold!}%s{!status!}' % self.upload
+        else:
+            self.bottombar += ' U: {!white,blue!}%s{!status!}' % self.upload
+
+        if self.config['max_upload_speed'] > -1:
+            self.bottombar += (
+                ' (%s ' % self.config['max_upload_speed'] + _('KiB/s') + ')'
+            )
+
+        if self.config['dht']:
+            self.bottombar += ' ' + _('DHT') + ': {!white,blue!}%s{!status!}' % self.dht
+
+        self.bottombar += ' ' + _('IP {!white,blue!}%s{!status!}') % (
+            self.external_ip if self.external_ip else _('n/a')
+        )