U
e5dn5 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ dddgZddd�Z dd d�Z
d
d� Z ddd�Z ddd�Z
dd� Zedkr�ee� �Ze�e� dS )� N)�partial� compile_dir� compile_file� compile_path�
c c s |dk r t | tj�r t�| �} |s0td�| �� zt�| �}W n. tk
rl |dk rdtd�| �� g }Y nX |�� |D ]�}|dkr�qztj �
| |�}|d k r�tj �
||�}nd }tj � |�s�||fV qz|dkrz|tj krz|tj
krztj � |�rztj �|�szt|||d |d�E d H qzd 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.8/compileall.pyr s:
�
� �r F�����r c
C s� d }
|dk rt d��|dkrFzddlm}
W n tk
rD d}Y nX t| |||d�} d} |dkr�|
d k r�|ppd }|
|d��0}
|
�tt|||||| d�| �}t|dd �} W 5 Q R X n(| D ]"\}}t |||||||| �s�d
} q�| S ) Nr z%workers must be greater or equal to 0r )�ProcessPoolExecutor)r r r
T)Z max_workers)�force�rxr �legacy�optimize�invalidation_mode)�defaultF)
�
ValueErrorZconcurrent.futuresr% � ImportErrorr �mapr �_compile_file_tuple�minr )r r r
r& r' r r( r) �workersr* r% Zfiles_and_ddirs�successZexecutorZresults�filer! r" r" r# r 2 sF
� �� �c K s | \}}t ||f|�S )N)r )Zfile_and_dfile�kwargsr3 r! r" r" r# r/ e s r/ c
C s� d}|dk r"t | tj�r"t�| �} tj�| �} |d k rFtj�|| �}
nd }
|d k rd|�| �} | rd|S tj�| ��r�|r�| d } nB|dkr�|dkr�|nd}
t j
j | |
d�} n t j
� | �} tj� | �}| d d� | dd � }}|d k�r�|�s\zXt
t�| �j�}t�d
t j
jd|�}t| d ��}|�d �}W 5 Q R X ||k�rB|W S W n tk
�rZ Y nX |�sptd
�| �� z tj| | |
d||d�}W �n tjk
�r } zjd}|dk�r�| W Y �RS |�r�td�| �� n tddd� |jj t j j dd�}|� t j j �}t|� W 5 d }~X Y n� t!t"tfk
�r� } zRd}|dk�rJ| W Y �:S |�r`td�| �� n tddd� t|j#j$d |� W 5 d }~X Y nX |dk�r�d}|S )NTr �cr r � )� optimization�����z.pyz<4sll�rb� zCompiling {!r}...)r) r* Fz*** Error compiling {!r}...z*** )�end�backslashreplace)�errors�:)%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) r* r2 r r! Zmo�cfile�optZ cache_dir�head�tail�mtimeZexpectZchandleZactual�ok�errrP �er" r" r# r j s�
�
�
�
�
$
c C sT d}t jD ]D}|r |tjkr2| r2|dk rNtd� q
|oLt||d |||||d�}q
|S )NTr zSkipping current directory)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� 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� d3d4� tjD �}|jd5t|�d6d7� |�� }|j}|j �r$dd l
}|� |j �|_ |j d k �r8|j }n|j
}|j�r�zF|jd8k�rZtjnt|j�� }|D ]}|�|�� � �qjW 5 Q R X W n4 tk
�r� |jd9k �r�td:�|j�� Y d;S X |j�r�|j�d8d<��� } tj| }
nd }
d=} z�|�rl|D ]h} tj � | ��r6t | |j |j |j |j|j!|
d>��sbd;} n,t"| ||j |j |j |j|j!|j#|
d?� �s�d;} �q�| W S t$|j!|j |j|
d@�W S W n, t%k
�r� |jd9k �r�tdA� Y d;S X d=S )BNr 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.)�typerc rd z-f�
store_truer& z/force rebuild even if timestamps are up to date)rb rc rd z-q�countr 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
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)�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-iZFILE�flistzzadd 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 d��qS )�_�-)r �lower�replace)�.0�moder" r" r# �
s �zmain..z--invalidation-modez�set .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])&�argparse�ArgumentParser� add_argumentrF rM �PycInvalidationMode�sorted�
parse_argsrk r' �rerN re r rj rR �stdinrK �append�stripr r r r r* rq �upperr r rA r r
r& r( r r1 r �KeyboardInterrupt)
rv �parserZinvalidation_modes�argsZ
compile_destsr| r �f�lineZivl_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)r rR �importlib.utilrB rM rI � functoolsr �__all__r r r/ r r r� rY rF Z exit_status�exitr" r" r" r# �
s:
�
3 �
V �
"i