File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/states/__pycache__/btrfs.cpython-310.pyc
o
�N�g�( � @ s� d Z ddlZddlZddlZddlZddlmZ e�e �Z
dZ dd� Z dd� Z
d d
� Zd d � Zd
d� Zdd� Zdd� Ze d dd��Zed dd��Zdd� Zed dd ��ZdS ) z�
Manage BTRFS file systems.
:maintainer: Alberto Planas <aplanas@suse.com>
:maturity: new
:depends: None
:platform: Linux
� N)�CommandExecutionError�btrfsc C sP |rdnd}t �� }td || d|dd�}|d s&t�d| |� t|� d S |S )
z0
Mount the device in a temporary place.
�defaultszsubvol=/z
mount.mountedr F)�deviceZfstype�optsZpersist�resultz Cannot mount device %s in %sN)�tempfileZmkdtemp�
__states__�log�error�_umount)r � use_defaultr �dest�res� r �E/opt/saltstack/salt/lib/python3.10/site-packages/salt/states/btrfs.py�_mount s
�r c C s t d | � td | � dS )z/
Umount and clean the temporary place.
zmount.unmountedz files.rm_rfN)r Z __utils__��pathr r r r & s r c C s, t d | �| d }t d |�d }||kS )z8
Check if the subvolume is the current default.
�btrfs.subvolume_show� subvolume idzbtrfs.subvolume_get_default�id��__salt__)r r �name� subvol_idZdef_idr r r � _is_default. s r c C s" t d | �| d }t d ||�S )z3
Set the subvolume as the current default.
r r zbtrfs.subvolume_set_defaultr )r r r r r r r � _set_default7 s r c C s t j�| �}dtd |�| vS )z1
Check if the subvolume is copy on write
�Cz file.lsattr)�osr �dirnamer )r r r r r �_is_cow? s r! c C s t d | ddd�S )z2
Disable the copy on write in a subvolume
z file.chattr�addr )�operatorZ
attributesr r r r r �
_unset_cowG s r$ c s t �� �� fdd��}|S )zf
Small decorator to makes sure that the mount and umount happends in
a transactional way.
c
s |� d| r | d nd �}|� dt| �dkr| d nd �}|� dd�}|di dgd�}zNz#|rFt||�}|sBd |� d
�}|d �|� ||d < � | i |��}W n! tyo } ztjd
|dd� |d �|� W Y d }~nd }~ww W |rwt|� |S |r�t|� w w )Nr r r � r
Fz)Some error happends during the operation.�r r �changes�commentzDevice z cannot be mountedr( �__destz*Exception raised while mounting device: %sT)�exc_info)�get�lenr �append� Exceptionr
r r ) �args�kwargsr r r
�retr �msg�exc��actionr r �wrapperT s8 �
����
�z __mount_device.<locals>.wrapper)� functools�wraps)r5 r6 r r4 r �__mount_deviceN s r9 FTc C sd | di g d�}t j�|| �}td |�} | r!|d �d| � d�� td r7d|d < | s5d| � d
�|d | < |S | s�t j�|�}
td |
d
d�} | d sX|d �d|
� d�� |S z td | ||d� W n tyx |d �d| � �� | Y S w d| � �|d | < | r�| r�|r�|r�t ||| �s�t
||| �|d | d < |s�t |�r�t |�|d | d < d
|d < |S )af
Makes sure that a btrfs subvolume is present.
name
Name of the subvolume to add
device
Device where to create the subvolume
qgroupids
Add the newly created subcolume to a qgroup. This parameter
is a list
set_default
If True, this new subvolume will be set as default when
mounted, unless subvol option in mount is used
copy_on_write
If false, set the subvolume with chattr +C
force_set_default
If false and the subvolume is already present, it will not
force it as default if ``set_default`` is True
Fr& �btrfs.subvolume_existsr( �
Subvolume z already present�testNr z will be createdr' zfile.directoryT)�makedirszError creating z
directoryzbtrfs.subvolume_create)r � qgroupidszError creating subvolume zCreated subvolume �_defaultZ_no_cow)
r r �joinr r- �__opts__r r r r r r! r$ ) r r r> Z set_defaultZ
copy_on_writeZforce_set_defaultr) r1 r �exists�_pathr r r r �subvolume_createds sT $� �����
� rD c C s� | di g d�}t j�|| �}td |�}|s!|d �d| � d�� td r7d|d < |r5d| � d
�|d | < |S |r;d nd}|shz
td
||d� W n ty^ |d �d| � �� | Y S w d| � �|d | < d|d < |S )z�
Makes sure that a btrfs subvolume is removed.
name
Name of the subvolume to remove
device
Device where to remove the subvolume
commit
Wait until the transaction is over
Fr& r: r( r; z already missingr< Nr z will be removedr' Zafterzbtrfs.subvolume_delete)�commitzError removing subvolume zRemoved subvolume T)r r r@ r r- rA r )r r rE r) r1 r rB r r r �subvolume_deleted� s2 � �rF c C sZ i }| � � D ]$\}}|| d }|du r|dkrd}n ||kr"d}nd}|r*|||< q|S )a# Calculate the difference between the current and the expected
properties
* 'expected' is expressed in a dictionary like: {'property': value}
* 'current' contains the same format retuned by 'btrfs.properties'
If the property is not available, will throw an exception.
�valueFzN/AT)�items)�expected�current�
differenceZ _propertyrG Z
current_valueZ needs_updater r r �_diff_properties� s �rL c
K s | di g d�}|r&t j�| �r t j�|t j�| t jj��}n
t j�|| �}n| }t j�|�s;|d �d| � d�� |S dd� |�� D �}i }zt d |�}W n( t
yv } z |d �d | � �� |d �d
|� �� |W Y d }~S d }~ww zt ||�} W n t y� |d �d | � �� | Y S w t
d
r�d |d< | r�| |d< |S d| � �}
|d �|
� |S | r�d�dd� | �� D ��} t d || d� t d |�}t ||�} | r�d�| | �}
|d �|
� |S |d �d| � �� | |d< n
|d �d| � �� d|d< |S )a5
Makes sure that a list of properties are set in a subvolume, file
or device.
name
Name of the object to change
device
Device where the object lives, if None, the device will be in
name
use_default
If True, this subvolume will be resolved to the default
subvolume assigned during the create operation
properties
Dictionary of properties
Valid properties are 'ro', 'label' or 'compression'. Check the
documentation to see where those properties are valid for each
object.
Fr& r( zObject z
not foundc S s. i | ]\}}|t |�tur|nt|��� �qS r )�type�bool�str�lower��.0�k�vr r r �
<dictcomp>G s "�z properties.<locals>.<dictcomp>zbtrfs.propertiesz Error reading properties from zCurrent error NzSome property not found in r< r r' z!No properties will be changed in �,c s s" � | ] \}}|� d |� �V qdS )�=Nr rQ r r r � <genexpr>c s � z properties.<locals>.<genexpr>)�setz(Properties {} failed to be changed in {}zProperties changed in zProperties not changed in T)r r �isabsr@ �relpath�seprB r- rH r r rL �KeyErrorrA �format)
r r r
r) �
propertiesr1 r Zcurrent_properties�eZproperties_to_setr2 Z _propertiesZproperties_failedr r r r_ sn �
��� �
�
�
r_ )NFTTN)FN)�__doc__r7 �loggingZos.pathr r Zsalt.exceptionsr � getLogger�__name__r
Z__virtualname__r r r r r! r$ r9 rD rF rL r_ r r r r �<module> s6
% �X 1