XylotrechusZ
3
�uAc� � @ s� d Z ddlmZ dee�kr eZdddddgZdd lZdd l Z dd l
Z
dd l Z dd l Z d
Z
d d � Zg fd
d�Zdd� Zddd�Zddd�Zddd�Zddd�Zd S )z=This module contains helper functions for dealing with files.� )�absolute_import�xrange�
issamefile�linkfile�copyfile�linkorcopyfile�overwrite_safelyNi c C s" t j| �}t j|�}t jj||�S )N)�os�stat�path�samestat)�path1�path2�s1�s2� r �/usr/lib/python3.6/files.py� _issamefile+ s
r c C s0 |dkr t }y
t| |�S |k
r* dS X dS )zECheck whether two paths point to the same file (i.e. are hardlinked).TFN)� Exceptionr )r
r �catch_stat_exceptionsr r r r 2 s
c C s� t | |td�rdS tjj|�}tjj|�}tjj|�}d}xpttj �D ]b}tj
|tj |d�}ytj | |� W n2 tk
r� } z|j
t
jkr�n� W Y dd}~X qFX d}P qFW |r�tj||� dS )zUHardlink srcpath to dstpath.
Attempt to atomically replace dstpath if it exists.)r NF)�prefix�dirT)r �OSErrorr r �abspath�dirname�basename�range�tempfile�TMP_MAX�mktemp�extsep�link�errno�EEXIST�rename)�srcpath�dstpath�dstdname�dstbnameZ
hardlinked�attempt�_dsttmp�er r r r > s$ Tc
C s8 t | |td�rdS tjj|�}tjj|�}tjj|�}t| d�}tj |tjj
|dd�}tj | �}|r�ytj |�}W n tk
r� Y nX tj |j
� t j|j�� d} xP| dkr�|jt�} y|j| � W q� |j� |j� tj|j� � Y q�X q�W |j� |j� tj|j|� |�r4tj� dk�r4tj|� dS )z�Copy srcpath to dstpath.
Abort operation if e.g. not enough space is available. Attempt to
atomically replace dstpath if it exists.)r N�rbF)r r �delete� r )r r r r r r r �openr ZNamedTemporaryFiler r
�fchmod�fileno�S_IMODE�st_mode�read� BLOCKSIZE�write�close�unlink�namer$ �selinux�is_selinux_enabled�
restorecon)
r% r& �copy_mode_from_dst�run_restoreconr' r( ZsrcfileZ
dsttmpfile�s�datar r r r _ s<
c C s^ yt | |� dS tk
rJ } z |jtjtjtjfkr:� n W Y dd}~X nX t| |||� dS )ztFirst attempt to hardlink srcpath to dstpath, if hardlinking isn't
possible, attempt copying srcpath to dstpath.N)r r r" ZEMLINKZEPERMZEXDEVr )r% r& r= r> r+ r r r r � s
Fc
C s� t jj|�}t jj|�}d}d}|r6tj� dkr6d}n^y&tj|�\}}|dk rZtd| ��W n6 tk
r� } z| j t j
kr�d}n� W Y dd} ~ X nX |s�t j | |� |r�tj |� n�d}
xtt
tj�D ]f} tj|t j |d�} yt j | | � W n6 tk
�r" } z| j t jk�rwĂ W Y dd} ~ X q�X | }
P q�W |
dk�rDtt jd��|�r^| �r^tj|
|� yt j|
|� W n t j|
� � Y nX |�r�tj |� dS )zpCreate a symlink, optionally replacing dstpath atomically, optionally
setting or preserving SELinux context.FNr zgetfilecon(%r) failedT)r r z/No suitable temporary symlink could be created.)r r r r r: r; Z lgetfilecon� RuntimeErrorr r" �ENOENT�symlinkr< r r r r r r# �IOErrorZ lsetfileconr$ �remove)
r% r& �force�preserve_contextr' r( r> �ctx�retr+ Zdsttmpr) r* r r r �symlink_atomically� sV
rJ c C s~ t jj| �} t jj| �}t jj| �}d}d}d} t jj| �}
|rPtj� dkrPd}z�tj |t jj
|d�\}} |
r�t j | �} |r�t j || j
| j� |r�t j|t j| j�� |r�tj| �\} }
| dk r�td| ��t j|d�}d}|j|� |j� d}t j| | � |�r$|
�rtj| |
� n
tj| � W d|�r8|j� n|�rHt j|� | �rxt jj| ��rxyt j| � W n Y nX X dS )z�Safely overwrite a file by creating a temporary file in the same
directory, writing it, moving it over the original file, eventually
preserving file mode, SELinux context and ownership.Nr F)r r zgetfilecon(%r) failed�w) r r �realpathr r �existsr: r; r Zmkstempr r
�fchown�st_uid�st_gidr0 r2 r3 Z
getfileconrA �fdopenr6 r7 r$ Z
setfileconr< �isfiler8 )r ZcontentZ
preserve_moderG Zpreserve_ownershipZdir_�base�fd�fZtmpnamerM r? rI rH r r r r � sR
)TT)TT)FT)TTT)�__doc__Z
__future__r r � __builtins__r r �__all__r r: r r" r
r5 r r r r r rJ r r r r r �<module> s& !
5
?