File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/freezer.cpython-310.pyc
o
�N�g�( � @ s� d Z ddlZddlZddlZddlm mZ ddlm Z ddl
m Z ddl m
Z
e�e�ZddiZdd � Zd
d � Zd d d
�Zd dd�Zdd� Zd!dd�Zdd� Zdd� Zdd� Zdd� Zd!d d �Z d d � Z dS )"z�
Module for freezer
:maintainer: Alberto Planas <aplanas@suse.com>
:maturity: new
:depends: None
:platform: Linux
� N)�CommandExecutionError)� clean_kwargs)�fopen�list_�listc C s dS )z�
Freezer is based on top of the pkg module.
Return True since the `pkg` module should always exist. This
avoids the overhead of loading all modules.
T� r r r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/freezer.py� __virtual__ s r c C s t j�td d�S )z9
Return the path where we will store the states.
Zcachedir�freezer)�os�path�joinZ__opts__r r r r � _states_path$ s r c C s: | sdn| } t � }tj�|| � d��tj�|| � d��fS )zR
Return the full path for the packages and repository freezer
files.
r
� -pkgs.yml� -reps.yml)r r r r
)�name� states_pathr r r �_paths+ s
�r c C s" | sdn| } t dd� t| �D ��S )a�
Return True if there is already a frozen state.
A frozen state is merely a list of packages (including the
version) in a specific time. This information can be used to
compare with the current list of packages, and revert the
installation of some extra packages that are in the system.
name
Name of the frozen state. Optional.
CLI Example:
.. code-block:: bash
salt '*' freezer.status
salt '*' freezer.status pre_install
r
c s s � | ] }t j�|�V qd S �N)r r �isfile)�.0�ir r r � <genexpr>N s � zstatus.<locals>.<genexpr>)�allr )r r r r �status9 s r c C sR g } t � }tj�|�s
| S t�|�D ]}|�d�r"| �|dd� � qtt| ��S )zw
Return the list of frozen states.
CLI Example:
.. code-block:: bash
salt '*' freezer.list
)r r Ni����) r r r �isdir�listdir�endswith�append�sorted�set)�retr �stater r r r Q s
� Fc
K s� t � }z
tj�|�st�|� W n ty- } zd}t�|||� t|||f ��d}~ww t | �r8|s8td��t
di |��}t d di |��}t d di |��}t t
| �||f�D ] \} }
t| d��} t�|
| � W d � n1 stw Y qZdS ) a~
Save the list of package and repos in a freeze file.
As this module is build on top of the pkg module, the user can
send extra attributes to the underlying pkg module via kwargs.
This function will call ``pkg.list_pkgs`` and ``pkg.list_repos``,
and any additional arguments will be passed through to those
functions.
name
Name of the frozen state. Optional.
force
If true, overwrite the state. Optional.
CLI Example:
.. code-block:: bash
salt '*' freezer.freeze
salt '*' freezer.freeze pre_install
salt '*' freezer.freeze force=True root=/chroot
z6Error when trying to create the freezer storage %s: %sNz?The state is already present. Use force parameter to overwrite.�
pkg.list_pkgs�pkg.list_repos�wTr )r r r �exists�makedirs�OSError�log�errorr r r �__salt__�zipr r �json�dump) r �force�kwargsr �e�msg� safe_kwargs�pkgs�repos�fname�content�fpr r r �freezei s0
��� � ��r9 c K s� t d d i |��}t| �t|� }|D ]N}z&| | �� }|�|� t d |fi |�� |d d �|� t�d|� W q tya } zd}t�|||� |d �|||f � W Y d}~qd}~ww dS )
z0Add missing repositories and update the ret dictr$ z pkg.mod_repor5 �addzAdded missing repository %sz Error adding %s repository: %s�commentNr ) r+ r �copy�updater r) �info� Exceptionr* ) � frozen_reposr! r0 r5 Z
missing_repos�repoZ _tmp_kwargsr1 r2 r r r �_add_missing_repositories� s
"���rB c K s� t d d i |��}t|�t| � }|D ]C}zt d |fi |�� |d d �|� t�d|� W q tyV } zd}t�|||� |d �|||f � W Y d}~qd}~ww dS )
z1Remove extra repositories and update the ret dictr$ z pkg.del_repor5 �removezRemoved extra repository %sz Error removing %s repository: %sr; Nr �r+ r r r) r> r? r* )r@ r! r0 r5 Z extra_reposrA r1 r2 r r r �_remove_extra_repositories� s "���rE c K s� t d d
i |��}t| �t|� }|D ]D}z t d d
d|i|�� |d d �|� t�d|� W q tyW } zd}t�|||� |d �|||f � W Y d }~qd }~ww d S ) z,Add missing packages and update the ret dictr# z pkg.installr r4 r: zAdded missing package %szError adding %s package: %sr; Nr rD )� frozen_pkgsr! r0 r4 Z missing_pkgs�pkgr1 r2 r r r �_add_missing_packages� s "���rH c K s� t d d
i |��}t|�t| � }|D ]D}z t d d
d|i|�� |d d �|� t�d|� W q tyW } zd}t�|||� |d �|||f � W Y d }~qd }~ww d S ) z-Remove extra packages and update the ret dictr# z
pkg.remover r4 rC zRemoved extra package %sz Error removing %s package: %sr; Nr rD )rF r! r0 r4 Z
extra_pkgsrG r1 r2 r r r �_remove_extra_packages� s "���rI c
K s t | �std��i }i }tt| �||f�D ] \}}t|��}|�t�|�� W d � n1 s0w Y qtdi |��}g g d�g g d�g d�} t || fi |�� t
|| fi |�� t || fi |�� t || fi |�� |r| d st| �D ]}t
�|� qw| S )a�
Make sure that the system contains the packages and repos from a
frozen state.
Read the list of packages and repositories from the freeze file,
and compare it with the current list of packages and repos. If
there is any difference, all the missing packages are repos will
be installed, and all the extra packages and repos will be
removed.
As this module is build on top of the pkg module, the user can
send extra attributes to the underlying pkg module via kwargs.
This function will call ``pkg.list_repos``, ``pkg.mod_repo``,
``pkg.list_pkgs``, ``pkg.install``, ``pkg.remove`` and
``pkg.del_repo``, and any additional arguments will be passed
through to those functions.
name
Name of the frozen state. Optional.
clean
If True remove the frozen information YAML from the cache
.. versionadded:: 3000
CLI Example:
.. code-block:: bash
salt '*' freezer.restore
salt '*' freezer.restore root=/chroot
�Frozen state not found.N)r: rC )r4 r5 r; r; r )r r r, r r r= r- �loadr rB rH rI rE r rC )
r Zcleanr0 rF r@ r6 r7 r8 r3 r! r r r �restore� s, "
��� rL c C s� i }t | �r
t |�std��tt| �t|��D ]W\}}t|d��
}t�|�}W d � n1 s0w Y t|d��
}t�|�}W d � n1 sJw Y |�d�r_tj j
� ||�|d< q|�d�rntj j
� ||�|d< q|S )a�
Display the difference between two frozen states. The results are shown as
as a dictionary with keys for packages and repositories. Each key may
contain a changes dictionary showing items that differ between the two
frozen states. Items shown in the "old" changes but not the "new" were
removed. Items in "new" but not "old" were added. Items shown in both
probably updated/changed versions between freezes.
old
Name of the "old" frozen state. Required.
new
Name of the "new" frozen state. Required.
CLI Example:
.. code-block:: bash
salt '*' freezer.freeze pre_install post_install
rJ �rNr r4 r r5 ) r r r, r r r- rK r �salt�utilsZ
dictdifferZ deep_diff) �old�newr! ZofileZnfileZofpZold_dictZnfpZnew_dictr r r �compare- s � �
�rR r )NF) �__doc__�loggingr Zsalt.utils.dictdifferrN Zsalt.utils.jsonrO r- Zsalt.exceptionsr Zsalt.utils.argsr Zsalt.utils.filesr � getLogger�__name__r) Z__func_alias__r r r r r r9 rB rE rH rI rL rR r r r r �<module> s.
�
0
P