Files
cibuildwheel/test/template_projects/c.py
T

84 lines
2.1 KiB
Python
Raw Normal View History

2020-05-02 09:59:55 +01:00
import jinja2
2020-05-02 16:54:19 +01:00
from .setuptools import SetuptoolsTemplateProject
2020-05-02 16:54:19 +01:00
spam_c_template = r'''
#include <Python.h>
2020-05-02 16:54:19 +01:00
{{ spam_c_top_level_add }}
2020-05-02 16:54:19 +01:00
static PyObject *
spam_system(PyObject *self, PyObject *args)
{
const char *command;
int sts;
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
if (!PyArg_ParseTuple(args, "s", &command))
return NULL;
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
sts = system(command);
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
{{ spam_c_function_add | indent(4) }}
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
return PyLong_FromLong(sts);
}
2020-05-02 16:54:19 +01:00
/* Module initialization */
2020-05-02 16:54:19 +01:00
#if PY_MAJOR_VERSION >= 3
#define MOD_INIT(name) PyMODINIT_FUNC PyInit_##name(void)
#define MOD_DEF(m, name, doc, methods, module_state_size) \
static struct PyModuleDef moduledef = { \
PyModuleDef_HEAD_INIT, name, doc, module_state_size, methods, }; \
m = PyModule_Create(&moduledef);
#define MOD_RETURN(m) return m;
#else
#define MOD_INIT(name) PyMODINIT_FUNC init##name(void)
#define MOD_DEF(m, name, doc, methods, module_state_size) \
m = Py_InitModule3(name, methods, doc);
#define MOD_RETURN(m) return;
#endif
2020-05-02 16:54:19 +01:00
static PyMethodDef module_methods[] = {
{"system", (PyCFunction)spam_system, METH_VARARGS,
"Execute a shell command."},
{NULL} /* Sentinel */
};
2020-05-02 16:54:19 +01:00
MOD_INIT(spam)
{
PyObject* m;
2020-05-02 16:54:19 +01:00
MOD_DEF(m,
"spam",
"Example module",
module_methods,
-1)
2020-05-02 16:54:19 +01:00
MOD_RETURN(m)
}
'''
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
class CTemplateProject(SetuptoolsTemplateProject):
def __init__(self, *, spam_c_top_level_add='', spam_c_function_add='', setup_py_add='',
setup_py_setup_args_add='', setup_cfg_add=''):
setup_py_setup_args_add += '''
ext_modules=[Extension('spam', sources=['spam.c'])],
'''
2020-05-02 09:59:55 +01:00
2020-05-02 16:54:19 +01:00
super().__init__(
setup_py_add=setup_py_add,
setup_py_setup_args_add=setup_py_setup_args_add,
setup_cfg_add=setup_cfg_add
)
self.files.update({
'spam.c': jinja2.Template(spam_c_template),
})
self.template_context.update({
'spam_c_top_level_add': spam_c_top_level_add,
'spam_c_function_add': spam_c_function_add,
})