File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/__pycache__/scripts.cpython-310.pyc
o
�N�gAH � @ sB d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z ddl
ZddlmZmZmZ e�e�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd � Z!d d � Z"d d � Z#d!d"� Z$d#d$� Z%d%d&� Z&d'd(� Z'd)d*� Z(d+d,� Z)d-d.� Z*d/d0� Z+d1d2� Z,dS )3zI
This module contains the function calls to execute command line scripts
� N)�randint)�SaltClientError�SaltReqTimeoutError�SaltSystemExitc C s� z| j j}W n
ttfy d}Y nw |tjkr6d}z| jjd }|d�|�7 }W n
ttfy5 Y nw d }|d u r@|r@d}|rJt |t
j dd� |riz t �
� }t�|� W n ty` Y nw t
�tjjj� t
�tjjj� d S )NFz
Exiting gracefully on Ctrl-c�jidz�
This job's jid is: {0}
The minions may not have all finished running and any remaining minions will return upon completion. To look up the return data for this job later, run the following command:
salt-run jobs.lookup_jid {0}z"
Exiting with hard crash on Ctrl-cT��file�flush)�options�
hard_crash�AttributeError�KeyError�signal�SIGINTZ local_clientZpub_data�format�print�sys�stderr� traceback�
format_exc�log�error�exit�salt�defaults� exitcodes�
EX_GENERIC�EX_OK)�client�signum�sigframeZ hardcrashZexit_msgr �trace� r" �@/opt/saltstack/salt/lib/python3.10/site-packages/salt/scripts.py�_handle_signals s< �
�� �r$ c C s^ t �t j�t jt jfv rt � t jt�t| �� t �t j�t ju r-t � t jt�t| �� d S d S �N) r � getsignalr �SIG_DFL�default_int_handler� functools�partialr$ �SIGTERM)r r" r" r# �_install_signal_handlers> s
�r, c C s8 ddl } tdkrtjt tjd< | jj�� }|�� dS )z
Start the salt master.
r N�__main__)�salt.cli.daemons�__name__r �modules�cli�daemonsZMaster�start)r Zmasterr" r" r# � salt_masterI s
r4 c
sr ddl �t�� �jj_ddl�ddl�dd� } t� � � � �fdd�}z�z+�j
j � � s8tj
|t�� fd�}|�� �jj�� }t�tjt�| |�� |�� W nQ tttfy� } zB� jdd � tj d
dd � t� d � d
}|dur}t |d�r}|j � dd
�}t d|�}t�!d|� t"�#|� t$�%�j&j'j(� W Y d}~nd}~ww W � jdd � dS W � jdd � dS � jdd � w )z
Start a minion process
r Nc S s | j �� d S r% )�minion�reload)�manager�sig�framer" r" r# �
handle_hupj s z"minion_process.<locals>.handle_hupc
s� � j dd�rL� �� t�d� zt�� dkr"�jj�� s"t� | d� W n t
yC } zt � d|� t�
�jjj� W Y d}~nd}~ww � j dd�sdS dS )z�
Have the minion suicide if the parent process is gone
NOTE: small race issue where the parent PID could be replace
with another process with same PID!
F��blocking� r z(Minion process encountered exception: %sN)�acquire�release�time�sleep�os�getuid�utils�platform�
is_windows�kill�OSErrorr r �_exitr r r )�
parent_pid�exc��lockr r" r# �suicide_when_without_parento s
� ���z3minion_process.<locals>.suicide_when_without_parent��target�argsTr; z4Fatal functionality error caught by minion handler:
��exc_infoz** Restarting minion **�< �config�random_reauth_delay� z waiting random_reauth_delay %ss))Z
salt._loggingrB �getpidZ_loggingZin_mainprocessZ__pid__r. �salt.utils.platform� threading�RLockrD rE rF �Thread�getppidr3 r1 r2 �Minionr �SIGHUPr) r* r r r r> r �warning�hasattrrU �getr �infor@ rA r r r r �SALT_KEEPALIVE)r: rN �threadr5 rK �delayr" rL r# �minion_processY sF
� �
���� rg c
C s� ddl } ddl}ddl}ddl}|jj�� |jj�� ddl }ddl
}t dkr0t j
t t j
d< dt jv r;t j�d� |jj�� rM|jj�� }|�� dS dt jv rdt j�d� |jj�� }|�� dS dd� }| �| j�}| �| j�} z1|jtd d
�}|�� | � | jt � ||j �� | � | jt � ||j �� | � | j t � ||j �� W n t y� |jj�� }|�� Y dS w |�!� | � | j|� | � | j|� |j"|j#j$j%ks�t �&|j"� t'�(d t)d d
� � t*�+� }|j,D ]} |�-| � q�t*�.� qu)zR
Start the salt minion in a subprocess.
Auto restart minion on error.
r Nr- � �--disable-keepalivec S s t �| |� dS )zy
Escalate the signal received to the multiprocessing process that
is actually running the minion
N)rB rG )�pidr r r" r" r# �escalate_signal_to_process� s z/salt_minion.<locals>.escalate_signal_to_processTZMinionKeepAlive)rP �name� rW �
)/r Zsalt.utils.debugrY �salt.utils.processrD �debugZenable_sigusr1_handler�process�notify_systemd�multiprocessingr. r/ r r0 �path�removerE rF r1 r2 r^ r3 �argvr& r r+ �Processrg r) r* rj r_ � Exception�join�exitcoder r rd r r@ rA r �logging� getLogger�handlers�
removeHandler� basicConfig)
r r rs r5 rk Zprev_sigint_handlerZprev_sigterm_handlerrq �rlogger�handlerr" r" r# � salt_minion� st
� � � � �
�r� c
s| ddl }ddl}t�� � � fdd�}zlz)|jj�� s(tj|t� � fd�}|�
� d}d}|j j j
}|jj�� }|�
� W n: ttttfy` } ztjddd � d}|j j j}W Y d}~nd}~w tyv } z d}|j}W Y d}~nd}~ww W � jdd
� n� jdd
� w |du r�t�d � d }|dur�t |d
�r�|j � dd �}t d|�} t� d| � | �!| � n| �!d� t"�#|� dS )z&
Start a proxy minion process
r Nc s` � j dd�r.� �� t�d� zt�| d� W n ty% t�d� Y nw � j dd�sdS dS )z�
Have the minion suicide if the parent process is gone
NOTE: there is a small race issue where the parent PID could be replace
with another process with the same PID!
Fr; r= r i� N)r> r? r@ rA rB rG rH rI )rJ �rM r" r# rN
s
��z9proxy_minion_process.<locals>.suicide_when_without_parentrO Fz Proxy Minion failed to start: TrR r; z ** Restarting proxy minion **rT rU rV rW z*Sleeping random_reauth_delay of %s seconds)$r. rY rZ r[ rD rE rF r\ rB r] r3 r r r r1 r2 � ProxyMinionrx r r r r r rd �
SystemExit�coder> r` ra rU rb r rc �putr r )
�queuer rN re Zrestart� proxyminion�statusrK rf Z random_delayr" r� r# �proxy_minion_process� sV
�
�����
r� c C st ddl } ddl}ddl}dtjv rtj�d� |jj�� r)|j j
� � }|� � dS dtj
v r@tj
�d� |j j
� � }|� � dS z| �� }W n ty[ |j j
� � }|� � Y dS w | jt|fdd�}|� � z2|�� z|jdd �}W n ty� |jdkr�Y W dS d
}Y nw |dkr�t�|j� t�|� W n
ty� Y dS w t�� }|jD ]}|� |� q�t� � qA) z
Start a proxy minion.
r Nrh ri Tr� )rP rQ rl F)�blockrT ) rs r. rY r rt ru rD rE rF r1 r2 r� r3 rv �Queuerx rw r� ry rb rz r r@ rA �KeyboardInterruptr{ r| r} r~ r )rs r r� r� rq Z
restart_delayr� r� r" r" r# �
salt_proxyI s\
��
� �
�r� c C s` ddl } | jj�� ddl} t�� }z
| jj� � }|�
� W dS t y/ t� |d� Y dS w )z
Start the salt syndic.
r N� )
ro rD rq rr r. rB rX r1 r2 ZSyndicr3 r� rG )r rj Zsyndicr" r" r# � salt_syndic� s �r� c
C sf ddl } z| jj�� }t|� |�� W dS ty2 } ztj� d|� d�� W Y d}~dS d}~ww )z7
Manage the authentication keys with salt-key.
r NzError: �
)
Z salt.cli.keyr1 �keyZSaltKeyr, �runrx r r �write)r r �errr" r" r# �salt_key� s "��r� c C s( ddl } | jj�� }t|� |�� dS )�V
Publish commands to the salt system from the command line on the
master.
r N)Z salt.cli.cpr1 ZcpZ SaltCPClir, r� �r r r" r" r# �salt_cp� s r� c C �> ddl } dtjv rtj�d� | jj�� }t|� |�� dS )zi
Directly call a salt command in the modules, does not require a running
salt minion to run.
r Nrh ) Z
salt.cli.callr rt ru r1 �callZSaltCallr, r� r� r" r" r# � salt_call� �
r� c C s> ddl } dtjv rtj�d� | jj�� }t|� |�� dS )z-
Execute a salt convenience routine.
r Nrh )Z salt.cli.runr rt ru r1 r� ZSaltRunr, r� r" r" r# �salt_run� s
r� c C s� ddl } dtjv rtj�d� z| jj�� }t|� |�� W dS t y` } z4t
t |�tj dd� z|j
jr@t�� }t�|� W n ttfyL Y nw t�| jjj� W Y d}~dS d}~ww )z%
Execute the salt-ssh system
r Nrh Tr )Z salt.cli.sshr rt ru r1 ZsshZSaltSSHr, r� r r �strr r
r r r r r r r
r r r r )r r r� r! r" r" r# �salt_ssh� s(
�� ��r� c
C s� z
ddl } ddl} W n( ty2 } z ddl} t�d|� td� t�| j j
j � W Y d}~nd}~ww dtj v r>tj �
d� | jj�� }t|� |�� dS )z*
The main function for salt-cloud
r Nz Error importing salt cloud: %sz*salt-cloud is not available in this systemrh )Z
salt.cloudZsalt.cloud.cli� ImportError�salt.defaults.exitcodesr r r r r r r �EX_UNAVAILABLErt ru Zcloudr1 Z SaltCloudr, r� )r �er r" r" r# �
salt_cloud� s ��
r� c C s4 ddl } | jj�� ddl} | jj�� }|�� dS )z(
The main function for salt-api
r N) ro rD rq rr Z salt.cli.apir1 ZapiZSaltAPIr3 )r Zsapir" r" r# �salt_api� s
r� c C r� )r� r Nrh ) Z
salt.cli.saltr rt ru r1 r ZSaltCMDr, r� r� r" r" r# � salt_main r� r� c C s ddl } | jj�� }|�� dS )z]
The main function for spm, the Salt Package Manager
.. versionadded:: 2015.8.0
r N)Z salt.cli.spmr1 �spmZSPMr� )r r� r" r" r# �salt_spm s r� c C s$ ddl }|jjj| ||||d� dS )zb
Quickstart for developing on the saltstack installation
.. versionadded:: 2016.11.0
r N)� extensionrl � description�salt_dir�merge)Zsalt.utils.extendrD �extendr� )r� rl r� r� r� r r" r" r# � salt_extend" s
�r� c C s� g } t tjt �D ]}|�d�r| �|dd� � q ttj�dk r8d}| D ] }|d|� �7 }q%t|� t� d� tjd }|| vrIdtjd< t
}nd |� �tjd< tj� d� t tjt d
|� ��}|� dS ) z=
Change the args and redirect to another salt script
r r= Nrm z4Must pass in a salt command, available commands are:r� rW r zsalt-Zsalt_)
�dirr r0 r/ �
startswith�append�lenrv r r r� �pop�getattr)ZavailZfun�msg�cmdZs_funr" r" r# �
salt_unity3 s&
�
r� c C sF | d d � }d}| D ]}d|v rd}q
d| v r!|s!|� d|� �� |S )NFz--targetT�installz --target=)r� )rQ rP �new_argsZtarget_in_args�argr" r" r# � _pip_argsM s � r� c C s: | � � }d| v r|� tj� | d � �|d< |S ||d< |S )N�
PYTHONPATH)�copyrB �pathsep)�env�extrasZnew_envr" r" r# �_pip_environmentX s �r� c C s4 t �t�� tjW d � S 1 sw Y d S r% )�
contextlib�suppressr r �RELENVr" r" r" r# �_get_onedir_env_patha s �r� c C s� t � } | du rtdtjdd� t�tjjj� n
t | dj
tj � �}t t
j�� |�}ttjdd� |�}tjddgttjdd� |� }tj|d d |d
�}t�|j� dS ) z'
Proxy to current python's pip
Nzi'salt-pip' is only meant to be used from a Salt onedir. You probably want to use the system 'pip` binary.Tr z extras-{}.{}rW z-mZpipF)�shell�checkr� )r� r r r r r r r r r� r � version_infor� rB �environr� r� rv �
executable�
subprocessr� �
returncode)Z relenv_pathr� r� rQ Zcommand�retr" r" r# �salt_piph s& ���r� )-�__doc__r� r) r{ rB r r� r rZ r@ r �randomr r� r Zsalt.exceptionsr r r r| r/ r r$ r, r4 rg r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# �<module> sL
' EaJ<