i fޛ 1 @ s d Z d d l Z d d l Z d d l Z d d l m 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
m Z Wn e k
r d Z Yn Xy d d l m Z Wn e k
r d 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 d0 d
Z e e d1 r4d. d d2 d3 Z n d4 d3 Z d. d d5 d Z d. d d6 d Z! d. d d7 d
Z" d8 d Z# d d e" d d9 d Z$ d: d; Z% d< d= Z& e j' e j e j( e j) h e j* k oe j+ e j, k oe j e j- k Z. d d d> d Z/ e. e/ _0 d? d@ Z1 dA d Z2 dB dC Z3 dD dE Z4 dF dG Z5 dH d d d d d dI dJ Z6 d d dK dL Z7 d d d dM dN Z8 i e6 d~ g dP f dQ 6e6 d g dR f dS 6e8 g dT f dU 6Z9 e re6 d g dW f e9 dX d_ d` Z? d dZ da d Z@ db d ZA dc dd ZB de df ZC dg dh ZD i di dj g eD g dP f dQ 6dk g eD g dR f dS 6dl g eC g dT f dU 6ZE e rdm g eD g dW f eE dX ._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 zENOTSUP)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 | r6 t j j | t j j | } n 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 | r6 t j j | t j j | } n 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 s t j | } | d k r- | | | } n t } t j | g } x| D]} | | k rh qP n t j j | | }
t j j | | } y t j j |
r=t j |
} | r t j | | t |
| d | qut j j
| r| rwP n t j j |
r-t |
| | | | qu| |
| n8 t j j |
rht |
| | | | 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 | | Wn\ t k
re} z< t | d d d k rS| j | | t | f n WYd d } ~ Xn X| r{t
| n | 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 symlinksignoreZ
copy_functionZignore_dangling_symlinksrJ rf errorsrL ZsrcnameZdstnamelinktor^ r] r! r! r" r sL $
&32c $ C s y% t j j | r$ t d n Wn2 t k
rY | 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
rQ| 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" rv _ s6
"
rv 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/ filenamers rt r. r` r8 r: ru r> O_RDONLYsamestatfstat_rmtree_safe_fdrx r; closerw )
topfdr. ry rJ r^ rL rz orig_str\ dirfdr! r! r" r sD ,
!
%
r c ! C s | r d d } n | d k r0 d d } n t rt | t rW t j | } n 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
| rZt | | | y t j | Wqt k
rV| 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" ry s zrmtree..onerrorNc W s d S)Nr! )rO r! r! r" ry s z%Cannot call rmtree on a symbolic link)_use_fd_functions
isinstancebytesr- fsdecoderC Exceptionrs rt r> r} r. r~ r r rx r/ r; r rv )r.
ignore_errorsry 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 | r~ t | | r; t j | | d St j j | t | } t j j | r~ t d | q~ n y t j | | Wn t k
r[t j j
| r t j | } t j | | t j
| nt t j j | r=t | | r t d | | f n t | | d d t | n t | | t j
| Yn X| S)a0 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.
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'.ro T)r- r. r_ r3 renamer` r rl r r/ r; r= r< rw
_destinsrcr r
r
)r1 r2 Zreal_dstrr r! r! r" r s.
c C su t | } t | } | j t j j s@ | t j j 7} n | j t j j sh | t j j 7} n | j | S)N)r endswithr- r. r
startswith)r1 r2 r! r! r" r 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_uid4 s
r gzipc
s i d d 6d d 6} i d d 6} t r> d | d ._set_uid_gidzw|%srd )_BZ2_SUPPORTED
ValueErrorr7 getr- r. dirnamerl infork r r tarfiler> addr ) base_namebase_dircompressverbosedry_runr r loggerZtar_compressionZ compress_extZ archive_name archive_dirr tarr! )r r r r r"
_make_tarball@ s4
r c C s~ | r d } n d } d d l m } d d l m } y | d | | | g d | Wn" | k
ry t d | Yn Xd S) Nz-rz-rqr )DistutilsExecError)spawnzipr zkunable to create zip file '%s': could neither import the 'zipfile' module nor find a standalone zip utility)Zdistutils.errorsr Zdistutils.spawnr r )r zip_filenamer r Z
zipoptionsr r r! r! r" _call_external_zip~ s
r c C sN | d } t j j | } | rm t j j | rm | d k rT | j d | n | sm t j | qm n y d d l } Wn t k
r d } Yn X| d k r t | | | | n| d k r | j d | | n | sJ| j | d d | j
H} t j j | } | j | | | d k rA| j d | n x t j
| D] \ }
} } xd t | D]V }
t j j t j j |
|
} | j | | | d k rm| j d | qmqmWxs | D]k }
t j j t j j |
|
} t j j | r| j | | | d k r9| j d | q9qqWqQWWd QXn | S) am 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 ExecError. Returns the name of the output zip
file.
z.zipNz creating %sr z#creating '%s' and adding '%s' to itwZ compressionz adding '%s')r- r. r rl r rk zipfile ImportErrorr ZipFileZ ZIP_DEFLATEDnormpathr( walksortedr` isfile)r r r r r r r r Zzfr. dirpathZdirnames filenamesrL r! r! r"
_make_zipfile sH
!
! 'r r zgzip'ed tar-fileZgztarzuncompressed tar filer zZIP filer r zbzip2'ed tar-fileZbztarc 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 } n t | s4 t d | n t | t t f sX t d n xE | D]= } t | t t f s t | d k r_ t d q_ q_ 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_zipfilee s0
r c C s^ y t j | } Wn% t j k
r: t d | Yn Xz | j | Wd | j Xd S)z:Unpack tar/tar.gz/tar.bz2 `filename` to `extract_dir`
z/%s is not a compressed or uncompressed tar fileN)r r> ZTarErrorr$ Z
extractallr )r| r Ztarobjr! r! r" _unpack_tarfile s r z.tar.gzz.tgzz.tarz.zipz.bz2c 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 } n | d k r y t | } Wn' t k
r^ t d j | Yn X| d } | | | t | d nb t | } | d k r t d j | n 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", or "gztar". 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 n | } | } | d k rH d } nB t | t r t | } | d k r t d j | q n | d k r d } nB t | t s t | } | d k r t d j | q n 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 ry t j t j j } Wn' t t
f k
r t j | } Yn X| d k r | j } n | d k r| j
} qn 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 rs
__stdout__fileno NameErrorr/
terminal_sizecolumnslines)Zfallbackr r sizer! r! r" r s$ c s d d } t j j r5 | | r1 Sd S| d k r\ t j j d t j } n | sf d S| j t j } t j d k rt j
| k r | j d t j
n t j j d d j t j } t f d d
| D r g } q f d d | D } n g } t
} xu | D]m } t j j | } | | k r-| j | x9 | D]. } t j j | | }
| |
| re|
SqeWq-q-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_checkF s $z which.._access_checkNPATHwin32r ZPATHEXTr c 3 s* | ] } j j | j Vq d S)N)lowerr )r r )cmdr! r" c s zwhich..c s g | ] } | q Sr! r! )r r )r r! r" r f s zwhich..)r- r. r r r defpathr pathseprs platformr insertanyre r0 r r` ) r r\ r. r ZpathextfilesseendirZnormdirZthefilerL r! )r r" r 9 s8
!
i @ )compresszgzip)r N)r zbzip2)r r )Sr# r- rs r8 Zos.pathr rc collectionsrF r r r r pwdr Zgrpr __all__r/ r r r r r$ r r% r r3 r r r, rN r r r
r r rv r r> rw rx 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 r rm
namedtupler r rL r r r F_OKX_OKr r! r! r! r" s
6 Y ! , 5 1 =6
6 %
%
+