File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/aliases.cpython-310.pyc
o
�N�g. � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
m Z dddddd�Z e�
d�Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )z,
Manage the information in the aliases file
� N)�SaltInvocationError�txtZyaml)�rm_alias�
has_target�
get_target�
set_target� list_aliasesz"([^:#]*)\s*:?\s*([^#]*?)(\s+#.*|$)c C s t j�td d��S )z9
Return the path to the appropriate aliases file
z
config.optionz aliases.file)�os�path�realpath�__salt__� r
r
�H/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/aliases.py�__get_aliases_filename s r c C s� t � } g }tj�| �s
|S tjj�| d��.}|D ]"}tjj� |�}t
� |�}|r0|� |�
� � q|� dd|�� f� qW d � |S 1 sFw Y |S )z�
Parse the aliases file, and return a list of line components:
[
(alias1, target1, comment1),
(alias2, target2, comment2),
]
�rN)r r r
�isfile�salt�utils�filesZfopenZ stringutilsZ
to_unicode�
__ALIAS_RE�match�append�groups�strip)�afn�retZifile�liner r
r
r �__parse_aliases" s
�
��r c
C sL t � }tj�|�}tj|dd��r}t�dd�sGtj�|�r8t� |�}t�
|j t � |j
�� t�|j |j|j� nt�
|j d� t�|j dd� | D ]1\}}}t|t�rXd�|�}|s\d}|rk|rk|� d|� |� d �}n|� d �}|�t�}|�|� qIW d
� n1 s�w Y tjj�|j |� tjj�d �} | d
ur�t d | � d
S )zm
Write a new copy of the aliases file. Lines is a list of lines
as returned by __parse_aliases.
F)�dir�deletezintegration.testi� r �, � z: �
N�
newaliaseszcmd.runT) r r r
�dirname�tempfileZNamedTemporaryFileZ__opts__�getr �stat�chmod�name�S_IMODE�st_mode�chown�st_uid�st_gid�
isinstance�list�join�encodeZ__salt_system_encoding__�writer r Z
atomicfileZ
atomic_rename�whichr )
�linesr Zadir�outZafn_st�
line_alias� line_target� line_commentZ
write_liner# r
r
r �__write_aliases_file: s6
� � r: c C s dd� t � D �} | S )z�
Return the aliases found in the aliases file in this format::
{'alias': 'target'}
CLI Example:
.. code-block:: bash
salt '*' aliases.list_aliases
c S s i | ]
\}}}|r||�qS r
r
)�.0�alias�target�_r
r
r �
<dictcomp>o s z list_aliases.<locals>.<dictcomp>)r )r r
r
r r c s r c C s t � }| |v r || S dS )z�
Return the target associated with an alias
CLI Example:
.. code-block:: bash
salt '*' aliases.get_target alias
r! )r )r< �aliasesr
r
r r s s
r c C sB |dkrt d��t� }| |vrdS t|t�rd�|�}||| kS )z�
Return true if the alias/target is set
CLI Example:
.. code-block:: bash
salt '*' aliases.has_target alias target
r! �!target can not be an empty stringFr )r r r/ r0 r1 )r< r= r@ r
r
r r � s
r c C s� | dkrt d��|dkrt d��t| �|krdS t� }g }d}|D ] \}}}|| kr7|s6|�| ||f� d}q!|�|||f� q!|sJ|�| |df� t|� dS )a
Set the entry in the aliases file for the given alias, this will overwrite
any previous entry for the given alias or create a new one if it does not
exist.
CLI Example:
.. code-block:: bash
salt '*' aliases.set_target alias target
r! z alias can not be an empty stringrA TF)r r r r r: )r< r= r5 r6 Zovrr7 r8 r9 r
r
r r � s(
�r c C sJ t | �sdS t� }g }|D ]\}}}|| kr |�|||f� q
t|� dS )z�
Remove an entry from the aliases file
CLI Example:
.. code-block:: bash
salt '*' aliases.rm_alias alias
T)r r r r: )r< r5 r6 r7 r8 r9 r
r
r r � s
�r )�__doc__r �rer' r% Zsalt.utils.atomicfiler Zsalt.utils.filesZsalt.utils.pathZsalt.utils.stringutilsZsalt.exceptionsr Z
__outputter__�compiler r r r: r r r r r r
r
r
r �<module> s2 �
) '