B 3\+@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)BoolnotebookbundlerextensionscCs0t|}t|ds td||}||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_pathsz>The Python module {} does not contain a valid bundlerextension)r hasattrKeyErrorformatr)modulembundlersrAlib/python3.7/site-packages/notebook/bundler/bundlerextensions.py_get_bundler_metadatas  rTFc Cs|rdn|}tjt||dd}t|d} |rL|d|r@dnd|||rn| tt ||||diin| tt |d ii| t t i |d k |kS) aFSet 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_dirz{} {} bundler {}...ZEnablingZ Disabling)label module_namegroupN) ospathjoinrrinforupdateBUNDLER_SECTIONBUNDLER_SUBSECTIONget) namerrrstaterrloggerrcmrrr_set_bundler_state)s,     r+cs$t|\}}fdd|DS)aPEnables 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 ysz-_set_bundler_state_python..)r)r(rrrr)rrr)r)r(rrr_set_bundler_state_pythones r.cCstd||||dS)aIEnables 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_pythons r/cCstd||||dS)aJDisables 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))ZconfighelpNcCsdS)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)r3rZtogglerrrtoggle_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)Z extra_argssysexitlenZpythonr7NotImplementedError)r3rrrstarts  zToggleBundlerExtensionApp.start)__name__ __module__ __qualname____doc__r'rversion descriptionr rr5r4r7r=rrrrr1sr1c@seZdZdZdZdZdZdS)EnableBundlerExtensionAppz%An App that enables bundlerextensionszjupyter bundlerextension enablez Enable a bundlerextension in frontend configuration. Usage jupyter bundlerextension enable [--system|--sys-prefix] TN)r>r?r@rAr'rCr5rrrrrDsrDc@seZdZdZdZdZdZdS)DisableBundlerExtensionAppz&An App that disables bundlerextensionsz jupyter bundlerextension disablez Disable a bundlerextension in frontend configuration. Usage jupyter bundlerextension disable [--system|--sys-prefix] N)r>r?r@rAr'rCr5rrrrrEsrEc@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 }x`|dD]P\}}|d } |d } | d ks| d krd |t} nd| | t} 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 {} {}z "{}" from {} {})r printrrr&itemsrr) r3Z config_dirsrheadZ head_shownr*dataZ bundler_idr"rrmsgrrrlist_nbextensionss&       z)ListBundlerExtensionApp.list_nbextensionscCs |dS)z)Perform the App's functions as configuredN)rN)r3rrrr=szListBundlerExtensionApp.startN) r>r?r@rAr'rrBrCrNr=rrrrrFs rFcsJeZdZdZdZeZdZdZe 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)enabledisablelistcs0tt|dt|j}td|dS)z)Perform the App's functions as configuredz, z)Please supply at least one subcommand: %sN)superrOr=r!sorted subcommandsr9r:)r3Zsubcmds) __class__rrr='szBundlerExtensionApp.start)r>r?r@rAr'rrBrCZexamplesdictrDrErFrUr= __classcell__rr)rVrrOs rO__main__)TFN)N)TFN)TFN) r9r extensionsrrrrZ_versionrZnotebook.config_managerrZjupyter_core.pathsr Ztraitlets.utils.importstringr Z traitletsr r$r%rr+r.r/r0r1rDrErFrOZlaunch_instancemainr>rrrrs,      ;   +  (