pyo3_ffi/
modsupport.rs

1use crate::methodobject::PyMethodDef;
2use crate::moduleobject::PyModuleDef;
3use crate::object::PyObject;
4use crate::pyport::Py_ssize_t;
5use std::ffi::{c_char, c_int, c_long};
6
7extern "C" {
8    #[cfg_attr(PyPy, link_name = "PyPyArg_Parse")]
9    pub fn PyArg_Parse(arg1: *mut PyObject, arg2: *const c_char, ...) -> c_int;
10    #[cfg_attr(PyPy, link_name = "PyPyArg_ParseTuple")]
11    pub fn PyArg_ParseTuple(arg1: *mut PyObject, arg2: *const c_char, ...) -> c_int;
12    #[cfg_attr(PyPy, link_name = "PyPyArg_ParseTupleAndKeywords")]
13    pub fn PyArg_ParseTupleAndKeywords(
14        arg1: *mut PyObject,
15        arg2: *mut PyObject,
16        arg3: *const c_char,
17        #[cfg(not(Py_3_13))] arg4: *mut *mut c_char,
18        #[cfg(Py_3_13)] arg4: *const *const c_char,
19        ...
20    ) -> c_int;
21
22    // skipped PyArg_VaParse
23    // skipped PyArg_VaParseTupleAndKeywords
24
25    pub fn PyArg_ValidateKeywordArguments(arg1: *mut PyObject) -> c_int;
26    #[cfg_attr(PyPy, link_name = "PyPyArg_UnpackTuple")]
27    pub fn PyArg_UnpackTuple(
28        arg1: *mut PyObject,
29        arg2: *const c_char,
30        arg3: Py_ssize_t,
31        arg4: Py_ssize_t,
32        ...
33    ) -> c_int;
34
35    #[cfg_attr(PyPy, link_name = "PyPy_BuildValue")]
36    pub fn Py_BuildValue(arg1: *const c_char, ...) -> *mut PyObject;
37    // skipped Py_VaBuildValue
38
39    #[cfg(Py_3_13)]
40    pub fn PyModule_Add(module: *mut PyObject, name: *const c_char, value: *mut PyObject) -> c_int;
41    #[cfg(Py_3_10)]
42    #[cfg_attr(PyPy, link_name = "PyPyModule_AddObjectRef")]
43    pub fn PyModule_AddObjectRef(
44        module: *mut PyObject,
45        name: *const c_char,
46        value: *mut PyObject,
47    ) -> c_int;
48    #[cfg_attr(PyPy, link_name = "PyPyModule_AddObject")]
49    pub fn PyModule_AddObject(
50        module: *mut PyObject,
51        name: *const c_char,
52        value: *mut PyObject,
53    ) -> c_int;
54    #[cfg_attr(PyPy, link_name = "PyPyModule_AddIntConstant")]
55    pub fn PyModule_AddIntConstant(
56        module: *mut PyObject,
57        name: *const c_char,
58        value: c_long,
59    ) -> c_int;
60    #[cfg_attr(PyPy, link_name = "PyPyModule_AddStringConstant")]
61    pub fn PyModule_AddStringConstant(
62        module: *mut PyObject,
63        name: *const c_char,
64        value: *const c_char,
65    ) -> c_int;
66    #[cfg(any(Py_3_10, all(Py_3_9, not(Py_LIMITED_API))))]
67    #[cfg_attr(PyPy, link_name = "PyPyModule_AddType")]
68    pub fn PyModule_AddType(
69        module: *mut PyObject,
70        type_: *mut crate::object::PyTypeObject,
71    ) -> c_int;
72    // skipped PyModule_AddIntMacro
73    // skipped PyModule_AddStringMacro
74    pub fn PyModule_SetDocString(arg1: *mut PyObject, arg2: *const c_char) -> c_int;
75    pub fn PyModule_AddFunctions(arg1: *mut PyObject, arg2: *mut PyMethodDef) -> c_int;
76    #[cfg_attr(PyPy, link_name = "PyPyModule_ExecDef")]
77    pub fn PyModule_ExecDef(module: *mut PyObject, def: *mut PyModuleDef) -> c_int;
78}
79
80pub const Py_CLEANUP_SUPPORTED: i32 = 0x2_0000;
81
82pub const PYTHON_API_VERSION: i32 = 1013;
83pub const PYTHON_ABI_VERSION: i32 = 3;
84
85extern "C" {
86    #[cfg(not(py_sys_config = "Py_TRACE_REFS"))]
87    #[cfg_attr(PyPy, link_name = "PyPyModule_Create2")]
88    pub fn PyModule_Create2(module: *mut PyModuleDef, apiver: c_int) -> *mut PyObject;
89
90    #[cfg(py_sys_config = "Py_TRACE_REFS")]
91    fn PyModule_Create2TraceRefs(module: *mut PyModuleDef, apiver: c_int) -> *mut PyObject;
92
93    #[cfg(not(py_sys_config = "Py_TRACE_REFS"))]
94    #[cfg_attr(PyPy, link_name = "PyPyModule_FromDefAndSpec2")]
95    pub fn PyModule_FromDefAndSpec2(
96        def: *mut PyModuleDef,
97        spec: *mut PyObject,
98        module_api_version: c_int,
99    ) -> *mut PyObject;
100
101    #[cfg(py_sys_config = "Py_TRACE_REFS")]
102    fn PyModule_FromDefAndSpec2TraceRefs(
103        def: *mut PyModuleDef,
104        spec: *mut PyObject,
105        module_api_version: c_int,
106    ) -> *mut PyObject;
107}
108
109#[cfg(py_sys_config = "Py_TRACE_REFS")]
110#[inline]
111pub unsafe fn PyModule_Create2(module: *mut PyModuleDef, apiver: c_int) -> *mut PyObject {
112    PyModule_Create2TraceRefs(module, apiver)
113}
114
115#[cfg(py_sys_config = "Py_TRACE_REFS")]
116#[inline]
117pub unsafe fn PyModule_FromDefAndSpec2(
118    def: *mut PyModuleDef,
119    spec: *mut PyObject,
120    module_api_version: c_int,
121) -> *mut PyObject {
122    PyModule_FromDefAndSpec2TraceRefs(def, spec, module_api_version)
123}
124
125#[inline]
126pub unsafe fn PyModule_Create(module: *mut PyModuleDef) -> *mut PyObject {
127    PyModule_Create2(
128        module,
129        if cfg!(Py_LIMITED_API) {
130            PYTHON_ABI_VERSION
131        } else {
132            PYTHON_API_VERSION
133        },
134    )
135}
136
137#[inline]
138pub unsafe fn PyModule_FromDefAndSpec(def: *mut PyModuleDef, spec: *mut PyObject) -> *mut PyObject {
139    PyModule_FromDefAndSpec2(
140        def,
141        spec,
142        if cfg!(Py_LIMITED_API) {
143            PYTHON_ABI_VERSION
144        } else {
145            PYTHON_API_VERSION
146        },
147    )
148}