o Jh@sdZddlmZddlmZddlmZddlmZm Z ddl m Z eGdddZ ed Z d d Zd d ZddZedddZdddZdS)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) dataclassfieldwrapsc@s(eZdZUdZeedZeeed<dS) 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__rrp/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/botocore/context.pyr s r _contextcCs tdS)z@Get the current ``_context`` context variable if set, else None.N)rgetrrrr get_context)s rcCst|}|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)ctxtokenrrr set_context.s rcCst|dS)zReset the current ``_context`` context variable. :type token: contextvars.Token :param token: Token object to reset the context variable. N)rreset)rrrr reset_context<srNccsL|pt}|durt}nt|}t|}z dVWt|dSt|w)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)rcurrentnewrrrrstart_as_current_contextEs rcsfdd}|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. cstfdd}|S)NcsBtr |i|WdS1swYdSN)r)argskwargs)funchookrrwrappervs  $z8with_current_context..decorator..wrapperr)r#r%r$)r#r decoratorusz'with_current_context..decoratorr)r$r'rr&rwith_current_contextds  r(r )r contextlibr contextvarsrcopyr dataclassesrr functoolsrr rrrrrr(rrrrs