HEX

File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/__pycache__/runner.cpython-310.pyc
o

;j�6�@s�dZddlZddlZddlZddlZddlZddlZddlZddl	Zddl
ZddlmZddl
mZddlmZe�e�ZGdd�dejej�ZGdd	�d	e�ZdS)
z#
Execute salt convenience routines
�N)�mixins)�display_output)�
verify_funcs|eZdZdZdZdZddd�Zdd�Zd	d
�Zdd�Z	e
d
d��Zdd�Zdd�Z
ddd�Z					d�fdd�	Z�ZS)�RunnerClienta7
    The interface used by the :command:`salt-run` CLI tool on the Salt Master

    It executes :ref:`runner modules <all-salt.runners>` which run on the Salt
    Master.

    Importing and using ``RunnerClient`` 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.

    Salt's :conf_master:`external_auth` can be used to authenticate calls. The
    eauth user must be authorized to execute runner modules: (``@runner``).
    Only the :py:meth:`master_call` below supports eauth.
    �runner�runNcCsltjj|||d�tjj|||d�||_|pi|_d|_tjj	�
�|_tjjjd|jd|jdd�|_dS)N��context�masterZsock_dirF)�opts�listen)
r�SyncClientMixin�__init__�AsyncClientMixinrr	�event�salt�utils�user�get_specific_userZ	salt_user�	get_event��selfrr	�r�?/opt/saltstack/salt/lib/python3.10/site-packages/salt/runner.pyr)s
�zRunnerClient.__init__cCs�|jdur
|j��d|_t|d�r%|jdur%t|jd�r"|j��i|_t|d�r?|jdurAt|jd�r:|j��i|_dSdSdS)N�
_functions�destroyr)rr�hasattrrr�rrrrr4s




�zRunnerClient.destroycCs|S�Nrrrrr�	__enter__AszRunnerClient.__enter__cGs|��dSr)r)r�argsrrr�__exit__DszRunnerClient.__exit__cCszt|d�s:t|d�stj�|j�|_ztjj|j|j|jd�|_W|jSty9tjj|j|jd�|_Y|jSw|jS)Nrr)rr	r)	rr�loaderrrrr	r�AttributeErrorrrrr�	functionsGs


���zRunnerClient.functionsc	s���d�}t|j|��fdd�dD�}tjjj��dg�dd�\}}|���di��|���g}|�|�|�	t
dd	d
i|���tjj|j||d
d�\}}t
d|||d�d
�|��S)a�
        Format the low data for RunnerClient()'s master_call() function

        This also normalizes the following low data formats to a single, common
        low data structure.

        Old-style low: ``{'fun': 'jobs.lookup_jid', 'jid': '1234'}``
        New-style: ``{'fun': 'jobs.lookup_jid', 'kwarg': {'jid': '1234'}}``
        CLI-style: ``{'fun': 'jobs.lookup_jid', 'arg': ['jid="1234"']}``
        �funcs i|]}|�vr|��|��qSr)�pop)�.0�i��lowrr�
<dictcomp>fs
	
�
z.RunnerClient._reformat_low.<locals>.<dictcomp>)�username�password�eauth�token�clientr�key�argF)Z	condition�kwargZ	__kwarg__T)Zignore_invalid)r3r2)r%r3Nr)
r&rr$rrr �parse_input�update�extend�append�dict�minion�load_args_and_kwargs)	rr*r%Zeauth_credsZ_argZ_kwargZmungedr2r3rr)r�
_reformat_lowXs

�


�zRunnerClient._reformat_lowcCs|�|�}tj�||�S)a�
        Execute a runner function asynchronously; eauth is respected

        This function requires that :conf_master:`external_auth` is configured
        and the user is authorized to execute runner functions: (``@runner``).

        .. code-block:: python

            runner.cmd_async({
                'fun': 'jobs.list_jobs',
                'username': 'saltdev',
                'password': 'saltdev',
                'eauth': 'pam',
            })
        )r;rr�	cmd_async)rr*�reformatted_lowrrrr<�s
zRunnerClient.cmd_asyncFcCs|�|�}tj�||||�S)a�
        Execute a runner function synchronously; eauth is respected

        This function requires that :conf_master:`external_auth` is configured
        and the user is authorized to execute runner functions: (``@runner``).

        .. code-block:: python

            runner.cmd_sync({
                'fun': 'jobs.list_jobs',
                'username': 'saltdev',
                'password': 'saltdev',
                'eauth': 'pam',
            })
        )r;rr
�cmd_sync)rr*�timeout�full_returnr=rrrr>�s
�zRunnerClient.cmd_syncTcst��||||||�S)a�
        Execute a function

        .. code-block:: python

            >>> opts = salt.config.master_config('/etc/salt/master')
            >>> runner = salt.runner.RunnerClient(opts)
            >>> runner.cmd('jobs.list_jobs', [])
            {
                '20131219215650131543': {
                    'Arguments': [300],
                    'Function': 'test.sleep',
                    'StartTime': '2013, Dec 19 21:56:50.131543',
                    'Target': '*',
                    'Target-type': 'glob',
                    'User': 'saltdev'
                },
                '20131219215921857715': {
                    'Arguments': [300],
                    'Function': 'test.sleep',
                    'StartTime': '2013, Dec 19 21:59:21.857715',
                    'Target': '*',
                    'Target-type': 'glob',
                    'User': 'saltdev'
                },
            }

        )�super�cmd)rr%r2Zpub_datar3Zprint_eventr@��	__class__rrrB�s%zRunnerClient.cmdr)NF)NNNTF)�__name__�
__module__�__qualname__�__doc__r0Z
tag_prefixrrrr!�propertyr$r;r<r>rB�
__classcell__rrrCrrs&


2
�rcsDeZdZdZd�fdd�	Z�fdd�Z�fdd�Zd
d
d�Z�ZS)�Runnerz+
    Execute the salt runner interface
    Ncs8t�j||d�tjj||j|d�|_tj�|�|_dS)Nr)rArrr"�	returnersr$�
outputtersrrCrrr�szRunner.__init__csnt|d�r|jdurt|jd�r|j��i|_t|d�r0|jdur0t|jd�r-|j��i|_t���dS)NrLrrM)rrLrrMrArrCrrr�s

zRunner.destroycsL|j�dd�}t��|�}t|�D]}t|�d�d|j�t||�qdS)z.
        Print out the documentation!
        r%N�:�text)r�getrA�get_docs�sortedr�print)rr2Zdocsr%rCrr�
print_docs�s�zRunner.print_docsFcCs�ddl}i}|j�dd�r|��dSd|jdi}�za|j|j�d�d�}|d|_|jjj|jd|j�d	g�d
�}t	|j
|d�|j�|j
|d|�\}}||d<||d<|j�d�r�d
|jvr�z-|jj
�tj�|jdd�d��}	|jj�|	���|d<Wd�n1s�wYWnty�|jd
|d
<Ynwd
|vr�d|vr�|jdr�ddl}|j�|j�}
|
�|jd�}|jdr�|r�|
�|jd|�}|r�|�d
d�|d
<|s�t�d�|WS|�|�|jd|d<n|jj��}
|ddv�r	|d|dd<|j�dd��r7|j�d��r|� |�}n|j!|jd||
|d�}t�"d|d�|dWS|j�d��re|�#|�}t$|t%��r[t&|�ddhk�r[|d}|d}nd}t'|||j�n|j(||j|jd||
|d|dd|d�	}Wnw|j)j*�y�}zh|jj+j,d|jd�� }|j-d|�d |jd||jd!�d"|j�d#�d$�Wd�n	1�s�wYd|v�r�|�.d%�/|d��}nd}|�s�|�}|j�d&d��s�t'|d'|j�WYd}~|SWYd}~|Sd}~wwt�0d(|�|S))z-
        Execute the runner sequence
        rN�docFr%�jid)rVr2�no_parse)rWr3r.r/Zcachedirz	.root_key�rr1Zmktoken�zAuthentication failed)zstate.orchestratez
state.orchz	state.slsZorchestration_jid�async)rZpubz�Running in asynchronous mode. Results of this execution may be collected by attaching to the master event bus or by examining the master job cache, if configured. This execution is running under tag %s�tag�data�	outputter)	�instancerr%r*rr[rVZ	daemonizer@r
)r�)�success�return�retcoder%�fun_argsrVz	salt/run/z/ret)r[z{}*�quiet�nestedzRunner return: %s)1�salt.minionrrPrTZ_gen_async_pubrVrr r4rr$r9r:�filesZfopen�os�path�joinZstringutilsZ
to_unicode�readline�OSErrorZ	salt.authZauthZResolverZcliZ	token_cli�log�errorr5rrr<Zasynchronous�warningr>�
isinstancer8�setrZ_proc_function�
exceptionsZ
SaltExceptionrrZ
fire_eventrQ�format�debug)rr@r�retr*Z	async_pubrcr �kwargsZfp_Zresolver�res�tokrr]�excZevtrrrr�s�
��
�
����

���


�����


���z
Runner.runr)F)	rErFrGrHrrrTrrJrrrCrrK�srK)rH�loggingrhZsalt.exceptionsrZsalt.loaderrfZsalt.utils.argsZsalt.utils.eventZsalt.utils.filesZsalt.utils.userZsalt.clientrZsalt.outputrZsalt.utils.lazyr�	getLoggerrErmr
rrrKrrrr�<module>s"
F