File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/context.cpython-310.pyc
o
�N�g8 � @ 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 @ sf e Zd ZdZddd�Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� 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 K s t | | j|d�}|S )zE
Clone this context, and return the ChildContextDict
)�parentr r )�ChildContextDictr )r �kwargs�childr
r
r �cloneW s �zContextDict.clonec C s$ | j r || jj|< d S || j|< d S r �r r r r �r �key�valr
r
r � __setitem__` s zContextDict.__setitem__c C s | j r
| jj|= d S | j|= d S r r# �r r% r
r
r � __delitem__f s zContextDict.__delitem__c C s | j r | jj| S | j| S r r# r( r
r
r � __getitem__l �
zContextDict.__getitem__c C � | j r t| jj�S t| j�S r )r �lenr r r r r
r
r �__len__r r+ zContextDict.__len__c C r, r )r �iterr r r r r
r
r �__iter__x 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 r1 )r2 r r r3 �deepcopyr r r )r �memor4 r
r
r � __deepcopy__� s �zContextDict.__deepcopy__N)F)�__name__�
__module__� __qualname__�__doc__r �propertyr r" r' r) r* r. r0 r5 r8 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 �itemsr3 r6 )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 rC r( r
r
r r) � � z ChildContextDict.__delitem__c C s
| j | S r rC r( r
r
r r* � �
z ChildContextDict.__getitem__c C �
t | j�S r )r- r> r r
r
r r. � rF zChildContextDict.__len__c C rG r )r/ r> r r
r
r r0 � rF zChildContextDict.__iter__c C s* t | jjd�r
| jjj| _| j| jj_d S )Nr )r r r r r? r> r r
r
r � __enter__� s zChildContextDict.__enter__c G s | j | jj_d S r )r? r r r )r �excr
r
r �__exit__� � zChildContextDict.__exit__)NF) r9 r: r; r< r r' r) r* r. r0 rH rJ 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 �drP �� __class__r
r r � s
z NamespacedDictWrapper.__init__c C s | j }| jD ]}|| }q|S r )rM rP )r �rrA r
r
r rR � s
zNamespacedDictWrapper._dictc C � t | �� �S r )�reprrR r r
r
r �__repr__� rE z NamespacedDictWrapper.__repr__c C s || � � |< d S r �rR r$ r
r
r r' � rK z!NamespacedDictWrapper.__setitem__c C s | � � |= d S r rZ r( r
r
r r) � rD z!NamespacedDictWrapper.__delitem__c C s | � � | S r rZ r( r
r
r r* � rE z!NamespacedDictWrapper.__getitem__c C rW r )r- rR r r
r
r r. � rE z NamespacedDictWrapper.__len__c C rW r )r/ rR r r
r
r r0 � rE z NamespacedDictWrapper.__iter__c C s t | �t�| j�t�| j��S r )r2 r3 rM rP r r
r
r r5 � s z NamespacedDictWrapper.__copy__c C s" t | �t�| j|�t�| j|��S r )r2 r3 r6 rM rP )r r7 r
r
r r8 � s �z"NamespacedDictWrapper.__deepcopy__c C s | � � �� S r )rR �__str__r r
r
r r[ � rE z NamespacedDictWrapper.__str__)r9 r: r; r< r rR rY r' r) r* r. r0 r5 r8 r[ �
__classcell__r
r
rT r rL � s rL ) r< r3 r �collections.abcr �
contextlibr r r r �dictrL r
r
r
r �<module> s
%W/