a hEb@sdZddlZddlZddlZddlZddgZddgddddZggfddZd d Zd d Z d dZ gggfddZ ddZ ddZ dS)zb Build a c-extension module on-the-fly in tests. See build_and_import_extensions for usage hints Nbuild_and_import_extensioncompile_extension_module)prologue build_dir include_dirs more_initc Csddlm}|t||}d}|s,td}|r@|d7}||7}|d7}t|||} zt|||| } Wn6|y} ztd|d| WYd } ~ n d } ~ 00dd l} | j || } | j | }| j ||S) a Build and imports a c-extension module `modname` from a list of function fragments `functions`. Parameters ---------- functions : list of fragments Each fragment is a sequence of func_name, calling convention, snippet. prologue : string Code to preceed the rest, usually extra ``#include`` or ``#define`` macros. build_dir : pathlib.Path Where to build the module, usually a temporary directory include_dirs : list Extra directories to find include files when compiling more_init : string Code to appear in the module PyMODINIT_FUNC Returns ------- out: module The module will have been loaded and is ready for use Examples -------- >>> functions = [("test_bytes", "METH_O", """ if ( !PyBytesCheck(args)) { Py_RETURN_FALSE; } Py_RETURN_TRUE; """)] >>> mod = build_and_import_extension("testme", functions) >>> assert not mod.test_bytes(u'abc') >>> assert mod.test_bytes(b'abc') r) CompileErrorz8PyObject *mod = PyModule_Create(&moduledef); .z.#define INITERROR return NULL z return mod;zcould not compile in :N)distutils.errorsr _make_methodspathlibPath _make_sourcer RuntimeErrorimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)modname functionsrrrrr bodyinit source_stringZmod_soe importlibspecZfoor _/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/numpy/testing/_private/extbuild.pyrs*'   (  c CsR|dd}||}|jddt||}tdg|}t||||ggdS)aH Build an extension module and return the filename of the resulting native code file. Parameters ---------- name : string name of the module, possibly including dots if it is a module inside a package. builddir : pathlib.Path Where to build the module, usually a temporary directory include_dirs : list Extra directories to find include files when compiling libraries : list Libraries to link into the extension module library_dirs: list Where to find the libraries, ``-L`` passed to the linker r T)exist_ok INCLUDEPY)outputfilenamer libraries library_dirs)splitmkdir_convert_str_to_file sysconfigget_config_var _c_compile) nameZbuilddirrrr&r'rdirnamecfiler r r!rPs  cCsD|d}|d}|t|Wdn1s60Y|S)zHelper function to create a file ``source.c`` in `dirname` that contains the string in `source`. Returns the file name zsource.cwN)openwritestr)sourcer/filenamefr r r!r*qs ,r*c Csg}g}|D]V\}}}d||f}d|vr0d}nd}|d|||fdj|||d} || q d|d td||d } | S) z Turns the name, signature, code in functions into complete functions and lists them in a methods_table. Then turns the methods_table into a ``PyMethodDef`` structure and returns the resulting code fragment ready for compilation z%s_%sZ METH_KEYWORDSz2(PyObject *self, PyObject *args, PyObject *kwargs)z (PyObject *self, PyObject *args)z{"%s", (PyCFunction)%s, %s},z^ static PyObject* {cfuncname}{signature} {{ {code} }} ) cfuncname signaturecode a6 static PyMethodDef methods[] = { %(methods)s { NULL } }; static struct PyModuleDef moduledef = { PyModuleDef_HEAD_INIT, "%(modname)s", /* m_name */ NULL, /* m_doc */ -1, /* m_size */ methods, /* m_methods */ }; )methodsr)appendformatjoindict) rrZ methods_tablecodesfuncnameflagsr:r8r9 func_coderr r r!r {s$     r cCsdt|||d}|S)zG Combines the code fragments into source code ready to be compiled zn #include %(body)s PyMODINIT_FUNC PyInit_%(name)s(void) { %(init)s } )r.rr)r@)r.rrr:r r r!rs   rc CsXtjdkr(dg}dtjtjdg}n"tjdrBgd}d}nd}}tjdkr^|dg}tjdkrd D]X}|d |vrtj|d r||d |d |vrltj|d rl||d ql| t }tj }zDt |||||||W|D]"\} } tj | | kr| tj | <qn2|D]&\} } tj | | kr*| tj | <q*0|S) Nwin32z/we4013z /LIBPATH:Zlibslinux)z-O0z-gz%-Werror=implicit-function-declarationz-fPICz/DEBUGdarwin)z/sw/z /opt/local/includelib)sysplatformospathr? base_prefix startswithexistsr= with_suffix get_so_suffixenvironcopybuilditemsget) r0r%rr&r' compile_extra link_extrasZ saved_environkeyvaluer r r!r-s@        r-c sddlm}|ddd}|dg} t} tjz<|jtj g||d} | fdd | D7} Wt| n t| 0|j | t||||d d S) z@cd into the directory where the cfile is, use distutils to buildr) new_compiler)forceverboser)r extra_preargscsg|]}tj|qSr )r4parent).0rr0r r! zbuild..)r&rbr'N) Znumpy.distutils.ccompilerr]Z customizerLgetcwdchdirrccompiler4r.Zlink_shared_object) r0r%rXrYrr&r'r]compilerobjectsoldresr rfr!rUs(     rUcCstd}|sJ|S)N EXT_SUFFIX)r+r,)retr r r!rRs rR)__doc__rLrrJr+__all__rrr*r rr-rUrRr r r r!s$ C ! ( '