ENH: Patch pooch and MNEBrowseFigure for JupyterLite compatibility

Author: natinew77-creator

doc/conf.py

@@ -480,22 +480,63 @@
     "first_notebook_cell": (
         "# 💡 This cell is automatically added to the start of each notebook.\n"
         "import micropip\n"
-        "await micropip.install(['mne', 'pyodide-http'])\n"
+        "await micropip.install(['mne'])\n"
+        "\n"
+        "# Mock lzma since it is missing in Pyodide but required by pooch\n"
+        "import sys\n"
+        "import types\n"
+        "if 'lzma' not in sys.modules:\n"
+        "    sys.modules['lzma'] = types.ModuleType('lzma')\n"
         "\n"
         "# 1. Patch networking so pooch can download datasets\n"
-        "import pyodide_http\n"
-        "pyodide_http.patch_all()\n"
+        "import pooch\n"
+        "import requests\n"
+        "import urllib.request\n"
+        "import urllib.error\n"
+        "import io\n"
+        "\n"
+        "orig_send = requests.Session.send\n"
+        "def pyodide_send(self, request, **kwargs):\n"
+        "    try:\n"
+        "        resp = urllib.request.urlopen(request.url)\n"
+        "        content = resp.read()\n"
+        "        status = resp.status\n"
+        "    except urllib.error.HTTPError as e:\n"
+        "        content = e.read()\n"
+        "        status = e.code\n"
+        "    except Exception:\n"
+        "        return orig_send(self, request, **kwargs)\n"
+        "    response = requests.Response()\n"
+        "    response.status_code = status\n"
+        "    response.url = request.url\n"
+        "    response.raw = io.BytesIO(content)\n"
+        "    return response\n"
+        "requests.Session.send = pyodide_send\n"
+        "\n"
+        "# Intercept pooch to provide helpful errors for large OSF datasets\n"
+        "orig_pooch_fetch = pooch.Pooch.fetch\n"
+        "def pyodide_pooch_fetch(self, fname, processor=None, downloader=None):\n"
+        "    url = self.get_url(fname)\n"
+        "    if 'osf.io' in url or 'files.osf.io' in url:\n"
+        "        raise RuntimeError(\n"
+        "            f'Cannot download {fname} natively in JupyterLite due to browser CORS restrictions '\n"
+        "            f'on OSF and memory limits. Please download the dataset locally and upload it '\n"
+        "            f'into the JupyterLite file browser.'\n"
+        "        )\n"
+        "    return orig_pooch_fetch(self, fname, processor=processor, downloader=downloader)\n"
+        "pooch.Pooch.fetch = pyodide_pooch_fetch\n"
         "\n"
         "# 2. Patch MNEBrowseFigure to auto-display in Pyodide's inline backend\n"
         "import mne\n"
-        "import mne.viz.utils\n"
         "import matplotlib.pyplot as plt\n"
-        "orig_plt_show = mne.viz.utils.plt_show\n"
-        "def pyodide_plt_show(*args, **kwargs):\n"
-        "    orig_plt_show(*args, **kwargs)\n"
+        "import importlib\n"
+        "viz_utils = importlib.import_module('mne.viz.utils')\n"
+        "orig_plt_show = viz_utils.plt_show\n"
+        "def pyodide_plt_show(fig=None, **kwargs):\n"
+        "    orig_plt_show(fig, **kwargs)\n"
         "    import IPython.display\n"
         "    IPython.display.display(plt.gcf())\n"
-        "mne.viz.utils.plt_show = pyodide_plt_show\n"
+        "viz_utils.plt_show = pyodide_plt_show\n"
     ),
     "doc_module": ("mne",),
     "reference_url": dict(mne=None),