U EZh@sdZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z eGdd d Zed Zd d Zd dZddZedddZdddZdS)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) dataclassfieldwraps)Setc@s(eZdZUdZeedZeee d<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 r str__annotations__rrW/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/botocore/context.pyr s r _contextcCs tdS)z@Get the current ``_context`` context variable if set, else None.N)rgetrrrr get_context*srcCst|}|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=srNccsD|pt}|dkrt}nt|}t|}z dVW5t|XdS)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_contextFs  r csfdd}|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)Nc s2t"r||W5QRSQRXdS)N)r )argskwargs)funchookrrwrapperwsz8with_current_context..decorator..wrapperr)r#r%r$)r#r decoratorvsz'with_current_context..decoratorr)r$r'rr&rwith_current_contextes r()N)N)r contextlibrZ contextvarsrcopyr dataclassesrr functoolsrtypingr r rrrrr r(rrrr s