File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/context.cpython-310.pyc
o
;j� � @ sn d Z ddlZddlZddlmZ ddlmZ edd� �ZG dd� de�ZG d d
� d
e�Z G d d � d ee
�Z dS )
z�
:codeauthor: Pedro Algarvio (pedro@algarvio.me)
:codeauthor: Thomas Jackson (jacksontj.89@gmail.com)
salt.utils.context
~~~~~~~~~~~~~~~~~~
Context managers used throughout Salt's source code.
� N)�MutableMapping)�contextmanagerc k s� � t | d�r | jr | j} | j}g }i }|D ]}||v r"|| ||< q|�|� q|�|� zdV W |�|� |D ]}||= q9dS |�|� |D ]}||= qHw )zI
Override specific variables within a function's global context.
�im_funcN)�hasattrr �__func__� __globals__�append�update)�func� overridesZ func_globalsZinjected_func_globalsZoverridden_func_globals�overrideZinjected� r
�F/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/context.py�func_globals_inject s* �
�
��r c @ sn e Zd ZdZddd�Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )� ContextDictz�
A context manager that saves some per-thread state globally.
Intended for use with Tornado's StackContext.
Provide arbitrary data as kwargs upon creation,
then allow any children to override the values of the parent.
Fc K s" t �� | _d | j_i | _|| _d S �N)� threading�local�_state�data� global_data� _threadsafe)�self�
threadsafer r
r
r �__init__A s
zContextDict.__init__c C s$ z| j jduW S ty Y dS w )z�Determine if this ContextDict is currently overridden
Since the ContextDict can be overridden in each thread, we check whether
the _state.data is set or not.
NF)r r �AttributeError�r r
r
r �activeK s
�zContextDict.activec C sL t | d�rd| j_z| j`W n ty Y nw t | d�r$| j�� dS dS )zB
Destroy the ContextDict and clear internal state
r Nr )r r r r r �clearr r
r
r �destroyV s
�
�zContextDict.destroyc K s t | | j|d�}|S )zE
Clone this context, and return the ChildContextDict
)�parentr r )�ChildContextDictr )r �kwargs�childr
r
r �cloned s �zContextDict.clonec C s$ | j r || jj|< d S || j|< d S r �r r r r �r �key�valr
r
r � __setitem__m s zContextDict.__setitem__c C s | j r
| jj|= d S | j|= d S r r% �r r' r
r
r � __delitem__s s zContextDict.__delitem__c C s | j r | jj| S | j| S r r% r* r
r
r � __getitem__y �
zContextDict.__getitem__c C � | j r t| jj�S t| j�S r )r �lenr r r r r
r
r �__len__ r- zContextDict.__len__c C r. r )r �iterr r r r r
r
r �__iter__� r- zContextDict.__iter__c C s< t | �| jd�}| jrt�| jj�|_|S t�| j�|_|S �N)r )�typer r �copyr r r )r �new_objr
r
r �__copy__� s �zContextDict.__copy__c C s@ t | �| jd�}| jrt�| jj|�|_|S t�| j|�|_|S r3 )r4 r r r5 �deepcopyr r r )r �memor6 r
r
r � __deepcopy__� s �zContextDict.__deepcopy__N)F)�__name__�
__module__� __qualname__�__doc__r �propertyr r r$ r) r+ r, r0 r2 r7 r: r
r
r
r r 8 s
r c @ sR e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )r! z$An overrideable child of ContextDictNFc C s� || _ |d u r i n|| _d | _|r+| j j�� D ]\}}|| jvr(t�|�| j|< qd S | j j�� D ]\}}|| jvr?|| j|< q1d S r )r �_data� _old_datar �itemsr5 r8 )r r r r �k�vr
r
r r � s
��
��zChildContextDict.__init__c C s || j |< d S r �r@ r& r
r
r r) � � z ChildContextDict.__setitem__c C s | j |= d S r rE r* r
r
r r+ � � z ChildContextDict.__delitem__c C s
| j | S r rE r* r
r
r r, � �
z ChildContextDict.__getitem__c C �
t | j�S r )r/ r@ r r
r
r r0 � rH zChildContextDict.__len__c C rI r )r1 r@ r r
r
r r2 � rH zChildContextDict.__iter__c C s* t | jjd�r
| jjj| _| j| jj_d S )Nr )r r r r rA r@ r r
r
r � __enter__� s zChildContextDict.__enter__c G s | j | jj_d S r )rA r r r )r �excr
r
r �__exit__� � zChildContextDict.__exit__)NF) r; r< r= r> r r) r+ r, r0 r2 rJ rL r
r
r
r r! � s
r! c sp e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�NamespacedDictWrapperz�
Create a dict which wraps another dict with a specific prefix of key(s)
MUST inherit from dict to serialize through msgpack correctly
c s4 || _ t|t�r
|f| _n|| _t� �| �� � d S r )� _NamespacedDictWrapper__dict�
isinstance�str�pre_keys�superr �_dict)r �drR �� __class__r
r r � s
z NamespacedDictWrapper.__init__c C s | j }| jD ]}|| }q|S r )rO rR )r �rrC r
r
r rT � s
zNamespacedDictWrapper._dictc C � t | �� �S r )�reprrT r r
r
r �__repr__� rG z NamespacedDictWrapper.__repr__c C s || � � |< d S r �rT r& r
r
r r) � rM z!NamespacedDictWrapper.__setitem__c C s | � � |= d S r r\ r* r
r
r r+ � rF z!NamespacedDictWrapper.__delitem__c C s | � � | S r r\ r* r
r
r r, � rG z!NamespacedDictWrapper.__getitem__c C rY r )r/ rT r r
r
r r0 � rG z NamespacedDictWrapper.__len__c C rY r )r1 rT r r
r
r r2 � rG z NamespacedDictWrapper.__iter__c C s t | �t�| j�t�| j��S r )r4 r5 rO rR r r
r
r r7 � s z NamespacedDictWrapper.__copy__c C s" t | �t�| j|�t�| j|��S r )r4 r5 r8 rO rR )r r9 r
r
r r: � s �z"NamespacedDictWrapper.__deepcopy__c C s | � � �� S r )rT �__str__r r
r
r r] � rG z NamespacedDictWrapper.__str__)r; r< r= r> r rT r[ r) r+ r, r0 r2 r7 r: r] �
__classcell__r
r
rV r rN � s rN ) r> r5 r �collections.abcr �
contextlibr r r r! �dictrN r
r
r
r �<module> s
%d/