o +wi4@s&dZddlmZddlZddlZddlZddlZddlZ ddl m Z ddl m Z mZmZddlmZmZmZmZmZddlmZdd lmZdd lmZdd lmZGd d d e ZGddde ZGddde Z Gddde Z!Gddde Z"GdddeZ#d!ddZ$e%d kre#&dSdS)"zApps for managing kernel specs.) annotationsN)Path) JupyterApp base_aliases base_flags)BoolDictInstanceListUnicode) Application) __version__)KernelSpecManager)KernelProvisionerFactoryc@sxeZdZdZeZdZeeZ e ddddZ e ddddZ dddiidfdd diid fe d d ZdddZdddZdS)ListKernelSpecszAn app to list kernel specs.z%List installed kernel specifications.Fz7output spec name and location as machine-readable json.Thelpconfigz*List only specs with missing interpreters. json_outputmissing_kernelszoutput only missing kernelsdebug)jsonmissingrreturnrcCst||jdS)N)parentdata_dirrrselfr Y/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/jupyter_client/kernelspecapp.py_kernel_spec_manager_default3z,ListKernelSpecs._kernel_spec_manager_defaultdict[str, t.Any] | Nonecsj}j}jrt||\}}jsR|stddStt|dddd}dfd d }td t| |dD]\}}td | |d|q>|Stt j d|idd|S)Start the application.zNo kernels availableNcSt|SNlennamer r r!Cz'ListKernelSpecs.start..keyitemt.Anyrcs:|d}tjD]\}}||r||fSq d|fS)z+sort key function for Jupyter path priorityr r0) enumerate jupyter_path startswith)r1pathidxprefixrr r!path_keyEs   z'ListKernelSpecs.start..path_keyzAvailable kernels:  kernelspecs)indent)r1r2rr2) kernel_spec_managerfind_kernel_specs get_all_specsr_limit_to_missingrprintr)sorteditemsljustrdumps)rpathsspecsname_lenr9 kernelnamer6r rr!start6s   zListKernelSpecs.startNrr)rr$)__name__ __module__ __qualname____doc__rversion descriptionr rr?rrrrflagsr"rLr r r r!rs2   rcseZdZdZeZdZdZdZe e Z d%ddZ e Ze d d d d Zd&ddZedd dd Ze d d dd Zedd dd ZdddZeeddd iidfddd iidfddejiidfeddZd'fd!d" Zd(d#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]rrcC t|jdSN)rrrr r r!r"g z.InstallKernelSpec._kernel_spec_manager_defaultTz&Install the kernel spec with this namerrstrcCstj|jSr')osr6basename sourcedirrr r r!_kernel_name_defaultmr#z&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+r8user'Install to the per-user kernel registryreplacer8zEInstall to Python's sys.prefix. Useful in conda/virtual environments.r)rarcz sys-prefixrargvNone | list[str]Nonecs>t||jr|jd|_dStdtjd|ddS)Parse the command line args.rzNo source directory specified.filer N)superparse_command_line extra_argsr^rCsysstderrexitrrd __class__r r!rks z$InstallKernelSpec.parse_command_linec Cs|jr |jr |dz|jj|j|j|j|j|jdWdSty^}z4|j t j krCt |t j d|js=t dt j d|d|j t jkrYt d|jt j d|dd}~ww)r%zCCan't specify both user and prefix. Please choose one or the other.) kernel_namerar8rcrh4Perhaps you want to install with `sudo` or `--user`?r z$A kernel spec is already present at N)rar8ror?install_kernel_specr^rsrcOSErrorerrnoEACCESrCrmrnEEXISTfilename)rer r r!rLs.       zInstallKernelSpec.startrM)rr[)rdrerrfrrf)rNrOrPrQrrRrSexamplesusager rr?r"r r^rsr_rrar8rcaliasesupdaterrmrrTrkrL __classcell__r r rqr!rUVsN       rUcseZdZdZeZdZdZeddddZ e e Z edddd Z eeZdd d Zdddiie jfdddiidfdZeejdfdd ZdddZZS)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.rZzRemove missing specs.rrrcCst|j|dS)N)rrrrr r r!r"r#z-RemoveKernelSpec._kernel_spec_manager_defaultforcerzremove missing kernels)frrdlist[str] | Nonerfcs0t||jrtt|j|_dSg|_dS)rgN)rjrkrlrDset spec_namesrprqr r!rks  z#RemoveKernelSpec.parse_command_linec Csjd|j_|j}|jr"t||j\}}tt|jt ||_t|j t|}|r8| dd ||j sp|jsptd|jD]}|||}td|dd|dqEtdt|j}|d spd S|jD]?}z|j|}Wn-ty} z!| jtjkrt| tjd td tjd | d nWYd } ~ nd } ~ wwtd|qsd S)r%Fz Couldn't find kernel spec(s): %sz, zKernel specs to remove:r: zRemove %i kernel specs [y/N]: yNrhzPerhaps you want sudo?r zRemoved )r?ensure_native_kernelr@rrBrArDrrlist differencerojoinr answer_yesrCgetrFinputr)lowerr5remove_kernel_specrvrwrxrmrn) r spec_paths_specrr+r6answerrsr{r r r!rLsB    "    zRemoveKernelSpec.startrM)rdrrrfr|)rNrOrPrQrrRrSr}rrr r rrr rr?r"rrTrrrkrLrr r rqr!rs,     rc@s\eZdZdZeZdZeeZ dddZ e ddd d Z dd diid fe d dZdddZdS)InstallNativeKernelSpecz)An app to install the native kernel spec.zG[DEPRECATED] Install the IPython kernel spec directory for this Python.rrcCrVrWrrr r r!r"rXz4InstallNativeKernelSpec._kernel_spec_manager_defaultFTr`rZrarbr)rarrfc Cs|jdzddlm}Wnty#tdtjd|dYnwz |j |j |j dWd St yd}z(|j t jkrTt|tjd|j sOtdtjd|d||WYd }~d Sd }~ww) 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.rhr )rartN)logwarning ipykernelrModuleNotFoundErrorrCrmrnroinstallr?rarvrwrx)rrr{r r r!rL"s.   zInstallNativeKernelSpec.startNrMr|)rNrOrPrQrrRrSr rr?r"rrarrTrLr r r r!rs"  rc@s"eZdZdZeZdZdddZdS) ListProvisionerszAn app to list provisioners.z=List available provisioners for use in kernel specifications.rrfcCsdtj|d}td|}tt|dddd}t|D]}td||d||qd S) r%)rzAvailable kernel provisioners:cSr&r'r(r*r r r!r,Hr-z(ListProvisioners.start..r.r0r:r;N)rinstancerCget_provisioner_entriesr)rDrF)rkfp provisionersrJr+r r r!rLAs   zListProvisioners.startNr|)rNrOrPrQrrRrSrLr r r r!r;s rc @seZdZdZeZdZdZee e j dfe e j dfe dfe e j dfe e j dfeej dfdZiZiZd d d Zd S) KernelSpecAppzAn app to manage kernel specs.zjupyter kernelspecz%Manage Jupyter kernel specifications.rzAlias for remove)rr uninstallremovez install-selfrrrfcCsJ|jdur tdt|jt|||ddS|jS)r%Nz0No subcommand specified. Must specify one of: %sr )subapprCr subcommandsprint_descriptionprint_subcommandsrorLrr r r!rLis  zKernelSpecApp.startNr|)rNrOrPrQrrRr+rSrr splitlinesrUrrrrrrTrLr r r r!rNs*  rrHdict[str, str]rIdict[str, t.Any]r'tuple[dict[str, str], dict[str, t.Any]]csnddlm}i|D]\}}|ddd}t|s"||r#q ||<q fdd|D}|fS)Nr)whichrrdcsi|] \}}|vr||qSr r ).0kvrr r! sz%_limit_to_missing..)shutilrrErexists)rHrIrr+dataexepaths_r rr!rBus  rB__main__)rHrrIrrr)'rQ __future__rrwros.pathr\rmtypingtpathlibrjupyter_core.applicationrrr traitletsrrr r r traitlets.config.applicationr rYrrrprovisioning.factoryrrrUrrrrrBrNlaunch_instancer r r r!s0      ?cN4 '