U lu f@sUdZddlmZddlmZddlmZGdddeZGddde eZ Gd d d e Z Gd d d eZ Gd ddeZdZded<dZded<dZded<dS)zExceptions for nbclient.) annotations)Any) NotebookNodec@seZdZdZdS)CellControlSignalz 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). N__name__ __module__ __qualname____doc__r r 7/tmp/pip-unpacked-wheel-72rqhnyi/nbclient/exceptions.pyr src@s(eZdZdZedddddddZdS) CellTimeoutErrorzS A custom exception to capture when a cell has timed out during execution. strintr)msgtimeoutcellreturncCsd|rL|jrL|jd}t|dkr,|jn|ddd|dd}nd}|tj|||dS) z)Create an error from a timeout on a cell. Nz ... zCell contents not found.)rr cell_contents)sourcestripsplitlentimeout_err_msgformat)clsrrrZ src_by_linessrcr r r error_from_timeout_and_cells  z,CellTimeoutError.error_from_timeout_and_cellN)rrr r classmethodr!r r r r r sr c@seZdZdZdS)DeadKernelErrorzA dead kernel error.Nrr r r r r#)sr#c@seZdZdZdS)CellExecutionCompletez 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. Nrr r r r r$/sr$cs^eZdZdZdddddfdd Zddd d Zddd d Zed dddddZZ S)CellExecutionErrorz 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. rNone) tracebackenameevaluercs"t|||_||_||_dS)zInitialize the error.N)super__init__r'r(r))selfr'r(r) __class__r r r+Bs zCellExecutionError.__init__z tuple[Any])rcCst||j|j|jffS)zReduce implementation.)typer'r(r)r,r r r __reduce__IszCellExecutionError.__reduce__cCs"|jr |jS|jd|jSdS)z Str repr.z: N)r'r(r)r0r r r __str__MszCellExecutionError.__str__rzdict[str, Any])rrrcCsg}|jD]0}|ddkr |tj|d|ddq |rV|dd|dd |}d |d gprg}|tj|||d |d d |dddS)zvInstantiate from a code cell object and a message contents (message is either execute_reply or error) output_typestreamnametext)r5r6rz------------------rr')r stream_outputr'r(zr))r(r)) outputsappendstream_output_msgrrstripinsertjoinget exec_err_msg)rrrZstream_outputsoutputr8tbr r r from_cell_and_msgTs(       z$CellExecutionError.from_cell_and_msg) rrr r r+r1r2r"rC __classcell__r r r-r r%:s r%z----- {name} ----- {text}rr;zAn error occurred while executing the following cell: ------------------ {cell.source} ------------------ {stream_output} {traceback} r@zA 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} ------------------- rN)r __future__rtypingrZnbformatr Exceptionr TimeoutErrorr RuntimeErrorr#r$r%r;__annotations__r@rr r r r s     :