diff --git a/hivemind_bus_client/client.py b/hivemind_bus_client/client.py index 9f9b4c8..487f056 100644 --- a/hivemind_bus_client/client.py +++ b/hivemind_bus_client/client.py @@ -188,6 +188,17 @@ def connect(self, bus=FakeBus(), protocol=None, site_id=None): self.protocol.bind(bus) self.wait_for_handshake() + def on_open(self, *args): + """ + Handle the "open" event from the websocket. + A Basic message with the name "open" is forwarded to the emitter. + """ + LOG.debug("Connected") + self.connected_event.set() + self.emitter.emit("open") + # Restore reconnect timer to 5 seconds on sucessful connect + self.retry = 5 + def on_error(self, *args): self.handshake_event.clear() self.crypto_key = None