a 0hW@sPddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddlm Z ddl mZejrdd lmZdd lmZdd lmZed Zejd ejdejfdZdddddZGdddZGdddejeZGdddZGdddZGdddeZd d dd d!Z d d d"d#d$Z!Gd%d&d&ejeZ"dS)') annotationsN) ContextVarpartial)update_wrapper) attrgetter)ClosingIterator) StartResponse)WSGIApplication)WSGIEnvironmentTF.)boundzLocal | LocalStack[t.Any]None)localreturncCs |dS)zRelease the data for the current context in a :class:`Local` or :class:`LocalStack` without using a :class:`LocalManager`. This should not be needed for modern use cases, and may be removed in the future. .. versionadded:: 0.6.1 N)__release_local__)rr.storage_Local__storage)ridobject __setattr__selfrrrr__init__6szLocal.__init__zt.Iterator[tuple[str, t.Any]]rcCst|jiSN)iterrgetitemsr rrr__iter__@szLocal.__iter__unbound_messagestr str | NoneLocalProxy[t.Any]namer*rcCst|||dS)zCreate a :class:`LocalProxy` that access an attribute on this local namespace. :param name: Proxy this attribute. :param unbound_message: The error message that the proxy will show if the attribute isn't set. r) LocalProxyr r/r*rrr__call__Cs zLocal.__call__cCs|jidSr#)rsetr'rrrrOszLocal.__release_local__t.Any)r/rcCs(|ji}||vr||St|dSr#)rr%AttributeErrorr r/valuesrrr __getattr__Rs zLocal.__getattr__)r/valuercCs(|ji}|||<|j|dSr#)rr%copyr4)r r/r:r8rrrrZszLocal.__setattr__cCs<|ji}||vr0|}||=|j|nt|dSr#)rr%r;r4r6r7rrr __delattr___s  zLocal.__delattr__)N) __name__ __module__ __qualname____doc__ __slots__r!r(r3rr9rr<rrrrr#s  rc@s~eZdZdZdZddddddZdd d d Zd d dddZdd ddZe dd ddZ dddddddddZ dS) LocalStackaCreate a stack of context-local data. This wraps a :class:`ContextVar` containing a :class:`list` value. This may incur a performance penalty compared to using individual context vars, as it has to copy data to avoid mutating the list between nested contexts. :param context_var: The :class:`~contextvars.ContextVar` to use as storage for this local. If not given, one will be created. Context vars not created at the global scope may interfere with garbage collection. .. versionchanged:: 2.0 Uses ``ContextVar`` instead of a custom storage implementation. .. versionadded:: 0.6.1 )_storageNzContextVar[list[T]] | NonerrcCs&|durtdt|d}||_dS)Nzwerkzeug.LocalStackr?r@rAr!rrGrKpropertyrLr3rrrrrBjs   rBc@sbeZdZdZdZddddddZdd d d Zd d d ddZd d dddZdd ddZ dS) LocalManageraManage releasing the data for the current context in one or more :class:`Local` and :class:`LocalStack` objects. This should not be needed for modern use cases, and may be removed in the future. :param locals: A local or list of locals to manage. .. versionchanged:: 2.1 The ``ident_func`` was removed. .. versionchanged:: 0.7 The ``ident_func`` parameter was added. .. versionchanged:: 0.6.1 The :func:`release_local` function can be used instead of a manager. )localsNzJNone | (Local | LocalStack[t.Any] | t.Iterable[Local | LocalStack[t.Any]])r)rOrcCs2|durg|_nt|tr$|g|_n t||_dSr#)rO isinstancerlist)r rOrrrr!s   zLocalManager.__init__r"cCs|jD] }t|qdS)zRelease the data in the locals for this context. Call this at the end of each request or use :meth:`make_middleware`. N)rOr)r rrrrcleanups zLocalManager.cleanupr )apprcsddddfdd }|S)zWrap a WSGI application so that local data is released automatically after the response has been sent for a request. r r zt.Iterable[bytes])environstart_responsercst||jSr#)r rR)rTrUrSr rr applicationsz1LocalManager.make_middleware..applicationr)r rSrWrrVrmake_middlewareszLocalManager.make_middleware)funcrcCst|||S)zLike :meth:`make_middleware` but used as a decorator on the WSGI application function. .. code-block:: python @manager.middleware def application(environ, start_response): ... )rrX)r rYrrr middlewares zLocalManager.middlewarer+cCsdt|jdt|jdS)N)typer=rIrOr'rrr__repr__szLocalManager.__repr__)N) r=r>r?r@rAr!rRrXrZr^rrrrrNs   rNc@steZdZdZdZdddddd d d d Zd dd dddZdd dddddZddddZd ddddddZ dS) _ProxyLookupaDescriptor that handles proxied attribute lookup for :class:`LocalProxy`. :param f: The built-in function this attribute is accessed through. Instead of looking up the special method, the function call is redone on the object. :param fallback: Return this function if the proxy is unbound instead of raising a :exc:`RuntimeError`. :param is_attr: This proxied name is an attribute, not a function. Call the fallback immediately to get the value. :param class_value: Value to return when accessed from the ``LocalProxy`` class directly. Used for ``__doc__`` so building docs still works. )bind_ffallbackis_attr class_valuer/NFt.Callable[..., t.Any] | None-t.Callable[[LocalProxy[t.Any]], t.Any] | Nonez t.Any | Noneboolr)frarcrbrcsbtdr"ddddfdd }n$durBddddfdd }nd}||_||_||_||_dS) N__get__r-r5t.Callable[..., t.Any]instancerDrcs|t|Sr#rhr]rkrDrgrrr`sz%_ProxyLookup.__init__..bind_fcs t|Sr#rrmrnrrr`%s)hasattrr`rarcrb)r rgrarcrbr`rrnrr!s z_ProxyLookup.__init__r-r+)ownerr/rcCs ||_dSr#r/)r rpr/rrr __set_name__3sz_ProxyLookup.__set_name__z type | Noner5)rkrprcCs|dur|jdur|jS|Sz |}Wn@tyh|jdurB|j||}|jr`|YS|YS0|jdur|||St||jSr#) rc_get_current_object RuntimeErrorrarhrbr`getattrr/)r rkrprDrarrrrh6s         z_ProxyLookup.__get__r"cCs d|jS)Nzproxy rqr'rrrr^Qsz_ProxyLookup.__repr__)rkargskwargsrcOs||t||i|S)zSupport calling unbound methods from the class. For example, this happens with ``copy.copy``, which does ``type(x).__copy__(x)``. ``type(x)`` can't be proxied, so it returns the proxy type and descriptor. rl)r rkrvrwrrrr3Tsz_ProxyLookup.__call__)NNNF)N) r=r>r?r@rAr!rrrhr^r3rrrrr_s"r_cs0eZdZdZdZd ddddfdd ZZS) _ProxyIOpzLook up an augmented assignment method on a proxied object. The method is wrapped to return the proxy instead of the object. rNrdrer)rgrarcs.t|ddddfdd }||_dS)Nr-r5rirjcs(ddddfdd }||t|S)Nr5r-)r otherrcs||Sr#rr ry)rgrkrri_opns z0_ProxyIOp.__init__..bind_f..i_oprl)rkrDr{rn)rkrr`msz"_ProxyIOp.__init__..bind_f)superr!r`)r rgrar` __class__rnrr!fsz_ProxyIOp.__init__)NN)r=r>r?r@rAr! __classcell__rrr}rrx_s rx)oprcs"ddddfdd }tt|S)z5Swap the argument order to turn an l-op into an r-op.r5)rDryrcs ||Sr#r)rDryrrrr_opzsz_l_to_r_op..r_op)tcastr)rrrrr _l_to_r_opwsr)orcCs|Sr#r)rrrr _identitysrc@seZdZUdZdZded<ddddddd d d d Zeed dddZeddddZee dddZ ee Z ee ZeZeejZeejZeejZeejZeejZeejZeeZeedddZ ee!Z"ee#Z$ee%Z&ee'dddZ(eddddZ)eddZ*eddZ+eddZ,ee-Z.eej/Z0eej1Z2eej3Z4eej5Z6ee7Z8ee9Z:ee;Zeej?Z@eejAZBeejCZDeejEZFeejGZHeejIZJeejKZLeeMZNeeOZPeejQZReejSZTeejUZVeejWZXeejYZZee[ej?Z\ee[ejAZ]ee[ejCZ^ee[ejEZ_ee[ejGZ`ee[ejIZaee[ejKZbee[eMZcee[eOZdee[ejQZeee[ejSZfee[ejUZgee[ejWZhee[ejYZiejejkZlejejmZnejejoZpejejqZrejejsZtejejuZvejejwZxejejyZzejej{Z|ejej}Z~ejejZejejZejejZeejZeejZeeZeejZeeZeeZeeZeejZeeZeejZeejZeejZeZeZeZeZeZeZeZeejZeejZdS)r1as A proxy to the object bound to a context-local object. All operations on the proxy are forwarded to the bound object. If no object is bound, a ``RuntimeError`` is raised. :param local: The context-local object that provides the proxied object. :param name: Proxy this attribute from the proxied object. :param unbound_message: The error message to show if the context-local object is unbound. Proxy a :class:`~contextvars.ContextVar` to make it easier to access. Pass a name to proxy that attribute. .. code-block:: python _request_var = ContextVar("request") request = LocalProxy(_request_var) session = LocalProxy(_request_var, "session") Proxy an attribute on a :class:`Local` namespace by calling the local with the attribute name: .. code-block:: python data = Local() user = data("user") Proxy the top item on a :class:`LocalStack` by calling the local. Pass a name to proxy that attribute. .. code-block:: app_stack = LocalStack() current_app = app_stack() g = app_stack("g") Pass a function to proxy the return value from that function. This was previously used to access attributes of local objects before that was supported directly. .. code-block:: python session = LocalProxy(lambda: request.session) ``__repr__`` and ``__class__`` are proxied, so ``repr(x)`` and ``isinstance(x, cls)`` will look like the proxied object. Use ``issubclass(type(x), LocalProxy)`` to check if an object is a proxy. .. code-block:: python repr(user) # isinstance(user, User) # True issubclass(type(user), LocalProxy) # True .. versionchanged:: 2.2.2 ``__wrapped__`` is set when wrapping an object, not only when wrapping a function, to prevent doctest from failing. .. versionchanged:: 2.2 Can proxy a ``ContextVar`` or ``LocalStack`` directly. .. versionchanged:: 2.2 The ``name`` parameter can be used with any proxied object, not only ``Local``. .. versionchanged:: 2.2 Added the ``unbound_message`` parameter. .. versionchanged:: 2.0 Updated proxied attributes and methods to reflect the current data model. .. versionchanged:: 0.6.1 The class can be instantiated with a callable. )Z __wrappedrszt.Callable[[], T]rsNr)z9ContextVar[T] | Local | LocalStack[T] | t.Callable[[], T]r,r)rr/r*rcs|durtnt|dur"dttrT|dur._get_current_objectcsj}|durt|Sr#)rLrtrDrrrrsscs2z }Wnty(tdYn0|Sr#)r% LookupErrorrtrrrrrss   cs Sr#rr)rrrrrs szDon't know how to proxy 'z'._LocalProxy__wrappedrs) rrrPr TypeErrorrBrcallabler]rr)r rr/r*rsrrrr!s$   zLocalProxy.__init__cCs t|jSr#)r]r@r'rrrzLocalProxy.T)rcrarbcCs|jSr#)rr'rrrrr)rarbcCsdt|jdS)Nr[z unbound>)r]r=r'rrrrr)racCsdS)NFrr'rrrr+rcCsgSr#rr'rrrr0rcCst|Sr#)r]r'rrrr;rcCs t||Sr#)rPrzrrrr<rcCs t||Sr#) issubclassrzrrrr=rcOs||i|Sr#r)r rvrwrrrr?r)N)r=r>r?r@rA__annotations__r!r_ __wrapped__reprr^r+__str__bytes __bytes__ __format__operatorlt__lt__le__le__eq__eq__ne__ne__gt__gt__ge__ge__hash__hash__rf__bool__rur9setattrrdelattrr<dir__dir__r~__instancecheck____subclasscheck__r3rI__len__ length_hint__length_hint__getitem __getitem__setitem __setitem__delitem __delitem__r$r(next__next__reversed __reversed__contains __contains__add__add__sub__sub__mul__mul__matmul __matmul__truediv __truediv__floordiv __floordiv__mod__mod__divmod __divmod__pow__pow__lshift __lshift__rshift __rshift__and___and__xor__xor__or___or__r__radd____rsub____rmul__ __rmatmul__ __rtruediv__ __rfloordiv____rmod__ __rdivmod____rpow__ __rlshift__ __rrshift____rand____rxor____ror__rxiadd__iadd__isub__isub__imul__imul__imatmul __imatmul__itruediv __itruediv__ ifloordiv __ifloordiv__imod__imod__ipow__ipow__ilshift __ilshift__irshift __irshift__iand__iand__ixor__ixor__ior__ior__neg__neg__pos__pos__abs__abs__invert __invert__complex __complex__int__int__float __float__index __index__round __round__mathtrunc __trunc__floor __floor__ceil__ceil__ __enter____exit__ __await__ __aiter__ __anext__ __aenter__ __aexit__r;__copy__deepcopy __deepcopy__rrrrr1s M 8                                                   r1)# __future__rr;rrtypingrZ contextvarsr functoolsrrrZwsgir TYPE_CHECKINGZ_typeshed.wsgir r r TypeVarr CallableAnyrrrGenericrBrNr_rxrrr1rrrrs0           GPE`