fix top-level await in ipython

prompt-toolkit · Nov 3, 2023 · 9ea3238 · 9ea3238
1 parent 945426b
commit 9ea3238
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ptpython/ipython.py b/ptpython/ipython.py
@@ -38,6 +38,7 @@
 
 from .completer import PythonCompleter
 from .python_input import PythonInput
+from .repl import PyCF_ALLOW_TOP_LEVEL_AWAIT
 from .style import default_ui_style
 from .validator import PythonValidator
 
@@ -211,6 +212,12 @@ def __init__(self, ipython_shell, *a, **kw):
         self.ui_styles = {"default": Style.from_dict(style_dict)}
         self.use_ui_colorscheme("default")
 
+    def get_compiler_flags(self):
+        flags = super().get_compiler_flags()
+        if self.ipython_shell.autoawait:
+            flags |= PyCF_ALLOW_TOP_LEVEL_AWAIT
+        return flags
+
 
 class InteractiveShellEmbed(_InteractiveShellEmbed):
     """