Optional wait for timeout.

eerimoq · Jun 26, 2020 · 799e0a1 · 799e0a1
1 parent 930b92b
commit 799e0a1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/bunga/client.py b/bunga/client.py
@@ -189,9 +189,9 @@ async def on_get_file_rsp(self, message):
     async def on_put_file_rsp(self, message):
         await self._write_completed(message)
 
-    async def wait_for_connection(self):
+    async def wait_for_connection(self, timeout=None):
         if not self._is_connected:
-            await self._connected_event.wait()
+            await asyncio.wait_for(self._connected_event.wait(), timeout)
 
             if self._connect_exception:
                 raise self._connect_exception
@@ -325,8 +325,8 @@ def run(self):
     def stop(self):
         pass
 
-    def wait_for_connection(self):
-        asyncio.run_coroutine_threadsafe(self._client.wait_for_connection(),
+    def wait_for_connection(self, timeout=None):
+        asyncio.run_coroutine_threadsafe(self._client.wait_for_connection(timeout),
                                          self._loop).result()
 
     def execute_command(self, command):

diff --git a/bunga/version.py b/bunga/version.py
@@ -1 +1 @@
-__version__ = '0.39.0'
+__version__ = '0.40.0'