a  d@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)selfrrre/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/altair/utils/plugin_registry.py__init__ szNoSuchEntryPoint.__init__cCsd|jd|jS)NzNo z entry point found in group )rrrrrr__str__szNoSuchEntryPoint.__str__N)__name__ __module__ __qualname__rrrrrrr sr c@sPeZdZdZdedddZddddZeee d d d d Z edd dZ 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 PluginRegistry)registryrcKs4||_||_||_||_|jj|fi|dSr)rroptions _get_stateoriginal_state_enable)rrrrrrrr s  zPluginEnabler.__init__returncCs|Srrrrrr __enter__'szPluginEnabler.__enter__N)typvalue tracebackr"cCs|j|jdSr)r _set_stater)rr$r%r&rrr__exit__*szPluginEnabler.__exit__cCsd|jjj|jS)Nz{}.enable({!r}))formatr __class__rrrrrr__repr__-szPluginEnabler.__repr__) rrr__doc__strrr#type Exceptionr r(r+rrrrrs  rc@seZdZdZiZiZdefeedddZ ee e e e dddZ e ed d d Zeeefd d d ZeeefddddZeddddZde eedddZeed ddZeeeefd ddZe e d ddZed ddZdS) 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"cCs8|dur|j|dSt||js&J||j|<|SdS)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.registerr!cCs<t|j}t|j}dd|D}||tt|S)z:List the names of the registered and entry points plugins.cSsg|] }|jqSrr.0eprrr zz(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_stateN)stater"cCs:t|hdksJ|D]\}}t|||q dS)zReset the state of the registry>r6r7r3r5r4N)rGrCitemssetattr)rrJkeyvalrrrr'szPluginRegistry._set_state)rr"c s|jvrzfddt|jD\}WnNtyv}z6|jvrTt|j|nt|j|WYd}~n d}~00tt|}| ||_ |j|_ t | t |j @D]}|||j|<q||_dS)Ncsg|]}|jkr|qSrr<r=r<rrr@s z*PluginRegistry._enable..)r5rDr1 ValueErrorentrypoint_err_messagesr rr loadr;r4r3rGrCr7r9r6)rrrr?errr%rMrr<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.enablecCs|jS)z.Return the name of the currently active plugin)r4rrrrrSszPluginRegistry.activecCs|jS)z%Return the current options dictionary)r6rrrrrszPluginRegistry.optionscCs$|jrt|jfi|jS|jSdS)z#Return the currently active plugin.N)r6r r3rrrrgetszPluginRegistry.getcCsd|jj|jt|S)Nz {}(active={!r}, registered={!r}))r)r*rr4rBrIrrrrr+szPluginRegistry.__repr__)N)rrrr,rPr7objectr-r.rrr r;rrIrrrr'r rrTpropertyrSrrUr+rrrrr1s   rcCs,t}t|dr|j|dS||gSdS)Nselect)r)r hasattrrXrU)rr?rrrrDs  rDN)typingrrrrrrrtypesr importlib.metadatar Ztoolzr r r/r rrrDrrrrs$    '