X-PhX tUdZddlmZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z mZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ ddl,Z,n #e-$rYnwxYwe r3ej.dkrddl,m/Z/n!ddl0m/Z1ddl2m3Z3ddl,m/Z4ee4e1fZ/de5d<ej.dkr ddl6m7Z7m8Z8nddl6m7Z7m8Z8didZ9djdZ:dkd!Z;dldmd+Zdpd;Z?dqd<Z@drdCZAdsdIZBdtdJZCdudMZDdvdQZEdwdSZFd"ddd"dTdxdZZGdd"d[d\d"d"d"dddd"d] dydhZHdS)zaSupport for building extensions using mypyc with distutils or setuptools The main entry point is mypycify, which produces a list of extension modules to be passed to setup. A trivial setup.py for a mypyc built project, then, looks like: from setuptools import setup from mypyc.build import mypycify setup(name='test_module', ext_modules=mypycify(['foo.py']), ) See the mypycify docs for additional arguments. mypycify can integrate with either distutils or setuptools, but needs to know at import-time whether it is using distutils or setuputils. We hackily decide based on whether setuptools has been imported already. ) annotationsN)Iterable) TYPE_CHECKINGAnyNoReturnUnioncast) BuildSource) CompileError)FileSystemCache)process_options)Options)write_junit_xml)generate_annotated_html) emitmodule)RUNTIME_C_FILESshared_lib_name)Errors)format_modules) exported_name)CompilerOptions ) Extension) TypeAliasrr) ccompiler sysconfigreturntype[Extension]cdtjv}tjdkr|sddl}|jj}n"|stjdtj}|S)N setuptoolsrrzerror: setuptools not installed)sysmodules version_infodistutils.corecorerexitr")use_setuptools distutilsextension_classs K/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypyc/build.py get_extensionr-Esd"S[0N '!!.!#.2 8 H6 7 7 7$. NonecTttttftj}t jdkr_|ddd|d<|ddd|d<|ddd|d<dSdS)z6Rewrite a bunch of config vars in pretty dubious ways.darwinLDSHAREDz -arch i386LDFLAGSCFLAGSN)r dictstrrget_config_varsr#platformreplace)varss r,setup_mypycify_varsr<Ws S#X 9 ; ; < z#get_mypy_config..s%GGGq16Y3F3F3F3F3Fr.cTg|]%}|j |jd#|&S) __init__.py)r]endswith)r^srcs r,raz#get_mypy_config..sG   CH =@X=N=N}=]=]    r.rzPython 2 not supportedz0Disabling strict optional checking not supportedTmypyc)r setseparater#r%rMr@strict_optionalshow_traceback export_types incremental preserve_astsper_module_options setdefaultmodule) rRrSrUrV all_sourcesrA mypyc_sourcessourcer`s @r,get_mypy_configruzsF+<IIIK%*++ GGGGKGGG #     (    3k722!-bqb1Ga A%% %&&&  "A ?@@@!GG*3G GQQLP"--fmR@@II +w ..r.full_module_name module_namedir_name group_namec d|dtjz}tj||}t tjt d5}|}dddn #1swxYwYt|| |t|t||S)a#Create a C extension shim with a passthrough PyInit function. Arguments: full_module_name: the dotted full module name module_name: the final component of the module name dir_name: the directory to place source code group_name: the name of the group z%s.c.zmodule_shim.tmplN)modnamelibname full_modname) r:ossepr]rOopen include_dirread write_fileformatrr)rvrwrxrycnamecpathf shim_templates r,generate_c_extension_shimrs  %--c26:: :E GLL5 ) )E bgll;==*<== > >!! !!!!!!!!!!!!!!! #J//&'788    Ls=BB"%B"r$c t|dkr|dStj}|d||ddS)zGProduce a probably unique name for a group from a list of module names.rJr,N)lenhashlibsha1updaterOencode hexdigest)r$hs r,ryrysi 7||qqz AHHSXXg   % % ' '((( ;;==" r.ctjtjtjt dS)z9Find the path of the lib-rt dir that needs to be includedzlib-rt)rr]rOabspathdirname__file__r[r.r,rrs4 7<<(A(ABBH M MMr.sourceslist[BuildSource]groupsemitmodule.Groupsr 'tuple[list[list[tuple[str, str]]], str]ctj} tj|||||}n]#t$rP}t ||jtj|z |j tjdYd}~nd}~wwxYwtj}|j r-t ||j ||z tjd|j rtd||z ddt|} tj ||| |\} } } tj} t || | |z | jrtjd|j rtd| |z dd|jrt#|j|| | | d t'| fS) aDrive the actual core compilation step. The groups argument describes how modules are assigned to C extension modules. See the comments on the Groups type in mypyc.emitmodule for details. Returns the C source code and (for debugging) the pretty printed IR. )rFrJNzParsed and typechecked in z.3fr_)rUerrorsrzCompiled to C in rK)timerparse_and_typecheckr rQrB use_stdoutr#r(rverboserNrcompile_modules_to_c new_messages num_errorsmypyc_annotation_filerrOr)rrArrVrUt0resultet1rr$ctextmappert2s r, generate_crs B/ W.   gqz49;;+;!,FVXXXX   B }gv}b2g666  ; 9279999::: G__F'<!1&GUF B'6..00"r':::   2 0"r'0000111$X =vwPVWWW $))N73344 44s. BABBcfilesdeps build_dirextra_compile_argslist[Extension]ctt||t|g||g}|D]}|jdd}t |j|||} |j} |jsJtj|jddkr| dz } |t| | g||S)aProduce the list of extension modules when a shared library is needed. This creates one shared library extension module that all of the others import and then one shim extension module for each module in the build, that simply calls an initialization function in the shared library. The shared library (which lib_name is the name of) is a python extension module that exports the real initialization functions in Capsules stored in module attributes. )r include_dirsdependsrr{rJrcz .__init__)rr) r-rrrqsplitrr]rappend) rryrrrr extensionsrtrw shim_filervs r,build_using_shared_librs(   J ' '%--31    J  m))#..r2 -fm[)U_`` "={ 7== % %a (M 9 9  +  MOO 9+J\       r.cjt|dj|tg|gS)zProduce the list of extension modules for a standalone extension. This contains just one module, since there is no need for a shared module. r)rrr)r-rqr)rrrs r,build_single_moduler7s?   AJ %--1     r.r]contentsc(|d} t|d5}|}dddn #1swxYwYn#t$rd}YnwxYw||krt jtj|dt|d5}||dddn #1swxYwYt j |j dz}t j |||fdSdS) zWrite data into a file. If the file already exists and has the same contents we want to write, skip writing so as to preserve the mtime and avoid triggering recompilation. utf-8rbNT)exist_okwbrJ)times) rrrOSErrorrmakedirsr]rwritestatst_mtimeutime)r]rencoded_contentsr old_contentsg new_mtimes r,rrHs w// $   2)*L 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2  ''' BGOOD))D9999 $   & GG$ % % % & & & & & & & & & & & & & & &GDMM*Q.  i3444444('sEAA AA  AA A A#"A#/CCCri)bool | list[tuple[list[str], str | None]]use_shared_libc |dur d|D}nt|trg}t |D]?\ } fd|D}|||f |@ fd|D}|r|d|Dn|dfg}t |D],\}\}}|r|std|D}||f||<-|S)alCompute Groups given the input source list and separate configs. separate is the user-specified configuration for how to assign modules to compilation groups (see mypycify docstring for details). This takes that and expands it into our internal representation of group configuration, documented in mypyc.emitmodule's definition of Group. Tcg|]}|gdfSr?r[r^rts r,raz$construct_groups..ts$L$L$L&vh%5$L$L$Lr.c&g|] }|jv |Sr[r\)r^refiless r,raz$construct_groups..ys%IIISsx57H7HS7H7H7Hr.cg|]}|v| Sr[r[)r^re used_sourcess r,raz$construct_groups..|s#LLL#C|4K4K#4K4K4Kr.cg|]}|gdfSr?r[rs r,raz$construct_groups..~sIIIVHd+IIIr.Ncg|] }|j Sr[rqrs r,raz$construct_groups..sAAAv}AAAr.) isinstancelistrhrrextend enumeratery) rrirrname group_sourcesunused_sourcesigrouprrs @@r,construct_groupsrdsS4$L$LG$L$L$L Hd # # #uu # / /KE4IIIIGIIIM MM=$/ 0 0 0    . . . .LLLLLLL  K MMII.III J J JD/"&f--""=E4  C$ CAA5AAABBDDMq Mr.list[tuple[str, str]]ct}|D]-\}}|tjd|.t |S)zFind all the headers used by a group of cfiles. We do this by just regexping the source, which is a bit simpler than properly plumbing the data through. Arguments: cfiles: A list of (file name, file contents) pairs. z#include "(.*)")rhrrefindallsorted)rheadersrIrs r,get_header_depsrsPGAA 8rz"4h??@@@@ '??r.)rirSskip_cgen_inputalways_use_shared_libpathsr Any | Noner;tuple[emitmodule.Groups, list[tuple[list[str], list[str]]]]ct}t|||\}}} t|dkptd|Dp|} t ||| } |sKt || | |\} } t tj j d| n|} g}| D]}g}|D]\}}tj j |}| j st ||tj |ddkr| |fdt|D}| ||f| |fS)zRDo the front and middle end of mypyc building, producing and writing out C source.rJc3(K|] }d|jvVdS)r{Nr)r^xs r, zmypyc_build..s(661sah666666r.)rUzops.txtz.ccZg|]'}tjj|(Sr[)rr]rO target_dir)r^deprUs r,razmypyc_build..s-bbb3 -8#>>bbbr.)r ruranyrrrrr]rOrmypyc_skip_c_generationsplitextrr)rrUrirSrrrVrsrrrArr group_cfilesops_textgroup_cfilenamesr cfilenamescfilerrs ` r, mypyc_buildrsG*9 !#3W++'M; MQ ! 66 666 6 6 !  mX~ F FF '!+ &'DT" " "  h 27<< 0 ;YGGRRRR& ;= 4 4 " ) )LE5GLL!1!r8s!(#""""" $$$$$$<<<<<<<<<<<<<<""""""$$$$$$((((((%%%%%% %%%%%%222222$$$$$$99999999******''''''))))))    D  R 7""(((((((DDDDDD//////AAAAAA$%:NNNN 15151515h----`"55558$$$$N    (;@/3"&"'4$4$4$4$4$4$t04:?"&!)-!&aaaaaaaasBBB