File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/netapi/__pycache__/__init__.cpython-310.pyc
o
;j2. � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlmZ e�e�Zdd� Zdd� ZG dd � d �Zd
d � ejedd �D �ZdS )
z
Make api awesomeness
� N)�DEFAULT_TARGET_DELIMc C sZ g }g }| D ] }t |t�r||v rq|�|� q||v r q|�|� qt|�t|td� S )zZ
Return a sorted list of the passed in permissions, de-duplicating in the process
��key)�
isinstance�str�append�sorted�repr)�permsZ
_str_permsZ_non_str_perms�entry� r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/netapi/__init__.py�sorted_permissions s
r c C sv |� | d g �}|�|� dg �� d| v r9| d r9t| d �}dd� |�� D �}||@ D ] }|�||� d� � q,|S )zN
Returns the sum of '*', user-specific and group specific permissions
�name�*�groupsc S s h | ] }|� d �r|�d ��qS )�%)�endswith�rstrip)�.0�ir r r
� <setcomp>5 s z"sum_permissions.<locals>.<setcomp>r )�get�extend�set�keys)�token�eauthr
Z user_groupsZ eauth_groups�groupr r r
�sum_permissions, s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd&d d �Zd d!� Zd"d#� Zd$d%� Zd S )'� NetapiClienta'
Provide a uniform method of accessing the various client interfaces in Salt
in the form of low-data data structures. For example:
>>> client = NetapiClient(__opts__)
>>> lowstate = {'client': 'local', 'tgt': '*', 'fun': 'test.ping', 'arg': ''}
>>> client.run(lowstate)
c C s� || _ d | _d | _d | _t�| j �}d|d< tj�|d d�|d< tj� |d �s0t�
|d � t j �
|�| _t j �|�| _t jj�|�| _t jj�|�| _d S )NTZenable_ssh_minionsZcachedirZsaltapi)�opts�resolver�loadauth� ckminions�copy�deepcopy�os�path�join�exists�makedirs�salt�authZResolverZLoadAuthZdaemonsZ masterapiZ access_keysr �utils�minionsZ CkMinions)�selfr! Zapioptsr r r
�__init__F s zNetapiClient.__init__c C s� | j dur t| j d�r t| j jd�r| j j�� i | j _d| _ | jdur2t| jd�r/| j�� d| _| jdurYt| jd�rT| jjdurTt| jjd�rP| jj�� d| j_d| _dS dS )z$
Clean up resources
Nr- �destroy�cache)r" �hasattrr- r2 r# r$ r3 �r0 r r r
r2 U s"
�zNetapiClient.destroyc C s | S �Nr r5 r r r
� __enter__j s zNetapiClient.__enter__c G s | � � d S r6 )r2 )r0 �argsr r r
�__exit__m s zNetapiClient.__exit__c C sD t jj�� rdS | jd dkrd}nd}tj�tj�| jd |��S )z�
Perform a lightweight check to see if the master daemon is running
Note, this will return an invalid success if the master crashed or was
not shut down cleanly.
TZ transportZtcpzpublish_pull.ipcz workers.ipcZsock_dir) r, r. �platformZ
is_windowsr! r' r( r* r) )r0 Zipc_filer r r
�_is_master_runningp s z NetapiClient._is_master_runningc C sZ g }d }d|v rd}d}dg}||||fS d|v r'd}d}ddg}||||fS t j�d��)Nr ZTokenAuthenticationErrorr �EauthAuthenticationError�username�password�#No authentication credentials given)r, �
exceptionsr< )r0 Z
clear_load�sensitive_load_keysr � auth_type�err_namer r r
�_prep_auth_info� s �z NetapiClient._prep_auth_infoc C s� | � |�\}}}}| jj|||d�}|�dg �}|�d�}|r%tj�|��|�di ��dt�} | j� |d |�dd�| �}
|
�d t
� �} |
�d
t
� �} | jj ||d |�d g �|d |�dd�| d
�}
|
sktj�d��d S )Nr � auth_list�error�kwargs� delimiterZtgtZtgt_type�globr/ �missing�fun�arg)r/ z Authorization error occurred.) rD r# Zcheck_authenticationr r, r@ r< r r$ Z
check_minions�list�
auth_check)r0 �lowrB rC r rA rN rE rF rH Z_resr/ rJ Z
authorizedr r r
�_authorize_ssh� s2
�
���zNetapiClient._authorize_sshc C s� | � � s
tj�d��|�d�tvr tj�d�|�d����|�d�| j�d�vr4tj�d�|�d����d|v sBd|v sBtj� d��|�d �rS| j�d
�sStj� d ��|d d kr^| �
|� t | |d �}tj j
�||�}||�d
d�i |�di ���S )zh
Execute the specified function in the specified client by passing the
lowstate
z Salt Master is not available.�clientz Invalid client specified: '{}'Znetapi_enable_clientszUClient disabled: '{}'. Add to 'netapi_enable_clients' master config option to enable.r r r? Z raw_shellZnetapi_allow_raw_shellz Raw shell option not allowed.�sshr8 r rG )r; r, r@ ZSaltDaemonNotRunningr �CLIENTSZSaltInvocationError�formatr! r< rP �getattrr. r8 Z format_call)r0 rO Zl_funZf_callr r r
�run� s2 �����
zNetapiClient.runc O �D t jj| jd��}|j|i |��W d � S 1 sw Y dS )z�
Run :ref:`execution modules <all-salt.modules>` asynchronously
Wraps :py:meth:`salt.client.LocalClient.run_job`.
:return: job ID
��moptsN)r, rQ �get_local_clientr! Zrun_job�r0 r8 rG rQ r r r
� local_async� � $�zNetapiClient.local_asyncc O rW )a
Run :ref:`execution modules <all-salt.modules>` synchronously
See :py:meth:`salt.client.LocalClient.cmd` for all available
parameters.
Sends a command from the master to the targeted minions. This is the
same interface that Salt's own CLI uses. Note the ``arg`` and ``kwarg``
parameters are sent down to the minion(s) and the given function,
``fun``, is called with those parameters.
:return: Returns the result from the execution module
rX N)r, rQ rZ r! �cmdr[ r r r
�local� s $�zNetapiClient.localc O rW )z�
Run :ref:`execution modules <all-salt.modules>` against subsets of minions
.. versionadded:: 2016.3.0
Wraps :py:meth:`salt.client.LocalClient.cmd_subset`
rX N)r, rQ rZ r! Z
cmd_subsetr[ r r r
� local_subset� r] zNetapiClient.local_subsetc O rW )a
Run :ref:`execution modules <all-salt.modules>` against batches of minions
.. versionadded:: 0.8.4
Wraps :py:meth:`salt.client.LocalClient.cmd_batch`
:return: Returns the result from the exeuction module for each batch of
returns
rX N)r, rQ rZ r! Z cmd_batchr[ r r r
� local_batch� s $�zNetapiClient.local_batchc O sD t jjjj| jdd��
}|�|�W d � S 1 sw Y dS )z�
Run salt-ssh commands synchronously
Wraps :py:meth:`salt.client.ssh.client.SSHClient.cmd_sync`.
:return: Returns the result from the salt-ssh command
T)rY Zdisable_custom_rosterN)r, rQ rR Z SSHClientr! �cmd_syncr[ r r r
rR
s
�$�zNetapiClient.sshNFc K sZ ||d< |dur t |�}tj�| j��}|j|||d�W d � S 1 s&w Y dS )a<
Run `runner modules <all-salt.runners>` synchronously
Wraps :py:meth:`salt.runner.RunnerClient.cmd_sync`.
Note that runner functions must be called using keyword arguments.
Positional arguments are not supported.
:return: Returns the result from the runner module
rK N)�timeout� full_return)�floatr, �runner� RunnerClientr! rb )r0 rK rc rd rG rf r r r
rf s $�zNetapiClient.runnerc K �D ||d< t j�| j��
}|�|�W d � S 1 sw Y dS )aG
Run `runner modules <all-salt.runners>` asynchronously
Wraps :py:meth:`salt.runner.RunnerClient.cmd_async`.
Note that runner functions must be called using keyword arguments.
Positional arguments are not supported.
:return: event data and a job ID for the executed function.
rK N)r, rf rg r! � cmd_async)r0 rK rG rf r r r
� runner_async* � $�zNetapiClient.runner_asyncc K rh )a=
Run :ref:`wheel modules <all-salt.wheel>` synchronously
Wraps :py:meth:`salt.wheel.WheelClient.master_call`.
Note that wheel functions must be called using keyword arguments.
Positional arguments are not supported.
:return: Returns the result from the wheel module
rK N)r, �wheel� WheelClientr! rb �r0 rK rG rl r r r
rl 9 rk zNetapiClient.wheelc K rh )a>
Run :ref:`wheel modules <all-salt.wheel>` asynchronously
Wraps :py:meth:`salt.wheel.WheelClient.master_call`.
Note that wheel functions must be called using keyword arguments.
Positional arguments are not supported.
:return: Returns the result from the wheel module
rK N)r, rl rm r! ri rn r r r
� wheel_asyncH rk zNetapiClient.wheel_async)NF)�__name__�
__module__� __qualname__�__doc__r1 r2 r7 r9 r; rD rP rV r\ r_ r` ra rR rf rj rl ro r r r r
r < s& (
r c C s& g | ]\}}|d ks|� d�s|�qS )rV �_)�
startswith)r r rt r r r
�
<listcomp>X s
�rv )� predicate)rs r% �inspect�loggingr' Z salt.authr, Z salt.clientZsalt.client.ssh.clientZ salt.configZsalt.daemons.masterapiZsalt.exceptionsZ salt.runnerZ
salt.syspathsZsalt.utils.argsZsalt.utils.minionsZ
salt.wheelZ
salt.defaultsr � getLoggerrp �logr r r �
getmembersrS r r r r
�<module> s4
�