B 9dh*@sddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZd Zd Zd d Zd"ddZd#ddZd$ddZd%ddZGdddeZGdddeZGdddeZGdddeZGdd d eZejZed!kredS)&N)BaseExtensionApp_get_config_dir GREEN_ENABLED RED_DISABLED) __version__)BaseJSONConfigManager)jupyter_config_path) import_item)BoolnotebookbundlerextensionscCs2t|}t|ds"td|d|}||fS)aGets the list of bundlers associated with a Python package. Returns a tuple of (the module, [{ 'name': 'unique name of the bundler', 'label': 'file menu item label for the bundler', 'module_name': 'dotted package/module name containing the bundler', 'group': 'download or deploy parent menu item' }]) Parameters ---------- module : str Importable Python module exposing the magic-named `_jupyter_bundlerextension_paths` function _jupyter_bundlerextension_pathszThe Python module z* does not contain a valid bundlerextension)r hasattrKeyErrorr)modulembundlersro/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/notebook/bundler/bundlerextensions.py_get_bundler_metadatas  rTFc Cs|rdn|}tjt||dd}t|d} |rT||rSet whether a bundler is enabled or disabled. Returns True if the final state is the one requested. Parameters ---------- name : string Unique name of the bundler label : string Human-readable label for the bundler menu item in the notebook UI module_name : string Dotted module/package name containing the bundler group : string 'download' or 'deploy' indicating the parent menu containing the label state : bool The state in which to leave the extension user : bool [default: True] Whether to update the user's .jupyter/nbconfig directory sys_prefix : bool [default: False] Whether to update the sys.prefix, i.e. environment. Will override `user`. logger : Jupyter logger [optional] Logger instance to use F)user sys_prefixnbconfig) config_dirZEnablingZ Disabling z bundler z...)label module_namegroupN) ospathjoinrrinfoupdateBUNDLER_SECTIONBUNDLER_SUBSECTIONget) namerrrstaterrloggerrcmrrr_set_bundler_state)s&  $  r+cs$t|\}}fdd|DS)aHEnables or disables bundlers defined in a Python package. Returns a list of whether the state was achieved for each bundler. Parameters ---------- state : Bool Whether the extensions should be enabled module : str Importable Python module exposing the magic-named `_jupyter_bundlerextension_paths` function user : bool Whether to enable in the user's nbconfig directory. sys_prefix : bool Enable/disable in the sys.prefix, i.e. environment logger : Jupyter logger [optional] Logger instance to use c s4g|],}t|d|d|d|ddqS)r'rrr)r'rrrr(rrr))r+).0Zbundler)r)r(rrrr usz-_set_bundler_state_python..)r)r(rrrr)rrr)r)r(rrr_set_bundler_state_pythonas r.cCstd||||dS)aAEnables bundlers defined in a Python package. Returns whether each bundle defined in the packaged was enabled or not. Parameters ---------- module : str Importable Python module exposing the magic-named `_jupyter_bundlerextension_paths` function user : bool [default: True] Whether to enable in the user's nbconfig directory. sys_prefix : bool [default: False] Whether to enable in the sys.prefix, i.e. environment. Will override `user` logger : Jupyter logger [optional] Logger instance to use T)r))r.)rrrr)rrrenable_bundler_python~s r/cCstd||||dS)aBDisables bundlers defined in a Python package. Returns whether each bundle defined in the packaged was enabled or not. Parameters ---------- module : str Importable Python module exposing the magic-named `_jupyter_bundlerextension_paths` function user : bool [default: True] Whether to enable in the user's nbconfig directory. sys_prefix : bool [default: False] Whether to enable in the sys.prefix, i.e. environment. Will override `user` logger : Jupyter logger [optional] Logger instance to use F)r))r.)rrrr)rrrdisable_bundler_pythons r0c@sFeZdZdZdZeZdZeddddZ dZ dd Z d d Z d d Z dS)ToggleBundlerExtensionAppz;A base class for apps that enable/disable bundlerextensionsz'jupyter bundlerextension enable/disablez3Enable/disable a bundlerextension in configuration.Tz;Apply the configuration only for the current user (default))confighelpNcCsdS)zThe default config file name.Zjupyter_notebook_configr)selfrrr_config_file_name_defaultsz3ToggleBundlerExtensionApp._config_file_name_defaultcCs$|jr tnt}|||j|j|jdS)aSToggle some extensions in an importable Python module. Returns a list of booleans indicating whether the state was changed as requested. Parameters ---------- module : str Importable Python module exposing the magic-named `_jupyter_bundlerextension_paths` function )rrr)) _toggle_valuer/r0rrlog)r4rtogglerrrtoggle_bundler_pythons z/ToggleBundlerExtensionApp.toggle_bundler_pythoncCsN|jstdnt|jdkr*td|jrB||jdntddS)Nz?Please specify an bundlerextension/package to enable or disablez5Please specify one bundlerextension/package at a timerz0Cannot install bundlers from non-Python packages) extra_argssysexitlenpythonr9NotImplementedError)r4rrrstarts  zToggleBundlerExtensionApp.start)__name__ __module__ __qualname____doc__r'rversion descriptionr rr6r5r9rArrrrr1sr1c@seZdZdZdZdZdZdS)EnableBundlerExtensionAppz%An App that enables bundlerextensionszjupyter bundlerextension enablez Enable a bundlerextension in frontend configuration. Usage jupyter bundlerextension enable [--system|--sys-prefix] TN)rBrCrDrEr'rGr6rrrrrHsrHc@seZdZdZdZdZdZdS)DisableBundlerExtensionAppz&An App that disables bundlerextensionsz jupyter bundlerextension disablez Disable a bundlerextension in frontend configuration. Usage jupyter bundlerextension disable [--system|--sys-prefix] N)rBrCrDrEr'rGr6rrrrrIsrIc@s,eZdZdZdZeZdZddZddZ dS) ListBundlerExtensionAppz,An App that lists and validates nbextensionszjupyter nbextension listz7List all nbextensions known by the configuration systemc CsddtD}tdx|D]}d|}d}t||d}|d}d|kr|s^t|d }xl|dD]\\}}|d } |d } | d ks| d krd |dt} nd| d| dt} t| qlWqWd S)zList all the nbextensionscSsg|]}tj|dqS)r)rr r!)r,prrrr-sz=ListBundlerExtensionApp.list_nbextensions..zKnown bundlerextensions:z config dir: F)parentrr r TrrNz rz "z" from )r printrr&itemsrr) r4Z config_dirsrheadZ head_shownr*dataZ bundler_idr"rrmsgrrrlist_nbextensionss$      z)ListBundlerExtensionApp.list_nbextensionscCs |dS)z)Perform the App's functions as configuredN)rR)r4rrrrA szListBundlerExtensionApp.startN) rBrCrDrEr'rrFrGrRrArrrrrJs rJcsJeZdZdZdZeZdZdZe e dfe dfe dfdZ fd d ZZS) BundlerExtensionAppz1Base jupyter bundlerextension command entry pointzjupyter bundlerextensionz$Work with Jupyter bundler extensionsa jupyter bundlerextension list # list all configured bundlers jupyter bundlerextension enable --py # enable all bundlers in a Python package jupyter bundlerextension disable --py # disable all bundlers in a Python package zEnable a bundler extensionzDisable a bundler extensionzList bundler extensions)enabledisablelistcs.tdt|j}td|dS)z)Perform the App's functions as configuredz, z'Please supply at least one subcommand: N)superrAr!sorted subcommandsr<r=)r4Zsubcmds) __class__rrrA!s zBundlerExtensionApp.start)rBrCrDrEr'rrFrGZexamplesdictrHrIrJrYrA __classcell__rr)rZrrSs rS__main__)TFN)N)TFN)TFN) r<r extensionsrrrr_versionrZnotebook.config_managerrZjupyter_core.pathsr Ztraitlets.utils.importstringr Z traitletsr r$r%rr+r.r/r0r1rHrIrJrSZlaunch_instancemainrBrrrrs,      7   +  &