U
e5dn5 @ s d Z ddlZddlZddlZddlZddlZddlmZ dddgZ ddd Z
dd
dZ dd Z dddZ
dddZdd Zedkree Zee dS )a Module/script to byte-compile all .py files to .pyc files.
When called as a script with arguments, this compiles the directories
given as arguments recursively; the -l option prevents it from
recursing into directories.
Without arguments, if compiles all modules on sys.path, without
recursing into subdirectories. (Even though it should do so for
packages -- for now, you'll have to deal with packages separately.)
See module py_compile for details of the actual byte-compilation.
N)partial compile_dir compile_file compile_path
c c s |dk r t | tjr t| } |s0td| zt| }W n. tk
rl |dk rdtd| g }Y nX | |D ]}|dkrqztj
| |}|d k rtj
||}nd }tj |s||fV qz|dkrz|tj krz|tj
krztj |rztj |szt|||d |dE d H qzd S )N zListing {!r}...zCan't list {!r} __pycache__r )ddir maxlevelsquiet)
isinstanceosPathLikefspathprintformatlistdirOSErrorsortpathjoinisdircurdirpardirislink _walk_dir)dirr
r r namesnamefullnamedfile r" "/usr/lib64/python3.8/compileall.pyr s:
r Fr c
C s d}
|dk rt d|dkrFzddlm}
W n tk
rD d}Y nX t| |||d} d} |dkr|
dk r|ppd}|
|d0}
|
tt|||||| d | }t|dd
} W 5 Q R X n(| D ]"\}}t |||||||| sd } q| S ) a Byte-compile all modules in the given directory tree.
Arguments (only dir is required):
dir: the directory to byte-compile
maxlevels: maximum recursion level (default 10)
ddir: the directory that will be prepended to the path to the
file as it is compiled into each byte-code file.
force: if True, force compilation, even if timestamps are up-to-date
quiet: full output with False or 0, errors only with 1,
no output with 2
legacy: if True, produce legacy pyc paths instead of PEP 3147 paths
optimize: optimization level or -1 for level of the interpreter
workers: maximum number of parallel workers
invalidation_mode: how the up-to-dateness of the pyc will be checked
Nr z%workers must be greater or equal to 0r )ProcessPoolExecutor)r r r
T)Z max_workers)forcerxr legacyoptimizeinvalidation_mode)defaultF)
ValueErrorZconcurrent.futuresr% ImportErrorr mapr _compile_file_tupleminr )r r r
r& r' r r( r) workersr* r% Zfiles_and_ddirssuccessZexecutorZresultsfiler! r" r" r# r 2 sF
c K s | \}}t ||f|S )z-Needs to be toplevel for ProcessPoolExecutor.)r )Zfile_and_dfilekwargsr3 r! r" r" r# r/ e s r/ c
C s d}|dk r"t | tjr"t| } tj| } |dk rFtj|| }
nd}
|dk rd|| } | rd|S tj| r|r| d } nB|dkr|dkr|nd}
t j
j | |
d} n t j
| } tj | }| dd | d d }}|d
kr|s\zXt
t| j}td t j
jd|}t| d }|d
}W 5 Q R X ||krB|W S W n tk
rZ Y nX |sptd| z tj| | |
d||d}W n tjk
r } zjd}|dkr| W Y RS |rtd| n tddd |jj t j j dd}| t j j }t| W 5 d}~X Y n t!t"tfk
r } zRd}|dkrJ| W Y :S |r`td| n tddd t|j#j$d | W 5 d}~X Y nX |dkrd}|S )aT Byte-compile one file.
Arguments (only fullname is required):
fullname: the file to byte-compile
ddir: if given, the directory name compiled in to the
byte-code file.
force: if True, force compilation, even if timestamps are up-to-date
quiet: full output with False or 0, errors only with 1,
no output with 2
legacy: if True, produce legacy pyc paths instead of PEP 3147 paths
optimize: optimization level or -1 for level of the interpreter
invalidation_mode: how the up-to-dateness of the pyc will be checked
Tr Ncr r ) optimizationz.pyz<4sllrb zCompiling {!r}...)r) r* Fz*** Error compiling {!r}...z*** )endbackslashreplace)errors:)%r
r r r r basenamer searchisfile importlibutilcache_from_sourcedirnameintstatst_mtimestructZpack MAGIC_NUMBERopenreadr r r
py_compilecompilePyCompileErrormsgencodesysstdoutencodingdecode SyntaxError UnicodeError __class____name__)r r
r& r' r r( r) r* r2 r r! ZmocfileoptZ cache_dirheadtailmtimeZexpectZchandleZactualokerrrP er" r" r# r j s
$
c C sT d}t jD ]D}|r |tjkr2| r2|dk rNtd q
|oLt||d|||||d}q
|S )a Byte-compile all module on sys.path.
Arguments (all optional):
skip_curdir: if true, skip current directory (default True)
maxlevels: max recursion level (default 0)
force: as for compile_dir() (default False)
quiet: as for compile_dir() (default 0)
legacy: as for compile_dir() (default False)
optimize: as for compile_dir() (default -1)
invalidation_mode: as for compiler_dir()
Tr zSkipping current directoryN)r r( r) r* )rR r r r r r ) Z skip_curdirr r& r r( r) r* r2 r r" r" r# r s
c
C s ddl } | jdd}|jdddddd d
|jd td d
d |jddddd |jdddddd |jddddd |jd d d dd d |jd!d"d#dd$d |jd%d&d'd(d) |jd*d+d,d-d. |jd/d0d1td2d3 d4d5 tjD }|jd6t|d7d8 | }|j}|j r$ddl
}| |j |_ |j dk r8|j }n|j
}|jrzF|jd9krZtjnt|j }|D ]}|| qjW 5 Q R X W n4 tk
r |jd:k rtd;|j Y d} z|rl|D ]h} tj | r6t | |j |j |j |j|j!|
d?sbd<} n,t"| ||j |j |j |j|j!|j#|
d@ sd<} q| W S t$|j!|j |j|
dAW S W n, t%k
r |jd:k rtdB Y dS )CzScript main program.r Nz1Utilities to support installing Python libraries.)Z descriptionz-lZ store_constr r z!don't recurse into subdirectories)actionZconstr+ desthelpz-r recursionzhcontrol the maximum recursion level. if `-l` and `-r` options are specified, then `-r` takes precedence.)typerc rd z-f
store_truer& z/force rebuild even if timestamps are up to date)rb rc rd z-qcountr zIoutput only error messages; -qq will suppress the error messages as well.)rb rc r+ rd z-br( z0use legacy (pre-PEP3147) compiled file locationsz-dZDESTDIRr
zdirectory to prepend to file paths for use in compile-time tracebacks and in runtime tracebacks in cases where the source file is unavailable)metavarrc r+ rd z-xZREGEXPr' zskip files matching the regular expression; the regexp is searched for in the full path of each file considered for compilationz-iZFILEflistzzadd all the files and directories listed in FILE to the list considered for compilation; if "-", names are read from stdin)ri rc rd compile_destzFILE|DIR*zrzero or more file and directory names to compile; if no arguments given, defaults to the equivalent of -l sys.path)ri nargsrd z-jz --workersr zRun compileall concurrently)r+ rf rd c S s g | ]}|j d dqS )_-)r lowerreplace).0moder" r" r#
s zmain..z--invalidation-modezset .pyc invalidation mode; defaults to "checked-hash" if the SOURCE_DATE_EPOCH environment variable is set, and "timestamp" otherwise.)choicesrd ro r zError reading file list {}Frn T)r* )r1 r* )r( r& r r* z
[interrupted])&argparseArgumentParser add_argumentrF rM PycInvalidationModesorted
parse_argsrk r' rerN re r rj rR stdinrK appendstripr r r r r* rq upperr r rA r r
r& r( r r1 r KeyboardInterrupt)
rv parserZinvalidation_modesargsZ
compile_destsr| r flineZivl_moder* r2 rc r" r" r# main s
r __main__)Nr r ) r NFNr Fr$ r N)NFNr Fr$ N)r r Fr Fr$ N)__doc__r rR importlib.utilrB rM rI functoolsr __all__r r r/ r r r rY rF Z exit_statusexitr" r" r" r# s<
3
V
"i