o tf0h@sdZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z m Z mZmZmZmZddlZddlmZddlmZddlmZmZdd lmZmZmZmZmZd d d d Z ddZ!e"de"de"ddZ#ddZ$eddGdddZ%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)Dict FrozenSetListOptionalSetTupleUnion) 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)rZ_get_extension_compat)info app_optionshandlerrf/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/jupyterlab/extensions/manager.py_ensure_compat_errorssr z*(?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)fastr!name)rZ build_check _message_mapitemsmatchappendgroup)rrmessagesstatusmsgkeypatternr)rrr_build_check_info)s   r1T)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__r7boolr8r9r r:r;r"dictr<r>r?r-rArBrCrDrErFrrrrr37s,        r3c@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) rGrHrIrJrKrLrPr rlistrSr rrrrrOhs  rOc@s4eZdZUdZeedZeee d<dZ e e d<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`. rQ lock_rulesFlock_allN) rGrHrIrJr frozensetrVrrKrLrWrMrrrrrUys  rUc@s\eZdZUdZeedZeee d<eedZ eee d<dZ e e d<ee dZe e d<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 rQallowed_extensions_urisblocked_extensions_urisilistings_refresh_secondslistings_tornado_optionsN)rGrHrIrJrsetrZr rKrLr[r\intrNr]rrrrrYs   rYc@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) rGrHrIrJrKrLrarMrbr rrrrr`s  r`c@sDeZdZUdZeedZeee ee e ffe d<dZ ee d<dS)ExtensionsCachez{Extensions cache Attributes: cache: Extension list per page last_page: Last available page result rQcache last_pageN)rGrHrIrJrrNrdrr_r rKr3rLrfrrrrrcs &rcc 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_prefixusersystemrhz:Level at which to manage plugins: sys_prefix, user, system)values default_valuehelpT)configNr ext_optionsparentreturncsntj|d|jd|jj|jt||_ddt t Dfdd|p(i D}t di||_ dS)N)rpz*Plugins in %s will managed on the %s levelcSsh|]}|jqSrr&).0frrr sz)PluginManager.__init__..csi|] \}}|vr||qSrr)rsoptionvalueZplugin_options_fieldrr s z*PluginManager.__init__..r) super__init__logdebug __class__rGlevelrrrrUr(options)selfrrorpZplugin_optionsr~rxrr{s    zPluginManager.__init__cst|jj|jjdS)z8Get information about locks on plugin enabling/disabling)Z lockRulesZ allLocked)rTrrVrWrrrr 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)rspluginrrrrusz-PluginManager._find_locked..r)rrWr^rVadd)rrZ locked_subsetZextensions_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-rPrrr@frontendr-rSN) isinstancerTrrloadrOgettextjoinrrr ExceptionreprrrlockedZtransrerrrrrdisable*  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)rrrrrrr"szExtensionManager.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 rrrrrr#rzExtensionManager.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)Nre)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 rrrrrrget_normalized_names z$ExtensionManager.get_normalized_namerec 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)r7zBlocked extension 'z' is installed.FzNot allowed extension ')rrdrefreshrrcallbackrTrkrr(r*rr>r|rrf) rrrrrd extensionsZlistingr&extrrrlist_extensionss:  z ExtensionManager.list_extensionscs4||jvrd|j|j|<||||IdHdS)zRefresh the list of extensions.N)rrd_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 NZblocked_extensionsz!Fetching allowed extensions from Zallowed_extensionscSsi|]}|d|qSrrr)rsrrrrrysz4ExtensionManager._fetch_listings..)rZ httpclientZAsyncHTTPClientrrrr[r|rfetchr]jsonloadsbodyextendgetrZr)rrulesclientZblocked_extensions_urirjZallowed_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} )rZfederated_extensionsr@rNrr&r4r=r5urlr;disabledr:Fr?rr<Tr>r-r"r6Zprebuiltr9rArBrCbugsrF repository)r?rZ shadowed_extsrsource)r&r4r5r;r:r?r<r>r-r6r9rArBrCrFr#Zhomepage)r&r4r5r<r;r:r?r>r-r6rArBrCrFr)rrr1r r(r_normalize_namer3rr_get_companionrrr&rk_get_scheduled_uninstall_info) rrrrZbuild_check_inforr&datar-Zpkg_infonormalized_namepkgpackagesrrr_get_installed_extensionss                             z*ExtensionManager._get_installed_extensionsrcCs@d}d|dvrd|ddvrd}|Sd|ddvrd}|S)NZ discoveryrserverZkernelr)rrr9rrrrs zExtensionManager._get_companioncCsX|jdd|d}|r*| }t|WdS1s#wYdSdS)zDGet information about a package that is scheduled for uninstallationZstagingZ 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 extensionsNre)rrrrdrfrc)rrrrrfrrrrrs z(ExtensionManager._update_extensions_listrr)Nrer)rqN)T)#rGrHrIrJr rNrr{rpropertyr`rrKrr_r rr3rrOr"r# staticmethodrrr rrrrrrrrrrrrrr,sz   .   j   r)-rJrr dataclassesrrrrpathlibrtypingrrr r r r r rZ#jupyterlab_server.translation_utilsrZ traitletsrZtraitlets.configrrZjupyterlab.commandsrrrrrrr compiler'r1r3rOrUrYr`rcrgrrrrrs@ $   0 w