a zeN @sddlZddlZddlZddlZddlZddlZddlmZddl m Z gdZ dddZ dddddd d d Z dddddd d dZdddZddZedkree ZeedS)N)partial)Path) compile_dir compile_file compile_pathccs|dkrt|tjrt|}|s0td|zt|}Wn,tyj|dkrbtd|g}Yn0||D]|}|dkrqxtj ||}tj |s|Vqx|dkrx|tj krx|tj krxtj |rxtj |sxt||d|dEdHqxdS)NzListing {!r}...zCan't list {!r} __pycache__r) maxlevelsquiet) isinstanceosPathLikefspathprintformatlistdirOSErrorsortpathjoinisdircurdirpardirislink _walk_dir)dirr r namesnamefullnamer /usr/lib/python3.9/compileall.pyrs2       rFr stripdir prependdir limit_sl_desthardlink_dupesc CsNd}|dur$| dus| dur$td|dur8|} |} d}|dkrHtd|dkrxzddlm}Wntyvd}Yn0|durt}t|||d}d}|dkr|dur|pd}||dD}|tt ||||||| | | | | d |}t |dd }Wdn1s 0Yn2|D],}t |||||||| | | | | d sd }q|S) NPDestination dir (ddir) cannot be used in combination with stripdir or prependdirrz%workers must be greater or equal to 0r )ProcessPoolExecutor)r r T) max_workers) ddirforcerxr legacyoptimizeinvalidation_moder$r%r&r')defaultr#F) ValueErrorconcurrent.futuresr) ImportErrorsysgetrecursionlimitrmaprrmin)rr r+r,r-r r.r/workersr0r$r%r&r'r)filessuccessexecutorresultsfiler r r!r0sX     . rc& Cs|dur |dus| dur tdd} |dkrBt|tjrBt|}tj|} d}|durhtj|| }|dur|tjj }|tjj }t |}t ||D]\}}||kr| |qtjj|}| dur|durtj| |}ntj| |}t|t r|g}tt|}| r.t|dkr.td|durL||}|rL| S| durtj|rt| t|jvr| Si}tj|r|D]d}|r|d||<nJ|dkr|dkr|nd}tjj||d }|||<ntj|}|||<q| dd | d d}}|d kr|szt t|j}td tjjd|d @}|D]L}t |d}|!d}Wdn1s0Y||kr\qq\| WSWnt"yYn0|st#d$|zt%|D]p\}}||}t&j'|||d||d} |dkr| r|||d}!t(j)||!ddrt*|t+|!|qWnt&j,y}"z|d} |dkr| WYd}"~"S|rt#d$|n t#dddt-j.j/pt-0}#|"j1j2|#dd3|#}$t#|$WYd}"~"nd}"~"0t4t5t"fyp}%z^d} |dkr(| WYd}%~%S|r>t#d$|n t#dddt#|%j6j7d|%WYd}%~%nd}%~%00| dkrd} | S)Nr(TrzXHardlinking of duplicated bytecode makes sense only for more than one optimization levelcrr ) optimizationz.pyz<4sLLlrb zCompiling {!r}...)r/r0F)shallowz*** Error compiling {!r}...z*** )endbackslashreplace)errors:)8r2r r rrrbasenamersplitseplistzipremoveintsortedsetlensearchrrresolveparentsisfile importlibutilcache_from_sourcestatst_mtimestructpack MAGIC_NUMBERvaluesopenreadrrr enumerate py_compilecompilefilecmpcmpunlinklinkPyCompileErrorr5stdoutencodinggetdefaultencodingmsgencodedecode SyntaxError UnicodeError __class____name__)&rr+r,r-r r.r/r0r$r%r&r'r;rdfilefullname_partsstripdir_parts ddir_partsspartopartmo opt_cfiles opt_leveloptcfileheadtailmtimeexpectchandleactualindexokprevious_cfileerrrlrner r r!r{s                *         ( rc CsTd}tjD]D}|r|tjkr2|r2|dkrNtdq |oLt||d|||||d}q |S)NTrzSkipping current directory)r r.r/r0)r5rr rrr) skip_curdirr r,r r.r/r0r;rr r r!rs    rc Csddl}|jdd}|jddddddd|jd td d d |jd dddd|jdddddd|jddddd|jdddddd|jdd d!dd"d|jd#d$d%dd&d|jd'd(d)dd*d|jd+d,d-d.d/|jd0d1d2d3d4|jd5d6d7td8d9d:d;tjD}|jd|jd?d@tdAdBdC|jdDdEdFdGd/|jdHddIdJd|}|j}|j rddl }| |j |_ |j dKkrd|_ |j dur|j }n|j}|jdurdLg|_t|jd7kr|jr|dM|jdur|jdus |jdur|dN|jrz\|jdOkr0tjnt|j*}|D]}||q@Wdn1sl0YWn2ty|jdPkrtdQ|jYdRS0|j r|j !dOdS"} tj| } nd} dT} z|r~|D]} t#j$%| r4t&| |j|j'|j |j|j(| |j|j|j|j |jdU stdR} n@t)| ||j|j'|j |j|j(|j*| |j|j|j|j |jdVsdR} q| WSt+|j(|j'|j| dWWSWn*t,y|jdPkrtdXYdRS0dTS)YNrz1Utilities to support installing Python libraries.) descriptionz-l store_constr z!don't recurse into subdirectories)actionconstr1desthelpz-r recursionzhcontrol the maximum recursion level. if `-l` and `-r` options are specified, then `-r` takes precedence.)typerrz-f store_truer,z/force rebuild even if timestamps are up to date)rrrz-qcountr zIoutput only error messages; -qq will suppress the error messages as well.)rrr1rz-br.z0use legacy (pre-PEP3147) compiled file locationsz-dDESTDIRr+zdirectory to prepend to file paths for use in compile-time tracebacks and in runtime tracebacks in cases where the source file is unavailable)metavarrr1rz-sSTRIPDIRr$zpart of path to left-strip from path to source file - for example buildroot. `-d` and `-s` options cannot be specified together.z-p PREPENDDIRr%zpath to add as prefix to path to source file - for example / to make it absolute when some part is removed by `-s` option. `-d` and `-p` options cannot be specified together.z-xREGEXPr-zskip files matching the regular expression; the regexp is searched for in the full path of each file considered for compilationz-iFILEflistzzadd all the files and directories listed in FILE to the list considered for compilation; if "-", names are read from stdin)rrr compile_destzFILE|DIR*zrzero or more file and directory names to compile; if no arguments given, defaults to the equivalent of -l sys.path)rnargsrz-jz --workersr zRun compileall concurrently)r1rrcSsg|]}|jddqS)_-)rlowerreplace).0moder r r! gszmain..z--invalidation-modezset .pyc invalidation mode; defaults to "checked-hash" if the SOURCE_DATE_EPOCH environment variable is set, and "timestamp" otherwise.)choicesrz-oappend opt_levelszOptimization levels to run compilation with. Default is -1 which uses the optimization level of the Python interpreter itself (see -O).)rrrrz-eDIRr&z+Ignore symlinks pointing outsite of the DIRz--hardlink-dupesr'zHardlink duplicated pyc filesr@r"zYHardlinking of duplicated bytecode makes sense only for more than one optimization level.z.-d cannot be used in combination with -s or -prrzError reading file list {}FrT)r0r$r%r/r&r')r9r0r$r%r/r&r')r.r,r r0z [interrupted])-argparseArgumentParser add_argumentrPrdPycInvalidationModerQ parse_argsrr-rerer&rr rrSr'errorr+r$r%rr5stdinrarstriprr rrr0rupperr rrWrr,r.rr9rKeyboardInterrupt) rparserinvalidation_modesargs compile_destsrr flineivl_moder0r;rr r r!main2s                    6        r__main__)r) NNFNrFr"r N)NFNrFr"N)r rFrFr"N)r r5importlib.utilrXrdr]rf functoolsrpathlibr__all__rrrrrrtrP exit_statusexitr r r r! sB    K  "