o e@sddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z edZGdddeZGdd d ZGd d d eeZd d ZdS))AnyDictListOptionalGenericTypeVarcast) TracebackType) entry_points)curry PluginTypec@seZdZddZddZdS)NoSuchEntryPointcCs||_||_dSN)groupname)selfrrrLD:\Projects\ConvertPro\env\Lib\site-packages\altair/utils/plugin_registry.py__init__ s zNoSuchEntryPoint.__init__cCsd|jd|jS)NzNo z entry point found in group )rrrrrr__str__zNoSuchEntryPoint.__str__N)__name__ __module__ __qualname__rrrrrrr s r c@sTeZdZdZdddefddZddd Zd ed ed e dd fddZ defddZ d S) PluginEnableraContext manager for enabling plugins This object lets you use enable() as a context manager to temporarily enable a given plugin:: with plugins.enable('name'): do_something() # 'name' plugin temporarily enabled # plugins back to original state registryPluginRegistryrcKs4||_||_||_||_|jj|fi|dSr)rroptions _get_stateoriginal_state_enable)rrrrrrrr s  zPluginEnabler.__init__returncCs|Srrrrrr __enter__'szPluginEnabler.__enter__typvalue tracebackNcCs|j|jdSr)r _set_stater )rr$r%r&rrr__exit__*szPluginEnabler.__exit__cCsd|jjj|jS)Nz{}.enable({!r}))formatr __class__rrrrrr__repr__-rzPluginEnabler.__repr__)r"r) rrr__doc__strrr#type Exceptionr r(r+rrrrrs  rc@seZdZdZiZiZdefdedefddZ dede e d e e fd d Z d e efd d Zd eeeffddZdeeefd dfddZded dfddZd de ed efddZed efddZed eeeffddZd e e fddZd efddZdS)!raA registry for plugins. This is a plugin registry that allows plugins to be loaded/registered in two ways: 1. Through an explicit call to ``.register(name, value)``. 2. By looking for other Python packages that are installed and provide a setuptools entry point group. When you create an instance of this class, provide the name of the entry point group to use:: reg = PluginRegister('my_entrypoint_group') entry_point_group plugin_typecCs6||_||_d|_d|_i|_i|_|jj|_dS)aJCreate a PluginRegistry for a named entry point group. Parameters ========== entry_point_group: str The name of the entry point group. plugin_type: object A type that will optionally be used for runtime type checking of loaded plugins using isinstance. Nr0) r1r2_active _active_name_plugins_optionsr*_global_settingscopy)rr1r2rrrrJs zPluginRegistry.__init__rr%r"cCs4|dur |j|dSt||jsJ||j|<|S)aRegister a plugin by name and value. This method is used for explicit registration of a plugin and shouldn't be used to manage entry point managed plugins, which are auto-loaded. Parameters ========== name: str The name of the plugin. value: PluginType or None The actual plugin object to register or None to unregister that plugin. Returns ======= plugin: PluginType or None The plugin that was registered or unregistered. N)r5pop isinstancer2)rrr%rrrregister]s  zPluginRegistry.registercCs<t|j}t|j}dd|D}||tt|S)z:List the names of the registered and entry points plugins.cSsg|]}|jqSrr.0eprrr zsz(PluginRegistry.names..)listr5keysimportlib_metadata_getr1extendsortedset)rextsZe_pointsZ more_extsrrrnamesvs    zPluginRegistry.namescCs&|j|j|j|j|jdS)zBReturn a dictionary representing the current state of the registry)r3r4r5r6r7)r3r4r5r8r6r7rrrrr~s zPluginRegistry._get_statestateNcCs:t|hdks J|D] \}}t|||qdS)zReset the state of the registry>r7r3r5r4r6N)rFrBitemssetattr)rrIkeyvalrrrr'szPluginRegistry._set_statec s|jvrBzfddt|jD\}Wn ty4}z|jvr)t|j|t|j|d}~wwtt|}| ||_ |j|_ t | t |j @D] }|||j|<qX||_dS)Ncsg|] }|jkr|qSrr<r=r<rrr@s  z*PluginRegistry._enable..)r5rCr1 ValueErrorentrypoint_err_messagesr rr loadr;r4r3rFrBr7r9r6)rrrr?errr%rLrr<rr!s$       zPluginRegistry._enablecKs |dur|j}t||fi|S)aEnable a plugin by name. This can be either called directly, or used as a context manager. Parameters ---------- name : string (optional) The name of the plugin to enable. If not specified, then use the current active name. **options : Any additional parameters will be passed to the plugin as keyword arguments Returns ------- PluginEnabler: An object that allows enable() to be used as a context manager N)activer)rrrrrrenableszPluginRegistry.enablecC|jS)z.Return the name of the currently active plugin)r4rrrrrRzPluginRegistry.activecCrT)z%Return the current options dictionary)r6rrrrrrUzPluginRegistry.optionscCs |jr t|jfi|jS|jS)z#Return the currently active plugin.)r6r r3rrrrgetszPluginRegistry.getcCsd|jj|jt|S)Nz {}(active={!r}, registered={!r}))r)r*rr4rArHrrrrr+szPluginRegistry.__repr__r)rrrr,rOr7objectr-r.rrr r;rrHrrrr'r!rrSpropertyrRrrVr+rrrrr1s"  rcCs(t}t|dr|j|dS||gS)Nselect)r)r hasattrrYrV)rr?rrrrCs   rCN)typingrrrrrrrtypesr importlib.metadatar Ztoolzr r r/r rrrCrrrrs$     '