3
\]/ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ dddgZddd�Z dd d�Z
dd d�Z dd
d�Z dd� Z
edkr�ee
� �Zeje� dS )� 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 } |d k rP|dk r t d��n2|dkrPyddlm} W n tk
rN d}Y nX t| |||d�}
d} |d k r�|dkr�| d k r�|p�d }| |d��0} | jtt||||||d�|
�}
t|
dd �} W d Q R X n&x$|
D ] }t|||||||�s�d
} q�W | S ) 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| �}|d k rFtjj||�} nd } |d k rd|j| �}
|
rd|S tjj| ��r�|r�| d } nB|dkr�|dkr�|nd} t j
j | | d�} n t j
j | �} tjj | �}
|d d� |dd � }}|d k�r�|�sXyTt
tj| �j�}tjd
t j
j|�}t| d ��}|jd �}W d Q R X ||k�r>|S W n tk
�rV Y nX |�sltd
j| �� 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 d d }~X n~ t!t"tfk
�rv } zLd}|dk�r2|S |�rHtdj| �� n tddd� t|j#j$d |� W Y d d }~X nX |dk�r�d}|S )NTr �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 )NTr zSkipping current directory)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 dd l } | jdd�}|jddddddd � |jd
td d d
� |jddddd� |jdddddd� |jddddd� |jdd d d d d � |jd d!d"d d#d � |jd$d%d&d'd(� |jd)d*d+d,d-� |jd.d/d0td1d2� |j� }|j}|jr�dd l}|j|j�|_|j d k �r|j }n|j
}|j �r�yJ|j d3k�r4t j
nt|j ��$}x |D ]}|j|j� � �qFW W d Q R X W n2 tk
�r� |jd4k �r�td5j|j �� d6S X |jd k �r�|j�p�d |_d7}y�|�r4xj|D ]b} tjj| ��r t| |j|j|j|j|j ��s*d6}n*t | ||j|j|j|j|j |jd8��s�d6}�q�W |S t |j |j|jd9�S W n* t k
�rt |jd4k �rptd:� d6S X d7S );Nr z1Utilities 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| )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
[