You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.1 KiB
115 lines
3.1 KiB
from typing import Dict
|
|
|
|
from nbformat import NotebookNode
|
|
|
|
|
|
class CellControlSignal(Exception):
|
|
"""
|
|
A custom exception used to indicate that the exception is used for cell
|
|
control actions (not the best model, but it's needed to cover existing
|
|
behavior without major refactors).
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
class CellTimeoutError(TimeoutError, CellControlSignal):
|
|
"""
|
|
A custom exception to capture when a cell has timed out during execution.
|
|
"""
|
|
|
|
@classmethod
|
|
def error_from_timeout_and_cell(
|
|
cls, msg: str, timeout: int, cell: NotebookNode
|
|
) -> "CellTimeoutError":
|
|
if cell and cell.source:
|
|
src_by_lines = cell.source.strip().split("\n")
|
|
src = (
|
|
cell.source
|
|
if len(src_by_lines) < 11
|
|
else f"{src_by_lines[:5]}\n...\n{src_by_lines[-5:]}"
|
|
)
|
|
else:
|
|
src = "Cell contents not found."
|
|
return cls(timeout_err_msg.format(timeout=timeout, msg=msg, cell_contents=src))
|
|
|
|
|
|
class DeadKernelError(RuntimeError):
|
|
pass
|
|
|
|
|
|
class CellExecutionComplete(CellControlSignal):
|
|
"""
|
|
Used as a control signal for cell execution across execute_cell and
|
|
process_message function calls. Raised when all execution requests
|
|
are completed and no further messages are expected from the kernel
|
|
over zeromq channels.
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
class CellExecutionError(CellControlSignal):
|
|
"""
|
|
Custom exception to propagate exceptions that are raised during
|
|
notebook execution to the caller. This is mostly useful when
|
|
using nbconvert as a library, since it allows to deal with
|
|
failures gracefully.
|
|
"""
|
|
|
|
def __init__(self, traceback: str, ename: str, evalue: str) -> None:
|
|
super().__init__(traceback)
|
|
self.traceback = traceback
|
|
self.ename = ename
|
|
self.evalue = evalue
|
|
|
|
def __reduce__(self) -> tuple:
|
|
return type(self), (self.traceback, self.ename, self.evalue)
|
|
|
|
def __str__(self) -> str:
|
|
s = self.__unicode__()
|
|
if not isinstance(s, str):
|
|
s = s.encode('utf8', 'replace')
|
|
return s
|
|
|
|
def __unicode__(self) -> str:
|
|
return self.traceback
|
|
|
|
@classmethod
|
|
def from_cell_and_msg(cls, cell: NotebookNode, msg: Dict) -> "CellExecutionError":
|
|
"""Instantiate from a code cell object and a message contents
|
|
(message is either execute_reply or error)
|
|
"""
|
|
tb = '\n'.join(msg.get('traceback', []) or [])
|
|
return cls(
|
|
exec_err_msg.format(
|
|
cell=cell,
|
|
traceback=tb,
|
|
ename=msg.get('ename', '<Error>'),
|
|
evalue=msg.get('evalue', ''),
|
|
),
|
|
ename=msg.get('ename', '<Error>'),
|
|
evalue=msg.get('evalue', ''),
|
|
)
|
|
|
|
|
|
exec_err_msg: str = """\
|
|
An error occurred while executing the following cell:
|
|
------------------
|
|
{cell.source}
|
|
------------------
|
|
|
|
{traceback}
|
|
{ename}: {evalue}
|
|
"""
|
|
|
|
|
|
timeout_err_msg: str = """\
|
|
A cell timed out while it was being executed, after {timeout} seconds.
|
|
The message was: {msg}.
|
|
Here is a preview of the cell contents:
|
|
-------------------
|
|
{cell_contents}
|
|
-------------------
|
|
"""
|