Merge pull request #286 from blink1073/fix-handle_external_iopub-again

Author: blink1073

jupyter_console/ptshell.py

@@ -76,7 +76,7 @@
 from pygments.util import ClassNotFound
 from pygments.token import Token
 
-from jupyter_console.utils import run_sync
+from jupyter_console.utils import run_sync, ensure_async
 
 
 def ask_yes_no(prompt, default=None, interrupt=None):
@@ -839,9 +839,10 @@ def include_output(self, msg):
     async def handle_external_iopub(self, loop=None):
         while self.keep_running:
             # we need to check for keep_running from time to time
-            poll_result = run_sync(self.client.iopub_channel.socket.poll)(500)
+            poll_result = await ensure_async(self.client.iopub_channel.socket.poll(0))
             if poll_result:
                 self.handle_iopub()
+            await asyncio.sleep(0.5)
 
     def handle_iopub(self, msg_id=''):
         """Process messages on the IOPub channel

jupyter_console/utils.py

@@ -1,6 +1,6 @@
 import inspect
 import typing as t
-from jupyter_core.utils import run_sync as _run_sync
+from jupyter_core.utils import run_sync as _run_sync, ensure_async  # noqa
 
 
 T = t.TypeVar("T")