3
uAc @ s d Z ddlmZ deekr eZdddddgZdd lZdd l Z dd l
Z
dd l Z dd l Z d
Z
d d Zg fd
dZdd ZdddZdddZdddZdddZd S )z=This module contains helper functions for dealing with files. )absolute_importxrange
issamefilelinkfilecopyfilelinkorcopyfileoverwrite_safelyNi c C s" t j| }t j|}t jj||S )N)osstatpathsamestat)path1path2s1s2 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 | |tdrdS tjj|}tjj|}tjj|}d}xpttj D ]b}tj
|tj |d}ytj | | W n2 tk
r } z|j
t
jkrn W Y dd}~X qFX d}P qFW |rtj|| dS )zUHardlink srcpath to dstpath.
Attempt to atomically replace dstpath if it exists.)r NF)prefixdirT)r OSErrorr r abspathdirnamebasenamerangetempfileTMP_MAXmktempextseplinkerrnoEEXISTrename)srcpathdstpathdstdnamedstbnameZ
hardlinkedattempt_dsttmper r r r > s$ Tc
C s8 t | |tdrdS tjj|}tjj|}tjj|}t| d}tj |tjj
|dd}tj | }|rytj |}W n tk
r Y nX tj |j
t j|j d} xP| dkr|jt} y|j| W q |j |j tj|j Y qX qW |j |j tj|j| |r4tj dkr4tj| dS )zCopy srcpath to dstpath.
Abort operation if e.g. not enough space is available. Attempt to
atomically replace dstpath if it exists.)r NrbF)r r delete r )r r r r r r r openr ZNamedTemporaryFiler r
fchmodfilenoS_IMODEst_moderead BLOCKSIZEwritecloseunlinknamer$ selinuxis_selinux_enabled
restorecon)
r% r& copy_mode_from_dstrun_restoreconr' r( ZsrcfileZ
dsttmpfilesdatar 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
krd}n W Y dd} ~ X nX |st j | | |rtj | nd}
xtt
tjD ]f} tj|t j |d} yt j | | W n6 tk
r" } z| j t jkrwĂ W Y dd} ~ X qX | }
P qW |
dkrDtt jd|r^| r^tj|
| yt j|
| W n t j|
Y nX |rtj | 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" ENOENTsymlinkr< r r r r r r# IOErrorZ lsetfileconr$ remove)
r% r& forcepreserve_contextr' r( r> ctxretr+ 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}ztj |t jj
|d\}} |
rt j | } |rt j || j
| j |rt j|t j| j |rtj| \} }
| dk rtd| 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 )zSafely 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) failedw) r r realpathr r existsr: r; r Zmkstempr r
fchownst_uidst_gidr0 r2 r3 Z
getfileconrA fdopenr6 r7 r$ Z
setfileconr< isfiler8 )r ZcontentZ
preserve_moderG Zpreserve_ownershipZdir_basefdfZtmpnamerM 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 s& !
5
?