fix await for futures; see #712

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -2022,7 +2022,7 @@ async def ast_formattedvalue(self, arg):
     async def ast_await(self, arg):
         """Evaluate await expr."""
         coro = await self.aeval(arg.value)
-        if coro and asyncio.iscoroutine(coro):
+        if coro and (asyncio.iscoroutine(coro) or asyncio.isfuture(coro)):
             return await coro
         return coro
 

tests/test_unit_eval.py

@@ -1472,6 +1472,16 @@ async def handler():
 """,
         [20, 20],
     ],
+    [
+        """
+import asyncio
+
+future = asyncio.Future()
+future.set_result(True)
+await future
+""",
+        True,
+    ],
 ]