U e5dn5 @sddlZddlZddlZddlZddlZddlmZdddgZdddZ dd dZ d dZ dddZ dddZ ddZedkree ZeedS)N)partial compile_dir compile_file compile_path ccs|dkrt|tjrt|}|s0td|zt|}Wn.tk rl|dkrdtd|g}YnX||D]}|dkrqztj ||}|dk rtj ||}nd}tj |s||fVqz|dkrz|tj krz|tj krztj |rztj |szt|||d|dEdHqzdS)NzListing {!r}...zCan't list {!r} __pycache__r)ddir maxlevelsquiet) isinstanceosPathLikefspathprintformatlistdirOSErrorsortpathjoinisdircurdirpardirislink _walk_dir)dirr r r namesnamefullnamedfiler""/usr/lib64/python3.8/compileall.pyrs:      rFr c  Csd} |dkrtd|dkrFzddlm} Wntk rDd}YnXt||||d} d} |dkr| dk r|ppd}| |d0} | tt|||||| d| }t|dd } W5QRXn(| D]"\}}t |||||||| sd } q| S) Nrz%workers must be greater or equal to 0r )ProcessPoolExecutor)r r r T)Z max_workers)forcerxr legacyoptimizeinvalidation_mode)defaultF) ValueErrorZconcurrent.futuresr% ImportErrorrmapr_compile_file_tupleminr)rr r r&r'r r(r)workersr*r%Zfiles_and_ddirssuccessZexecutorZresultsfiler!r"r"r#r2sF    cKs|\}}t||f|S)N)r)Zfile_and_dfilekwargsr3r!r"r"r#r/esr/c Csd}|dkr"t|tjr"t|}tj|} |dk rFtj|| } nd} |dk rd||} | rd|Stj|r|r|d} nB|dkr|dkr|nd} t j j || d} n t j |} tj | }| dd| dd}}|d kr|s\zXt t|j}td t j jd|}t| d }|d }W5QRX||krB|WSWntk rZYnX|sptd |ztj|| | d||d}Wntjk r}zjd}|dkr|WYRS|rtd|n tddd|jjtjjdd}| tjj}t|W5d}~XYnt!t"tfk r}zRd}|dkrJ|WY:S|r`td|n tdddt|j#j$d|W5d}~XYnX|dkrd}|S)NTrcrr ) optimizationz.pyz<4sllrb zCompiling {!r}...)r)r*Fz*** Error compiling {!r}...z*** )endbackslashreplace)errors:)%r rrrrbasenamersearchisfile importlibutilcache_from_sourcedirnameintstatst_mtimestructZpack MAGIC_NUMBERopenreadrrr py_compilecompilePyCompileErrormsgencodesysstdoutencodingdecode SyntaxError UnicodeError __class____name__)r r r&r'r r(r)r*r2rr!ZmocfileoptZ cache_dirheadtailmtimeZexpectZchandleZactualokerrrPer"r"r#rjs                    $ c CsTd}tjD]D}|r|tjkr2|r2|dkrNtdq |oLt||d|||||d}q |S)NTrzSkipping current directory)r r(r)r*)rRrrrrr) Z skip_curdirr r&r r(r)r*r2rr"r"r#rs    c Csddl}|jdd}|jddddddd |jd td d d |jddddd|jdddddd|jddddd|jdddddd|jd d!d"dd#d|jd$d%d&d'd(|jd)d*d+d,d-|jd.d/d0td1d2d3d4tjD}|jd5t|d6d7|}|j}|j r$ddl }| |j |_ |j dk r8|j }n|j }|jrzF|jd8krZtjnt|j }|D]}||qjW5QRXWn4tk r|jd9krtd:|jYd;SX|jr|jd8d<} 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? sd;} q| WSt$|j!|j |j| d@WSWn,t%k r|jd9krtdAYd;SXd=S)BNrz1Utilities to support installing Python libraries.)Z descriptionz-lZ store_constrr 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.)typercrdz-f store_truer&z/force rebuild even if timestamps are up to date)rbrcrdz-qcountr zIoutput only error messages; -qq will suppress the error messages as well.)rbrcr+rdz-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)metavarrcr+rdz-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)rircrd compile_destzFILE|DIR*zrzero or more file and directory names to compile; if no arguments given, defaults to the equivalent of -l sys.path)rinargsrdz-jz --workersr zRun compileall concurrently)r+rfrdcSsg|]}|jddqS)_-)rlowerreplace).0moder"r"r# szmain..z--invalidation-modezset .pyc invalidation mode; defaults to "checked-hash" if the SOURCE_DATE_EPOCH environment variable is set, and "timestamp" otherwise.)choicesrdrorzError reading file list {}FrnT)r*)r1r*)r(r&r r*z [interrupted])&argparseArgumentParser add_argumentrFrMPycInvalidationModesorted parse_argsrkr'rerNrer rjrRstdinrKappendstriprr rrr*rqupperrrrArr r&r(rr1rKeyboardInterrupt) rvparserZinvalidation_modesargsZ compile_destsr|r flineZivl_moder*r2rcr"r"r#mains                  r__main__)Nrr) rNFNrFr$r N)NFNrFr$N)r rFrFr$N)rrRimportlib.utilrBrMrI functoolsr__all__rrr/rrrrYrFZ exit_statusexitr"r"r"r# s:    3 V "i