await awaitable object

tinylambda · tinylambda · commit 69a50bdcd98f · 2022-08-24T18:12:33.000+08:00
diff --git a/module_asyncio/samples/await_awaitable.py b/module_asyncio/samples/await_awaitable.py
@@ -0,0 +1,27 @@
+import asyncio
+from typing import Awaitable, Generator
+
+
+class MyAwaitable(Awaitable):
+    def __init__(self):
+        print("init")
+
+    def __await__(self) -> Generator:
+        async def closure():
+            print("await")
+            return self
+
+        return closure().__await__()
+
+    async def method(self):
+        print("method")
+
+
+async def main():
+    m = MyAwaitable()
+    obj = await m
+    await obj.method()
+
+
+if __name__ == "__main__":
+    asyncio.get_event_loop().run_until_complete(main())
