Yfp A @ sr d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l Z [ d Z Wn e k
r d Z Yn Xy d d l
Z
[
d Z Wn e k
r d Z Yn Xy d d l Z [ d Z
Wn e k
r d Z
Yn Xy d d l m Z Wn e k
rd Z Yn Xy d d l m Z Wn e k
rGd Z Yn Xd d d d
d d d
d d d d d d d d d d d d d d d d d d g Z Gd d d e Z Gd! d d e Z Gd" d d e Z Gd# d d e Z Gd$ d% d% e Z Gd& d' d' e Z d d* d Z d+ d, Z d- d d. d Z d- d d/ d Z e e d0 rd- d d1 d2 Z n d3 d2 Z d- d d4 d
Z! d- d d5 d Z" d- d d6 d Z# d7 d Z$ d d e# d d8 d
Z% d9 d: Z& d; d< Z' e j( e j e j) e j* h e j+ k oFe j, e j- k oFe j e j. k Z/ d d d= d Z0 e/ e0 _1 d> d? Z2 e# d@ d Z3 dA dB Z4 dC dD Z5 dE dF Z6 dG d d d d d dH dI Z7 d d d dJ dK Z8 dL e7 d g dN f i Z9 e r e7 d g dO f e9 dP d_ d` Z? d dZ da d Z@ db d ZA dc dd ZB de df ZC dg dh ZD dL di g eD g dN f dR dj g eC g dQ f i ZE e rLdk dl g eD g dO f eE dP ._nopc s t t | S)N)getattrr, )rL )rR r r! lookup s zcopystat..lookupc s) t t | } | t j k r% | S S)N)rS r, supports_follow_symlinks)rL r? )rR r r! rT s r8 utimerB st_flagsZchflags
EOPNOTSUPPrG )rX rG )r, r- r; r8 rD r: st_atime_ns st_mtime_nsNotImplementedErrorr+ rW r. rF rS rN ) r1 r2 r4 ZfollowrT r@ modewhyerrr )rR r! r s, +!
'c C s] t j j | r3 t j j | t j j | } t | | d | t | | d | | S)a3 Copy data and mode bits ("cp src dst"). Return the file's destination.
The destination may be a directory.
If follow_symlinks is false, symlinks won't be followed. This
resembles GNU's "cp -P src dst".
If source and destination are the same file, a SameFileError will be
raised.
r4 )r, r- isdirjoinbasenamer r )r1 r2 r4 r r r! r s
!c C s] t j j | r3 t j j | t j j | } t | | d | t | | d | | S)zCopy data and all stat info ("cp -p src dst"). Return the file's
destination."
The destination may be a directory.
If follow_symlinks is false, symlinks won't be followed. This
resembles GNU's "cp -P src dst".
r4 )r, r- r_ r` ra r r )r1 r2 r4 r r r! r s
!c s f d d } | S)zFunction that can be used as copytree() ignore parameter.
Patterns is a sequence of glob-style patterns
that are used to exclude filesc s: g } x' D] } | j t j | | q
Wt | S)N)extendfnmatchfilterset)r- rJ
ignored_namespattern)patternsr r! _ignore_patterns
s
z)ignore_patterns.._ignore_patternsr )rh ri r )rh r! r s c C ss t j | } | d k r- | | | } n t } t j | g } x| D]} | | k re qP t j j | | }
t j j | | } y t j j |
r7t j |
} | r t j | | t |
| d | qot j j
| r | r wP t j j |
r't |
| | | | qo| |
| n8 t j j |
rbt |
| | | | n
| |
| WqP t
k
r}
z | j |
j d WYd d }
~
XqP t k
r} z! | j |
| t | f WYd d } ~ XqP XqP Wy t | | WnY t k
r\} z9 t | d d d k rJ| j | | t | f WYd d } ~ Xn X| rot
| | S)a Recursively copy a directory tree.
The destination directory must not already exist.
If exception(s) occur, an Error is raised with a list of reasons.
If the optional symlinks flag is true, symbolic links in the
source tree result in symbolic links in the destination tree; if
it is false, the contents of the files pointed to by symbolic
links are copied. If the file pointed by the symlink doesn't
exist, an exception will be added in the list of errors raised in
an Error exception at the end of the copy process.
You can set the optional ignore_dangling_symlinks flag to true if you
want to silence this exception. Notice that this has no effect on
platforms that don't support os.symlink.
The optional ignore argument is a callable. If given, it
is called with the `src` parameter, which is the directory
being visited by copytree(), and `names` which is the list of
`src` contents, as returned by os.listdir():
callable(src, names) -> ignored_names
Since copytree() is called recursively, the callable will be
called once for each directory that is copied. It returns a
list of names relative to the `src` directory that should
not be copied.
The optional copy_function argument is a callable that will be used
to copy each file. It will be called with the source path and the
destination path as arguments. By default, copy2() is used, but any
function that supports the same signature (like copy()) can be used.
Nr4 r Zwinerror)r, listdirre makedirsr- r` r; r= r< r existsr_ r
r
rb rO r. appendstrrS )r1 r2 symlinksignore
copy_functionZignore_dangling_symlinksrJ rf errorsrL ZsrcnameZdstnamelinktor^ r] r r r! r
sL $
&3/ c $ C s y" t j j | r! t d Wn2 t k
rV | t j j | t j d SYn Xg } y t j | } Wn+ t k
r | t j | t j Yn Xx | D] } t j j | | } y t j | j } Wn t k
r d } Yn Xt
j | rt | | q y t j
| Wq t k
rN| t j
| t j Yq Xq Wy t j | Wn+ t k
r| t j | t j Yn Xd S)Nz%Cannot call rmtree on a symbolic linkr )r, r- r; r. sysexc_inforj r` rC r: r8 S_ISDIR_rmtree_unsafeunlinkrmdir)r- onerrorrJ rL fullnamer\ r r r! rw k s6
"
rw c
0 C s@ g } y t j | } WnG t k
rb } z' | | _ | t j | t j WYd d } ~ Xn Xx| D]} t j j | | } y( t j | d | d d } | j } Wn t k
r d } Yn Xt j
| ry t j | t j d | } Wn+ t k
r#| t j | t j Yq8Xz t j j
| t j | rt | | | y t j | d | Wqt k
r| t j | t j YqXnA y t d Wn. t k
r| t j j | t j Yn XWd t j | Xqj y t j | d | Wqj t k
r7| t j | t j Yqj Xqj Wd S)Ndir_fdr4 Fr z%Cannot call rmtree on a symbolic link)r, rj r. filenamert ru r- r` r8 r: rv r> O_RDONLYsamestatfstat_rmtree_safe_fdry r; closerx )
topfdr- rz rJ r^ rL r{ orig_str\ dirfdr r r! r sD ,
!
%
r c ! C s | r d d } n | d k r- d d } t rt | t rQ t j | } y t j | } Wn/ t k
r | t j | t j d SYn Xy t j | t j
} Wn/ t k
r | t j | t j d SYn Xz t j j | t j
| rTt | | | y t j | Wqt k
rP| t j | t j YqXnA y t d Wn. t k
r| t j j | t j Yn XWd t j | Xn
t | | Sd S)a Recursively delete a directory tree.
If ignore_errors is set, errors are ignored; otherwise, if onerror
is set, it is called to handle the error with arguments (func,
path, exc_info) where func is platform and implementation dependent;
path is the argument to that function that caused it to fail; and
exc_info is a tuple returned by sys.exc_info(). If ignore_errors
is false and onerror is None, an exception is raised.
c W s d S)Nr )rO r r r! rz s zrmtree..onerrorNc W s d S)Nr )rO r r r! rz s z%Cannot call rmtree on a symbolic link)_use_fd_functions
isinstancebytesr, fsdecoderC Exceptionrt ru r> r~ r- r r r ry r. r; r rw )r-
ignore_errorsrz r fdr r r! r s<
!
%c C s5 t j j t j j p d } t j j | j | S)N )r, r- sepaltsepra rstrip)r- r r r r! _basename s r c
C s] | } t j j | rx t | | r; t j | | d St j j | t | } t j j | rx t d | y t j | | Wn t k
rXt j j
| r t j | } t j | | t j
| nw t j j | r:t | | rt d | | f t | | d | d d t | n | | | t j
| Yn X| S)a+ Recursively move a file or directory to another location. This is
similar to the Unix "mv" command. Return the file or directory's
destination.
If the destination is a directory or a symlink to a directory, the source
is moved inside the directory. The destination path must not already
exist.
If the destination already exists but is not a directory, it may be
overwritten depending on os.rename() semantics.
If the destination is on our current filesystem, then rename() is used.
Otherwise, src is copied to the destination and then removed. Symlinks are
recreated under the new name if os.rename() fails because of cross
filesystem renames.
The optional `copy_function` argument is a callable that will be used
to copy the source or it will be delegated to `copytree`.
By default, copy2() is used, but any function that supports the same
signature (like copy()) can be used.
A lot more could be done here... A look at a mv.c shows a lot of
the issues this implementation glosses over.
Nz$Destination path '%s' already existsz.Cannot move a directory '%s' into itself '%s'.rq ro T)r, r- r_ r3 renamer` r rl r
r. r; r= r< rx
_destinsrcr
r )r1 r2 rq Zreal_dstrs r r r! r s2
c C s{ t j j | } t j j | } | j t j j sI | t j j 7} | j t j j sn | t j j 7} | j | S)N)r, r- r0 endswithr
startswith)r1 r2 r r r! r 2 s r c C s_ t d k s | d k r d Sy t | } Wn t k
rF d } Yn X| d k r[ | d Sd S)z"Returns a gid, given a group name.N )r KeyError)rL resultr r r! _get_gid; s
r c C s_ t d k s | d k r d Sy t | } Wn t k
rF d } Yn X| d k r[ | d Sd S)z"Returns an uid, given a user name.Nr )r r )rL r r r r! _get_uidG s
r gzipc
s | d k r d } nf t r0 | d k r0 d } nK t rK | d k rK d } n0 t rf | d k rf d } n t d j | d d l } | r d
| n d }
| d |
} t j j | } | rt j j | r| d k r | j
d | | st j | | d k r | j
d
t t
f d d }
| s| j | d | } z | j | d |
Wd | j X| S)a Create a (possibly compressed) tar file from all the files under
'base_dir'.
'compress' must be "gzip" (the default), "bzip2", "xz", or None.
'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_name' + ".tar", possibly plus
the appropriate compression extension (".gz", ".bz2", or ".xz").
Returns the output filename.
Nr r Zgzbzip2bz2xzzCbad value for 'compress', or compression format not supported : {0}r .z.tarz creating %szCreating tar archivec s@ d k r | _ | _ d k r< | _ | _ | S)N)gidZgnameuiduname)Ztarinfo)r groupownerr r r! _set_uid_gid s z#_make_tarball.._set_uid_gidzw|%srd )_ZLIB_SUPPORTED_BZ2_SUPPORTED_LZMA_SUPPORTED
ValueErrorr7 tarfiler, r- dirnamerl infork r r r> addr ) base_namebase_dircompressverbosedry_runr r loggerZtar_compressionr Z compress_extZ archive_name archive_dirr tarr )r r r r r!
_make_tarballS s<
r c
C s d d l } | d } t j j | } | rp t j j | rp | d k r] | j d | | sp t j | | d k r | j d | | | s| j | d d | j K} t j j | } | t j
k r | j | | | d k r | j d | x t j | D] \ }
} } xa t
| D]S }
t j j t j j |
|
} | j | | | d k r,| j d | q,Wxm | D]e }
t j j t j j |
|
} t j j | r| j | | | d k r| j d | qWqWWd QRX| S) zCreate a zip file from all the files under 'base_dir'.
The output zip file will be named 'base_name' + ".zip". Returns the
name of the output zip file.
r Nz.zipz creating %sz#creating '%s' and adding '%s' to itwZ compressionz adding '%s')zipfiler, r- r rl r rk ZipFileZ ZIP_DEFLATEDnormpathcurdirr' walksortedr` isfile)r r r r r r Z zip_filenamer Zzfr- dirpathZdirnames filenamesrL r r r!
_make_zipfile s@
!
! r r r zuncompressed tar filezgzip'ed tar-fileZgztarzZIP filezipr zbzip2'ed tar-fileZbztarr zxz'ed tar-fileZxztarc C s' d d t j D } | j | S)zReturns a list of supported formats for archiving and unarchiving.
Each element of the returned sequence is a tuple (name, description)
c S s& g | ] \ } } | | d f q S)r r ).0rL registryr r r!
s z'get_archive_formats..)_ARCHIVE_FORMATSitemssort)formatsr r r! r s
r c C s | d k r g } t | s. t d | t | t t f sO t d xB | D]: } t | t t f s t | d k rV t d qV W| | | f t | .)_UNPACK_FORMATSr r )r r r r! r * s
c C s i } x9 t j D]+ \ } } x | d D] } | | | r' r ) r} extract_dirr r r rL targetdatafr r r! _unpack_zipfileg s*
r c C sj d d l } y | j | } Wn% | j k
rF t d | Yn Xz | j | Wd | j Xd S)zAUnpack tar/tar.gz/tar.bz2/tar.xz `filename` to `extract_dir`
r Nz/%s is not a compressed or uncompressed tar file)r r> ZTarErrorr# Z
extractallr )r} r r Ztarobjr r r! _unpack_tarfile s r z.tarz.zipz.tar.gzz.tgzz.tar.bz2z.tbz2z.tar.xzz.txzc C sI xB t j D]4 \ } } x% | d D] } | j | r$ | Sq$ Wq
Wd S)Nr )r r r )r} rL r r r r r! _find_unpack_format s
r c C s | d k r t j } | d k r y t | } Wn' t k
r[ t d j | Yn X| d } | | | t | d n_ t | } | d k r t d j | t | d } t t | d } | | | | d S)a
Unpack an archive.
`filename` is the name of the archive.
`extract_dir` is the name of the target directory, where the archive
is unpacked. If not provided, the current working directory is used.
`format` is the archive format: one of "zip", "tar", "gztar", "bztar",
or "xztar". Or any other registered format. If not provided,
unpack_archive will use the filename extension and see if an unpacker
was registered for that extension.
In case none is found, a ValueError is raised.
NzUnknown unpack format '{0}'r r z Unknown archive format '{0}') r, r r r r r7 dictr r# )r} r r7 r r rP r r r! r s
statvfs
disk_usageZusageztotal used freec C sV t j | } | j | j } | j | j } | j | j | j } t | | | S)zReturn disk usage statistics about the given path.
Returned value is a named tuple with attributes 'total', 'used' and
'free', which are the amount of total, used and free space, in bytes.
)r, r f_bavailf_frsizef_blocksf_bfree_ntuple_diskusage)r- r@ freetotalusedr r r! r s
ntc C s/ t j | \ } } | | } t | | | S)zReturn disk usage statistics about the given path.
Returned values is a named tuple with attributes 'total', 'used' and
'free', which are the amount of total, used and free space, in bytes.
)r Z
_getdiskusager )r- r r r r r r! r s
c C s | d k r$ | d k r$ t d | } | } | d k rE d } n< t | t r t | } | d k r t d j | | d k r d } n< t | t s t | } | d k r t d j | t j | | | d S)zChange owner user and group of the given path.
user and group can be the uid/gid or the user/group names, and in that case,
they are converted to their respective uid/gid.
Nz user and/or group must be setr zno such user: {!r}zno such group: {!r}r )
r r rn r LookupErrorr7 intr r, r )r- userr Z_userZ_groupr r r! r s" P c C s y t t j d } Wn t t f k
r7 d } Yn Xy t t j d } Wn t t f k
ro d } Yn X| d k s | d k r y t j t j j } Wn* t t t
f k
r t j | } Yn X| d k r | j } | d k r | j
} t j | | f S)a Get the size of the terminal window.
For each of the two dimensions, the environment variable, COLUMNS
and LINES respectively, is checked. If the variable is defined and
the value is a positive integer, it is used.
When COLUMNS or LINES is not defined, which is the common case,
the terminal connected to sys.__stdout__ is queried
by invoking os.get_terminal_size.
If the terminal size cannot be successfully queried, either because
the system doesn't support querying, or because we are not
connected to a terminal, the value given in fallback parameter
is used. Fallback defaults to (80, 24) which is the default
size used by many terminal emulators.
The value returned is a named tuple of type os.terminal_size.
ZCOLUMNSr ZLINES)r r, environr r r rt
__stdout__filenoAttributeErrorr.
terminal_sizecolumnslines)Zfallbackr r sizer r r! r s$ c s d d } t j j r5 | | r1 Sd S| d k rY t j j d t j } | sc d S| j t j } t j d k rt j
| k r | j d t j
t j j d d j t j } t f d d
| D r g } q f d d | D } n g } t
} xr | D]j } t j j | } | | k r'| j | x6 | D]. } t j j | | }
| |
| r_|
Sq_Wq'Wd S)
a3 Given a command, mode, and a PATH string, return the path which
conforms to the given mode on the PATH, or None if there is no such
file.
`mode` defaults to os.F_OK | os.X_OK. `path` defaults to the result
of os.environ.get("PATH"), or can be overridden with a custom search
path.
c S s5 t j j | o4 t j | | o4 t j j | S)N)r, r- rl accessr_ )r? r\ r r r!
_access_checkO s $z which.._access_checkNPATHwin32r ZPATHEXTr c 3 s* | ] } j j | j Vq d S)N)lowerr )r r )cmdr r! l s zwhich..c s g | ] } | q Sr r )r r )r r r! r o s zwhich..)r, r- r r getdefpathr pathseprt platformr insertanyre r/ r r` ) r r\ r- r ZpathextfilesseendirZnormdirZthefilerL r )r r! r B s8
!
i @ )r N)r r )r r )r r )r r )Sr" r, rt r8 rc collectionsrF zlibr ImportErrorr r Zlzmar pwdr Zgrpr __all__r. r
r r r r# r r$ r r3 r r r+ rN r r r r r
rw r r> rx ry supports_dir_fdrj supports_fdrU r r Zavoids_symlink_attacksr r r r r r r r r r r r r r r r r r r r r r rm
namedtupler r rL r r r F_OKX_OKr r r r r! s
6 Y ! , 5 8 @-
6 "
%
-