ipython-smart-await

Auto-await coroutine results in the IPython REPL — drive a pure-async API without typing await everywhere.

When loaded, the extension rewrites each input cell so that expressions which evaluate to a coroutine are awaited automatically on IPython's loop runner.

Install

pip install ipython-smart-await

Usage

%load_ext smart_await

Then, given some async API bound to p:

p.get_pid()          # -> awaited automatically (no `await` needed)
a[0]                 # -> awaited if `a.__getitem__` returns a coroutine
a[0] = 7             # -> routed to `a.setindex(0, 7)` (awaited) when `a` has an async `setindex`
proxy.length         # -> awaited if attribute access returns a coroutine (dynamic proxies)

Opting out

Wrap a call in ...( ) to get the raw, un-awaited value:

coro = ...(p.get_pid())   # `coro` is the coroutine itself

Cells that already use await / async constructs are left untouched.

How it works

The extension installs an ast transformer (an IPython AST transformer) that wraps:

• calls (foo()), subscript reads (a[0]), and attribute reads (a.b) in a helper that awaits the result only if it is a coroutine (non-coroutines pass through unchanged);
• single-target subscript assignment (a[0] = v) into a call that routes to an async setindex(key, value) when the target exposes one, otherwise a normal item assignment — so dicts, lists, numpy arrays, etc. are unaffected.

License

MIT