File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/wheel/__pycache__/__init__.cpython-310.pyc
o
;j� � @ sf d Z ddlmZ ddlZddlZddlZddlZddlZddl ZG dd� dej
j j ej
j j
�ZeZdS )z+
Modules used to control the master itself
� )�MappingNc sx e Zd ZdZdZdZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z ddd�Z dd� Z
d� fdd� Z� ZS )� WheelClienta�
An interface to Salt's wheel modules
:ref:`Wheel modules <all-salt.wheel>` interact with various parts of the
Salt Master.
Importing and using ``WheelClient`` must be done on the same machine as the
Salt Master and it must be done using the same user that the Salt Master is
running as. Unless :conf_master:`external_auth` is configured and the user
is authorized to execute wheel functions: (``@wheel``).
Usage:
.. code-block:: python
import salt.config
import salt.wheel
opts = salt.config.master_config('/etc/salt/master')
wheel = salt.wheel.WheelClient(opts)
�wheelNc C s� t jjjj| ||d� t jjjj| ||d� || _|p i | _d | _t j j
� � | _ t j jj
d| jd | jdd�| _t jj|| jd�| _d S )N)�contextZmasterZsock_dirF)�opts�listen)�salt�client�mixins�SyncClientMixin�__init__�AsyncClientMixinr r �event�utils�userZget_specific_userZ salt_userZ get_event�loaderZwheels� functions)�selfr r � r �G/opt/saltstack/salt/lib/python3.10/site-packages/salt/wheel/__init__.pyr * s
�zWheelClient.__init__c C sV | j d ur
| j �� d | _ t| d�r'| jd ur)t| jd�r"| j�� i | _d S d S d S )Nr �destroy)r r �hasattrr �r r r r r 6 s
�zWheelClient.destroyc C s | S �Nr r r r r � __enter__? s zWheelClient.__enter__c G s | � � d S r )r )r �argsr r r �__exit__B s zWheelClient.__exit__c K s" | j |||�dd�|�dd�d�S )z)
Backwards compatibility
� print_eventT� full_returnF)r r )�low�get)r �fun�kwargsr r r � call_funcF s
�zWheelClient.call_funcc K s� |}d|d< | j d }|dkrd}|dkrd}d�tjj�|�t| j d ��}tjjj j
| j d
|d d ��
}|� |�}W d
� n1 sDw Y t |t
�r^d|v r^tjjjdi |d �� |S )zX
Execute a wheel function through the master network interface (eauth).
r �cmd� interfacez0.0.0.0z 127.0.0.1z::z::1z tcp://{}:{}Zret_port�clear� master_call)Zcrypt�
master_uri�usageN�errorr )r �formatr r �networkZ
ip_bracket�str�channelr Z
ReqChannel�factory�send�
isinstancer r* Z raise_error)r r" �loadr% r( r. �retr r r r' S s*
�
� �
zWheelClient.master_callFc C s | j di |��S )a3
Execute a wheel function synchronously; eauth is respected
This function requires that :conf_master:`external_auth` is configured
and the user is authorized to execute runner functions: (``@wheel``).
.. code-block:: python
>>> wheel.cmd_sync({
'fun': 'key.finger',
'match': 'jerry',
'eauth': 'auto',
'username': 'saltdev',
'password': 'saltdev',
})
{'minions': {'jerry': '5d:f6:79:43:5e:d4:42:3f:57:b8:45:a8:7e:a4:6e:ca'}}
Nr )r' )r r �timeoutr r r r �cmd_syncm s zWheelClient.cmd_syncc C s |� d�}| j||dd�S )a
Execute a function asynchronously; eauth is respected
This function requires that :conf_master:`external_auth` is configured
and the user is authorized
.. code-block:: python
>>> wheel.cmd_async({
'fun': 'key.finger',
'match': 'jerry',
'eauth': 'auto',
'username': 'saltdev',
'password': 'saltdev',
})
{'jid': '20131219224744416681', 'tag': 'salt/wheel/20131219224744416681'}
r! F)�local)r Z asynchronous)r r r! r r r � cmd_async� s
zWheelClient.cmd_asyncTc s t � �||||||�S )z�
Execute a function
.. code-block:: python
>>> wheel.cmd('key.finger', ['jerry'])
{'minions': {'jerry': '5d:f6:79:43:5e:d4:42:3f:57:b8:45:a8:7e:a4:6e:ca'}}
)�superr$ )r r! �argZpub_data�kwargr r �� __class__r r r$ � s zWheelClient.cmdr )NF)NNNTF)�__name__�
__module__� __qualname__�__doc__r Z
tag_prefixr r r r r# r' r5 r7 r$ �
__classcell__r r r; r r s$
�r )r@ �collections.abcr Zsalt.channel.clientr Zsalt.client.mixinsZ salt.configZ salt.loaderZsalt.utils.errorZsalt.utils.networkr r
r r
r ZWheelr r r r �<module> s
�