hdZddlmZddlmZddlmZddlmZm Z ddl m Z eGddZ ed Z d Zd Zd ZeddZddZy )z NOTE: All classes and functions in this module are considered private and are subject to abrupt breaking changes. Please do not use them directly. )contextmanager) ContextVar)deepcopy) dataclassfieldwrapsc4eZdZUdZeeZeeed<y) ClientContexta  Encapsulation of objects tracked within the ``_context`` context variable. ``features`` is a set responsible for storing features used during preparation of an AWS request. ``botocore.useragent.register_feature_id`` is used to add to this set. )default_factoryfeaturesN) __name__ __module__ __qualname____doc__rsetr str__annotations__l/home/developers/rajanand/mypropertyqr-fmb-refixing-v2/venv/lib/python3.12/site-packages/botocore/context.pyr r ss3Hc#h3rr _contextc,tjdS)z@Get the current ``_context`` context variable if set, else None.N)rgetrrr get_contextr)s << rc0tj|}|S)a2Set the current ``_context`` context variable. :type ctx: ClientContext :param ctx: Client context object to set as the current context variable. :rtype: contextvars.Token :returns: Token object used to revert the context variable to what it was before the corresponding set. )rr)ctxtokens r set_contextr.s LL E Lrc.tj|y)zReset the current ``_context`` context variable. :type token: contextvars.Token :param token: Token object to reset the context variable. N)rreset)rs r reset_contextr"<s  NN5rNc#K|xs t}| t}n t|}t|} dt |y#t |wxYww)a Context manager that copies the passed or current context object and sets it as the current context variable. If no context is found, a new ``ClientContext`` object is created. It mainly ensures the context variable is reset to the previous value once the executed code returns. Example usage: def my_feature(): with start_as_current_context(): register_feature_id('MY_FEATURE') pass :type ctx: ClientContext :param ctx: The client context object to set as the new context variable. If not provided, the current or a new context variable is used. N)rr rrr")rcurrentnewrs rstart_as_current_contextr&EsL&"[]Gow  E e es2AA A AAcfd}|S)a Decorator that wraps ``start_as_current_context`` and optionally invokes a hook within the newly-set context. This is just syntactic sugar to avoid indenting existing code under the context manager. Example usage: @with_current_context(partial(register_feature_id, 'MY_FEATURE')) def my_feature(): pass :type hook: callable :param hook: A callable that will be invoked within the scope of the ``start_as_current_context`` context manager. c2tfd}|S)Ncft5r|i|cdddS#1swYyxYwN)r&)argskwargsfunchooks rwrapperz8with_current_context..decorator..wrappervs3)+ -FT,V, - - -s'0r)r-r/r.s` r decoratorz'with_current_context..decoratorus t -  - rr)r.r0s` rwith_current_contextr1ds" rr*)r contextlibr contextvarsrcopyr dataclassesrr functoolsr r rrrr"r&r1rrrr7sh &"(  4 4  4 j ! <r