feat: auto-installs also have toasts now

src/python/patches/imports.py

@@ -0,0 +1,19 @@
+from functools import cache
+from sys import modules
+
+from micropip._compat import REPODATA_PACKAGES
+from pyodide.code import find_imports
+
+
+@cache
+def build_reversed_index() -> dict[str, str]:
+    return {import_name: package_name for package_name, info in REPODATA_PACKAGES.items() for import_name in info["imports"]}
+
+
+def import_name_to_package_name(import_name: str):
+    if import_name not in modules:
+        return build_reversed_index().get(import_name)
+
+
+def find_packages_to_install(source: str):
+    return list(filter(None, map(import_name_to_package_name, find_imports(source))))

src/python/patches/patches.py

@@ -49,9 +49,20 @@ def patch_linecache():
 
 @cache
 def patch_console():
-    from pyodide.console import PyodideConsole
+    from pyodide.console import Console, PyodideConsole
 
-    PyodideConsole.runcode = with_lock(PyodideConsole.runcode)
+    @with_lock
+    @wraps(PyodideConsole.runcode)
+    async def runcode(self: PyodideConsole, source: str, code):
+        from .imports import find_packages_to_install
+
+        packages = find_packages_to_install(source)
+        if packages:
+            await micropip.install(packages)
+
+        return await Console.runcode(self, source, code)
+
+    PyodideConsole.runcode = runcode
 
 
 @cache

src/python/patches/pyodide_js.pyi

@@ -0,0 +1 @@
+async def loadPackagesFromImports(source: str): ...  # noqa: N802