o IAfO@s ddlZddlZddlZddlZddlZddlZddlmZddl m Z gdZ dddZ  dddddd d d Z  dddddd d dZ  dddZddZedkrgee ZeedSdS)N)partial)Path) compile_dir compile_file compile_pathccs|dkrt|tjrt|}|std|zt|}Wnty6|dkr2td|g}Ynw||D]>}|dkrDq=tj ||}tj |sU|Vq=|dkr{|tj kr{|tj kr{tj |r{tj |s{t||d|dEdHq=dS)NzListing {!r}...zCan't list {!r} __pycache__r) maxlevelsquiet) isinstanceosPathLikefspathprintformatlistdirOSErrorsortpathjoinisdircurdirpardirislink _walk_dir)dirr r namesnamefullnamer !/usr/lib/python3.10/compileall.pyrs:       rFr stripdir prependdir limit_sl_desthardlink_dupesc Cs\ d}|dur| dus| durtd|dur|} |} d}|dkr%td|dkrFddlm}z|Wn ty?d}Ynwddlm}|durNt}t|||d}d}|dkr|dur|pbd}||d #}| t t ||||||| | | | | d |}t |dd }Wd|S1swY|S|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 )_check_system_limits)ProcessPoolExecutor)r r T) max_workers) ddirforcerxr legacyoptimizeinvalidation_moder$r%r&r')defaultr#F) ValueErrorconcurrent.futures.processr)NotImplementedErrorconcurrent.futuresr*sysgetrecursionlimitrmaprrmin)rr r,r-r.r r/r0workersr1r$r%r&r'r*r)filessuccessexecutorresultsfiler r r!r0sj        rc& Csn |dur|dus | durtdd} |dkr"t|tjr"t|}tj|} d}|dur5tj|| }|durd|tjj }|tjj }t |}t ||D] \}}||kr]| |qPtjj|}| dur{|durttj| |}ntj| |}t|t r|g}tt|}| rt|dkrtd|dur||}|r| S| durtj|rt| t|jvr| Si}tj|r5|D].}|r|d||<q|dkr|dkr|nd}tjj||d }|||<qtj|}|||<q| dd | d d}}|d kr5|sXzCt t|j}td tjjd|d @}|D]%}t |d }|!d}Wdn 1s;wY||krGnq#| WSWn t"yWYnw|sbt#d$|z?t%|D]8\}}||}t&j'|||d||d} |dkr| r|||d}!t(j)||!ddrt*|t+|!|qgWnt&j,y}"z?d} |dkr| WYd}"~"S|rt#d$|nt#dddt-j.j/pt-0}#|"j1j2|#dd3|#}$t#|$WYd}"~"| Sd}"~"wt4t5t"fy-}%z0d} |dkr| WYd}%~%S|rt#d$|nt#dddt#|%j6j7d|%WYd}%~%| Sd}%~%ww| dkr5d} | S)Nr(TrzXHardlinking of duplicated bytecode makes sense only for more than one optimization levelcrr ) optimizationz.pyz<4sLLlrb zCompiling {!r}...)r0r1F)shallowz*** Error compiling {!r}...z*** )endbackslashreplace)errors:)8r3r r rrrbasenamersplitseplistzipremoveintsortedsetlensearchrrresolveparentsisfile importlibutilcache_from_sourcestatst_mtimestructpack MAGIC_NUMBERvaluesopenreadrrr enumerate py_compilecompilefilecmpcmpunlinklinkPyCompileErrorr7stdoutencodinggetdefaultencodingmsgencodedecode SyntaxError UnicodeError __class____name__)&rr,r-r.r r/r0r1r$r%r&r'r=rdfilefullname_partsstripdir_parts ddir_partsspartopartmo opt_cfiles opt_leveloptcfileheadtailmtimeexpectchandleactualindexokprevious_cfileerrrnrper r r!r}s                          rc CsV d}tjD]"}|r|tjkr|r|dkrtdq|o't||d|||||d}q|S)NTrzSkipping current directory)r r/r0r1)r7rr rrr) skip_curdirr r-r r/r0r1r=rr r r!rs$  rc Cs ddl}|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|jrRz0|jdOkrtjnt|jdPdQ}|D] }||qWdn 1s2wYWntyQ|jdRkrNtdS|jYdTSw|j re|j !dOdU"} tj| } nd} dV} z^|r|D]I} t#j$%| rt&| |j|j'|j |j|j(| |j|j|j|j |jdW sdT} qot)| ||j|j'|j |j|j(|j*| |j|j|j|j |jdXsdT} qo| WSt+|j(|j'|j| dYWSt,y|jdRkrtdZYdTSw)[Nrz1Utilities to support installing Python libraries.) descriptionz-l store_constr z!don't recurse into subdirectories)actionconstr2desthelpz-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.)rrr2rz-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)metavarrr2rz-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)r2rrcSsg|] }|jddqS)_-)rlowerreplace).0moder r r! iszmain..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 filesrBr"zYHardlinking of duplicated bytecode makes sense only for more than one optimization level.z.-d cannot be used in combination with -s or -przutf-8)rnrzError reading file list {}FrT)r1r$r%r0r&r')r;r1r$r%r0r&r')r/r-r r1z [interrupted])-argparseArgumentParser add_argumentrRrfPycInvalidationModerS parse_argsrr.rergr&rr rrUr'errorr,r$r%rr7stdinrcrstriprr rrr1rupperr rrYrr-r/rr;rKeyboardInterrupt) rparserinvalidation_modesargs compile_destsrr flineivl_moder1r=rr r r!main4s                            r__main__)r) NNFNrFr"r N)NFNrFr"N)r rFrFr"N)r r7importlib.utilrZrfr_rh functoolsrpathlibr__all__rrrrrrvrR exit_statusexitr r r r!sH     M  "