Updated clear_worker

2019-07-04 19:36:23 +08:00 · 2019-07-04 19:36:23 +08:00 · 5bbf0408ce
parent 976b00ca85
commit 5bbf0408ce
1 changed files with 7 additions and 10 deletions
--- a/webssh/worker.py
+++ b/webssh/worker.py
@ -13,16 +13,13 @@ clients = {}  # {ip: {id: worker}}
 def clear_worker(worker, clients):
    ip = worker.src_addr[0]
    workers = clients.get(ip)
-    if workers:
-        try:
-            workers.pop(worker.id)
-        except KeyError:
-            pass
-        else:
-            if not workers:
-                clients.pop(ip)
-    if not clients:
-        clients.clear()
+    assert worker.id in workers
+    workers.pop(worker.id)
+
+    if not workers:
+        clients.pop(ip)
+        if not clients:
+            clients.clear()


 def recycle_worker(worker):