File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/atomicfile.cpython-310.pyc
o
�N�g � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZ ddl
Z dZ ej dkrqdd� Z
dd� Zz-ddlZd Zd
ZejjjZd d� Z
ejjjZejjjZejjjZ ejjj Z d Z d
d� ZW n e yk Y nw dd� Z nej!Z d Z G dd� d�Z"ddd�Z#dS )zY
A module written originally by Armin Ronacher to manage file transfers in an
atomic way
� NF�ntc C � dS �NF� ��src�dstr r �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/atomicfile.py�_rename � r
c C r r r r r r r �_rename_atomic r r � � c C s� t | t�s t| t�� �} t |t�st|t�� �}t| |�r dS d}d}|sB|dk rBt| |ttB �}|s<t� d� |d7 }|sB|dk s)|S )NTr F�d ���MbP?r
)
�
isinstance�str�sys�getfilesystemencodingr � _MoveFileEx�_MOVEFILE_REPLACE_EXISTING�_MOVEFILE_WRITE_THROUGH�time�sleep)r r �retry�rvalr r r r
! s"
�
�Tc C s� t d dddddd�}|dkrdS z2d}d}|s<|dk r<t| |d d ttB |�}|r-t|�}nt�d� |d7 }|s<|dk s|W t|� S t|� w ) Nr i� z
Atomic rename���Fr r r
)�_CreateTransaction�_MoveFileTransactedr r �_CommitTransactionr r � _CloseHandle)r r Ztrar r r r r r < s. �
�c C s� t | |�rd S z t�| |� W d S ty[ } z?|jtjkr � |� dt�dtj �d��}t�||� t�| |� zt�
|� W n t yH Y nw W Y d }~d S W Y d }~d S d }~ww )N�-r Z08x) r
�os�rename�OSError�errnoZEEXIST�random�randintr Zmaxint�unlink� Exception)r r �err�oldr r r �
atomic_renameY s&
����r, c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� _AtomicWFilez/
Helper class for :func:`atomic_open`.
c C s || _ || _|| _d S �N)�_fh�
_tmp_filename� _filename)�selfZfhanleZ tmp_filename�filenamer r r �__init__u s
z_AtomicWFile.__init__c C s t | j|�S r. )�getattrr/ )r2 �attrr r r � __getattr__z s z_AtomicWFile.__getattr__c C s | S r. r �r2 r r r � __enter__} r z_AtomicWFile.__enter__c C s� | j jrd S | j �� tjjjr#tj� | j
�r"tjjj | j
| j d� n.tj� | j
�rCt
�| j
| j � t�| j
�}t�| j |j|j� nt�| j dtjj�� @ � t| j | j
� d S )N)�source�targeti� )r/ �closed�close�salt�utilsZwin_daclZ HAS_WIN32r" �path�isfiler1 Z
copy_securityr0 �shutil�copymode�stat�chown�st_uid�st_gid�chmod�filesZ get_umaskr, )r2 �str r r r= � s
�� z_AtomicWFile.closec C sF |d u r
| � � d S | j� � z t�| j� W d S ty" Y d S w r. )r= r/ r" �remover0 r$ )r2 �exc_type� exc_value� tracebackr r r �__exit__� s
�z_AtomicWFile.__exit__c C s& d� | jj| jjr dp d| j| jj�S )Nz<{} {}{}, mode {}>zclosed � )�format� __class__�__name__r/ r< r1 �moder8 r r r �__repr__� s �z_AtomicWFile.__repr__N)
rS �
__module__� __qualname__�__doc__r4 r7 r9 r= rO rU r r r r r- p s
r- �wc C sT |dv rt d��dtj�| �dd�}d|vrd|d< tj|fi |��}t||j| �S ) z�
Works like a regular `open()` but writes updates into a temporary
file instead of the given file and moves it over when the file is
closed. The file returned behaves as if it was a regular Python
)�r�rbzr+zrb+�a�abz0Read or append modes don't work with atomic_openz.___atomic_writeF)�prefix�dir�delete�brP �newline)� TypeErrorr" r@ �dirname�tempfileZNamedTemporaryFiler- �name)r3 rT �kwargsZntfr r r � atomic_open� s
�rh )rY )$rX r% r" r& rB r re r Zsalt.utils.filesr> Zsalt.utils.win_daclZCAN_RENAME_OPEN_FILErf r
r �ctypesr r ZwindllZkernel32Z MoveFileExWr Zktmw32ZCreateTransactionr ZCommitTransactionr ZMoveFileTransactedWr Z CloseHandler r) r, r# r- rh r r r r �<module> sD
�
6