File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/parsers.cpython-310.pyc
o
�N�g~� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl Z
ddlmZ ddlZ
ddlZ
ddlmZ ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddl Z
ddl Z
ddl Z
ddl m Z ddl!m"Z" ddl#m$Z$ ddl%m&Z&m'Z'm(Z( e�)e*�Z+dd� Z,G d d
� d
e-�Z.G d d � d e/�Z0G d
d� de0�Z1G dd� dej2�Z3G dd� dej4�Z4G dd� de0d�Z5G dd� de0d�Z6G dd� de0d�Z7G dd� de0d�Z8G d d � d e0d�Z9G d d � d e0d�Z:G d d!� d!e0d�Z;G d"d#� d#e0d�Z<G d$d%� d%e0d�Z=G d&d'� d'e=�Z>G d(d)� d)e0d�Z?G d*d+� d+e0d�Z@G d,d-� d-e0d�ZAG d.d/� d/e0d�ZBG d0d1� d1e0d�ZCG d2d3� d3e0d�ZDG d4d5� d5e0d�ZEG d6d7� d7e0d�ZFG d8d9� d9e0d�ZGG d:d;� d;e0d�ZHG d<d=� d=e0d�ZIG d>d?� d?e0d�ZJG d@dA� dA�ZKG dBdC� dCe4e9e5e:e;e<e6e1d�
ZLG dDdE� dEeLe1d�ZMG dFdG� dGe4eAe9e5e:e;e<e6e1d� ZNG dHdI� dIe4e9e5e:e;e<e6e1d�
ZOG dJdK� dKe4e9e5e?e>eDe:eBe7e6e@eJe8e1d�ZPG dLdM� dMe4eDe9e5e?e=e:e7e6e1d� ZQG dNdO� dOe4e9e5e:eDe;e7e6eJe1d� ZRG dPdQ� dQe4eAe9eBe5e:eDe7e6e@eHe8eCe1d�ZSG dRdS� dSe4e9e5e?e:e7e6eDe@eHeJe8eKe1d�ZTG dTdU� dUe4e9e5e:e=eDe6e7e8eKe1d�
ZUG dVdW� dWe4e:e5eDe9eFeEeGeIe7e6e1d�ZVG dXdY� dYe4e9e:e5e6e1d�ZWG dZd[� d[e4e9e:e<e5e1d�ZXdS )\z�
:codeauthor: Pedro Algarvio (pedro@algarvio.me)
salt.utils.parsers
~~~~~~~~~~~~~~~~~~
This is where all the black magic happens on all of salt's CLI tools.
� N)�partial)�DEFAULT_TARGET_DELIM)� is_writeable)� insecure_log�
verify_log�verify_log_filesc C s t | dd� d�S )Nc S s t | dd�S )N� _mixin_prio_�� )�getattr)Zmf� r �F/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/parsers.py�<lambda>2 s z_sorted.<locals>.<lambda>)�key)�sorted)Zmixins_or_funcsr r r �_sorted1 � r c @ � e Zd Zdd� ZdS )�MixinFuncsContainerc C s. t |tj�r |j}|| vrt�| |� d S d S �N)�
isinstance�types�
MethodType�__func__�list�append)�self�funcr r r r 6 s
�zMixinFuncsContainer.appendN)�__name__�
__module__� __qualname__r r r r r r 5 � r c s e Zd ZdZ� fdd�Z� ZS )� MixInMetar c s2 t � �| |||�}t|d�std�| j|���|S )N� _mixin_setupzNDon't subclass {} in {} if you're not going to use it as a salt parser mix-in.)�super�__new__�hasattr� RuntimeError�formatr )�mcs�name�bases�attrs�instance�� __class__r r r$ E s
�zMixInMeta.__new__)r r r r r$ �
__classcell__r r r- r r! ? s r! c s e Zd Z� fdd�Z� ZS )�OptionParserMetac s4 t � �| |||�}t|d�st� |_t|d�st� |_t|d�s$t� |_t|d�s-t� |_t||f �D ]c}t |dd �}|d urK||jvrK|j�
|� t |dd �}|d ur`||jvr`|j�
|� t |dd �}|d uru||jvru|j�
|� t |�D ] }|� d�s�qyt ||�}t |d d �d ur�qyt |d d
�|_
qyq4|S ) N�_mixin_setup_funcs�_mixin_process_funcs�_mixin_after_parsed_funcs�_mixin_before_exit_funcsr" �_mixin_after_parsed�_mixin_before_exit�process_r r )r# r$ r% r r1 r2 r3 r4 r r
r �dir�
startswithr )r( r) r* r+ r, �baser r- r r r$ P s8
� zOptionParserMeta.__new__)r r r r$ r/ r r r- r r0 O s r0 c @ r )� CustomOptionc O s$ d| _ tjj| ||g|�R i |��S )NT)�explicit�optparse�Option� take_action)r �action�dest�args�kwargsr r r r? x s zCustomOption.take_actionN)r r r r? r r r r r; w r r; c @ sv e Zd ZejjZdZdZdZ e
j d Z dd� Z
dd� Zdd d
�Zdd d
�Zdd� Ze
jfdd�Zddd�Zdd� ZdS )� OptionParserz%prog [options]z^You can find additional help about %prog issuing "man %prog" or on https://docs.saltproject.ioN�� c O s� |� dd| j� �� |� d| j� | jr|� d| j� | jr%|� d| j� |� dt� tjj| g|�R i |�� | jrLd| jv rN| j� d| �
� �| _d S d S d S )N�versionz%prog �usage� description�epilog� option_classz%prog) �
setdefault�VERSIONrG rH rI r; r= rD �__init__�replace�
get_prog_name)r rB rC r r r rM � s �zOptionParser.__init__c O s$ t jj| g|�R i |��}t|_|S r )r= rD �add_option_groupr; rJ )r rB rC Z option_groupr r r rP � s z OptionParser.add_option_groupc
C s� t j�| ||�\}}d|jv r5|jdu r5tj�� }dd� |D �}t j�| |�\}}|j�|j� |� |� |j
r<| � � ||| _ | _
t| j dd �}tj�|d u rSdn|� g }|jD ]} t| d| � �d �}
|
d uro|�|
� q[t|�D ]/}
t�d|
� z|
� W qt ty� } zt�| � | �d �|
t�� �� W Y d } ~ qtd } ~ ww | j D ]0} t�d| � z| | � W q� ty� } zt�| � | �d �| t�� �� W Y d } ~ q�d } ~ ww | j � d
d �d ur�t� d | j d
� tj j!�"d � ||fS )
N�
args_stdinTc S s g | ]}|� d ��qS )z
)�rstrip)�.0�argr r r �
<listcomp>� s z+OptionParser.parse_args.<locals>.<listcomp>� log_level�errorr7 �
Processing %sz Error while processing {}: {}Z conf_filezConfiguration file path: %s� MainProcess)#r= rD �
parse_args�__dict__rQ �sys�stdin� readlines�update�extend�versions_report�print_versions_report�optionsrB r
�salt�_loggingZsetup_temp_handlerr r �log�trace� Exception� exceptionrW r' � traceback�
format_excr3 �config�get�debug�utils�process�appendproctitle)
r rB �valuesrc Z
new_inargsZ new_options�new_argsZtemp_log_levelZprocess_option_funcsZ
option_keyZprocess_option_func�errZmixin_after_parsed_funcr r r rZ � sn
�
�
���� �
�����zOptionParser.parse_argsTc C s6 t jj| ||d� | jD ] }t�d|� || � q d S )N)�add_helprX )r= rD �_populate_option_listr1 rf rg )r � option_listru Zmixin_setup_funcr r r rv � s �
�z"OptionParser._populate_option_listc C s" t j�| � | jddddd� d S )Nz--versions-reportz-V�
store_truez4Show program's dependencies version number and exit.)r@ �help)r= rD �_add_version_option�
add_option�r r r r rz � s
�z OptionParser._add_version_optionc C s, t d�t�� �|dd� | �tjjj� d S )N�
T)�file�flush� �print�joinrF ra �exitrd �defaults� exitcodes�EX_OK�r r~ r r r rb � s z"OptionParser.print_versions_reportr c C s� | j D ]1}t�d|� z|| � W q ty4 } zt�|� tjd|t�� tj d� W Y d }~qd }~ww t
j � � }|d urB|�
� t
j �� t|t�rY|rY|d dkrY|� d�}tj�| ||� d S )NrX z Error while processing %s: %s�Zexc_info_on_loglevel���r} )r4 rf rg rh ri rW rj rk �logging�DEBUGrd re Zget_temp_handlerr Zshutdown_temp_handlerr �strr= rD r� )r �status�msgZmixin_before_exit_funcrt Ztemp_log_handlerr r r r� � s. �
���
zOptionParser.exitc C s2 | � tj� | �tjjj| �� � d|� d�� dS )z�
error(msg : string)
Print a usage message incorporating 'msg' to stderr and exit.
This keeps option parsing exit status uniform for all parsing errors.
� : error: r} N) � print_usager\ �stderrr� rd r� r� �EX_USAGErO )r r� r r r rW s
�zOptionParser.error�NN�T�r N)r r r rF Z__saltstack_version__Zformatted_versionrL rG rI rH r\ �maxsizer rM rP rZ rv rz �stdoutrb r� rW r r r r rD � s �
D
�
rD c @ s& e Zd ZdZejZdd� Zdd� ZdS )�MergeConfigMixInz�
This mix-in will simply merge the CLI-passed options, by overriding the
configuration file loaded settings.
This mix-in should run last.
c C s* t | d�s t | d�s d S | j�| j� d S )N� setup_configrl )r% r3 r �(_MergeConfigMixIn__merge_config_with_clir| r r r r" 1 s z MergeConfigMixIn._mixin_setupc C sL | j D ]K}|jd u r q| j�|j�}t| j|j|�}|j| jvr+|d ur*|| j|j< q|d ur<t|dd�r<|| j|j< q|j| jv rNt| j|j| j|j � q| jD ]Q}|j D ]K}|jd u r_qW| j�|j�}t| j|j|�}|j| jvr|d ur~|| j|j< qW|d ur�t|dd�r�|| j|j< qW|j| jv r�t| j|j| j|j � qWqRd S )Nr< F) rw rA r� rm r
rc rl �setattr�
option_groups)r �option�default�value�groupr r r Z__merge_config_with_cli; s>
� �
� ���z(MergeConfigMixIn.__merge_config_with_cliN) r r r �__doc__r\ r� r r" r� r r r r r� & s
r� )� metaclassc @ � e Zd ZdZdd� Zdd� ZdS )�
SaltfileMixIni��c C � | j dd dd� d S )Nz
--saltfileziSpecify the path to a Saltfile. If not passed, one will be searched for in the current working directory.�r� ry �r{ r| r r r r" v s
�zSaltfileMixIn._mixin_setupc
C s� | j jd u rtj�dd �| j _| j jd u rNz tj�t�� d�}W n ty+ d}Y nw tj� |�r7|| j _ntj�tj�
d�dd�}tj� |�rM|| j _n| j j}| j jsXd S tj� | j j�sk| � d| j j� d�� tj� | j j�| j _t
�d| j j� zt�|�}W n, tjjy� } z | � |j� | �tjjj| �� � d |j� d
�� W Y d }~nd }~ww |s�d S | �� |vr�d S || �� }| jD ]6}|jd u r�q�|j|vr�q�| j�|j�}t | j |j|�}||kr�|� |j� q�t | j |j||j � d |_ q�| j D ]D}|jD ]=}|jd u �r�q|j|v�r�q| j�|j�}t | j |j|�}||k�r3|� |j� �qt | j |j||j � d |_ �q�q |D ] } t | j | || � �qGd S ) NZ
SALT_SALTFILEZSaltfile� �~z.salt�'z' file does not exist.
zLoading Saltfile from '%s'r� r} T)!rc �saltfile�os�environrm �pathr� �getcwd�OSError�isfile�
expanduserrW �abspathrf �inforl Z_read_conf_filerd �
exceptionsZSaltConfigurationError�messager� r� r� �
EX_GENERICrO rw rA r
�popr� r< r� )
r r� Zsaltfile_configrW Z
cli_configr� r� r� r� r r r r �process_saltfile� s� �
� ���
��z SaltfileMixIn.process_saltfileN)r r r r r" r� r r r r r� s s
r� c @ s e Zd ZdZdZdd� ZdS )�HardCrashMixin�( Nc C s$ t j�dd�}| jdd|dd� d S )NZSALT_HARD_CRASHFz --hard-crashrx zORaise any original exception rather than exiting gracefully. Default: %default.)r@ r� ry �r� r� rm r{ )r Z
hard_crashr r r r" � s
�zHardCrashMixin._mixin_setup)r r r r �_config_filename_r" r r r r r� � s r� c @ r� )� NoParseMixin�2 c C s$ t j�dd�}| jd|ddd� d S )NZ
SALT_NO_PARSEr� z
--no-parsezpComma-separated list of named CLI arguments (i.e. argname=value) which should not be parsed as Python data typeszargname1,argname2,...)r� ry �metavarr� )r �no_parser r r r" s
�zNoParseMixin._mixin_setupc C sT | j jr$zdd� | j j�d�D �| j _W d S ty# g | j _Y d S w g | j _d S )Nc S � g | ]}|� � �qS r )�strip�rS �xr r r rU s �z1NoParseMixin.process_no_parse.<locals>.<listcomp>�,)rc r� �split�AttributeErrorr| r r r �process_no_parse s � � z NoParseMixin.process_no_parseN)r r r r r" r� r r r r r� � s r� c @ s8 e Zd ZdZdZejZdZdd� Z dd� Z
d
dd �Z dS ) �ConfigDirMixIni����NZSALT_CONFIG_DIRc C s> t j�| jd �}|s| j}t�dtj� | j dd|dd� d S )NzSYSPATHS setup as: %sz-cz --config-dirzDPass in an alternative configuration directory. Default: '%default'.r� )
r� r� rm � _default_config_dir_env_var_�_default_config_dir_rf rn �syspaths�
CONFIG_DIRr{ )r �
config_dirr r r r" s
�zConfigDirMixIn._mixin_setupc
C s� t j�| jj�| j_t j�| jj�s tj�d� | jj�� t j�
| jj�| j_t | d�r[t | d�s4i | _ z | j �
| �� � W d S tyZ } z| �d|� �� W Y d }~d S d }~ww d S )Nz/WARNING: CONFIG '{}' directory does not exist.
r� rl z Failed to load configuration: )r� r� r� rc r� �isdirr\ r� �writer' r� r% rl r_ r� r� rW �r �excr r r �process_config_dir* s$ ��
���z!ConfigDirMixIn.process_config_dirc C s |d u r| j }tj�| jj|�S r )r� r� r� r� rc r� )r �
configfiler r r �get_config_file_path? s z#ConfigDirMixIn.get_config_file_pathr ) r r r r r� r� r� r� r� r" r� r� r r r r r� s r� c @ s� e Zd Zejd ZdZdZdZdZ dZ
dZ dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) �
LogLevelMixIn�d �warningN�log_filerV Zlog_level_logfile)�-l� --log-levelc s � j d u rtd�� jj���t�� dd� }� _� �|� |j � j
� j t t
jj�d�d�dd� t
jjD ��� jd�t� ��d�� � fd d
�}|j d � jd d d
|d� j � d�d� |j d� jt t
jj�d�d�dd� t
jjD ��� jd�t� ��d� � j�� j� d S )Nz'Please set {}._default_logging_logfile_zLogging OptionszOLogging options which override any settings defined on the configuration files.z{Console logging log level. One of {}. Default: '{}'.
The following log levels are INSECURE and may log sensitive data: {}z, c S � g | ]}d |� d ��qS �r� r �rS �nr r r rU i � z.LogLevelMixIn._mixin_setup.<locals>.<listcomp>)rA �choicesry c s2 t j�|�st j�t jj|�}t|j� j|� d S r )r� r� �dirnamer� �curdirr� rr � _logfile_config_setting_name_)r� �optr� �parserrB rC r| r r �_logfile_callbacko s z5LogLevelMixIn._mixin_setup.<locals>._logfile_callbackz
--log-file�callback�stringzLog file path. Default: 'z'.)rA r� r@ �typer� ry z--log-file-levelz{Logfile logging log level. One of {}. Default: '{}'.
The following log levels are INSECURE and may log sensitive data: {}c S r� r� r r� r r r rU � r� )�_default_logging_logfile_r& r' r. r r= � OptionGroupZlogging_options_grouprP r{ � _console_log_level_cli_flags� _loglevel_config_setting_name_r rd re Z
LOG_LEVELSr� ZSORTED_LEVEL_NAMES�_default_logging_level_r r� �&_logfile_loglevel_config_setting_name_r3 r �&_LogLevelMixIn__setup_logging_routines)r r� r� r r| r r" R sZ
��
�
�� �
�� zLogLevelMixIn._mixin_setupc C sX | j �| j� | j �| j� | j �| j� | j �| j� | j �| j� | j�| j� d S r ) r3 r �+_LogLevelMixIn__setup_console_logger_config�+_LogLevelMixIn__setup_logfile_logger_config�$_LogLevelMixIn__setup_logging_config� _LogLevelMixIn__verify_logging� _LogLevelMixIn__setup_loggingr4 � _LogLevelMixIn__shutdown_loggingr| r r r Z__setup_logging_routines� s z&LogLevelMixIn.__setup_logging_routinesc C sX t | j| jd �s*| j�| jd �r t| j| j| j�| j�� d S t| j| j| j� d S d S r )r
rc r� rl rm r� r� r| r r r �process_log_level� s ���z LogLevelMixIn.process_log_levelc C s" t j�� tj�� tj�� d S r )rd re �shutdown_loggingr\ r� r r� r| r r r Z__shutdown_logging� s
z LogLevelMixIn.__shutdown_loggingc C sv t | j| jd �s7| j�| jd �r t| j| j| j�| j�� d S t| j| j| j� | j| jv r9| j�| j� d S d S d S r )r
rc r� rl rm r� r� r� r| r r r �process_log_file� s �� �z LogLevelMixIn.process_log_filec C s� t | j| jd �s=| j�| jd �r t| j| j| j�| j�� d S t| j| j| j�| j| j�� | j| jv r?| j�| j� d S d S d S r ) r
rc r� rl rm r� r� r� r� r| r r r �process_log_level_logfile� s& ��� �z'LogLevelMixIn.process_log_level_logfilec C sl | j �d| j �dtjj��}| j �dd �d u r | j �dd � | j �d| j �dd��}|| j d< || j d< d S )N�log_fmt_console�log_fmt�log_datefmt_console� log_datefmt�%Y-%m-%d %H:%M:%S)rl rm rd re �DFLT_LOG_FMT_CONSOLEr� )r Zlogfmt�datefmtr r r Z __setup_console_logger_config� s ��
z+LogLevelMixIn.__setup_console_logger_configc
C sv | j | jv r| j�| j �s| j�| j � t| j| j | j�}t| j| j| j�}d� | �
� � dd��}|| jv rB| j�|�sB| j�|� | j| jv rV| j�| j�sV| j�| j� | jd r�| jd dvr�| j| j dkr�|d ur�t j
j�d�� t|g| jd � W d � n1 s�w Y |d u r�| j}d
� | �
� � dd��}|| jv r�| j�|�s�| j�|� | j�d d �d u r�| j�d d � | j�d | j�d | j�d
t jj���}| j�dd �d u r�| j�dd � | j�dd �d u r�| j�dd � | j�d| j�d| j�dd���}t|dd��slt j
j�� }| jd |k�rl| jd }t j
jj||�� d��rlt j
j�� } tj � | ��r:| ntj � d�}
tj � |
��sMt� |
d� tj � | j�} t!�"dt#|�t#|�t#| �� tj �$|
| � d��}| j�dd�} | j�d d�}
t j
j%�&� �s�| dk�r�t!�'d � d} |
dk�r�t!�'d � d}
|| j| j< || j| j < || jd < || jd< | | jd< |
| jd < d S ) Nzcli_{}_log_file�-�_Z
verify_envrV )�quietr � �userzcli_{}_log_file_fmtZlog_fmt_logfiler� r� Zlog_datefmt_logfiler� r r T)Z check_parentZ
publisher_acl)Z whitelistz~/.salti� z`The user '%s' is not allowed to write to '%s'. The log file will be stored in '~/.salt/'%s'.log'z.log�log_rotate_max_bytesr �log_rotate_backup_countz3'log_rotate_max_bytes' is only supported on Windowsz6'log_rotate_backup_count' is only supported on Windows)(r� rl rm r� r
rc r� r� r� r' rO rN rd ro �filesZ set_umaskr re r r r Zget_userZ stringutilsZcheck_whitelist_blacklist�keysZxdgZxdg_config_dirr� r� r� r� �makedirs�basenamerf rn r� r� �platform�
is_windowsr� )r ZloglevelZlogfileZ cli_log_pathZcli_log_file_fmtZ log_file_fmtZlog_file_datefmt� current_userZaclZxdg_dirZ
user_salt_dirZlogfile_basenamer r
r r r Z __setup_logfile_logger_config� s� ��� � �� �� ��
�
�
� �
z+LogLevelMixIn.__setup_logfile_logger_configc C sj t �| j�}t| jdd�du |d< | j|d< t| jdd �dkr(d�|d �|d< tj� |� tj�
� d S ) N�daemonFZconfigure_console_loggerZ log_file_key�output�yamlz# {}r� ) �copy�deepcopyrl r
rc r� r' rd re Zset_logging_options_dictZfreeze_logging_options_dict)r Z logging_optsr r r Z__setup_logging_config} s �
� z$LogLevelMixIn.__setup_logging_configc
C sR zt j�� W d S t jjy( } z| �t jjjt |�� W Y d }~d S d }~ww r )
rd re �
setup_loggingr� ZLoggingRuntimeErrorr� r� r� �EX_UNAVAILABLEr� r� r r r Z__setup_logging� s $��z LogLevelMixIn.__setup_loggingc C s t | j� d S r )r rl r| r r r Z__verify_logging� � z LogLevelMixIn.__verify_logging)r r r r\ r� r r� r� r� r� r� r� r" r� r� r� r� r� r� r� r� r� r� r r r r r� E s*
�:
r� c @ � e Zd ZdZdd� ZdS )� RunUserMixin� c C s | j ddd| �� � d�d� d S )N�-u�--userzSpecify user to run �.)ry )r{ rO r| r r r r" � s
�zRunUserMixin._mixin_setupN�r r r r r" r r r r r � � 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 S )!� DaemonMixIn� c C sL | j ddddd| �� � d�d� | j dd tj�tj| �� � d
��d d � d S )
N�-dz--daemonFrx zRun the z
as a daemon.�r� r@ ry z
--pid-file�pidfilez.pidz9Specify the location of the pidfile. Default: '%default'.�rA r� ry )r{ rO r� r� r� r� Z PIDFILE_DIRr| r r r r" � s �
�zDaemonMixIn._mixin_setupc
C s� t | d�rb| j�d�rd| �� rfz t�| jd � W d S tya } z;d}tjj �
� r:tjj � � }tjj �
|�r9d}nt�� s@d}|rVtjd| jd |tjd� W Y d }~d S W Y d }~d S d }~ww d S d S d S )Nrl r& FTz$PIDfile(%s) could not be deleted: %sr� )r% rl rm �
check_pidfiler� �unlinkr� rd ro r r �
win_functionsZget_current_userZis_admin�getuidrf r� r� r� )r rt Z log_errorr r r r r6 � s2 ������z DaemonMixIn._mixin_before_exitc C s& ddl m} || jd | jd � d S )Nr )� set_pidfiler& r )�salt.utils.processr, rl )r r, r r r r, � s zDaemonMixIn.set_pidfilec C � ddl m} || jd �S )z1
Report whether a pidfile exists
r )r( r& )r- r( rl )r r( r r r r( � � zDaemonMixIn.check_pidfilec C r. )z5
Return a pid contained in a pidfile
r )� get_pidfiler& )r- r0 rl )r r0 r r r r0 � r/ zDaemonMixIn.get_pidfilec C sL | j jr$tj�� tjj�� t� � tjj
_ tj� � tjj�
d� d S d S )NrY )rc r rd re r� ro rp Z daemonizer� �getpidZin_mainprocessZ__pid__r rq r| r r r �daemonize_if_required� s
�z!DaemonMixIn.daemonize_if_requiredc C sl | � � r4| �� }tjj�� s!| � � r | �|�r t�� |kr dS dS | � � r4| �|�r4tjj �
� |kr4dS dS )zd
Check if a pid file exists and if it is associated with
a running process.
TF) r( r0 rd ro r r �
is_daemonizedr� �getppidr* Zget_parent_pid)r �pidr r r �
check_running� s" �� ���zDaemonMixIn.check_runningc C s: t j�| jd �}t j�|d d|d �}tjj�|�S )zS
This will stop from more than on prcoess from doing the same task
r& r Zprocess_responsibility_� ) r� r� r� rl r� rd ro rp Z claim_mantle_of_responsibility)r Zresponsibility_filer r r � claim_process_responsibility s
�z(DaemonMixIn.claim_process_responsibilityc C s ddl m} ||�S )Nr )�
os_is_running)r- r9 )r r5 r9 r r r r3 s zDaemonMixIn.is_daemonizedc C s$ t � t j| j� t � t j| j� d S r )�signal�SIGTERM�_handle_signals�SIGINTr| r r r �_install_signal_handlers s z$DaemonMixIn._install_signal_handlersc C s | � � d S r )rZ r| r r r �prepare s zDaemonMixIn.preparec C s | � � | �� d S r )r? r> r| r r r �start s zDaemonMixIn.startc C sV | j j}|tjkr|d7 }n |tjkr|d7 }t�t��d|� | j|� d�d� d S )Nz received a SIGINT.z received a SIGTERM.z %s Exiting.z Exited.)�exitmsg) r. r r: r= r; r� � getLoggerr� �shutdown)r �signumZsigframer� r r r r< s
zDaemonMixIn._handle_signalsr Nc C s | � ||� d S r )r� )r �exitcoderA r r r rC ) r zDaemonMixIn.shutdownr� )r r r r r" r6 r, r( r0 r2 r6 r8 r3 r> r? r@ r< rC r r r r r" � s
r" c @ �, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �TargetOptionsMixInr Nc C s� t �| dd� }| _| �|� |jddddddd � |jd
d ddd d
� |jdddddd
� |jdddddd
� |jdddddd
� |jdddddd
� |jddddd d
� t �| d d � }| _| �|� |jd td d!� | �� d S )"NzTarget OptionszTarget selection options.�-Hz--hostsFrx Z
list_hostszDList all known hosts to currently visible or other specified rosters�r� r@ rA ry z-Ez--pcrezZInstead of using shell globs to evaluate the target servers, use pcre regular expressions.r% �-L�--listzrInstead of using shell globs to evaluate the target servers, take a comma or whitespace delimited list of servers.z-Gz--grainz�Instead of using shell globs to evaluate the target use a grain value to identify targets, the syntax for the target is the grain key followed by a globexpression: "os:Arch*".�-Pz --grain-pcrez�Instead of using shell globs to evaluate the target use a grain value to identify targets, the syntax for the target is the grain key followed by a pcre regular expression: "os:Arch.*".z-Nz --nodegroupzwInstead of using shell globs to evaluate the target use one of the predefined nodegroups to identify a list of targets.�-Rz--rangez�Instead of using shell globs to evaluate the target use a range expression to identify targets. Range expressions look like %cluster.zAdditional Target Optionsz(Additional options for minion targeting.z --delimiterz^Change the default delimiter for matching in multi-level data structures. Default: '%default'.r� )r= r� �target_options_grouprP r{ Z additional_target_options_groupr �_create_process_functions�r r� r r r r" 3 s�
�
��
� � � � �
�
� z TargetOptionsMixIn._mixin_setupc �F � j jD ] }� fdd�}d|j� �}t� |�s t� |t||�� qd S )Nc s t � j| j�r
| j� _d S d S r )r
rc rA �selected_target_option)r� r| r r rp � s �z=TargetOptionsMixIn._create_process_functions.<locals>.processr7 )rN rw rA r% r� r �r r� rp �funcnamer r| r rO � s
��z,TargetOptionsMixIn._create_process_functionsc �R � fdd�� j jD �}t|�dkr!� �d�d�dd� |D ���� � j� jd< d S )Nc s" g | ]
}t � j|j�d u r|�qS r� �r
rc rA �rS r� r| r r rU � �
�z:TargetOptionsMixIn._mixin_after_parsed.<locals>.<listcomp>r7 �EThe options {} are mutually exclusive. Please only choose one of them�/c S r� r �Zget_opt_stringrW r r r rU � � rR )rN rw �lenrW r' r� rR rl �r Zgroup_options_selectedr r| r r5 � s
� ���z&TargetOptionsMixIn._mixin_after_parsed)r r r r rR r" rO r5 r r r r rG - s b rG c @ s e Zd Zdd� Zdd� ZdS )�ExtendedTargetOptionsMixInc C sn t �| � | j}|jdddddd� |jdddd dd
d � |jd d
dddd� |jdddddd� | �� d S )N�-Cz
--compoundFrx a? The compound target option allows for multiple target types to be evaluated, allowing for greater granularity in target matching. The compound target is space delimited, targets other than globs are preceded with an identifier matching the specific targets argument type: salt 'G@os:RedHat and webser* or E@database.*'.r% z-Iz--pillar�
pillar_targetz�Instead of using shell globs to evaluate the target use a pillar value to identify targets, the syntax for the target is the pillar key followed by a glob expression: "role:production*".�r� rA r@ ry z-Jz
--pillar-pcrez�Instead of using shell globs to evaluate the target use a pillar value to identify targets, the syntax for the target is the pillar key followed by a pcre regular expression: "role:prod.*".�-Sz--ipcidrz4Match based on Subnet (CIDR notation) or IP address.)rG r" rN r{ rO rP r r r r" � s@
��
� � z'ExtendedTargetOptionsMixIn._mixin_setupc C s | j jr d| _d S d S )NZpillar)rc ra rR r| r r r �process_pillar_target� s
�z0ExtendedTargetOptionsMixIn.process_pillar_targetN)r r r r" rd r r r r r_ � s 4r_ c @ r )� TimeoutMixIn�
c C s6 t | d�std�| jj���| jddt| jdd� d S )N�default_timeoutz8You need to define the 'default_timeout' attribute on {}�-t� --timeoutz[Change the timeout, if applicable, for the running command (in seconds). Default: %default.)r� r� ry )r% r& r' r. r r{ �intrg r| r r r r" � s
��
�zTimeoutMixIn._mixin_setupNr r r r r re � r! re c @ r )�ArgsStdinMixInrf c C s | j dddddd� d S )Nz --args-stdinFrQ rx zURead additional options and/or arguments from stdin. Each entry is newline separated.rb r� r| r r r r" s
�zArgsStdinMixIn._mixin_setupNr r r r r rk r! rk c @ r )� ProxyIdMixInr� c C s | j dd ddd� d S )Nz --proxyid�proxyidzId for this proxy.�r� rA ry r� r| r r r r" s
�zProxyIdMixIn._mixin_setupN�r r r Z _mixin_prior" r r r r rl r! rl c @ r )�ExecutorsMixInrf c C s, | j ddd ddd� | j ddd dd d� d S )
Nz--module-executorsZmodule_executorsZ
EXECUTOR_LISTzNSet an alternative list of executors to override the one set in minion config.�rA r� r� ry z--executor-optsZ
executor_optsZ
EXECUTOR_OPTSzjSet alternate executor options if supported by executor. Options set by minion config are used by default.r� r| r r r r" s �
�zExecutorsMixIn._mixin_setupNro r r r r rp r! rp c @ r )�
CacheDirMixInr� c C s | j ddddd� d S )Nz
--cachedirz/var/cache/salt/ZcachedirzCache Directoryrn r� r| r r r r" 8 s
�zCacheDirMixIn._mixin_setupNro r r r r rr 5 r! rr c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )�OutputOptionsMixInr� FNc s t �� dd� }� _� �|� |jdddd�� �� �d� |jdd d
d td d � |jd
ddd dd� |jddddddd� |jdddddd � |jd d ddd d � |jd d!d d"d#� |jd$d%d d&d#� � jjD ] }� fd'd(�}d)|j � �}t
� |�s�t � |t ||�� qgd S )*N�Output Options�'Configure your preferred output format.z--outz--outputr zEPrint the output from the '{}' command using the specified outputter.�rA ry z --out-indentz--output-indentZ
output_indentz�Print the output indented by the provided value in spaces. Negative values disables indentation. Only applicable in outputters that support indentation.�rA r� r� ry z
--out-filez
--output-file� output_filez'Write the output to the specified file.r' z--out-file-appendz--output-file-appendrx �output_file_appendFz(Append the output to the specified file.)r@ rA r� ry z
--no-colorz --no-colourzDisable all colored output.r% z
--force-colorz--force-colourzForce colored output.z--state-outputz--state_outputz�Override the configured state_output value for minion output. One of 'full', 'terse', 'mixed', 'changes' or 'filter'. Default: '%default'.r� z--state-verbosez--state_verbosezgOverride the configured state_verbose value for minion output. Set to True or False. Default: %default.c s2 � j �| j�}t� j| j|�du rd S | j� _d S )NF)r� rm rA r
rc �selected_output_option)r� r� r| r r rp � s z0OutputOptionsMixIn._mixin_setup.<locals>.processr7 )
r= r� �output_options_grouprP r{ r' rO rj rw rA r% r� r )r r� r� rp rT r r| r r" H s�
�
�� � �����
�
��z OutputOptionsMixIn._mixin_setupc C s | j j| _d S r )rc r rz r| r r r �process_output� r z!OutputOptionsMixIn.process_outputc
C s� | j jd urS| j jdu rUtj�| j j�rWz tjj� | j jd��
W d � W d S 1 s,w Y W d S t
yR } z| � | j j� d|� �� W Y d }~d S d }~ww d S d S d S )NF�wz: Access denied: ) rc rx ry r� r� r� rd ro r Zfopenr� rW r� r r r �process_output_file� s &�&���z&OutputOptionsMixIn.process_output_filec C sL | j jdks | j jdkrd| j _d S | j jdks | j jdkr$d| j _d S d S )N�True�trueT�False�falseF)rc Z
state_verboser| r r r �process_state_verbose� s �z(OutputOptionsMixIn.process_state_verbosec rU )Nc s4 g | ]}t � j|j�r|j�d �s|jdkr|�qS )Z_outr )r
rc rA �endswithrW r| r r rU � s �
�
�z:OutputOptionsMixIn._mixin_after_parsed.<locals>.<listcomp>r7 rY rZ c S r� r r[ rW r r r rU � r\ rz )r{ rw r] rW r' r� rz rl r^ r r| r r5 � s
� ���z&OutputOptionsMixIn._mixin_after_parsed) r r r r Z_include_text_out_rz r" r| r~ r� r5 r r r r rs A s Z rs c @ r� )�ExecutionOptionsMixInrf c C sB t �| d� }| _|jddd dd� |jddd dd� |jd d
d d d d
d� |jddd dd� |jddd dd� |jdddddd� |jdd ddd d� |jd d d d!d"d#� |jd$d%ddd&d� |jd'd(ddd)d� |jd*d+ddd,d� |jd-d.ddd/d� |jd0ddd1d� |jd2d d3d� |jd4d5d6dd7d8d� | �|� d S )9NzExecution OptionsrJ z
--locationz#Specify which region to connect to.r� �-az--actionz�Perform an action that may be specific to this cloud provider. This argument requires one or more instance names to be specified.�-fz
--function� z<FUNC-NAME> <PROVIDER>z�Perform a function that may be specific to this cloud provider, that does not apply to an instance. This argument requires a provider to be specified (i.e.: nova).)�nargsr� r� ry �-pz --profilez/Create an instance using the specified profile.�-mz--mapz�Specify a cloud map file to use for deployment. This option may be used alone, or in conjunction with -Q, -F, -S or -d. The map can also be filtered by a list of VM names.rH z--hardFrx z�Delete all VMs that are not defined in the map file. CAUTION!!! This operation can irrevocably destroy VMs! It must be explicitly enabled in the cloud config file.r% r$ z --destroyz"Destroy the specified instance(s).z --no-deployTZdeploy� store_falsez2Don't run a deploy script after instance creation.rb rL z
--parallelz1Build all of the specified instances in parallel.r z--update-bootstrapz=Update salt-bootstrap to the latest stable bootstrap release.�-y� --assume-yes�6Default "yes" in answer to all confirmation questions.z-kz
--keep-tmpz8Do not remove files from /tmp/ after deploy.sh finishes.z--show-deploy-argszCInclude the options used to deploy the minion in the data returned.z
--script-argszIScript arguments to be fed to the bootstrap script when deploying the VM.�-bz --bootstrapr7 z <HOST> [MINION_ID] [OPTIONS...]z Bootstrap an existing machine.)r= r� Zexecution_groupr{ rP rP r r r r" � s�
���
� ��
� ���������z"ExecutionOptionsMixIn._mixin_setupc C sD | j jr | j j\| _| _| j�d�sd| jv r | �d� d S d S d S )Nr �=z<--function expects two arguments: <function-name> <provider>)rc �functionZ
function_nameZfunction_providerr9 rW r| r r r �process_functionP s ��z&ExecutionOptionsMixIn.process_functionN)r r r r r" r� r r r r r� � s |r� c @ rF ) �CloudQueriesMixInr Nc C s� t �| d� }| _|jdddddd� |jdd ddd
d� |jd d ddd
d� |jddddd� |jdd ddd� | �|� | �� d S )Nz
Query Optionsz-Qz--queryFrx zbExecute a query and return some information about the nodes running on configured cloud providers.r% �-Fz --full-queryzaExecute a query and return all information about the nodes running on configured cloud providers.rc z--select-queryzdExecute a query and return select information about the nodes running on configured cloud providers.�--list-providersz'Display a list of configured providers.z--list-profiles�storez�Display a list of configured profiles. Pass in a cloud provider to view the provider's associated profiles, such as digitalocean, or pass in "all" to list all the configured profiles.)r= r� �cloud_queries_groupr{ rP rO rP r r r r" ^ sN
��
�
�
��
z CloudQueriesMixIn._mixin_setupc rQ )Nc s� t � j| j�rJd}| jdkr|d7 }n2| jdkr |d7 }n(| jdkr-d}� jr,� �d� n| jdkrEd}t� j�}|�d�d krE� �d
� |� _d S d S ) NZ
list_nodesZ
full_queryZ_fullZ select_query�_selectZlist_providersz0'--list-providers' does not accept any argumentsZ
list_profilesr� zC'--list-profiles' does not accept '--list-providers' as an argument)r
rc rA rB rW �varsrm �selected_query_option)r� �queryZ option_dictr| r r rp � s,
��
�
�z<CloudQueriesMixIn._create_process_functions.<locals>.processr7 )r� rw rA r% r� r rS r r| r rO � s
��z+CloudQueriesMixIn._create_process_functionsc rU )Nc s4 g | ]}t � j|j�d urt � j|j�dur|�qS )FNrV rW r| r r rU � s �z9CloudQueriesMixIn._mixin_after_parsed.<locals>.<listcomp>r7 rY rZ c S r� r r[ rW r r r rU � r\ r� )r� rw r] rW r' r� r� rl r^ r r| r r5 � s
� ���z%CloudQueriesMixIn._mixin_after_parsed)r r r r r� r" rO r5 r r r r r� Y s 8 r� c @ r� )�CloudProvidersListsMixInr# c C sP t �| d� }| _|jdd dd� |jdd dd� |jdd dd� | �|� d S ) NzCloud Providers Listingsz--list-locationsz�Display a list of locations available in configured cloud providers. Pass the cloud provider that available locations are desired on, such as "linode", or pass "all" to list locations for all configured cloud providers.r� z
--list-imagesz�Display a list of images available in configured cloud providers. Pass the cloud provider that available images are desired on, such as "linode", or pass "all" to list images for all configured cloud providers.z --list-sizesz�Display a list of sizes available in configured cloud providers. Pass the cloud provider that available sizes are desired on, such as "AWS", or pass "all" to list sizes for all configured cloud providers.)r= r� �providers_listings_groupr{ rP rP r r r r" � s(
��
�
�
z%CloudProvidersListsMixIn._mixin_setupc sJ � fdd�� j jD �}t|�dkr#� �d�d�dd� |D ���� d S d S )Nc s" g | ]
}t � j|j�d ur|�qS r rV rW r| r r rU � rX z@CloudProvidersListsMixIn._mixin_after_parsed.<locals>.<listcomp>r7 rY rZ c S r� r r[ rW r r r rU � r\ )r� rw r] rW r' r� )r Zlist_options_selectedr r| r r5 � s
� ����z,CloudProvidersListsMixIn._mixin_after_parsedN)r r r r r" r5 r r r r r� � s &r� c @ r )�ProfilingPMixIn� c C sF t �| d� }| _|jddddd� |jddd d
d d � | �|� d S )
NzProfiling supportz--profiling-pathZprofiling_pathz
/tmp/statszFFolder that will hold all stats generations path. Default: '%default'.r' z--enable-profilingZprofiling_enabledFrx z>Enable generating profiling stats. See also: --profiling-path.�rA r� r@ ry )r= r� Zprofiling_groupr{ rP rP r r r r" s$
���z ProfilingPMixIn._mixin_setupNr r r r r r� r! r� c @ r� )�CloudCredentialsMixInr# c C s4 t �| d� }| _|jdd dddd� | �|� d S )NzCloud Credentialsz--set-passwordr� z<USERNAME> <PROVIDER>z�Configure password for a cloud provider and save it to the keyring. PROVIDER can be specified with or without a driver, for example: "--set-password bob rackspace" or more specific "--set-password bob rackspace:openstack" Deprecated.)r� r� r� ry )r= r� Zcloud_credentials_groupr{ rP rP r r r r" s
��
z"CloudCredentialsMixIn._mixin_setupc C s | j jrtd��d S )NzwThis functionality is not supported; please see the keyring module at https://docs.saltproject.io/en/latest/topics/sdb/)rc Z set_passwordr& r| r r r �process_set_password5 s
��z*CloudCredentialsMixIn.process_set_passwordN)r r r r r" r� r r r r r� s r� c @ r )�
EAuthMixInr# c C sr t �| d� }| _|jdddddddd � |jd
d d d
ddd� |jddddd� |jddddd� | �|� d S )NzExternal Authenticationr� z--authz--eauthz--external-authr� Zeauthz1Specify an external authentication system to use.rn z-Tz --make-tokenFZmktokenrx z�Generate and save an authentication token for re-use. The token is generated and made available for the period defined in the Salt Master.rb z
--username�usernamer7 z%Username for external authentication.)rA r� ry z
--password�passwordz%Password for external authentication.)r= r� Z eauth_groupr{ rP rP r r r r" @ sD
�� � ��zEAuthMixIn._mixin_setupNr r r r r r� = r! r� c @ r� )�JIDMixinr# c C r� )Nz--jidz0Pass a JID to be used instead of generating one.r� r� r| r r r r" n s
�zJIDMixin._mixin_setupc C s@ | j jd ur tjj�| j j�s | �d| j j� d�� d S d S d S )Nr� z' is not a valid JID)rc Zjidrd ro Zis_jidrW r| r r r � process_jidu s
�zJIDMixin.process_jidN)r r r r r" r� r r r r r� j s r� c @ �& e Zd ZdZdZejd Zdd� ZdS )�MasterOptionParserz1The Salt Master, used to control the Salt Minions�masterr� c C � t �| �� �S r �rl �
master_configr� r| r r r r� � r z MasterOptionParser.setup_configN) r r r rH r� rl �DEFAULT_MASTER_OPTSr� r� r r r r r� { s
r� c @ r� )�MinionOptionParserz<The Salt Minion, receives commands from a remote Salt Master�minionr� c C s t j| �� ddd�S )NTF)�cache_minion_idZignore_config_errors)rl �
minion_configr� r| r r r r� � s
�z MinionOptionParser.setup_configN) r r r rH r� rl �DEFAULT_MINION_OPTSr� r� r r r r r� � s
r� c @ r� )�ProxyMinionOptionParserz~The Salt Proxy Minion, connects to and controls devices not able to run a minion.
Receives commands from a remote Salt Master.�proxyr� c C s8 z| j j}W n ty d }Y nw tj| �� d|d�S )NF�r� � minion_id)rr rm r� rl � proxy_configr� )r r� r r r r� � s �
�z$ProxyMinionOptionParser.setup_configN) r r r rH r� rl ZDEFAULT_PROXY_MINION_OPTSr� r� r r r r r� � s
�
r� c @ s4 e Zd ZdZdZdZejd Zeje Z dd� Z
dS )�SyndicOptionParserz�The Salt Syndic daemon, a special Minion that passes through commands from a
higher Master. Scale Salt to thousands of hosts or across many different networks.r� Zsyndic_log_filerV c C s t �| �� | �d��S )Nr� )rl Z
syndic_configr� r| r r r r� � s �z SyndicOptionParser.setup_configN) r r r rH r� r� rl r� r� r� r� r r r r r� � s �
� r� c @ sr e Zd ZdZdZdZdZejd Z ejd Z
ze � � W n e
y* e�d� Y nw dd � Zd
d � Zd d
� ZdS )�SaltCMDOptionParser� z�Salt allows for commands to be executed across a swath of remote systems in
parallel, so they can be both controlled and queried with ease.�1%prog [options] '<target>' <function> [arguments]r� rV r� z1Cannot access current working directory. Exiting!c C s� | j dddddd� | j ddddd d� | j d
ddd d� | j d dd
ddd� | j ddtdd� | j dddddd� | j dddddd � | j d ddd d� | j d d d!d"d#d$d%� | j d&dd'td(d)� | j d*dd+td,d)� | j d-d.d/d0d%� | j d1d"d2d3d4� | j d5d"d6d3d4� | j d7i d8d9d4� | j d:d;d<d=ddd>d � | j d?d@dAdBdC� | j dDdEdddFd � | j dGd"dHdId4� | j dJdKdddLdM� | j dNdOdddPdM� | j dQdRdddSdM� d S )TN�-s�--staticFrx �>Return the data from minions as a group after they all return.r% r� z
--progressz@Display a progress graph. Requires "progressbar" python package.z
--failhardz-Stop batch execution upon first "bad" return.�--async�asyncz0Run the salt command but don't wait for a reply.rb z--subsetr z�Execute the routine on a random subset of the targeted minions. The minions will be verified that they have the named function before executing.�r� r� ry �-v� --verbosez>Turn on command verbosity, display jid and active job queries.z--hide-timeoutZ show_timeoutTr� zHide minions that timeout.r� z
--show-jidz7Display jid without the additional output of --verbose.r� z--batchz --batch-sizer� Zbatchz�Execute the salt job in batch mode, pass either the number of minions to batch at a time, or the percentage of minions to have running.rn z --batch-waitZ
batch_waitzpWait the specified time in seconds after each job is done before freeing the slot in the batch for the next one.�r� rA r� ry z--batch-safe-limitZbatch_safe_limitz`Execute the salt job in batch mode if the job would have executed on at least this many minions.z--batch-safe-size� Zbatch_safe_sizez=Batch size to use for batch jobs created by batch-safe-limit.�--return�RETURNERz�Set an alternative return method. By default salt will send the return data from the command back to the master, but the return data can be redirected into any number of systems, databases or applications.�r� r� ry z--return_configZ
RETURNER_CONFz--return_kwargsZRETURNER_KWARGSz-Set any returner options at the command line.r$ �--doc�--documentation�doc�[Return the documentation for the specified module or for all modules if none are specified.z--args-separator�args_separatorr� z�Set the special argument used as a delimiter between command arguments of compound commands. This is useful when one wants to pass commas as arguments to some of the commands in a compound command.r' z --summaryZ cli_summaryz1Display summary information about a salt command.�
--metadata�METADATA�-Pass metadata into Salt, used to search jobs.�
--output-diff�state_output_diff�+Report only those states that have changed.�rA r@ r� ry z
--config-dump� config_dumpz$Dump the master configuration valuesz--preview-target�preview_targetzHShow the minions expected to match a target. Does not issue any command.)r{ rj �floatr| r r r r" s( �����
���� �
�
�� � �� � ����
�z SaltCMDOptionParser._mixin_setupc C s4 t | j�dkr-| jjs-| jjs-z| �� W n ty, tj� d� t�
t j j
j� Y nw | jjrMt�| �� �}tj� t jjj|dd�� t�
t j j
j� | jjrW| j�d� | jjr�| jse| j�dd� t | j�dk rs| j�dd � | jd d kr�| j�dd � t | j�d
kr�| �d � | jjr�z#d | jd v r�| jd �d
d�� d �| jd< n
| jd � � | jd< W n) t y� | �
dd� Y nw z
| jd | jd< W n t y� | �
dd� Y nw | j�r�z#d | jd v �r�| jd � d �| jd< g g| jd< d}| jdd � � | jj �t | jd �d k�r@| jdd � D ]"}|| jj k�r3|d7 }| jd �g � �q | jd | �|� �q n|| jdd � D ]I}| jj |v �r�|� | jj �}t |�D ]'\}}|�rl| jd | �|� |t |�d k�r�|d7 }| jd �g � �q[�qG| jd | �|� �qGt | jd �t | jd �k�r�| �
dd� nt | jd �t | jd �k �r�| �
dd� t!t | jd ��D ]}t jjj"| jd | | jj#d�| jd |< �q�W d S | jd | jd< | jdd � | jd< t jjj"| jd | jj#d�| jd< W d S t �y | �
dd� Y d S w d S )Nr7 z/Invalid options passed. Please try -h for help.F)Zdefault_flow_styleZnot_a_valid_commandr �*r� zsys.doc� z:You can only get documentation for one method at one time.r� � r� �tgt�* z4
Cannot execute command without defining a target.
�funrT z@Cannot execute compound command without defining all arguments.
zCCannot execute compound command with more arguments than commands.
)r� z
Incomplete options passed.
)$r] rB rc r� r� �
print_helprh r\ r� r� r� rd r� r� r� r� rl r� r� ro r Z safe_dumpr� r �insertrW r rN r� �
IndexError�countr� � enumerate�rangeZ parse_inputr� )r ZcfgZ cmd_indexrT Zsub_argsZ
sub_arg_indexZsub_arg�ir r r r5 � s� �� ��
� � � ���� � ������z'SaltCMDOptionParser._mixin_after_parsedc C r� r �rl �
client_configr� r| r r r r� 4 r z SaltCMDOptionParser.setup_configN)r r r rg rH rG r� rl r� r� r� r� r� r� r\ r� r" r5 r� r r r r r� � s" �
� : pr� c @ �H e Zd ZdZdZdZdZejd Z ejd Z
dd� Z d d
� Z d d � Z
d
S )�SaltCPOptionParserz�salt-cp is NOT intended to broadcast large files, it is intended to handle text
files. salt-cp can be used to distribute configuration files.z&%prog [options] '<target>' SOURCE DESTr� r� rV r� c C sF t �| d�}|jddddddd� |jd d
d d d
dd� | �|� d S )Nz File Optionsr` z --chunkedF�chunkedrx zZUse chunked files transfer. Supports big files, recursive lookup and directories creation.rb z-nz--no-compressionT�gzipr� zDisable gzip compression.)r= r� r{ rP )r Zfile_opts_groupr r r r" T s$ � �z SaltCPOptionParser._mixin_setupc C s� t | j�dkr| �� | �d� | jjr2d| jd v r'| jd �d�| jd< n| jd �� | jd< n| jd | jd< dd� | jdd� D �| jd < | jd | jd
< d S ) Nr7 �Insufficient argumentsr� r r� c S s g | ]}t j�|��qS r )r� r� �realpathr� r r r rU x r� z:SaltCPOptionParser._mixin_after_parsed.<locals>.<listcomp>r� �srcrA )r] rB r� rW rc r r� rl r| r r r r5 k s
z&SaltCPOptionParser._mixin_after_parsedc C r� r r� r| r r r r� { r z SaltCPOptionParser.setup_configN)r r r rH rG rg r� rl r� r� r� r" r5 r� r r r r r� 8 s
�
r� c s~ e Zd ZdZdZdZdZeje Z dd� Z
� fdd�Z e j
e _
d d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Z� ZS )�SaltKeyOptionParserz3salt-key is used to manage Salt authentication keysr� )r� Z key_logfilec C s< t �| d�}|�d� |jdddddd� |jd d
d d d
d� |jddddd� |jddd d dd� |jddddd� |jddd d dd� |jd d d d d� |jd d d d d� |jd d d d!d� |jd"d d d#d� |jd$d%dd&d� |jd'd(d d d)d� |jd*d+dd,d� |jd-d.d d d/d� |jd0d1dd2d� |jd3d4d d d5d� | �|� | jd6d7d d d8d� | jd9d:d d d;d� | jd<d=d>d� | jd?d d@d� t �| dA�}| �|� |jdBddCd� |jdDdEdFd� |jdGdHtdIdJ� |jdKd d dLd� |jdMdtdNdJ� |jdOdtdPdJ� |jdQdtdRdJ� |jdSd d dTd� d S )UNZActions�resolver� rK r� ZARGa
List the public keys. The args 'pre', 'un', and 'unaccepted' will list unaccepted/unsigned keys. 'acc' or 'accepted' will list accepted/signed keys. 'rej' or 'rejected' will list rejected keys. 'den' or 'denied' will list denied keys. Finally, 'all' will list all keys.r� rJ z
--list-allFrx z3List all public keys. Deprecated: use "--list all".r% r� z--acceptz�Accept the specified public key (use --include-rejected and --include-denied to match rejected and denied keys in addition to pending keys). Globs are supported.r� z-Az --accept-allzAccept all pending keys.�-rz--rejectz�Reject the specified public key. Use --include-accepted and --include-denied to match accepted and denied keys in addition to pending keys. Globs are supported.rM z --reject-allzReject all pending keys.z
--include-allzwInclude rejected/accepted keys when accepting/rejecting. Deprecated: use "--include-rejected" and "--include-accepted".z--include-acceptedz%Include accepted keys when rejecting.z--include-rejectedz%Include rejected keys when accepting.z--include-deniedz-Include denied keys when accepting/rejecting.r� z--printz Print the specified public key.rL z --print-allzPrint all public keys.r$ z--deletez.Delete the specified key. Globs are supported.z-Dz --delete-allzDelete all keys.r� z--fingerz&Print the specified key's fingerprint.r� z --finger-allz Print all keys' fingerprints.z-q�--quietzSuppress output.r� z--yesz;Answer "Yes" to all questions presented. Default: %default.z--rotate-aes-keyTz�Setting this to False prevents the master from refreshing the key session when keys are deleted or rejected, this lowers the security of the key deletion/rejection operation. Default: %default.z--preserve-minionsz�Setting this to True prevents the master from deleting the minion cache when keys are deleted, this may have security implications if compromised minions auth with a previous deleted minion ID. Default: %default.zKey Generation Optionsz
--gen-keysz3Set a name to generate a keypair for use with salt.z--gen-keys-dirr zjSet the directory to save the generated keypair, only works with "--gen-keys" option. Default: '%default'.z --keysize� z�Set the keysize for the generated key, only works with the "--gen-keys" option, the key size must be 2048 or higher, otherwise it will be rounded up to 2048. Default: %default.r� z--gen-signaturea< Create a signature file of the masters public-key named master_pubkey_signature. The signature can be send to a minion in the masters auth-reply and enables the minion to verify the masters public-key cryptographically. This requires a new signing-key-pair which can be auto-created with the --auto-create parameter.�--privz0The private-key file to create a signature with.z--signature-pathz4The path where the signature file should be written.z--pubz.The public-key file to create a signature for.z
--auto-createz8Auto-create a signing key-pair if it does not yet exist.)r= r� Zset_conflict_handlerr{ rP rj r� )r Z
actions_groupZkey_options_groupr r r r" � sP
��� �� ��
���������
��� �
�� � ����
�z SaltKeyOptionParser._mixin_setupc sR | j jr"t�| j jtj�s"tj�| j j�s t� | j j� | j j| j _t
� � � d S r ) rc �gen_keysr� �accessr� �R_OKr� r� � gen_keys_dirr
r# r� r| r- r r r� �
s z&SaltKeyOptionParser.process_config_dirc C s& t �| �� �}| jjr| jj|d< |S )NZpki_dir)rl r� r� rc r� r� )r Z keys_configr r r r� �
s z SaltKeyOptionParser.setup_configc C �^ t | jd�r)t| jjt�r+| jj�� dkrd| j_d S | jj�� dkr-d| j_d S d S d S d S )N�rotate_aes_keyr� Tr� F)r% rc r r� r� �lowerr| r r r �process_rotate_aes_key�
� � �z*SaltKeyOptionParser.process_rotate_aes_keyc C r� )N�preserve_minionsr� Tr� F)r% rc r r� r� r� r| r r r �process_preserve_minions�
r� z,SaltKeyOptionParser.process_preserve_minionsc C s8 | j jsd S | j j�d�s| �d| j j� d�� d S d S )N)�accZpreZunZrejZden�allr� z%' is not a valid argument to '--list')rc r r9 rW r| r r r � process_list�
s
�z SaltKeyOptionParser.process_listc C s8 | j jdk r
| �d� d S | j jdkr| �d� d S d S )Nr� z%The minimum value for keysize is 2048i � z&The maximum value for keysize is 32768)rc ZkeysizerW r| r r r �process_keysize�
s
�z#SaltKeyOptionParser.process_keysizec C s | j jr
| j�| j� d S d S r )rc r� r3 r �%_SaltKeyOptionParser__create_keys_dirr| r r r �process_gen_keys_dir�
s
��z(SaltKeyOptionParser.process_gen_keys_dirc C s* t j�| jd �st �| jd � d S d S )Nr� )r� r� r� rl r
r| r r r Z__create_keys_dir�
s �z%SaltKeyOptionParser.__create_keys_dir)r r r rH r� r� r� rl r� r� r" r� r� r r� r� r r r r r r/ r r r- r r� s&
� s r� c @ sL e Zd ZdZdZdZejd Zejd Z dd� Z
dd � Z d
d � Z d d
� Z
dS )�SaltCallOptionParserzFsalt-call is used to execute module functions locally on a Salt Minion�&%prog [options] <function> [arguments]r� rV r� c C sX | j ddddddd� | j dd g d
d d � | j d
ddddddd� | j ddddd� | j ddddd� | j dddd d � | j d d d d � | j d d d!d � | j d"d d#d � | j d$ddd%d � | j d&dd'dd(d)� | j d*d+d d,d-d.� | j d/dd0d1d� | j d2ddd3d � | j d4ddd5d � | j d6ddd7d � | j d8d9d:d;td<d=� | j d>d?ddd@dA� d S )BNz-gz--grains�
grains_runFrx z4Return the information generated by the salt grains.r� r� z
--module-dirsr z�Specify an additional directory to pull modules from. Multiple directories can be provided by passing `-m/--module-dirs` multiple times.r% r$ r� r� r� r� z--masterr� r� z�Specify the master to use. The minion must be authenticated with the master. If this option is omitted, the master options from the minion config will be used. If multi masters are set up the first listed master that responds will be used.rn r� r� zISet salt-call to pass the return data to one or many returner interfaces.r� z--localz9Run salt-call locally, as if there was no master running.z --file-rootz)Set this directory as the base file root.r� z
--pillar-rootz+Set this directory as the base pillar root.z --states-dirz3Set this directory to search for additional states.z--retcode-passthroughz@Exit with the salt call retcode and not the salt binary retcode.r� Zprint_metadatazuPrint out the execution metadata as well as the return. This will print out the outputter data, the return code, etc.rb z--set-metadata�metadatar� r� rq z--id�idzkSpecify the minion id to use. If this option is omitted, the id option from the minion config will be used.�
--skip-grains�Do not load grains.z--refresh-grains-cachez$Force a refresh of the grains cache.z--no-return-eventz/Do not produce the return event back to master.rh ri �< Z auth_timeoutzNChange the timeout, if applicable, for the running command. Default: %default.r� r� r� r� r� )r{ rj r| r r r r" �
s� �� � � � ������ �� ����
�z!SaltCallOptionParser._mixin_setupc C s� | j s| jjs| jjs| �� | �d� d S t| j �dkrJ| jjr&| �d� | jjr6t| j �dkr6| �d� | j d | jd< | j dd � | jd< d S d S )Nz$Requires function, --grains or --docr7 z)-g/--grains does not accept any arguments�9You can only get documentation for one method at one timer r� rT )rB rc r r� r� rW r] rl r| r r r r5 | s
�z(SaltCallOptionParser._mixin_after_parsedc C s> | j jrtj| jdd�d| j jd�}|S tj| �� dd�}|S )Nr� )r� Tr� )r� )rc rm rl r� r� r� )r �optsr r r r� � s
� ��z!SaltCallOptionParser.setup_configc C s\ | j jD ]'}d|v r | j�dg ��dd� |�d�D �� q| j�dg ��tj� |�� qd S )Nr� � module_dirsc s s � | ] }t j�|�V qd S r )r� r� r� r� r r r � <genexpr>� s �
�z;SaltCallOptionParser.process_module_dirs.<locals>.<genexpr>)
rc r rl rK r` r� r r� r� r� )r Z
module_dirr r r �process_module_dirs� s �
��z(SaltCallOptionParser.process_module_dirsN)r r r rH rG r� rl r� r� r� r" r5 r� r r r r r r �
s �
r c @ r� )�SaltRunOptionParserr7 z�salt-run is the frontend command for executing Salt Runners.
Salt Runners are modules used to execute convenience functions on the Salt Masterr r� rV r� c C sp | j dddddddd� | j d ddd
d � | j d ddd
d � t�| dd� }| _| �|� |j ddddd � d S )Nr$ r� r� r� Frx zyDisplay documentation for runners, pass a runner or runner.function to see documentation on only that runner or function.r� r� z:Start the runner operation and immediately return control.r% r r
rt ru r� z&Do not display the results of the run.)r{ r= r� r{ rP rP r r r r" � s> �
��
�
�z SaltRunOptionParser._mixin_setupc C sv | j jrt| j�dkr| �d� | jr | jd | jd< nd| jd< t| j�dkr4| jdd � | jd< d S g | jd< d S )Nr7 r r r� r� rT )rc r� r] rB rW rl r| r r r r5 � s
z'SaltRunOptionParser._mixin_after_parsedc C r� r r� r| r r r r� � r z SaltRunOptionParser.setup_configN)r r r rg rH rG r� rl r� r� r� r" r5 r� r r r r r � s �
&
r c @ sD e Zd ZdZdZdZejd Zeje Z dd� Z
dd� Z d d
� Z d S ) �SaltSSHOptionParserr� r� Z ssh_log_filerV c C s� | j dddddddd� | j d d
d d d
� | j ddddd
� | j ddddddd� | j dddtdd� | j dd d d d
� | j d d d d d
� | j d!d"d d#d
� | j d$d%ddd&d'� | j d(d)ddd*d'� | j d+d,ddd-d.d/� | j d0d1ddd2d'� | j d3d4d5d6ddd7d� | j d8d9d:d;� | j d<d=d>d;� | j d?ddd@dAd/� t�| dBdC�}|j dDdEdFdG� |j dHdIdJdKdL� | �|� t�| dMdN�}|j dOdPdQdG� |j dRdSddTd
� |j dUdVdWdddXd� |j dYdZddd[d� |j d\d]d^d_d
� |j d`daddbd
� |j dcddddded� |j dfdgdddhd� |j didjdddkd� |j dldmdddnd� |j dodpdddqd� | �|� t�| drds�}|j dtdudvdwdx� |j dydzd{d|dx� | �|� d S )}Nr� z--rawz --raw-shellZ raw_shellFrx zIDon't execute a salt routine on the targets, execute a raw shell command.r� z--rosterZrosterZflatz�Define which roster system to use, this defines if a database backend, scanner, or custom roster system is used. Default: 'flat'.r' z
--roster-fileZ roster_filer� z�Define an alternative location for the default roster file location. The default roster file is called roster and is found in the same directory as the master config file.z --refreshz--refresh-cacheZ
refresh_cachez�Force a refresh of the master side data cache of the target's data. This is needed if a target's grains have been changed and the auto refresh timeframe has not been reached.z --max-procsZ
ssh_max_procs� z�Set the number of concurrent minions to communicate with. This value defines how many processes are opened up at a time to manage connections, the more running processes the faster communication should be. Default: %default.rw z--extra-filerefsZextra_filerefsz4Pass in extra files to include in the state tarball.z--min-extra-modulesZmin_extra_modszUOne or comma-separated list of extra Python modules to be included into Minimal Salt.z--thin-extra-modulesZthin_extra_modszROne or comma-separated list of extra Python modules to be included into Thin Salt.r� r� z'Turn on command verbosity, display jid.r% r� r� r� z-wz--wipeZssh_wipez<Remove the deployment of the salt files when done executing.rI z-Wz--rand-thin-dirzeSelect a random temp dir to deploy on the remote system. The dir will be cleaned after the execution.rh z --regen-thinz--thinZ
regen_thinzoTrigger a thin tarball regeneration. This is needed if custom grains/modules/states have been added or updated.z
--python2-binZpython2z2Path to a python2 binary which has salt installed.r� z
--python3-binZpython3z2Path to a python3 binary which has salt installed.z --pre-flightZssh_run_pre_flightz3Run the defined ssh_pre_flight script in the rosterz SSH Optionsz Parameters for the SSH client.z--remote-port-forwardsZssh_remote_port_forwardsz�Setup remote port forwarding using the same syntax as with the -R parameter of ssh. A comma separated list of port forwarding definitions will be translated into multiple -R parameters.rv z --ssh-optionZ ssh_optionsr z�Equivalent to the -o ssh command option. Passes options to the SSH client in the format used in the client configuration file. Can be used multiple times.)rA r@ ry zAuthentication Optionsz$Parameters affecting authentication.r� Zssh_privzSsh private key file.z
--priv-passwdZssh_priv_passwdz$Passphrase for ssh private key file.z-iz--ignore-host-keysZignore_host_keysz}By default ssh host keys are honored and connections will ask for approval. Use this option to disable StrictHostKeyChecking.z--no-host-keysZ no_host_keysz=Removes all host key checking functionality from SSH session.r Zssh_user�rootz;Set the default user to attempt to use when authenticating.z--passwd�
ssh_passwdz?Set the default password to attempt to use when authenticating.z --askpass� ssh_askpasszlInteractively ask for the SSH password with no echo - avoids password in process args and stored in history.z --key-deployZssh_key_deployz�Set this flag to attempt to deploy the authorized ssh key with all minions. This combined with --passwd can make initial deployment of keys very fast and easy.z--identities-onlyZssh_identities_onlyzyUse the only authentication identity files configured in the ssh_config files. See IdentitiesOnly flag in man ssh_config.z--sudoZssh_sudozRun command via sudo.z--update-rosterZssh_update_rosterzbIf hostname is not found in the roster, store the information into the default roster file (flat).zScan Roster Optionsz!Parameters affecting scan roster.z --scan-portsZ22Zssh_scan_portsz9Comma-separated list of ports to scan in the scan roster.rn z--scan-timeoutg{�G�z�?Zssh_scan_timeoutz,Scanning socket timeout for the scan roster.)r{ rj r= r� rP )r Z ssh_groupZ
auth_groupZ
scan_groupr r r r" s� � �
� �
� �� � ����
� �����
�
��� ����
� �
��
���z SaltSSHOptionParser._mixin_setupc C s | j s | �� | �d� | jjr.d| j d v r#| j d �d�| jd< n| j d �� | jd< n| j d | jd< | j dd � | jd< | jd rJ| jd sS| �� | �d� | jjrf| jd �dd� | jj� � | jj
r�t � d�| j_ | j
D ]}|jD ] }|jd kr�d
|_ nqyqtd S d S ) Nr� r� r r� r7 �argvz --no-parse=z
Password: r T)rB r� rW rc r r� rl r� r r� r �getpassr r� rw rA r< )r r� r� r r r r5 /
s2
���z'SaltSSHOptionParser._mixin_after_parsedc C r� r r� r| r r r r� N
r z SaltSSHOptionParser.setup_configN)
r r r rG r� r� rl r� r� r� r" r5 r� r r r r r � s
� r c s\ e Zd ZdZdZdZejd Zejd Z e
j fdd�Z d� fd d
� Z
d d � Zd
d� Z� ZS )�SaltCloudParserz�Salt Cloud is the system used to provision virtual machines on various public
clouds via a cleanly controlled profile and mapping systemz8%prog [options] <-m MAP | -p PROFILE> <NAME> [NAME2 ...]ZcloudrV r� c C s. t d�tjdd��|d� | �tjjj� d S )Nr} T)Zinclude_salt_cloud)r~ r� r� r r r rb o
s z%SaltCloudParser.print_versions_reportNc
sV z ddl m} |�� W n ty# } z | �|� W Y d }~nd }~ww t� �||�S )Nr )�
libcloudfuncs)Z
salt.cloudr Zcheck_libcloud_version� ImportErrorrW r# rZ )r rB rr r r� r- r r rZ u
s ��zSaltCloudParser.parse_argsc C sP dt jv rdd l}td� |�| j� | �tjjj � | j
r&| j
| jd< d S d S )NZDUMP_SALT_CLOUD_CONFIGr z7Salt Cloud configuration dump (INCLUDES SENSIBLE DATA):�names) r� r� �pprintr� rl r� rd r� r� r� rB )r r r r r r5
s
�z#SaltCloudParser._mixin_after_parsedc
C sF zt �| �� �W S tjjy" } z | �|� W Y d }~d S d }~ww r )rl Z cloud_configr� rd r� ZSaltCloudConfigErrorrW r� r r r r� �
s ��z SaltCloudParser.setup_configr� )r r r rH rG r� rl ZDEFAULT_CLOUD_OPTSr� r� r\ r� rb rZ r5 r� r/ r r r- r r R
s �
�
r c @ sB e Zd ZdZdZdZdZdZej e Z
dd� Z dd � Z d
d � Z
d S )
� SPMParserzD
The CLI parser object used to fire up the Salt SPM system.
zBSPM is used to manage 3rd party formulas and other Salt componentsz%%prog [options] <function> <argument>ZspmZ spm_logfilec C s@ | j dddddd� | j dddddd� | j d d
ddd d� d S ) Nr� r� Frx r� r% r� z--forcer� r� z"Display more detailed information.r� r| r r r r" �
s* ��
�zSPMParser._mixin_setupc C s@ t | j�dkr | jr| jd dvr | �� | �d� d S d S d S )Nr7 r )Z update_repor� )r] rB r� rW r| r r r r5 �
s �z SPMParser._mixin_after_parsedc C � t j�| �� �S r )rd rl Z
spm_configr� r| r r r r� �
r zSPMParser.setup_configN)r r r r� rH rG r� r� rl ZDEFAULT_SPM_OPTSr� r" r5 r� r r r r r! �
s
r! c @ s. e Zd ZdZdZdZdZeje Z dd� Z
dS )�
SaltAPIParserzD
The CLI parser object used to fire up the Salt API system.
zFThe Salt API system manages network API connectors for the Salt Masterr� Z api_logfilec C r" r )rd rl Z
api_configr� r| r r r r� �
s �zSaltAPIParser.setup_configN) r r r r� rH r� r� rl ZDEFAULT_API_OPTSr� r� r r r r r# �
s �
r# )Yr� r r r� r= r� r: r\ rj r � functoolsr Z
salt._loggingrd Z salt.configrl Zsalt.defaults.exitcodesZsalt.exceptionsZ
salt.syspathsr� Zsalt.utils.argsZsalt.utils.dataZsalt.utils.filesZsalt.utils.jidZsalt.utils.platformr- Zsalt.utils.stringutilsZsalt.utils.userZsalt.utils.win_functionsZsalt.utils.xdgZsalt.utils.yamlZ salt.versionrF Z
salt.defaultsr Zsalt.utils.validate.pathr Zsalt.utils.verifyr r r rB r rf r r r r� r! r0 r> r; rD r� r� r� r� r� r� r r" rG r_ re rk rl rp rr rs r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r! r# r r r r �<module> s�
( 'My- R : n: -
�
�
�
!�
�
U�
G�
Q�
Y�
X�
V�
?�
8
�