execnb 0.1.8__tar.gz → 0.1.9__tar.gz

{execnb-0.1.8/execnb.egg-info → execnb-0.1.9}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.1
 Name: execnb
-Version: 0.1.8
+Version: 0.1.9
 Summary: A description of your project
-Home-page: https://github.com/fastai/execnb/
+Home-page: https://github.com/AnswerDotAI/execnb/
 Author: Jeremy Howard
 Author-email: j@fast.ai
 License: Apache Software License 2.0

execnb-0.1.9/execnb/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.9"

{execnb-0.1.8 → execnb-0.1.9}/execnb/_modidx.py

@@ -22,6 +22,7 @@ d = { 'settings': { 'branch': 'master',
                              'execnb.nbio.write_nb': ('nbio.html#write_nb', 'execnb/nbio.py')},
             'execnb.shell': { 'execnb.shell.CaptureShell': ('shell.html#captureshell', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.__init__': ('shell.html#captureshell.__init__', 'execnb/shell.py'),
+                              'execnb.shell.CaptureShell._run': ('shell.html#captureshell._run', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.cell': ('shell.html#captureshell.cell', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.complete': ('shell.html#captureshell.complete', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.enable_gui': ('shell.html#captureshell.enable_gui', 'execnb/shell.py'),
@@ -29,6 +30,7 @@ d = { 'settings': { 'branch': 'master',
                               'execnb.shell.CaptureShell.prettytb': ('shell.html#captureshell.prettytb', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.run': ('shell.html#captureshell.run', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.run_all': ('shell.html#captureshell.run_all', 'execnb/shell.py'),
+                              'execnb.shell.CaptureShell.run_async': ('shell.html#captureshell.run_async', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.run_cell': ('shell.html#captureshell.run_cell', 'execnb/shell.py'),
                               'execnb.shell.CaptureShell.set_path': ('shell.html#captureshell.set_path', 'execnb/shell.py'),
                               'execnb.shell.ExecutionInfo.__repr__': ('shell.html#executioninfo.__repr__', 'execnb/shell.py'),
@@ -48,6 +50,8 @@ d = { 'settings': { 'branch': 'master',
                               'execnb.shell._out_exc': ('shell.html#_out_exc', 'execnb/shell.py'),
                               'execnb.shell._out_nb': ('shell.html#_out_nb', 'execnb/shell.py'),
                               'execnb.shell._out_stream': ('shell.html#_out_stream', 'execnb/shell.py'),
+                              'execnb.shell._pre': ('shell.html#_pre', 'execnb/shell.py'),
+                              'execnb.shell._strip': ('shell.html#_strip', 'execnb/shell.py'),
                               'execnb.shell.exec_nb': ('shell.html#exec_nb', 'execnb/shell.py'),
                               'execnb.shell.find_output': ('shell.html#find_output', 'execnb/shell.py'),
                               'execnb.shell.format_exc': ('shell.html#format_exc', 'execnb/shell.py'),

{execnb-0.1.8 → execnb-0.1.9}/execnb/shell.py

@@ -53,7 +53,7 @@ def __repr__(self: ExecutionResult): return f'result: {self.result}; err: {self.
 class CaptureShell(InteractiveShell):
     displayhook_class = _CustDisplayHook
 
-    def __init__(self, path:str|Path=None, mpl_format='retina', history=False, timeout=None):
+    def __init__(self, path:str|Path=None, mpl_format='retina', history=False, timeout:Optional[int]=None):
         super().__init__()
         self.history_manager.enabled = history
         self.timeout = timeout
@@ -63,21 +63,15 @@ class CaptureShell(InteractiveShell):
         if not IN_NOTEBOOK: InteractiveShell._instance = self
         if set_matplotlib_formats:
             self.enable_matplotlib("inline")
-            self.run_cell("from matplotlib_inline.backend_inline import set_matplotlib_formats")
-            self.run_cell(f"set_matplotlib_formats('{mpl_format}')")
+            self._run("from matplotlib_inline.backend_inline import set_matplotlib_formats")
+            self._run(f"set_matplotlib_formats('{mpl_format}')")
 
-    def run_cell(self, raw_cell, store_history=False, silent=False, shell_futures=True, cell_id=None,
-                 stdout=True, stderr=True, display=True, timeout=None):
-        if not timeout: timeout = self.timeout
+    def _run(self, raw_cell, store_history=False, silent=False, shell_futures=True, cell_id=None,
+                 stdout=True, stderr=True, display=True):
         # TODO what if there's a comment?
         semic = raw_cell.rstrip().endswith(';')
-        if timeout:
-            def handler(*args): raise TimeoutError()
-            signal.signal(signal.SIGALRM, handler)
-            signal.alarm(timeout)
         with capture_output(display=display, stdout=stdout, stderr=stderr) as c:
             result = super().run_cell(raw_cell, store_history, silent, shell_futures=shell_futures, cell_id=cell_id)
-        if timeout: signal.alarm(0)
         return AttrDict(result=result, stdout='' if semic else c.stdout, stderr=c.stderr,
                         display_objects=c.outputs, exception=result.error_in_exec, quiet=semic)
     
@@ -85,10 +79,24 @@ class CaptureShell(InteractiveShell):
         "Add `path` to python path, or `path.parent` if it's a file"
         path = Path(path)
         if path.is_file(): path = path.parent
-        self.run_cell(f"import sys; sys.path.insert(0, '{path.as_posix()}')")
+        self._run(f"import sys; sys.path.insert(0, '{path.as_posix()}')")
     
     def enable_gui(self, gui=None): pass
 
+# %% ../nbs/02_shell.ipynb
+@patch
+def run_cell(self:CaptureShell, raw_cell, store_history=False, silent=False, shell_futures=True, cell_id=None,
+             stdout=True, stderr=True, display=True, timeout=None):
+    if not timeout: timeout = self.timeout
+    if timeout:
+        def handler(*args): raise TimeoutError()
+        signal.signal(signal.SIGALRM, handler)
+        signal.alarm(timeout)
+    try: return self._run(raw_cell, store_history, silent, shell_futures, cell_id=cell_id,
+                          stdout=stdout, stderr=stderr, display=display)
+    finally:
+        if timeout: signal.alarm(0)
+
 # %% ../nbs/02_shell.ipynb
 def format_exc(e):
     "Format exception `e` as a string"
@@ -122,6 +130,8 @@ def _out_nb(o, fmt):
     for x in o.display_objects: res.append(_mk_out(x.data, x.metadata))
     if r is not None and not o.quiet:
         res.append(_mk_out(*fmt.format(r), 'execute_result'))
+    if 'execution_count' not in o: o['execution_count']=None
+    for p in res: p['execution_count'] = o['execution_count']
     return res
 
 # %% ../nbs/02_shell.ipynb
@@ -129,33 +139,52 @@ def _out_nb(o, fmt):
 def run(self:CaptureShell,
         code:str, # Python/IPython code to run
         stdout=True, # Capture stdout and save as output?
-        stderr=True): # Capture stderr and save as output?
+        stderr=True, # Capture stderr and save as output?
+       timeout:Optional[int]=None): # Shell command will time out after {timeout} seconds
     "Run `code`, returning a list of all outputs in Jupyter notebook format"
-    res = self.run_cell(code, stdout=stdout, stderr=stderr)
+    res = self.run_cell(code, stdout=stdout, stderr=stderr, timeout=timeout)
     self.result = res.result.result
     self.exc = res.exception
     return _out_nb(res, self.display_formatter)
 
 # %% ../nbs/02_shell.ipynb
-def render_outputs(outputs, ansi_renderer=strip_ansi):
-    try: import mistletoe
+@patch
+async def run_async(self:CaptureShell,
+        code: str,  # Python/IPython code to run
+        stdout=True,  # Capture stdout and save as output?
+        stderr=True,  # Capture stderr and save as output?
+        timeout:Optional[int]=None): # Shell command will time out after {timeout} seconds
+    return self.run(code, stdout=stdout, stderr=stderr, timeout=timeout)
+
+# %% ../nbs/02_shell.ipynb
+def _pre(s, xtra=''): return f"<pre {xtra}><code>{escape(s)}</code></pre>"
+def _strip(s): return strip_ansi(escape(s))
+
+def render_outputs(outputs, ansi_renderer=_strip, include_imgs=True, pygments=False):
+    try:
+        from mistletoe import markdown, HTMLRenderer
+        from mistletoe.contrib.pygments_renderer import PygmentsRenderer
     except ImportError: return print('mistletoe not found -- please install it')
+    renderer = PygmentsRenderer if pygments else HTMLRenderer
     def render_output(out):
         otype = out['output_type']
         if otype == 'stream':
             txt = ansi_renderer(''.join(out['text']))
-            return f"<pre>{txt}</pre>" if out['name']=='stdout' else f"<pre class='stderr'>{txt}</pre>"
+            xtra = '' if out['name']=='stdout' else "class='stderr'"
+            return f"<pre {xtra}><code>{txt}</code></pre>"
         elif otype in ('display_data','execute_result'):
             data = out['data']
             _g = lambda t: ''.join(data[t]) if t in data else None
             if d := _g('text/html'): return d
             if d := _g('application/javascript'): return f'<script>{d}</script>'
-            if d := _g('text/markdown'): return mistletoe.markdown(d)
-            if d := _g('image/svg+xml'): return d
-            if d := _g('image/jpeg'): return f'<img src="data:image/jpeg;base64,{d}"/>'
-            if d := _g('image/png'): return f'<img src="data:image/png;base64,{d}"/>'
+            if d := _g('text/markdown'): return markdown(d, renderer=renderer)
             if d := _g('text/latex'): return f'<div class="math">${d}$</div>'
-            if d := _g('text/plain'): return f"<pre>{escape(d)}</pre>"
+            if include_imgs:
+                if d := _g('image/jpeg'): return f'<img src="data:image/jpeg;base64,{d}"/>'
+                if d := _g('image/png'): return f'<img src="data:image/png;base64,{d}"/>'
+            if d := _g('text/plain'): return _pre(d)
+            if d := _g('image/svg+xml'): return d
+            
         return ''
     
     return '\n'.join(map(render_output, outputs))
@@ -167,10 +196,7 @@ def cell(self:CaptureShell, cell, stdout=True, stderr=True):
     if cell.cell_type!='code': return
     self._cell_idx = cell.idx_ + 1
     outs = self.run(cell.source)
-    if outs:
-        cell.outputs = _dict2obj(outs)
-        for o in outs:
-            if 'execution_count' in o: cell['execution_count'] = o['execution_count']
+    if outs: cell.outputs = _dict2obj(outs)
 
 # %% ../nbs/02_shell.ipynb
 def find_output(outp, # Output from `run`

{execnb-0.1.8 → execnb-0.1.9/execnb.egg-info}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.1
 Name: execnb
-Version: 0.1.8
+Version: 0.1.9
 Summary: A description of your project
-Home-page: https://github.com/fastai/execnb/
+Home-page: https://github.com/AnswerDotAI/execnb/
 Author: Jeremy Howard
 Author-email: j@fast.ai
 License: Apache Software License 2.0

{execnb-0.1.8 → execnb-0.1.9}/settings.ini

@@ -5,11 +5,11 @@ repo = execnb
 description = A description of your project
 copyright = Put your copyright info here
 keywords = some keywords
-user = fastai
+user = AnswerDotAI
 author = Jeremy Howard
 author_email = j@fast.ai
 branch = master
-version = 0.1.8
+version = 0.1.9
 min_python = 3.7
 requirements = fastcore>=1.5.5 ipython
 dev_requirements = matplotlib Pillow mistletoe
@@ -23,9 +23,9 @@ nbs_path = nbs
 doc_path = _docs
 recursive = False
 tst_flags = notest
-doc_host = https://fastai.github.io
+doc_host = https://AnswerDotAI.github.io
 doc_baseurl = /execnb/
-git_url = https://github.com/fastai/execnb/
+git_url = https://github.com/AnswerDotAI/execnb/
 lib_path = execnb
 title = execnb
 black_formatting = False

execnb-0.1.8/execnb/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.8"