add Client.wait_until_ready (#1)

* add Client.wait_until_ready * add missing future import
circuitsacul · Dec 8, 2022 · 1c8e060 · 1c8e060
1 parent e896fdc
commit 1c8e060
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/example/client.py b/example/client.py
@@ -5,9 +5,10 @@
 
 
 async def send(client: Client) -> None:
+    await client.wait_until_ready()
     while True:
-        await asyncio.sleep(5)
         await client.send(Message(data="hi"), client.clients)
+        await asyncio.sleep(1)
 
 
 async def main(client: Client) -> None:

diff --git a/example/events.py b/example/events.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from socketapp import Client, Event, Server
 
 

diff --git a/socketapp/client.py b/socketapp/client.py
@@ -32,6 +32,10 @@ def __init__(
     def uri(self) -> str:
         return f"ws://{self.host}:{self.port}"
 
+    async def wait_until_ready(self) -> None:
+        while self.ws is None and self.future and not self.future.done():
+            await asyncio.sleep(0.25)
+
     async def run(self) -> None:
         asyncio.get_event_loop().add_signal_handler(signal.SIGINT, self.stop)