Allow printing while waiting for user input in readline mode

aioconsole/rlwrap.py

@@ -5,6 +5,7 @@
 import signal
 import builtins
 import subprocess
+from concurrent.futures import ThreadPoolExecutor
 
 from . import compat
 
@@ -30,22 +31,43 @@ def _rlwrap(process, prompt_control, use_stderr=False):
     # Check prompt control
     assert len(prompt_control) == 1
 
-    # Loop over prompts
-    while process.poll() is None:
-        try:
-            prompt = wait_for_prompt(
-                source, dest, prompt_control)
-            raw = input(prompt, use_stderr=use_stderr) + '\n'
-        except KeyboardInterrupt:
-            process.send_signal(signal.SIGINT)
-        except EOFError:
-            process.stdin.close()
-        else:
-            process.stdin.write(raw)
-
-    # Clean up
-    dest.write(source.read())
-    return process.returncode
+    # Run background task
+    with ThreadPoolExecutor(1) as executor:
+        future = executor.submit(
+            wait_for_prompt, source, dest, prompt_control)
+
+        # Loop over prompts
+        while process.poll() is None:
+
+            # Get prompt
+            try:
+                prompt = future.result()
+            except KeyboardInterrupt:
+                process.send_signal(signal.SIGINT)
+                continue
+            except EOFError:
+                break
+            else:
+                future = executor.submit(
+                    wait_for_prompt, source, dest, prompt_control)
+
+            # Get user input
+            try:
+                raw = input(prompt, use_stderr=use_stderr) + '\n'
+            except KeyboardInterrupt:
+                process.send_signal(signal.SIGINT)
+                continue
+            except EOFError:
+                break
+            else:
+                process.stdin.write(raw)
+
+        # Close and wait process streams
+        process.stdin.close()
+        future.exception()
+
+    # Wait process and return code
+    return process.wait()
 
 
 def wait_for_prompt(src, dest, prompt_control, buffersize=1):

setup.py

@@ -23,9 +23,10 @@
     entry_points={'console_scripts': ['apython = aioconsole:run_apython']},
 
     setup_requires=['pytest-runner' if TESTING else ''],
-    tests_require=['pytest',
-                   'pytest-asyncio<0.6;python_version=="3.4"',
+    tests_require=['pytest;python_version>"3.4"',
                    'pytest-asyncio>=0.8;python_version>"3.4"',
+                   'pytest<4;python_version=="3.4"',
+                   'pytest-asyncio<0.6;python_version=="3.4"',
                    'pytest-cov'],
 
     license="GPLv3",