]gSrSSKJr SSKJr SSKJr SSKJrJ r SSK J r SSK J r \"SS 55r\"S 5rS rS rS r\SSj5rSSjrg)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)Setc6\rSrSr%Sr\"\S9r\\ \ S'Sr g) ClientContextz 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____firstlineno____doc__rsetrr str__annotations____static_attributes__r2/usr/lib/python3/dist-packages/botocore/context.pyr r ss3Hc#h3rr _contextc,[RS5$)z@Get the current ``_context`` context variable if set, else None.N)rgetrrr get_contextr*s << rc0[RU5nU$)aSet 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.[RU5 g)zReset the current ``_context`` context variable. :type token: contextvars.Token :param token: Token object to reset the context variable. N)rreset)r"s r reset_contextr&=s  NN5rNc## U=(d [5nUc [5nO [U5n[U5nSv [ U5 g![ U5 f=f7f)aO 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 rr#r&)r!currentnewr"s rstart_as_current_contextr*FsK&"[]Gow  E e es8AA A AAc^U4SjnU$)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. c4>^[T5UU4Sj5nU$)Ncz>[5 T(aT"5 T"U0UD6sSSS5 $!,(df  g=fN)r*)argskwargsfunchooks rwrapper8with_current_context..decorator..wrapperws*)+FT,V,,++s, :r)r1r3r2s` r decorator'with_current_context..decoratorvs t -  - rr)r2r5s` rwith_current_contextr7es" rr.)r contextlibr contextvarsrcopyr dataclassesrr functoolsr typingr r rrr#r&r*r7rrrr>sk &"(  4 4  4 j ! <r