Expand await stmt handling

Fixes #107
2025-08-02 16:44:46 +08:00 · 2017-04-18 01:49:55 -04:00
parent cb6925beec
commit 4a898ff4c1
3 changed files with 5 additions and 2 deletions
--- a/test/bytecode_3.5/03_async_await.pyc
+++ b/test/bytecode_3.5/03_async_await.pyc
--- a/test/simple_source/bug35/03_async_await.py
+++ b/test/simple_source/bug35/03_async_await.py
@@ -24,3 +24,6 @@ async def awith_test():
 async def awith_as_test():
    async with 1 as i:
        print(i)
+
+async def f(z):
+    await z
--- a/uncompyle6/parsers/parse35.py
+++ b/uncompyle6/parsers/parse35.py
@@ -27,8 +27,8 @@ class Python35Parser(Python34Parser):
                           POP_BLOCK else_suite COME_FROM_LOOP

        # Python 3.5+ Await statement
-        stmt ::= await_stmt
-        await_stmt ::= call_function GET_AWAITABLE LOAD_CONST YIELD_FROM POP_TOP
+        stmt       ::= await_stmt
+        await_stmt ::= expr GET_AWAITABLE LOAD_CONST YIELD_FROM POP_TOP

        # Python 3.5+ has WITH_CLEANUP_START/FINISH