o ^i h@s~dZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl Z ddlmZddlmZddlmZmZdd lmZmZmZmZmZd d d d ZddZedededdZddZeddGdddZ eddGdddZ!eddGdddZ"eddGddde"Z#eddGd d!d!Z$eGd"d#d#Z%Gd$d%d%eZ&Gd&d'd'e&Z'dS)(z'Base classes for the extension manager.N) dataclassfieldfieldsreplace)Path)OptionalUnion) translator)Enum) ConfigurableLoggingConfigurable) _AppHandler_ensure_optionsdisable_extensionenable_extension get_app_infoz-alpha.z-beta.z-rc.)abrccCst|}||d<dS)z0Ensure that the app info has compat_errors field compat_errorsN)r _get_extension_compat)info app_optionshandlerrX/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyterlab/extensions/manager.py_ensure_compat_errorssrz*(?P.*) needs to be included in buildz+(?P.*) needs to be removed from buildz:(?P.*) changed from (?P.*) to (?P.*)install uninstallupdatecCsbt|}|jdd}gggd}|D]}tD]\}}||}|r-|||dqq|S)z8Get info about packages scheduled for (un)install/updateT)fastrname)r build_check _message_mapitemsmatchappendgroup)rrmessagesstatusmsgkeypatternr&rrr_build_check_info)s   r.T)frozenc@seZdZUdZeed<eed<eed<eed<dZeed<dZeed <d Z e eed <dZ eed <dZ eed <d Z e eed<d Ze eed<dZeed<dZeed<dZeed<d Ze eed<d Ze eed<d Ze eed<d Ze eed<d Ze eed<d Ze eed<d S)ExtensionPackageaExtension package entry. Attributes: name: Package name description: Package description homepage_url: Package home page pkg_type: Type of package - ["prebuilt", "source"] allowed: [optional] Whether this extension is allowed or not - default True approved: [optional] Whether the package is approved by your administrators - default False companion: [optional] Type of companion for the frontend extension - [None, "kernel", "server"]; default None core: [optional] Whether the package is a core package or not - default False enabled: [optional] Whether the package is enabled or not - default False install: [optional] Extension package installation instructions - default None installed: [optional] Whether the extension is currently installed - default None installed_version: [optional] Installed version - default "" latest_version: [optional] Latest available version - default "" status: [optional] Package status - ["ok", "warning", "error"]; default "ok" author: [optional] Package author - default None license: [optional] Package license - default None bug_tracker_url: [optional] Package bug tracker URL - default None documentation_url: [optional] Package documentation URL - default None package_manager_url: Package home page in the package manager - default None repository_url: [optional] Package code repository URL - default None r" description homepage_urlpkg_typeTallowedFapprovedN companioncoreenabledr installedinstalled_versionlatest_versionokr*authorlicensebug_tracker_urldocumentation_urlpackage_manager_urlrepository_url)__name__ __module__ __qualname____doc__str__annotations__r4boolr5r6rr7r8rdictr9r;r<r*r>r?r@rArBrCrrrrr07s,        r0c@s@eZdZUdZeed<dZeeed<ee dZ e eed<dS) ActionResultzAction result Attributes: status: Action status - ["ok", "warning", "error"] message: Action status explanation needs_restart: Required action follow-up - Valid follow-up are "frontend", "kernel" and "server" r*Nmessagedefault_factory needs_restart) rDrErFrGrHrIrMrrlistrPrrrrrLhs  rLc@s4eZdZUdZeedZeeed<dZ e ed<dS)PluginManagerOptionsalPlugin manager options. Attributes: lock_all: Whether to lock (prevent enabling/disabling) all plugins. lock_rules: A list of plugins or extensions that cannot be toggled. If extension name is provided, all its plugins will be disabled. The plugin names need to follow colon-separated format of `extension:plugin`. rN lock_rulesFlock_allN) rDrErFrGr frozensetrSrHrIrTrJrrrrrRys  rRc@s\eZdZUdZeedZeeed<eedZ eeed<dZ e ed<ee dZ e ed<dS) ExtensionManagerOptionsa Extension manager options. Attributes: allowed_extensions_uris: A list of comma-separated URIs to get the allowed extensions list blocked_extensions_uris: A list of comma-separated URIs to get the blocked extensions list listings_refresh_seconds: The interval delay in seconds to refresh the lists listings_tornado_options: The optional kwargs to use for the listings HTTP requests as described on https://www.tornadoweb.org/en/stable/httpclient.html#tornado.httpclient.HTTPRequest rNallowed_extensions_urisblocked_extensions_urisilistings_refresh_secondslistings_tornado_optionsN)rDrErFrGrsetrWrHrIrXrYintrKrZrrrrrVs   rVc@s6eZdZUdZeed<dZeed<dZe eed<dS)ExtensionManagerMetadataa#Extension manager metadata. Attributes: name: Extension manager name to be displayed can_install: Whether the extension manager can un-/install packages (default False) install_path: Installation path for the extensions (default None); e.g. environment path r"F can_installN install_path) rDrErFrGrHrIr^rJr_rrrrrr]s  r]c@sDeZdZUdZeedZeeeee e ffe d<dZ ee d<dS)ExtensionsCachez{Extensions cache Attributes: cache: Extension list per page last_page: Last available page result rNcache last_pageN) rDrErFrGrrKrar\rrHr0rIrcrrrrr`s &r`c seZdZdZegddddjddZ   dd eed eed ee d dffd d Z d efddZ de e d ee fddZdee e e fd efddZdee e e fd efddZZS) PluginManageraPlugin manager enables or disables plugins unless locked. It can also disable/enable all plugins in an extension. Args: app_options: Application options ext_options: Plugin manager (subset of extension manager) options parent: Configurable parent Attributes: app_options: Application options options: Plugin manager options ) sys_prefixusersystemrez:Level at which to manage plugins: sys_prefix, user, system)values default_valuehelpT)configNr ext_optionsparentreturncsxtj|d|jd|jjd|jdt||_ddt t Dfdd|p-i D}t d i||_ dS) N)rmz Plugins in z will managed on the z levelcSsh|]}|jqSrr").0frrr sz)PluginManager.__init__..csi|] \}}|vr||qSrr)rpoptionvalueplugin_options_fieldrr s z*PluginManager.__init__..r) super__init__logdebug __class__rDlevelrrrrRr%options)selfrrlrmplugin_optionsr|rurrys   zPluginManager.__init__cst|jj|jjdS)z8Get information about locks on plugin enabling/disabling) lockRules allLocked)rQr~rSrTrrrr plugin_lockss zPluginManager.plugin_locksplugins_or_extensionscCsh|jjrt|St}dd|jjD}|D]}d|vr(||jjvr'||q||vr1||q|S)z9Find a subset of plugins (or extensions) which are lockedcSsh|] }|ddqS):r)split)rppluginrrrrrsz-PluginManager._find_locked..r)r~rTr[rSadd)rr locked_subsetextensions_with_locked_pluginsrrrr _find_lockeds   zPluginManager._find_lockedpluginsc t|tr|n|g}||}td}|r%td|dd|dSz|D] }t||j |j dq(tddgd WSt yU}ztdt |dWYd }~Sd }~ww) zDisable a set of plugins (or an extension). Args: plugins: The list of plugins to disable Returns: The action result jupyterlaberrorz=The following plugins cannot be disabled as they are locked: , r*rMrr}r=frontendr*rPN) isinstancerQrr loadrLgettextjoinrrr} Exceptionreprrrlockedtransrerrrrrdisable*  zPluginManager.disablec r) zEnable a set of plugins (or an extension). Args: plugins: The list of plugins to enable Returns: The action result rrz"} Args: extension: The extension name version: The version to install; default None (i.e. the latest possible) Returns: The action result r)rrrrrrrszExtensionManager.installcr)ajUninstall the required extension. Note: If the user must be notified with a message (like asking to restart the server), the result should be {"status": "warning", "message": ""} Args: extension: The extension name Returns: The action result rrrrrrrzExtensionManager.uninstallc Cs tdddtdd|dS)a;Convert a Python version to Semver version. It: - drops ``.devN`` and ``.postN`` - converts ``aN``, ``bN`` and ``rcN`` to ``-alpha.N``, ``-beta.N``, ``-rc.N`` respectively Args: version: Version to convert Returns Semver compatible version z(a|b|rc)(\d+)$cSst|d|dS)Nrb)PYTHON_TO_SEMVERr()mrrrsz5ExtensionManager.get_semver_version..z\.(dev|post)\d+r:r)resubsubn)rrrrget_semver_versions z#ExtensionManager.get_semver_versioncCs|jS)aNormalize extension name. Extension have multiple parts, npm package, Python package,... Sub-classes may override this method to ensure the name of an extension from the service provider and the local installed listing is matching. Args: extension: The extension metadata Returns: The normalized name rorrrrget_normalized_names z$ExtensionManager.get_normalized_namerbc sh||jvs||j|jvr||||IdH|jdur*|jdur*|jIdH|j|j|}|dur8i}t|}|dur|jdurt|j}g}|jr| D](\}}||vrg| t |ddqU|j r}|j d|d| t |ddqUn-| D](\}}||vr| t |ddq|j r|j d|d| t |ddq||j|jfS)aLList extensions for a given ``query`` search term. This will return the extensions installed (if ``query`` is None) or available if allowed by the listing settings. Args: query: [optional] Query search term. Returns: The extensions Last page of results NT)r4zBlocked extension 'z' is installed.FzNot allowed extension ')rrarefreshrrcallbackrQrhrr%r'rr;rzrrc) rrrrra extensionslistingr"extrrrlist_extensionss:  z ExtensionManager.list_extensionscs4||jvrd|j|j|<||||IdHdS)zRefresh the list of extensions.N)rra_update_extensions_listrrrrrs zExtensionManager.refreshcsg}tj}|jrAt|jjr@|jd|jj|jjD]}|j |fi|jj IdH}t |j }||dgq n5t|jjrv|jd|jj|jjD]}|j |fi|jj IdH}t |j }||dgqVdd|D|_dS)z-Fetch the listings for the extension manager.z!Fetching blocked extensions from Nblocked_extensionsz!Fetching allowed extensions from allowed_extensionscSsi|]}|d|qSror)rprrrrrwsz4ExtensionManager._fetch_listings..)r httpclientAsyncHTTPClientrrr~rXrzrfetchrZjsonloadsbodyextendgetrWr)rrulesclientblocked_extensions_urirjallowed_extensions_urirrrrs<         z ExtensionManager._fetch_listingsTc sN|j}t|d}t|}t||i}|dD]\}}d}|} |d|dr,d}||} td)id| d| dd d |d d d ||d vdddt |ddddt |dd|d|diddd| |d|did|dd|dd|did d|did |d} |rt | | | j IdHd } | || <q|d!D]\}}||d"vrqd}|d|drd}n |D]} || vrd#}q||} t| |dd |d ||d vdt |ddt |d|d$| ||did|d|d|did |did |dd%} |rAt | | | j IdHd } | || <q|d&D]Z}||}|dur||} t| |dd |d'd dddt |dt |dd#d|did|d|d|did |did |dd(} | || <qJ|S)*zGet the installed extensions. Args: get_latest_version: Whether to fetch the latest extension version or not. Returns: The installed extensions as a mapping {name: metadata} )rfederated_extensionsr=rNrr"r1r:r2urlr8disabledr7Fr<rr9Tr;r*rr3prebuiltr6r>r?r@bugsrC repository)r<r shadowed_extsrsource)r"r1r2r8r7r<r9r;r*r3r6r>r?r@rCrhomepage)r"r1r2r9r8r7r<r;r*r3r>r?r@rCr)rrr.rr%r_normalize_namer0rr_get_companionrrr"rh_get_scheduled_uninstall_info) rrrrbuild_check_inforr"datar*pkg_infonormalized_namepkgpackagesrrr_get_installed_extensionss                             z*ExtensionManager._get_installed_extensionsrcCs@d}d|dvrd|ddvrd}|Sd|ddvrd}|S)N discoveryrserverkernelr)rrr6rrrrs zExtensionManager._get_companioncCsX|jdd|d}|r*| }t|WdS1s#wYdSdS)zDGet information about a package that is scheduled for uninstallationstaging node_modulesz package.jsonN)rexistsopenrr)rr"targetfidrrrrs  $z.ExtensionManager._get_scheduled_uninstall_infor"cCs|S)zNormalize extension name; by default does nothing. Args: name: Extension name Returns: Normalized name r)rr"rrrrsz ExtensionManager._normalize_namecsd}|dur||||IdH\}}n|IdH}||jvr2||j|j|<|p+d|j|_dSt||i|p9d|j|<dS)zUpdate the list of extensionsNrb)rrrrarcr`)rrrrrcrrrrrs z(ExtensionManager._update_extensions_listrr)Nrbr)rnN)T)"rDrErFrGrrKr ryrpropertyr]rrHrr\tupler0rrLrr staticmethodrrrQrrrrrrrrrrrrrr,sz   .   j   r)(rGrr dataclassesrrrrpathlibrtypingrrr#jupyterlab_server.translation_utilsr traitletsr traitlets.configr r jupyterlab.commandsr rrrrrrcompiler$r.r0rLrRrVr]r`rdrrrrrs@    0 w