add to pythonpath

Author: jph00

execnb/_modidx.py

@@ -21,12 +21,12 @@
                 'min_python': '3.7',
                 'nbs_path': 'nbs',
                 'recursive': 'False',
-                'requirements': 'fastcore>=1.3.27',
+                'requirements': 'fastcore>=1.3.27 ipython',
                 'status': '2',
                 'title': 'execnb',
                 'tst_flags': 'notest',
                 'user': 'fastai',
-                'version': '0.0.2'},
+                'version': '0.0.3'},
   'syms': { 'execnb.fastshell': { 'execnb.fastshell.FastInteractiveShell': 'https://fastai.github.io/execnb/fastshell#FastInteractiveShell',
                                   'execnb.fastshell.FastInteractiveShell.ask_exit': 'https://fastai.github.io/execnb/fastshell#FastInteractiveShell.ask_exit',
                                   'execnb.fastshell.FastInteractiveShell.data_pub': 'https://fastai.github.io/execnb/fastshell#FastInteractiveShell.data_pub',
@@ -57,4 +57,5 @@
                               'execnb.shell.CaptureShell.execute': 'https://fastai.github.io/execnb/shell#CaptureShell.execute',
                               'execnb.shell.CaptureShell.run': 'https://fastai.github.io/execnb/shell#CaptureShell.run',
                               'execnb.shell.CaptureShell.run_all': 'https://fastai.github.io/execnb/shell#CaptureShell.run_all',
+                              'execnb.shell.CaptureShell.set_path': 'https://fastai.github.io/execnb/shell#CaptureShell.set_path',
                               'execnb.shell.exec_nb': 'https://fastai.github.io/execnb/shell#exec_nb'}}}

execnb/shell.py

@@ -47,7 +47,13 @@ def __init__(self):
         self.exc,self.result,self.out,self.count = None,None,[],1
         self.run_cell('%matplotlib inline')
 
-    def enable_gui(self, gui=None): pass
+    def set_path(self, path):
+        "Add `path` to python path"
+        self.run_cell(f"import sys; sys.path.insert(0, {str(path)})")
+
+    def enable_gui(self, gui=None):
+        "Disable GUI (over-ridden; called by IPython)"
+        pass
 
     def _showtraceback(self, etype, evalue, stb: str):
         self.out.append(_out_exc(etype, evalue, stb))
@@ -116,7 +122,7 @@ def run_all(self:CaptureShell,
             postproc(cell)
         if self.exc and exc_stop: raise self.exc[1] from None
 
-# %% ../nbs/02_shell.ipynb 37
+# %% ../nbs/02_shell.ipynb 38
 @patch
 def execute(self:CaptureShell,
             src:str|Path, # Notebook path to read from
@@ -130,12 +136,13 @@ def execute(self:CaptureShell,
 ):
     "Execute notebook from `src` and save with outputs to `dest"
     nb = read_nb(src)
+    self.set_path(Path(src).parent.resolve())
     if inject_path is not None: inject_code = Path(inject_path).read_text()
     self.run_all(nb, exc_stop=exc_stop, preproc=preproc, postproc=postproc,
                  inject_code=inject_code, inject_idx=inject_idx)
     if dest: write_nb(nb, dest)
 
-# %% ../nbs/02_shell.ipynb 47
+# %% ../nbs/02_shell.ipynb 49
 @call_parse
 def exec_nb(
     src:str, # Notebook path to read from

nbs/02_shell.ipynb

@@ -97,7 +97,13 @@
     "        self.exc,self.result,self.out,self.count = None,None,[],1\n",
     "        self.run_cell('%matplotlib inline')\n",
     "\n",
-    "    def enable_gui(self, gui=None): pass\n",
+    "    def set_path(self, path):\n",
+    "        \"Add `path` to python path\"\n",
+    "        self.run_cell(f\"import sys; sys.path.insert(0, {str(path)})\")\n",
+    "\n",
+    "    def enable_gui(self, gui=None):\n",
+    "        \"Disable GUI (over-ridden; called by IPython)\"\n",
+    "        pass\n",
     "\n",
     "    def _showtraceback(self, etype, evalue, stb: str):\n",
     "        self.out.append(_out_exc(etype, evalue, stb))\n",
@@ -200,8 +206,8 @@
        "  'execution_count': 1},\n",
        " {'name': 'stdout',\n",
        "  'output_type': 'stream',\n",
-       "  'text': ['CPU times: user 2 us, sys: 0 ns, total: 2 us',\n",
-       "   'Wall time: 3.81 us']}]"
+       "  'text': ['CPU times: user 3 us, sys: 1 us, total: 4 us',\n",
+       "   'Wall time: 5.25 us']}]"
       ]
      },
      "execution_count": null,
@@ -531,6 +537,15 @@
     "except Exception as e: print(f\"got exception: {e}\")"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "nb = read_nb('../../testw/index.ipynb')"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -626,12 +641,22 @@
     "):\n",
     "    \"Execute notebook from `src` and save with outputs to `dest\"\n",
     "    nb = read_nb(src)\n",
+    "    self.set_path(Path(src).parent.resolve())\n",
     "    if inject_path is not None: inject_code = Path(inject_path).read_text()\n",
     "    self.run_all(nb, exc_stop=exc_stop, preproc=preproc, postproc=postproc,\n",
     "                 inject_code=inject_code, inject_idx=inject_idx)\n",
     "    if dest: write_nb(nb, dest)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "s.execute(clean, '../../testw/index.ipynb')"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},