:vj~>dZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZddlmZddlmZej:j<Zd Z Gd d eZ!ejDd ejFZ$dd Z%dZ&ddZ'ddZ(Gdde)Z*GddeZ+ddZ,ddZ- d ddZ.e+j\je._d d!dZ/e+j^je/_y)"zTools for managing kernel specs) annotationsN)SYSTEM_JUPYTER_PATHjupyter_data_dir jupyter_path) BoolCaselessStrEnumDict HasTraitsListSetTypeUnicodeobserve)LoggingConfigurable)KernelProvisionerFactorypython3ceZdZUdZeZded<eZeZ eZ eZ eZ e ZeZeddgdZe Zed dZd dZd d Zy ) KernelSpeczA kernel spec model object. List[str]argvmessagesignal) default_valuect|d}t|d5}tj|}ddd|dd|iS#1swYxYw)z|Create a KernelSpec object by reading kernel.json Pass the path to the *directory* containing kernel.json. kernel.jsonzutf-8)encodingN resource_dir)pjoinopenjsonload)clsr kernel_filef kernel_dicts j/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_client/kernelspec.pyfrom_resource_dirzKernelSpec.from_resource_dir'sJ L-8 + 0A))A,K1< < <<1 0s AA c|j|j|j|j|j|j |j d}|S)z"Convert the kernel spec to a dict.renv display_namelanguageinterrupt_modemetadatakernel_protocol_versionr+)selfds r(to_dictzKernelSpec.to_dict2sIII88 -- "11 '+'C'C cHtj|jS)zOSerialise this kernelspec to a JSON object. Returns a string. )r"dumpsr4r2s r(to_jsonzKernelSpec.to_json@s zz$,,.))r5N)r$ztype[KernelSpec]rstrreturnrr;zdict[str, t.Any]r;r:)__name__ __module__ __qualname____doc__r r__annotations__rnamemimetyper-r.r1r r,rrr/r0 classmethodr)r4r9rr5r(rrsw%fD) 9DyH9LyH%i &C9L$i%:(SNvH== *r5rz^[a-z0-9._\-]+$c,tj|S)z"Check that a kernel name is valid.)_kernel_name_patmatchrCs r(_is_valid_kernel_namerJKs  ! !$ ''r5zuKernel names can only contain ASCII letters and numbers and these separators: - . _ (hyphen, period, and underscore).ctjj|xr)tjjt |dS)zIs ``path`` a kernel directory?r)ospathisdirisfiler )rMs r(_is_kernel_dirrPWs. 77==  M277>>%m2L#MMr5c6|tjj|siSi}tj|D][}t ||}t |s|j }t|s!tjdtd|d|||<]|S)zReturn a mapping of kernel names to resource directories from dir. If dir is None or does not exist, returns an empty dict. z#Invalid kernelspec directory name (z):  stacklevel) rLrMrNlistdirr rPlowerrJwarningswarn_kernel_name_description)dirkernelsr&rMkeys r(_list_kernels_inr]\s  {"''--, G ZZ_S!}d# ggi$S) MM56N5OsSWRXY   Nr5c eZdZdZddZddZy) NoSuchKernelz7An error raised when there is no kernel of a give name.c||_y)zInitialize the error.NrI)r2rCs r(__init__zNoSuchKernel.__init__us  r5c d|jS)NzNo such kernel named rIr8s r(__str__zNoSuchKernel.__str__ys&tyyk22r5N)rCr:r;Noner=)r>r?r@rArarcrr5r(r_r_rsA3r5r_c^eZdZUdZeeddZedddZe Z ddZ e Z ddZ eddZedd Zed Zd ed <ddiZeeeddZddZddZd dZd!dZd"dZd#dZd$dZ d% d&dZ d' d(dZ d)d*dZ!y)+KernelSpecManagerzA manager for kernel specs.TzThe kernel spec class. This is configurable to allow subclassing of the KernelSpecManager for customized behavior. )confighelpzIf there is no Python kernelspec registered and the IPython kernel is available, ensure it is added to the spec list. ctS)N)rr8s r(_data_dir_defaultz#KernelSpecManager._data_dir_defaults !!r5c.t|jdS)Nr[)r data_dirr8s r(_user_kernel_dir_defaultz*KernelSpecManager._user_kernel_dir_defaultsT]]I..r5z@Deprecated, use `KernelSpecManager.allowed_kernelspecs` z^List of allowed kernel names. By default, all installed kernels are allowed. az List of kernel directories to search. Directories earlier in the list take priority: when two directories contain a kernelspec with the same name, the one from the earlier directory wins. This matches the precedence of ``$JUPYTER_PATH``, where the first entry has the highest priority (see ``jupyter_path("kernels")``). )rhr kernel_dirs whitelist)allowed_kernelspecsz7.0c J|j}|j|\}}t||}||jk7rj|jj |j jd|d|d|j jd|d t|||jyy)zobserver for deprecated traits.z! is deprecated in jupyter_client z, use z insteadN) rC_deprecated_aliasesgetattrnewlogwarning __class__r>setattr)r2changeold_attrnew_attrversion new_values r(_deprecated_traitz#KernelSpecManager._deprecated_traits;; 44X>'D(+  " HH  >>**+1XJ6W)6$.."9"9!:!H:XO  D(FJJ / #r5ctd} ddlm}|jtj j |d|S#t$rY|SwxYw)Nr[r)get_ipython_dir)r IPython.pathsrappendrLrMjoinModuleNotFoundError)r2dirsrs r(_kernel_dirs_defaultz&KernelSpecManager._kernel_dirs_defaultsRI&  5 KK _%6 B C #    s:A AAc0i}|jD]L}t|}|jD],\}}||vs |jj d|||||<.N|j r9t |vr1 ddlm}|jj dt |||t <|jr2|jDcic]\}}||jvs||}}}|S#t$r#|jjdt YkwxYwcc}}w)HHNN#:E:N#AeH /+  $ $);1)D \::& )2$%  # #./ggi\i d44C[C[;[tiA\  \  !FHZ[ \ ]s30C#DD#)DDc<d}|tk(r* ddlm}m}||k(r|}|jdd|i|} |s|jj |}tj|jj|s t||S#t $rYdwxYw)zaReturns a :class:`KernelSpec` instance for a given kernel_name and resource_dir. Nr)rget_kernel_dictr)parentr) rrrrkernel_spec_classrr)KPFinstanceris_provisioner_availabler_)r2 kernel_namerkspecrrkdicts r(_get_kernel_spec_by_namez*KernelSpecManager._get_kernel_spec_by_names , , WK  9,+-E2D22V VPUVE**<<\JE||4;;/HHO{+ +   sB BBch|jDcgc]$}tjj|s#|&c}D]O}tj|}|D]3}t ||}|j |k(s#t|s/|ccSQ|tk(r ddl m }|Sycc}w#t$rYywxYw)z2Find the resource directory of a named kernel specrrN) rnrLrMrNrUr rVrPrrrr)r2rkdrfilesr&rMrs r(_find_spec_directoryz&KernelSpecManager._find_spec_directory s(,(8(8N(8"BGGMM">%6688 MD  //6D.:DLLNSE $-    V  !>PT U VsA A:: B.&B.-B.cL|j} d|_|j}||_||}|jjd|tj j |rt j||Stj||S#||_wxYw)z\Remove a kernel spec directory by name. Returns the path that was deleted. Fz Removing %s) rrrvrrLrMislinkremoveshutilrmtree)r2rC save_nativespecsspec_dirs r(remove_kernel_specz$KernelSpecManager.remove_kernel_specKs //  4(-D %**,E(3D %; }h/ 77>>( # IIh  MM( #)4D %s B B#Nc.|r*tjj|j|S|r@tjjtjj |ddd|Stjjt dd|S)Nsharejupyterr[r)rLrMruser_kernel_dirabspathr)r2ruserprefixs r(_get_destination_dirz&KernelSpecManager._get_destination_dir^sk 77<< 4 4kB B 77<< 7)YXcd d77<< 3A 6 ;O Or5c:|jd}|stjj|}|j }t |sd|dt }t||r|r d}t||tjdtd|j|||}|jjd |tjj|}||jvr'|jj!d ||jtjj#|r1|jj%d |t'j(|t'j*|||jj%d |||S) a#Install a kernel spec by copying its directory. If ``kernel_name`` is not given, the basename of ``source_dir`` will be used. If ``user`` is False, it will attempt to install into the systemwide kernel registry. If the process does not have appropriate permissions, an :exc:`OSError` will be raised. If ``prefix`` is given, the kernelspec will be installed to PREFIX/share/jupyter/kernels/KERNEL_NAME. This can be sys.prefix for installation inside virtual or conda envs. z/\zInvalid kernel name z. zCCan't specify both user and prefix. Please choose one or the other.zTreplace is ignored. Installing a kernelspec always replaces an existing installationrS)rrzInstalling kernelspec in %szFInstalling to %s, which is not in %s. The kernelspec may not be found.z"Removing existing kernelspec in %szInstalled kernelspec %s in %s)rstriprLrMbasenamerVrJrY ValueErrorrWrXDeprecationWarningrrvrdirnamernrwrNinforrcopytree) r2 source_dirrrreplacermsg destinationrs r(install_kernel_specz%KernelSpecManager.install_kernel_spechsW* &&u- ''**:6K!'') $[1(s;S:TUCS/ ! FWCS/ !   MM"  // $v/V  4kBWW__[1 T-- - HH  X    77== % HHMM> L MM+ & K0  5{KPr5cRtjddddlm}|||y)z,DEPRECATED: Use ipykernel.kernelspec.installzRinstall_native_kernel_spec is deprecated. Use ipykernel.kernelspec import install.rrSr)installrN)rWrXrr)r2rrs r(install_native_kernel_specz,KernelSpecManager.install_native_kernel_specs# `  14 r5r=)rzt.Anyr;rd)r;z list[str]r;dict[str, str])rr:rr:r;r)rr:r; str | Nonerr:r;rr<)rCr:r;r:)FN)rr:rboolrrr;r:)NFNN rr:rrrrrz bool | Nonerrr;r:Frrr;rd)"r>r?r@rAr rrrrrrlrjrrmr rorpr rnrBrsrlistrrrrrrrrrrrrr5r(rfrf}sh%    yH"iO/ I  "  K  3  d&'( 0) 0 :.$H @(JNPP&*Prs3%# RR___09  ,*,*^2::0"--@(/ N ,383o!+o!d 3 <#  ccc c c  c  c0CCKK> &7%Q%Q%Y%Y"r5