o h'@sTddlmZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZejdkr8ddlmZnddlmZejd krJdd lmZndd lmZerXdd lmZe d Ze d Zede defefdZe dee dZe egeee fZd!ddZGdddeZGddde eefZGddde eefZ dd Z!dS)") annotationsN)partial) entry_points) TYPE_CHECKINGAnyCallableGenericTypeVarcast)deprecated_warn) )TypeIs)r ) TypeAliasType) TracebackTypeTRPlugin.) type_paramsPluginT)boundtptype[T]return#Callable[[object], TypeIs[type[T]]]csdfdd }|S) zq Converts a type to guard function. Added for compatibility with original `PluginRegistry` default. objobjectrTypeIs[type[T]]cs t|SN) isinstance)rrP/var/www/vscode/kcb/lib/python3.10/site-packages/altair/utils/plugin_registry.pyfunc$s z_is_type..funcN)rrrrr")rr$r"r!r#_is_typesr%c@seZdZddZddZdS)NoSuchEntryPointcCs||_||_dSr)groupname)selfr'r(r"r"r#__init__+s zNoSuchEntryPoint.__init__cCsd|jd|jS)NzNo z entry point found in group )r(r'r)r"r"r#__str__/szNoSuchEntryPoint.__str__N)__name__ __module__ __qualname__r*r,r"r"r"r#r&*s r&c@s8eZdZdZdd d Zdd dZdddZdddZdS) PluginEnablera% Context 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 registryPluginRegistry[PluginT, R]r(stroptionsrrNonecKs4||_||_||_||_|jj|fi|dSr)r1r(r4 _get_stateoriginal_state_enable)r)r1r(r4r"r"r#r*?s  zPluginEnabler.__init__PluginEnabler[PluginT, R]cCs|Srr"r+r"r"r# __enter__HszPluginEnabler.__enter__typtypevalue Exception tracebackrcCs|j|jdSr)r1 _set_stater7)r)r;r=r?r"r"r#__exit__KszPluginEnabler.__exit__cCst|jjd|jdS)Nz.enable())r<r1r-r(r+r"r"r#__repr__NszPluginEnabler.__repr__N)r1r2r(r3r4rrr5)rr9)r;r<r=r>r?rrr5rr3)r-r.r/__doc__r*r:rArCr"r"r"r#r03s  r0c@seZdZUdZiZded<iZded<defd.d dZd/ddZ d0ddZ d1ddZ d2ddZ d3ddZ d4d5d#d$Zed6d%d&Zed1d'd(Zd7d*d+Zd6d,d-ZdS)8PluginRegistrya A 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") zdict[str, str]entrypoint_err_messagesdict[str, Any]_global_settingsentry_point_groupr3 plugin_typeIsPluginrr5cCs|||_||tur&t|tr&dt|jd}t|ddttt||_ n||_ d|_ d|_ i|_ i|_ |jj|_dS)a Create a PluginRegistry for a named entry point group. Parameters ---------- entry_point_group: str The name of the entry point group. plugin_type A type narrowing function that will optionally be used for runtime type checking loaded plugins. References ---------- https://typing.readthedocs.io/en/latest/spec/narrowing.html zsz(PluginRegistry.names..)listrRkeysimportlib_metadata_getrKextendsortedset)r)extse_points more_extsr"r"r#namess    zPluginRegistry.namescCs&|j|j|j|j|jdS)zCReturn a dictionary representing the current state of the registry.)rPrQrRrSrI)rPrQrRrUrSrIr+r"r"r#r6s zPluginRegistry._get_statestatecCs:t|hdks J|D] \}}t|||qdS)z Reset the state of the registry.>rPrSrRrQrIN)rfrbitemssetattr)r)rkkeyvalr"r"r#r@szPluginRegistry._set_statec s|jvrBzfddt|jD\}Wn ty4}z|jvr)t|j|t|j|d}~wwtt|}| ||_ |j|_ t | t |j @D] }|||j|<qX||_dS)Nc3s|] }|jkr|VqdSrr\r]r\r"r# s z)PluginRegistry._enable..)rRrcrK ValueErrorrGr&r rloadrZrQrPrfrbrIrXrS)r)r(r4r_errr=rnr"r\r#r8s$       zPluginRegistry._enableN str | Noner4rr9cKs |dur|j}t||fi|S)a& Enable 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)activer0)r)r(r4r"r"r#enableszPluginRegistry.enablecC|jS)z/Return the name of the currently active plugin.)rQr+r"r"r#ruzPluginRegistry.activecCrw)z&Return the current options dictionary.)rSr+r"r"r#r4rxzPluginRegistry.optionspartial[R] | Plugin[R] | NonecCsl|j}r||r|jrt|fi|jS|S|jdur0t|jdt|jjd}t|tr4tdS)z#Return the currently active plugin.Nz2 requires all plugins to be callable objects, but z is not callable.) rPrLrSrr<r-rYrNotImplementedError)r)r$rVr"r"r#gets   zPluginRegistry.getcCs"t|jd|jd|dS)Nz(active=z , registered=rB)r<r-rurjr+r"r"r#rCs"zPluginRegistry.__repr__)rKr3rLrMrr5)r(r3r=rWrrW)rr[)rrH)rkrHrr5)r(r3rr5r)r(rtr4rrr9rD)rry)r-r.r/rErG__annotations__rIrOr*rZrjr6r@r8rvpropertyrur4r{rCr"r"r"r#rFRs&    &       rFcCs(t}t|dr|j|dS||gS)Nselect)r')rhasattrr~r{)r'r_r"r"r#rcs   rc)rrrr)" __future__rsys functoolsrimportlib.metadatartypingrrrrr r altair.utils.deprecationr version_infortyping_extensionsrtypesrrrrrrrMr%r>r&r0rFrcr"r"r"r#s2              H