ipython-smart-await 0.1.0__tar.gz → 0.2.0__tar.gz

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/.github/workflows/test.yml

@@ -24,7 +24,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ipython-smart-await
-Version: 0.1.0
+Version: 0.2.0
 Summary: Auto-await coroutine results (calls, subscripts, item-assignment) in the IPython REPL.
 Project-URL: Homepage, https://github.com/doronz88/ipython-smart-await
 Project-URL: Repository, https://github.com/doronz88/ipython-smart-await
@@ -51,6 +51,7 @@ 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
@@ -67,8 +68,8 @@ Cells that already use `await` / `async` constructs are left untouched.
 
 The extension installs an `ast` transformer (an IPython AST transformer) that wraps:
 
-- **calls** (`foo()`) and **subscript reads** (`a[0]`) in a helper that awaits the result only if
-  it is a coroutine (non-coroutines pass through unchanged);
+- **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.

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/README.md

@@ -24,6 +24,7 @@ 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
@@ -40,8 +41,8 @@ Cells that already use `await` / `async` constructs are left untouched.
 
 The extension installs an `ast` transformer (an IPython AST transformer) that wraps:
 
-- **calls** (`foo()`) and **subscript reads** (`a[0]`) in a helper that awaits the result only if
-  it is a coroutine (non-coroutines pass through unchanged);
+- **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.

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ipython-smart-await"
-version = "0.1.0"
+version = "0.2.0"
 description = "Auto-await coroutine results (calls, subscripts, item-assignment) in the IPython REPL."
 readme = "README.md"
 requires-python = ">=3.10"

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/smart_await/__init__.py

@@ -24,7 +24,7 @@ from typing import Any
 from IPython.core.interactiveshell import InteractiveShell
 
 
-__version__ = "0.1.0"
+__version__ = "0.2.0"
 __all__ = ["load_ipython_extension"]
 
 
@@ -80,6 +80,15 @@ class _SmartAwaitTransformer(ast.NodeTransformer):
             return visited_node
         return ast.Call(func=ast.Name(_maybe_await.__name__, ctx=ast.Load()), args=[visited_node], keywords=[])
 
+    def visit_Attribute(self, node: ast.Attribute) -> ast.AST:
+        visited_node = self.generic_visit(node)
+        assert isinstance(visited_node, ast.expr)
+        # Only reads yield a (possibly awaitable) value; never wrap an assignment/deletion target.
+        # Supports objects that dispatch attribute access to a coroutine (e.g. dynamic proxies).
+        if not isinstance(node.ctx, ast.Load):
+            return visited_node
+        return ast.Call(func=ast.Name(_maybe_await.__name__, ctx=ast.Load()), args=[visited_node], keywords=[])
+
     def visit_Assign(self, node: ast.Assign) -> ast.AST:
         visited_node = self.generic_visit(node)
         assert isinstance(visited_node, ast.Assign)

{ipython_smart_await-0.1.0 → ipython_smart_await-0.2.0}/tests/test_smart_await.py

@@ -53,6 +53,10 @@ def test_subscript_read_is_wrapped(ip: InteractiveShell) -> None:
     assert _rewrite(ip, "a[0]") == "_maybe_await(a[0])"
 
 
+def test_attribute_read_is_wrapped(ip: InteractiveShell) -> None:
+    assert _rewrite(ip, "a.length") == "_maybe_await(a.length)"
+
+
 def test_subscript_assign_is_rewritten(ip: InteractiveShell) -> None:
     assert _rewrite(ip, "a[0] = 7") == "_setitem(a, 0, 7)"
 
@@ -91,6 +95,18 @@ def test_subscript_assign_routes_to_setindex(ip: InteractiveShell) -> None:
     assert a.store == {0: 7}
 
 
+def test_attribute_auto_awaited(ip: InteractiveShell) -> None:
+    class Proxy:
+        def __getattr__(self, name):
+            async def _resolve():
+                return f"resolved:{name}"
+
+            return _resolve()
+
+    ip.user_ns["a"] = Proxy()
+    assert ip.run_cell("a.length").result == "resolved:length"
+
+
 def test_dict_assign_unaffected(ip: InteractiveShell) -> None:
     ip.user_ns["d"] = {}
     assert ip.run_cell("d['k'] = 5").success