U
e5d|! @ sD d Z ddlZddlmZ ddlZz ddlZW n ek
rD dZY nX ddlmZ ddl m
Z
ddl m Z ddl
mZ zddlmZ W n ek
r dZY nX zdd lmZ W n ek
r dZY nX d
d Zd d
Zd#ddZd$ddZedgdfedgdfedgdfedgdfedgd feg d fd Zd d Zd%d!d"ZdS )&zodistutils.archive_util
Utility functions for creating archive files (tarballs, zip files,
that sort of thing). N)warn)DistutilsExecError)spawn)mkpath)log)getpwnam)getgrnamc C sN t dks| dkrdS z t | }W n tk
r8 d}Y nX |dk rJ|d S dS )z"Returns a gid, given a group name.N )r KeyErrornameresult r ./usr/lib64/python3.8/distutils/archive_util.py_get_gid s
r c C sN t dks| dkrdS z t | }W n tk
r8 d}Y nX |dk rJ|d S dS )z"Returns an uid, given a user name.Nr )r r
r r r r _get_uid+ s
r gzipc s. dddddd}dddd d
}|d k r:|| kr:td | d
} |dkrZ| ||d7 } ttj| |d dd l}
t d t
t fdd} |s|
| d|| } z| j|| d W 5 |
X |dkr*tdt | || }
tjdkr|| |
g}n
|d| g}t||d |
S | S )a= Create a (possibly compressed) tar file from all the files under
'base_dir'.
'compress' must be "gzip" (the default), "bzip2", "xz", "compress", or
None. ("compress" will be deprecated in Python 3.2)
'owner' and 'group' can be used to define an owner and a group for the
archive that is being built. If not provided, the current owner and group
will be used.
The output tar file will be named 'base_dir' + ".tar", possibly plus
the appropriate compression extension (".gz", ".bz2", ".xz" or ".Z").
Returns the output filename.
Zgzbz2xz )r bzip2r Ncompressz.gzz.bz2z.xzz.Z)r r r r NzKbad value for 'compress': must be None, 'gzip', 'bzip2', 'xz' or 'compress'z.tarr dry_runr zCreating tar archivec s, d k r | _ | _d k r(| _| _| S )N)gidZgnameuiduname)Ztarinfor groupownerr r r _set_uid_gida s z"make_tarball.._set_uid_gidzw|%s)filterz 'compress' will be deprecated.Zwin32z-f)keys
ValueErrorgetr ospathdirnametarfiler infor r opencloseaddr PendingDeprecationWarningsysplatformr ) base_namebase_dirr verboser r r Ztar_compressionZ compress_extZ archive_namer( r tarZcompressed_namecmdr r r make_tarball7 sB
r5 c
C s | d }t tj||d tdkrp|r.d}nd}ztd|||g|d W n tk
rj td| Y nX n8td|| |sztj |d tj
d
}W n& t k
r tj |d tj d
}Y nX | |tj
krtjtj|d }||| td | t|D ]\}} }
| D ]6} tjtj|| d }||| td | q|
D ]B} tjtj|| }tj|rV||| td | qVq W 5 Q R X |S )
av Create a zip file from all the files under 'base_dir'.
The output zip file will be named 'base_name' + ".zip". Uses either the
"zipfile" Python module (if available) or the InfoZIP "zip" utility
(if installed and found on the default search path). If neither tool is
available, raises DistutilsExecError. Returns the name of the output zip
file.
z.zipr Nz-rz-rqzipzkunable to create zip file '%s': could neither import the 'zipfile' module nor find a standalone zip utilityz#creating '%s' and adding '%s' to itw)Z compressionr z adding '%s')r r% r& r' zipfiler r r r) ZZipFileZ ZIP_DEFLATED RuntimeErrorZ
ZIP_STOREDcurdirnormpathjoinwritewalkisfile) r0 r1 r2 r Z zip_filenameZ
zipoptionsr6 r& dirpathZdirnames filenamesr r r r make_zipfile sV
rB )r r zgzip'ed tar-file)r r zbzip2'ed tar-file)r r zxz'ed tar-file)r r zcompressed tar file)r Nzuncompressed tar filezZIP file)ZgztarZbztarZxztarZztarr3 r6 c C s | D ]}|t kr| S qdS )zqReturns the first format from the 'format' list that is unknown.
If all formats are known, returns None
N)ARCHIVE_FORMATS)Zformatsformatr r r check_archive_formats s
rE c
C s t }|dk r6td| t j| } |s6t | |dkrDt j}d|i} z t| }
W n t k
rx t
d| Y nX |
d } |
d D ]\} }
|
| | < q|dkr|| d< || d < z| | |f| }W 5 |dk rtd
| t | X |S ) a Create an archive file (eg. zip or tar).
'base_name' is the name of the file to create, minus any format-specific
extension; 'format' is the archive format: one of "zip", "tar", "gztar",
"bztar", "xztar", or "ztar".
'root_dir' is a directory that will be the root directory of the
archive; ie. we typically chdir into 'root_dir' before creating the
archive. 'base_dir' is the directory where we start archiving from;
ie. 'base_dir' will be the common prefix of all files and
directories in the archive. 'root_dir' and 'base_dir' both default
to the current directory. Returns the name of the archive file.
'owner' and 'group' are used when creating a tar archive. By default,
uses the current owner and group.
Nzchanging into '%s'r zunknown archive format '%s'r r6 r r zchanging back to '%s') r% getcwdr debugr& abspathchdirr: rC r
r# )r0 rD Zroot_dirr1 r2 r r r Zsave_cwdkwargsZ format_infofuncargvalfilenamer r r make_archive s2
rP )r r r NN)r r )NNr r NN)__doc__r% warningsr r. r8 ImportErrorZdistutils.errorsr Zdistutils.spawnr Zdistutils.dir_utilr Z distutilsr pwdr Zgrpr r r r5 rB rC rE rP r r r r sN
H
=