Add block=True back to get_msg()

Author: davidbrochart

jupyter_client/channels.py

@@ -254,12 +254,17 @@ async def _recv(self, **kwargs) -> t.Dict[str, t.Any]:
         ident, smsg = self.session.feed_identities(msg)
         return self.session.deserialize(smsg)
 
-    async def get_msg(self, timeout: t.Optional[float] = None) -> t.Dict[str, t.Any]:
+    async def get_msg(
+        self, block: bool = True, timeout: t.Optional[float] = None
+    ) -> t.Dict[str, t.Any]:
         """ Gets a message if there is one that is ready. """
-        if timeout is not None:
-            timeout *= 1000  # seconds to ms
         assert self.socket is not None
-        ready = await self.socket.poll(timeout)
+        if block:
+            if timeout is not None:
+                timeout *= 1000  # seconds to ms
+            ready = await self.socket.poll(timeout)
+        else:
+            ready = await self.socket.poll(0)
 
         if ready:
             res = await self._recv()