Changeset e99dd0 for deluge/ui/web/server.py

Timestamp:

02/16/2009 12:46:16 AM (16 years ago)

Author:

Damien Churchill <damoc@gmail.com>

Branches:

2.0.x, develop, extjs4-port, master

Children:

990790

Parents:

5bd5db

Message:

more progress on the connection manager

File:

• deluge/ui/web/server.py (modified) (10 diffs)

deluge/ui/web/server.py

@@ -25 +25 @@
 import os
 import sys
+import time
 import locale
 import shutil
@@ -49 +50 @@
 from deluge.log import setupLogger, LOG as _log
 from deluge.ui import common as uicommon
-from deluge.ui.client import client
+from deluge.ui.client import client, Client
 from deluge.ui.tracker_icons import TrackerIcons
 log = logging.getLogger(__name__)
@@ -65 +66 @@
 except Exception, e:
     log.error("Unable to initialize gettext/locale: %s", e)
+
+_ = gettext.gettext
 
 current_dir = os.path.dirname(__file__)
@@ -89 +92 @@
     "refresh_secs": 10
 }
+
+DEFAULT_HOST = "127.0.0.1"
+DEFAULT_PORT = 58846
+DEFAULT_HOSTS = {
+    "hosts": [(hashlib.sha1(str(time.time())).hexdigest(), DEFAULT_HOST, DEFAULT_PORT, "", "")]
+}
+
+HOSTLIST_COL_ID = 0
+HOSTLIST_COL_HOST = 1
+HOSTLIST_COL_PORT = 2
+HOSTLIST_COL_STATUS = 3
+HOSTLIST_COL_USER = 4
+HOSTLIST_COL_PASS = 5
+HOSTLIST_COL_VERSION = 6
+
 config = ConfigManager("webui06.conf", CONFIG_DEFAULTS)
+hostlist = ConfigManager("hostlist.conf.1.2", DEFAULT_HOSTS)
 
 def rpath(path):
@@ -100 +119 @@
     
     builtins = {
-        "_": gettext.gettext,
+        "_": _,
         "version": common.get_version()
     }
@@ -127 +146 @@
             "web.get_torrent_info": self.get_torrent_info,
             "web.add_torrents": self.add_torrents,
-            "web.login": self.login
+            "web.login": self.login,
+            "web.get_hosts": self.get_hosts
         }
         for entry in open(common.get_default_config_dir("auth")):
@@ -140 +160 @@
             self.local_username = username
             self.local_password = password
-        self.connect()
-    
-    def connect(self, host="localhost", username=None, password=None):
+    
+    def connect(self, host="localhost", port=58846, username=None, password=None):
         """
         Connects the client to a daemon
@@ -215 +234 @@
                 return self._exec_remote(method, params), request_id
         except Exception, e:
+            log.exception(e)
             raise JSONException(e)
     
@@ -353 +373 @@
     
     def login(self, password):
-        """
+        """Method to allow the webui to authenticate
         """
         m = hashlib.md5()
@@ -362 +382 @@
         return d
     
+    def get_hosts(self):
+        """Return the hosts in the hostlist"""
+        hosts = dict((host[0], host[:]) for host in hostlist["hosts"])
+        
+        main_deferred = Deferred()
+        def run_check():
+            if all(map(lambda x: x[5] is not None, hosts.values())):
+                main_deferred.callback(hosts.values())
+        
+        def on_connect(result, c, host_id):
+            def on_info(info, c):
+                hosts[host_id][5] = _("Online")
+                hosts[host_id][6] = info
+                c.disconnect()
+                run_check()
+            
+            def on_info_fail(reason):
+                hosts[host_id][5] = _("Offline")
+                run_check()
+            
+            d = c.daemon.info()
+            d.addCallback(on_info, c)
+            d.addErrback(on_info_fail, c)
+            
+        def on_connect_failed(reason, host_id):
+            print reason
+            hosts[host_id][5] = _("Offline")
+            run_check()
+        
+        for host in hosts.values():
+            host_id, host, port, user, password = host[0:5]
+            hosts[host_id].append(None)
+            hosts[host_id].append(None)
+            
+            if client.connected() and (host, port, user) == client.connection_info():
+                def on_info(info):
+                    hosts[host_id][6] = info
+                    run_check()
+                host[5] = _("Connected")
+                client.daemon.info().addCallback(on_info)
+                hosts[host_id] = host
+                continue
+            
+            c = Client()
+            d = c.connect(host, port, user, password)
+            d.addCallback(on_connect, c, host_id)
+            d.addErrback(on_connect_failed, host_id)
+        return main_deferred
 
 class GetText(resource.Resource):