feat: async console extras

Author: antazoey

src/ape_console/_cli.py

@@ -158,6 +158,10 @@ def _load_extras_file(self, extras_file: Path) -> dict:
             func_spec = inspect.getfullargspec(ape_init_extras)
             init_kwargs: dict[str, Any] = {k: self._get_from_ape(k) for k in func_spec.args}
             extras = ape_init_extras(**init_kwargs)
+            if inspect.iscoroutine(extras):
+                from ape.utils import run_until_complete
+
+                extras = run_until_complete(extras)
 
             if isinstance(extras, dict):
                 all_extras.update(extras)

tests/functional/test_console.py

@@ -85,6 +85,25 @@ def test_extras_load_using_ape_namespace(self, scope):
         extras_content = """
 def ape_init_extras(project):
     return {"foo": type(project)}
+"""
+        with create_tempdir() as temp:
+            extras_file = temp / CONSOLE_EXTRAS_FILENAME
+            extras_file.write_text(extras_content)
+            extras.__dict__[f"_{scope}_path"] = extras_file
+            extras.__dict__.pop(f"_{scope}_extras", None)
+            assert extras["foo"] is LocalProject
+
+    @pytest.mark.parametrize("scope", ("local", "global"))
+    def test_extras_async_ape_init_extras(self, scope):
+        """
+        ``ape_init_extras`` is allowed to be defined as an ``async`` function;
+        it gets awaited before its result is merged into the namespace.
+        """
+        extras = ApeConsoleNamespace()
+        _ = getattr(extras, f"_{scope}_path")
+        extras_content = """
+async def ape_init_extras(project):
+    return {"foo": type(project)}
 """
         with create_tempdir() as temp:
             extras_file = temp / CONSOLE_EXTRAS_FILENAME