Merge pull request #19 from piccolo-orm/ipython_improvements

iPython improvements
piccolo-orm · Nov 14, 2020 · 7c4bcb9 · 7c4bcb9
2 parents f24365a + ca29dc9
commit 7c4bcb9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/piccolo/apps/playground/commands/run.py b/piccolo/apps/playground/commands/run.py
@@ -132,8 +132,7 @@ def run(
         import IPython
     except ImportError:
         print(
-            "Install iPython using `pip install ipython==7.6.1` to use this "
-            "feature."
+            "Install iPython using `pip install ipython` to use this feature."
         )
         sys.exit(1)
 
@@ -176,4 +175,7 @@ def run(
     print("\n")
 
     populate()
-    IPython.embed()
+
+    from IPython.core.interactiveshell import _asyncio_runner
+
+    IPython.embed(using=_asyncio_runner)
diff --git a/piccolo/apps/shell/commands/run.py b/piccolo/apps/shell/commands/run.py
@@ -10,8 +10,7 @@ def start_ipython_shell(**tables: t.Dict[str, t.Type[Table]]):
         import IPython
     except ImportError:
         print(
-            "Install iPython using `pip install ipython==7.6.1` to use this "
-            "feature."
+            "Install iPython using `pip install ipython` to use this feature."
         )
         sys.exit(1)
 
@@ -20,7 +19,9 @@ def start_ipython_shell(**tables: t.Dict[str, t.Type[Table]]):
         if table_class_name not in existing_global_names:
             globals()[table_class_name] = table_class
 
-    IPython.embed()
+    from IPython.core.interactiveshell import _asyncio_runner
+
+    IPython.embed(using=_asyncio_runner)
 
 
 def run():