tnidZddlmZddlmZddlmZddlmZm Z ddl m Z eGddZ ed Z d Zd Zd ZeddZddZd S)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) dataclassfieldwrapscBeZdZUdZeeZeeed<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__M/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/botocore/context.pyr r s?s333Hc#h33333rr _contextc6tdS)z@Get the current ``_context`` context variable if set, else None.N)rgetrrr get_contextr)s <<  rc:t|}|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:t|dS)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|p t}|t}nt|}t|} dVt |dS#t |wxYw)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&Esv&"[]]Goow   E e es AA)cfd}|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. c@tfd}|S)Nc|t5r |i|cdddS#1swxYwYdSN)r&)argskwargsfunchooks rwrapperz8with_current_context..decorator..wrappervs)++ - -DFFFtT,V,, - - - - - - - - - - - - - - - - - -s 155r)r-r/r.s` r decoratorz'with_current_context..decoratorus9 t - - - - -  - rr)r.r0s` rwith_current_contextr1ds$" rr*)r contextlibr contextvarsrcopyr dataclassesrr functoolsr r rrrr"r&r1rrrr7s! &%%%%%""""""((((((((  4 4 4 4 4 4 4  4 :j ! !   <r