3
\]/ � @ s� d Z ddlZddlZddlZddlZddlZddlmZ dddgZ ddd �Z
dd d�Z dd
d�Z ddd�Z
dd� Zedkr�ee� �Zeje� 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 | tj�r tj| �} |s0tdj| �� ytj| �}W n. tk
rl |dk rdtdj| �� g }Y nX |j� x�|D ]�}|dkr�q|tj j
| |�}|d k r�tj j
||�}nd }tj j |�s�|V q||dkr||tj kr||tj
kr|tj j |�r|tj j|� r|t|||d |d�E d H q|W d S )N� zListing {!r}...zCan't list {!r}� __pycache__r � )�ddir� maxlevels�quiet)�
isinstance�os�PathLike�fspath�print�format�listdir�OSError�sort�path�join�isdir�curdir�pardir�islink� _walk_dir)�dirr
r r �names�name�fullname�dfile� r" �"/usr/lib64/python3.6/compileall.pyr s0
r Fr c C s� d} |dk rP|dk r t d��n2|dkrPyddlm} W n tk
rN d}Y nX t| |||d�}
d} |dk r�|dkr�| dk r�|p�d}| |d��0} | jtt||||||d �|
�}
t|
dd
�} W dQ R X n&x$|
D ] }t|||||||�s�d } q�W | 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
Nr z%workers must be greater or equal to 0r )�ProcessPoolExecutor)r r r
T)Z max_workers)r
�force�rxr �legacy�optimize)�defaultF) �
ValueErrorZconcurrent.futuresr$ � ImportErrorr �mapr r �min)r r r
r% r& r r' r( �workersr$ �files�successZexecutorZresults�filer" r" r# r 2 s6
c C s� d}|dk r"t | tj�r"tj| �} tjj| �}|dk rFtjj||�} nd} |dk rd|j| �}
|
rd|S tjj| ��r�|r�| d } nB|dkr�|dkr�|nd} t j
j | | d�} n t j
j | �} tjj | �}
|dd� |dd� }}|d
k�r�|�sXyTt
tj| �j�}tjd t j
j|�}t| d ��}|jd
�}W dQ R X ||k�r>|S W n tk
�rV Y nX |�sltdj| �� ytj| | | d|d�}W n� tjk
�r } zdd}|dk�r�|S |�r�tdj| �� n tddd� |jj t j j dd�}|j t j j �}t|� W Y dd}~X n~ t!t"tfk
�rv } zLd}|dk�r2|S |�rHtdj| �� n tddd� t|j#j$d |� W Y dd}~X nX |dk�r�d}|S )a 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
Tr N�cr r � )� optimization� z.pyz<4sl�rb� zCompiling {!r}...)r( Fz*** Error compiling {!r}...z*** )�end�backslashreplace)�errors�:���r< )%r
r r r r �basenamer �search�isfile� importlib�util�cache_from_source�dirname�int�stat�st_mtime�structZpack� MAGIC_NUMBER�open�readr r r �
py_compile�compile�PyCompileError�msg�encode�sys�stdout�encoding�decode� SyntaxError� UnicodeError� __class__�__name__)r r
r% r& r r' r( r0 r r! Zmo�cfile�optZ cache_dir�head�tail�mtimeZexpectZchandleZactual�ok�errrN �er" r" r# r b sx
$
c
C sX d}xNt jD ]D}| s |tjkr6| r6|dk rPtd� q |oNt||d||||d�}q W |S )aw 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)
Tr zSkipping current directoryN)r r' r( )rP r r r r r )Z skip_curdirr r% r r' r( r0 r r" r" r# r � s
c
C sz 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� |j� }|j}|jr�ddl}|j|j�|_|j dk �r|j }n|j
}|j �r�yJ|j d4k�r4t j
nt|j ��$}x |D ]}|j|j� � �qFW W dQ R X W n2 tk
�r� |jd5k �r�td6j|j �� d7S X |jdk �r�|j�p�d|_d8}y�|�r4xj|D ]b} tjj| ��r t| |j|j|j|j|j ��s*d7}n*t | ||j|j|j|j|j |jd9��s�d7}�q�W |S t |j |j|jd:�S W n* t k
�rt |jd5k �rptd;� d7S X d8S )<zScript main program.r Nz1Utilities to support installing Python libraries.)Z descriptionz-lZ store_constr r z!don't recurse into subdirectories)�actionZconstr) �dest�helpz-r� recursionzhcontrol the maximum recursion level. if `-l` and `-r` options are specified, then `-r` takes precedence.)�typera rb z-f�
store_truer% z/force rebuild even if timestamps are up to date)r` ra rb z-q�countr zIoutput only error messages; -qq will suppress the error messages as well.)r` ra r) rb z-br' z0use legacy (pre-PEP3147) compiled file locationsz-dZDESTDIRr
z�directory to prepend to file paths for use in compile-time tracebacks and in runtime tracebacks in cases where the source file is unavailable)�metavarra r) rb z-xZREGEXPr& zskip files matching the regular expression; the regexp is searched for in the full path of each file considered for compilationz-iZFILE�flistzzadd all the files and directories listed in FILE to the list considered for compilation; if "-", names are read from stdin)rg ra rb � compile_destzFILE|DIR�*zrzero or more file and directory names to compile; if no arguments given, defaults to the equivalent of -l sys.path)rg �nargsrb z-jz --workersr zRun compileall concurrently)r) rd rb �-r zError reading file list {}FT)r. )r' r% r z
[interrupted]) �argparse�ArgumentParser� add_argumentrD �
parse_argsri r& �rerL rc r rh rP �stdinrI �append�stripr r r r r. r r r? r r
r% r' r r �KeyboardInterrupt)
rm �parser�argsZ
compile_destsrq r �f�liner0 ra r" r" r# �main� s~
"
rz �__main__)Nr r ���)r NFNr Fr| r r| )NFNr Fr| r| )r r Fr Fr| )�__doc__r rP �importlib.utilr@ rK rG � functoolsr �__all__r r r r rz rW rD Z exit_status�exitr" r" r" r# �<module> s$
/
R
[