dH ddlZddlmZmZmZmZmZmZmZddl m Z ej dkrddl m Z nddlm Z ddlmZedZGdd eZGd d ZGd d eeZdZdS)N)AnyDictListOptionalGenericTypeVarcast) TracebackType)) entry_points)curry PluginTypeceZdZdZdZdS)NoSuchEntryPointc"||_||_dSN)groupname)selfrrs _active _active_name_plugins_optionsr5_global_settingscopy)rr=r>s rrzPluginRegistry.__init__NsP"3&   $ ? D D F Frrr.r*c||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)rBpop isinstancer>)rrr.s rregisterzPluginRegistry.registerasK$ ==$$T400 0eT%566 6 6 6"'DM$ Lrct|j}t|j}d|D}||t t|S)z:List the names of the registered and entry points plugins.cg|] }|j Sr r).0eps r z(PluginRegistry.names..~s000RW000r)listrBkeysimportlib_metadata_getr=extendsortedset)rextse_points more_extss rnameszPluginRegistry.nameszsfDM&&(()))$*@AA00x000  Ic$ii   rc|j|j|j|j|jdS)zBReturn a dictionary representing the current state of the registry)r@rArBrCrD)r@rArBrErCrDrs rr'zPluginRegistry._get_statesR| - **,, **,, $ 5 : : < <    rstateNct|hdksJ|D]\}}t|||dS)zReset the state of the registry>r@rCrBrArDN)rUrQitemssetattr)rr[keyvals rr1zPluginRegistry._set_statest5::<<  % % %       $ $HC D#s # # # # $ $rc |jvr fdt|jD\}nL#t$r?}|jvrt|j|t |j|d}~wwxYwt t|}| ||_ |j|_ t| t|j zD]}|||j|< ||_dS)Nc*g|]}|jk |Sr rL)rMrNrs rrOz*PluginRegistry._enable..s,w$&r)rBrRr= ValueErrorentrypoint_err_messagesrr rloadrIrAr@rUrQrDrGrC)rrr&rNerrr.r_s ` rr)zPluginRegistry._enablesR t} $ $ R4T5KLL  R R R4777$T%A$%GHHcQ*4+A4HHcQ  R RWWYY//E MM$ & & & }T* w||~~&&T-B-G-G-I-I)J)JJ : :C)0S)9)9D !# & & s"/ A8:A33A8c 0||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 )activer")rrr&s renablezPluginRegistry.enables(& <;DT4337333rc|jS)z.Return the name of the currently active plugin)rArs rrhzPluginRegistry.actives   rc|jS)z%Return the current options dictionary)rCrs rr&zPluginRegistry.optionss }rcL|jrt|jfi|jS|jS)z#Return the currently active plugin.)rCrr@rs rgetzPluginRegistry.gets. = 7777 7< rcd|jj|jt |S)Nz {}(active={!r}, registered={!r}))r4r5rrArPrYrs rr6zPluginRegistry.__repr__s94;; N #T%6TZZ\\8J8J   rr)rrrr7rdrDobjectr8r9rrrrIrrYrrr'r1r)r"ripropertyrhr&rmr6r rrr$r$5s$!02GG#GGGGG&S*)=(:BV2!tCy!!!! DcN     $S#X $4 $ $ $ $ C t    *448C=4}4444.!!!!X!c3hX Xj)     #      rr$ct}t|dr||S||gS)Nselect)r)r hasattrrrrm)rrNs rrRrRsG B r8!yyuy%%%vveR   r)systypingrrrrrrr typesr version_infoimportlib.metadatar importlib_metadatatoolzrrr:rr"r$rRr rrr{sa DDDDDDDDDDDDDDDDDDv/////////////W\ " " MMMMMyMMMUUUUUUUU8c c c c c WZ(c c c L ! ! ! ! !r