:vj5|UdZddlmZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZddlmZdd lmZdd lmZmZd d?dZGddeZiZded<ej?e j@ejCddejCddej?dd diidfdd d didfdd diidfdddiidfdeded<iZ"ded <e"j?e jFGd!de Z$ejJd"k7rd#nd$Z&ejJd"k7rd%nd&Z'ejJd"k7rd'nd(Z(ejJd"k7rd)nd*Z) d@ dAd+Z*iZ e j?e$j@e jCdde jCdde j?d,d diid-fd,d d did.fd,d diid/fd,ddiidfde de d<d0Z+Gd1d,e$Z,Gd2d3e,Z-Gd4d5e,Z.Gd6d7e$Z/d8Z0Gd9d:e$Z1e1jdZ3e4d;k(re3yy)Bz#Utilities for installing extensions) annotationsN) JupyterApp)ENV_CONFIG_PATHSYSTEM_CONFIG_PATHjupyter_config_dir) LogFormatter)Bool) __version__)ExtensionConfigManager)ExtensionManagerExtensionPackageFuser sys_prefixcZ|r|rd}|r t}|S|r td}|Std}|S)a8Get the location of config files for the current context Returns the string to the environment Parameters ---------- user : bool [default: False] Get the user's .jupyter config directory sys_prefix : bool [default: False] Get sys.prefix, i.e. ~/.envs/my-env/etc/jupyter Fr)rrr)rrextdirs y/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_server/extension/serverextension.py_get_config_dirrsG   #% M  # M$A& Mct||}t|gtjj ||}t |}||fS)aGet an extension manager pointing at the current context Returns the path to the current context and an ExtensionManager object. Parameters ---------- write_dir : str [default: 'jupyter_server_config.d'] Name of config directory to write extension config. user : bool [default: False] Get the user's .jupyter config directory sys_prefix : bool [default: False] Get sys.prefix, i.e. ~/.envs/my-env/etc/jupyter rrread_config_pathwrite_config_dir)config_manager)rr ospathjoinr ) write_dirrr config_dirrextension_managers r_get_extmanager_for_contextr!-sO !dzBJ+$j)<N)% ( ((rc eZdZy)ArgumentConflictN)__name__ __module__ __qualname__rrr#r#Hsrr#dict[str, t.Any] _base_flagsyzgenerate-configBaseExtensionAppTz+Apply the operation only for the given userrzApply the operation system-widezTUse sys.prefix as the prefix for installing extensions (for environments, packaging)pythonInstall from a Python package)rsystemz sys-prefixpyr/ _base_aliasesc|eZdZdZeZeZeZ e Z e dddZ e dddZe dddZd dZed d Zy ) r+zBase extension installer appFTzWhether to do a user install)confighelpz Use the sys.prefix as the prefixr-cy)zA default format for messagesz %(message)sr'selfs r_log_format_defaultz$BaseExtensionApp._log_format_defaultsrcDt|j|jS)Nr)rrrr5s rrzBaseExtensionApp.config_dirsDII$//JJrN)returnstr)r$r%r&__doc__r_log_formatter_clsr)flagsr0aliasesr versionr rrr,r7propertyrr'rrr+r+{sb&% EGG d)G HDd4.PQJ %+J KFKKrntzenabledenabledzdisableddisabledz OKokz  Xz Xc|rdn|}t||}t|gtjj |d}|r|j |y|j |y)zZToggle the boolean setting for a given server extension in a Jupyter config file. Frjupyter_server_config.drN)rr rrrenabledisable) import_namerBparentrrrmanagers rtoggle_server_extension_pythonrLsXJJ dzBJ$$j2KLG{# $rToggleServerExtensionAppz*Perform the operation for the current userz!Perform the operation system-widez=Use sys.prefix as the prefix for installing server extensionszEEnable/disable a server extension using frontend configuration files.cBeZdZdZdZeZeZeZ dZ dZ ddZ ddZ y) rMz.A base class for enabling/disabling extensionsz'jupyter server extension enable/disablect|j|j\}} |jj |j j d||jj d||jj d|d|j}d}|r*|j}||vrd|d}t|||}t|| }|js d }t||j} |jj d |d | d t|j}|r1|jd ur|j|n|j!||jj d|j"dy#t$$r/} |jj'dt(d| Yd} ~ yd} ~ wwxYw)a.Change the status of a named server extension. Uses the value of `self._toggle_value`. Parameters --------- import_name : str Importable Python module (dotted-notation) exposing the magic-named `load_jupyter_server_extension` function rz: z- Writing config:  - Validating ...Fz The module 'z>' could not be found. Are you sure the extension is installed?namerBvalidation failed  Tz - Extension successfully .z z Validation failed: N)r!rrloginfo_toggle_pre_message capitalizerget_jpserver_extensions ValueErrorr validater?GREEN_OK _toggle_valuerGrH_toggle_post_message ExceptionerrorRED_X) r6rIrr r2rBjpserver_extensionsmsgextpkgr?errs rtoggle_server_extensionz0ToggleServerExtensionApp.toggle_server_extensions)Dt) % %$ E HHMMT55@@BC2k]S T HHMM.zl; < HHMM-k]#> ?&55FG&,&D&D&F#&99&{m4;;%S/)-k:&;HF??$) o%nnG HHMMF;-q 8*E F'55F%%-MM+.NN;/ HHMM9$:S:S9TTUV W E HHNNU5')=cUC D D EsE5F G%%GGc|jstjd|jD]}|j|y)'Perform the App's actions as configuredz>Please specify a server extension/package to enable or disableN) extra_argssysexitrj)r6args rstartzToggleServerExtensionApp.start's2 HHU V??C  ( ( -#rN)rIr:r9Noner9rr)r$r%r&r;rT_desc descriptionr=r rar[rbrjrqr'rrrMrMs38 4DK EFM4El.rc$eZdZdZdZdZdZdZdZy)EnableServerExtensionAppz5An App that enables (and validates) Server Extensionszjupyter server extension enablez Enable a server extension in configuration. Usage jupyter server extension enable [--system|--sys-prefix] TenablingrBN r$r%r&r;rTrurar[rbr'rrrwrw/s$? ,DK M$$rrwc$eZdZdZdZdZdZdZdZy)DisableServerExtensionAppz&An App that disables Server Extensionsz jupyter server extension disablez Disable a server extension in configuration. Usage jupyter server extension disable [--system|--sys-prefix] F disablingrCNryr'rrr{r{>s$0 -DK M%%rr{c,eZdZdZdZeZdZddZddZ y)ListServerExtensionsAppz3An App that lists (and validates) Server Extensionszjupyter server extension listz?"= HHMM" 1%(!2HHNN2N5F5'3%0112s AD E#'2EE#c$|jy)rlN)rr5s rrqzListServerExtensionsApp.startys ##%rNrs) r$r%r&r;rTr r?rurrqr'rrr~r~Ms= *DGPK#J&rr~a/ jupyter server extension list # list all configured server extensions jupyter server extension enable --py # enable all server extensions in a Python package jupyter server extension disable --py # disable all server extensions in a Python package cbeZdZUdZdZeZdZded<e Z e dfe dfe dfd Zd ed <d fd ZxZS)ServerExtensionAppzRoot level server extension appzjupyter server extensionz#Work with Jupyter server extensionsr:ruzEnable a server extensionzDisable a server extensionzList server extensions)rGrHlistr( subcommandsct|djt|j}t j d|zy)rlz, z)Please supply at least one subcommand: %sN)superrqrsortedrrnro)r6subcmds __class__s rrqzServerExtensionApp.starts9  ))F4#3#345 r+rTrrr`rerLrtrMrwr{r~rrlaunch_instancemainr$r'rrrsV)# /VV$/BO0X]))6:)PT)!)6 z !# " :##$ T !4( #D%  :  #!"'%  . # $%  c  #d%  ,  5"$J$D) H"$ $ Z''(KzK.-/GGtO( ,.GGtO( "$''T/ t!ww$D  %% % %  %  %  %4   # #$ #t T" +D-  9  +!"'-  0 + $-  L  +d-  ,  5"$J+h PG./G.T %7 % & 8 &.&..&b  H)H0)) zFr