U lu f}@sUdZddlmZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z ddl mZmZmZmZmZddlmZdd lmZd d lmZd d dddZded<dddiidfdddiidfdZded<Gddde ZejZdS)z nbclient cli.) annotationsN)Path)dedent) JupyterApp)BoolIntegerListUnicodedefault)catch_config_error) __version__)NotebookClientzNbClientApp.timeoutzNbClientApp.startup_timeoutzNbClientApp.kernel_namezNbClientApp.output_base)timeoutstartup_timeout kernel_nameoutputzdict[str, str]nbclient_aliases NbClientApp allow_errorsTzLErrors are ignored and execution is continued until the end of the notebook.inplacez/Overwrite input notebook with executed results.)z allow-errorsrzdict[str, typing.Any]nbclient_flagscs<eZdZdZeeZdZeZ e Z dZ e eddjddZedded d jddZed ed djddZed eddjddZededdjddZededdjddZed eddjddZeddedd jddZedddddZed&dddfdd Zddd d!Zd"dd#d$d%ZZ S)'rzE An application used to execute notebook files (``*.ipynb``) zjupyter-executez7An application used to execute notebook files (*.ipynb)zPath of notebooks to convert)helpT)configNz The time to wait (in seconds) for output from executions. If a cell execution takes longer, a TimeoutError is raised. ``-1`` will disable the timeout. ) allow_noner<z The time to wait (in seconds) for the kernel to start. If kernel startup takes longer, a RuntimeError is raised. FaB When a cell raises an error the default behavior is that execution is stopped and a :py:class:`nbclient.exceptions.CellExecutionError` is raised. If this flag is provided, errors are ignored and execution is continued until the end of the notebook. zskip-executionz^ Name of the cell tag to use to denote a cell that should be skipped. z Name of kernel to use to execute the cells. If not set, use the kernel_spec embedded in the notebook. z Default is execute notebook without writing the newly executed notebook. If this flag is provided, the newly generated notebook will overwrite the input notebook. a8 Write executed notebook to this file base name. Supports pattern replacements ``'{notebook_name}'``, the name of the input notebook file without extension. Note that output is always relative to the parent directory of the input notebook. Z log_levelint)returncCstjS)N)loggingINFO)selfr"0/tmp/pip-unpacked-wheel-72rqhnyi/nbclient/cli.py_log_level_defaultszNbClientApp._log_level_defaultzlist[str] | NoneNone)argvrcsnt|||_|js&tdt|jdkrT|jdk rTd|jkrTd}t||jD]}| |qZdS)zInitialize the app.r Nz{notebook_name}zgIf passing multiple notebooks with `--output=output` option, output string must contain {notebook_name}) super initialize get_notebooks notebookssysexitlen output_base ValueError run_notebook)r!r&msgpath __class__r"r#r)s     zNbClientApp.initializez list[str]cCs|jr|j}n|j}|S)zGet the notebooks for the app.) extra_argsr+)r!r+r"r"r#r*szNbClientApp.get_notebooksstr) notebook_pathrc Cs|jd|t|d}|j}|jr6|}n0|jrb|j|jj |dj dd}nd}|r|j sd|jd}t || }tj|dd }W5QRXt||j|j|j|j|jd d |iid }||r|jd |t||dS)zRun a notebook by path.z Executing z.ipynbr)Z notebook_nameNzCannot write to directory=z that does not exist)Z as_versionmetadatar3)rrskip_cells_with_tagrr resourceszSave executed results to )loginfor with_suffixparentabsoluterr/joinpathformatnameis_dirr0opennbformatreadrrrr;rrexecutewrite) r!r8Z input_pathr3Z output_pathr2fnbclientr"r"r#r1s<    zNbClientApp.run_notebook)N)!__name__ __module__ __qualname____doc__r r versionrDraliasesrflags descriptionrtagr+rrrrrrr;rrr/r r$r r)r*r1 __classcell__r"r"r4r#r0s           )rQ __future__rrr,typingpathlibrtextwraprrGZjupyter_core.applicationrZ traitletsrrrr r Ztraitlets.configr Znbclientr rMrr__annotations__rrZlaunch_instancemainr"r"r"r#sF         0