o tf@sUdZddlmZddlmZddlmZGdddeZGddde eZ Gd d d e Z Gd d d eZ Gd ddeZdZded<dZded<dZded<dS)zExceptions for nbclient.) annotations)Any) NotebookNodec@eZdZdZdS)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 \/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/nbclient/exceptions.pyr src@seZdZdZed d d Zd S) CellTimeoutErrorzS A custom exception to capture when a cell has timed out during execution. msgstrtimeoutintcellrreturncCsd|r&|jr&|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)rrrrrrrr)rr r r classmethodr"r r r r rsrc@r)DeadKernelErrorzA dead kernel error.Nrr r r r r$)sr$c@r)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%csDeZdZdZdfdd Zdd d Zdd dZedddZZ 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. tracebackrenameevaluerNonecs"t|||_||_||_dS)zInitialize the error.N)super__init__r'r(r))selfr'r(r) __class__r r r,Bs  zCellExecutionError.__init__ tuple[Any]cCst||j|j|jffS)zReduce implementation.)typer'r(r)r-r r r __reduce__IszCellExecutionError.__reduce__cCs|jr|jS|jd|jS)z Str repr.z: )r'r(r)r2r r r __str__MszCellExecutionError.__str__rrrdict[str, Any]cCsg}|jD]}|ddkr|tj|d|ddq|r+|dd|dd |}d |d gp9g}|tj|||d |d d |dddS)zvInstantiate from a code cell object and a message contents (message is either execute_reply or error) output_typestreamnametext)r8r9rz------------------rr')r stream_outputr'r(zr))r(r)) Zoutputsappendstream_output_msgrrstripinsertjoinget exec_err_msg)r rrZstream_outputsoutputr;tbr r r from_cell_and_msgTs*       z$CellExecutionError.from_cell_and_msg)r'rr(rr)rrr*)rr0)rr)rrrr5rr&) rr r r r,r3r4r#rE __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} rBzA 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__rBrr r r r s     :