U lu f/@sdZddlmZddlZddlZddlZddlZddlZ ddl m Z m Z m Z ddlmZmZmZmZmZddlmZddlmZdd lmZdd lmZGd d d e ZGd dde ZGddde ZGddde ZGddde Z GdddeZ!e"dkr e!#dS)zApps for managing kernel specs.) annotationsN) JupyterApp base_aliases base_flags)BoolDictInstanceListUnicode) Application) __version__)KernelSpecManager)KernelProvisionerFactoryc@sdeZdZdZeZdZeeZ e ddddZ dddiidfe dd Z d d d d Zdd ddZdS)ListKernelSpecszAn app to list kernel specs.z%List installed kernel specifications.Fz7output spec name and location as machine-readable json.T)helpconfig json_outputdebug)jsonrrreturncCst||jdS)N)parentdata_dirrrselfr@/tmp/pip-unpacked-wheel-naub1w99/jupyter_client/kernelspecapp.py_kernel_spec_manager_default)sz,ListKernelSpecs._kernel_spec_manager_defaultzdict[str, t.Any] | Nonecsj}j}js|s*tddStt|dddd}dddfd d }td t||dD]"\}}td ||d |qnntt j d|idd|S)Start the application.zNo kernels availableNcSst|SNlennamerrr5z'ListKernelSpecs.start..keyzt.Any)itemrcs:|d}tjD]\}}||r||fSqd|fS)z+sort key function for Jupyter path priorityr r*) enumerateZ jupyter_path startswith)r+pathidxprefixrrrpath_key7s  z'ListKernelSpecs.start..path_keyzAvailable kernels:  Z kernelspecs)indent) kernel_spec_managerfind_kernel_specsZ get_all_specsrprintr#sorteditemsljustrdumps)rpathsspecsname_lenr1Z kernelnamer.rrrstart,s   zListKernelSpecs.startN)__name__ __module__ __qualname____doc__r version descriptionrrr6rrrflagsrr@rrrrrs  rcseZdZdZeZdZdZdZe e Z ddddZ e Ze d d d d Zd dddZedd dd Ze d d dd Zedd dd ZdddZeeddd iidfddd iidfddejiidfeddZddd fd!d" Zddd#d$ZZS)%InstallKernelSpecz An app to install a kernel spec.a=Install a kernel specification directory. Given a SOURCE DIRECTORY containing a kernel spec, jupyter will copy that directory into one of the Jupyter kernel directories. The default is to install kernelspecs for all users. `--user` can be specified to install a kernel only for the current user. z> jupyter kernelspec install /path/to/my_kernel --user z1jupyter kernelspec install SOURCE_DIR [--options]rrcCs t|jdSN)rrrrrrrYsz.InstallKernelSpec._kernel_spec_manager_defaultTz&Install the kernel spec with this namerrstrcCstj|jSr!)osr.basename sourcedirrrrr_kernel_name_default_sz&InstallKernelSpec._kernel_name_defaultF Try to install the kernel spec to the per-user directory instead of the system or environment directory. zSpecify a prefix to install to, e.g. an env. The kernelspec will be installed in PREFIX/share/jupyter/kernels/ z0Replace any existing kernel spec with this name.zInstallKernelSpec.kernel_namezInstallKernelSpec.prefix)r%r0user'Install to the per-user kernel registryreplacer0zEInstall to Python's sys.prefix. Useful in conda/virtual environments.r)rRrTz sys-prefixrzNone | list[str]Noneargvrcs<t||jr |jd|_ntdtjd|ddS)Parse the command line args.rzNo source directory specified.filer N)superparse_command_line extra_argsrOr8sysstderrexitrrW __class__rrr\s  z$InstallKernelSpec.parse_command_linec Cs|jr|jr|dz$|jj|j|j|j|j|jdWntk r}zl|j t j krt |t j d|jszt dt j d|dn,|j t jkrt d|jt j d|dW5d}~XYnXdS)r zCCan't specify both user and prefix. Please choose one or the other.) kernel_namerRr0rTrY4Perhaps you want to install with `sudo` or `--user`?r z$A kernel spec is already present at N)rRr0r`r6Zinstall_kernel_specrOrdrTOSErrorerrnoEACCESr8r^r_EEXISTfilename)rerrrr@s(       zInstallKernelSpec.start)rArBrCrDr rErFexamplesusagerrr6rr rOrdrPrrRr0rTaliasesupdaterr^rrGr\r@ __classcell__rrrbrrHHsL     rHcseZdZdZeZdZdZeddddZ e e Z e eZdd d d Zd dd diie jfiZeejdddfdd Zdd ddZZS)RemoveKernelSpeczAn app to remove a kernel spec.z/Remove one or more Jupyter kernelspecs by name.z1jupyter kernelspec remove python2 [my_kernel ...]FTz-Force removal, don't prompt for confirmation.rKrrcCst|j|dS)N)rrrrrrrrsz-RemoveKernelSpec._kernel_spec_manager_defaultfforcezlist[str] | NonerUrVcs2t||jr$tt|j|_n |ddS)rXzNo kernelspec specified.N)r[r\r]r9set spec_namesr`rarbrrr\s z#RemoveKernelSpec.parse_command_linec Cs:d|j_|j}t|jt|}|r>|dd||js|j st d|jD]0}| ||}t d| dd| dqXt dt|j}|d sd S|jD]}z|j|}Wn\tk r$}z<|jtjkrt |tjd t d tjd |d nW5d }~XYnXt d|qd S)r Fz Couldn't find kernel spec(s): %sz, zKernel specs to remove:r2 zRemove %i kernel specs [y/N]: yNrYzPerhaps you want sudo?r zRemoved )r6Zensure_native_kernelr7rtru differencer`joinrsZ answer_yesr8getr;inputr#lowerr-Zremove_kernel_specrfrgrhr^r_)rZ spec_pathsmissingr%r.Zanswerrdrkrrrr@s.    "  zRemoveKernelSpec.start)rArBrCrDr rErFrlrrsr r rurrr6rrrGrorr\r@rprrrbrrqs   rqc@sdeZdZdZeZdZeeZ ddddZ e ddd d Z dd diid fe d dZddddZdS)InstallNativeKernelSpecz)An app to install the native kernel spec.zG[DEPRECATED] Install the IPython kernel spec directory for this Python.rrcCs t|jdSrIrrrrrrsz4InstallNativeKernelSpec._kernel_spec_manager_defaultFTrQrKrRrSr)rRrrUc Cs|jdzddlm}Wn,tk rHtdtjd|dYnXz|j |j |j dWndt k r}zF|j t jkrt|tjd|j stdtjd|d||W5d }~XYnXd S) r z`jupyter kernelspec install-self` is DEPRECATED as of 4.0. You probably want `ipython kernel install` to install the IPython kernelspec.r) kernelspecz0ipykernel not available, can't install its spec.rYr )rRreN)logwarningZ ipykernelrModuleNotFoundErrorr8r^r_r`installr6rRrfrgrh)rrrkrrrr@s(  zInstallNativeKernelSpec.startN)rArBrCrDr rErFrrr6rrrRrrGr@rrrrrs  rc@s&eZdZdZeZdZddddZdS)ListProvisionerszAn app to list provisioners.z=List available provisioners for use in kernel specifications.rUrcCsdtj|d}td|}tt|dddd}t|D]"}td||d||q.r(r*r2r3N)rinstancer8Zget_provisioner_entriesr#r9r;)rZkfp provisionersr?r%rrrr@ s   zListProvisioners.startN)rArBrCrDr rErFr@rrrrrsrc @seZdZdZeZdZdZee e j dfe e j dfe dfe e j dfe e j dfeej dfdZiZiZddd d Zd S) KernelSpecAppzAn app to manage kernel specs.zjupyter kernelspecz%Manage Jupyter kernel specifications.rzAlias for remove)listr uninstallremovez install-selfrrUrcCsL|jdkr>tdt|jt|||dn |jSdS)r Nz0No subcommand specified. Must specify one of: %sr )Zsubappr8r subcommandsZprint_descriptionZprint_subcommandsr`r@rrrrr@Hs  zKernelSpecApp.startN)rArBrCrDr rEr%rFrr splitlinesrHrqrrrrnrGr@rrrrr-s(  r__main__)$rD __future__rrgros.pathrMr^typingtZjupyter_core.applicationrrrZ traitletsrrrr r Ztraitlets.config.applicationr rJr rrZprovisioning.factoryrrrHrqrrrrAZlaunch_instancerrrrs(     3c;4'