Updated clear_worker

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