blob: d13b5962c13c445475775958e87671836132e394 [file] [log] [blame]
Yury Selivanovf23746a2018-01-22 19:11:18 -05001#include "Python.h"
2
3#include "clinic/_contextvarsmodule.c.h"
4
5/*[clinic input]
6module _contextvars
7[clinic start generated code]*/
8/*[clinic end generated code: output=da39a3ee5e6b4b0d input=a0955718c8b8cea6]*/
9
10
11/*[clinic input]
12_contextvars.copy_context
13[clinic start generated code]*/
14
15static PyObject *
16_contextvars_copy_context_impl(PyObject *module)
17/*[clinic end generated code: output=1fcd5da7225c4fa9 input=89bb9ae485888440]*/
18{
Yury Selivanov2ec872b2018-09-21 15:33:56 -040019 return PyContext_CopyCurrent();
Yury Selivanovf23746a2018-01-22 19:11:18 -050020}
21
22
23PyDoc_STRVAR(module_doc, "Context Variables");
24
25static PyMethodDef _contextvars_methods[] = {
26 _CONTEXTVARS_COPY_CONTEXT_METHODDEF
27 {NULL, NULL}
28};
29
Hai Shi7d795682020-02-17 21:49:26 +080030static int
31_contextvars_exec(PyObject *m)
32{
Victor Stinner988f1ec2020-11-04 16:33:55 +010033 if (PyModule_AddType(m, &PyContext_Type) < 0) {
Hai Shi7d795682020-02-17 21:49:26 +080034 return -1;
35 }
Victor Stinner988f1ec2020-11-04 16:33:55 +010036 if (PyModule_AddType(m, &PyContextVar_Type) < 0) {
Hai Shi7d795682020-02-17 21:49:26 +080037 return -1;
38 }
Victor Stinner988f1ec2020-11-04 16:33:55 +010039 if (PyModule_AddType(m, &PyContextToken_Type) < 0) {
Hai Shi7d795682020-02-17 21:49:26 +080040 return -1;
41 }
Hai Shi7d795682020-02-17 21:49:26 +080042 return 0;
43}
44
45static struct PyModuleDef_Slot _contextvars_slots[] = {
46 {Py_mod_exec, _contextvars_exec},
47 {0, NULL}
48};
49
Yury Selivanovf23746a2018-01-22 19:11:18 -050050static struct PyModuleDef _contextvarsmodule = {
51 PyModuleDef_HEAD_INIT, /* m_base */
52 "_contextvars", /* m_name */
53 module_doc, /* m_doc */
Hai Shi7d795682020-02-17 21:49:26 +080054 0, /* m_size */
Yury Selivanovf23746a2018-01-22 19:11:18 -050055 _contextvars_methods, /* m_methods */
Hai Shi7d795682020-02-17 21:49:26 +080056 _contextvars_slots, /* m_slots */
Yury Selivanovf23746a2018-01-22 19:11:18 -050057 NULL, /* m_traverse */
58 NULL, /* m_clear */
59 NULL, /* m_free */
60};
61
62PyMODINIT_FUNC
63PyInit__contextvars(void)
64{
Hai Shi7d795682020-02-17 21:49:26 +080065 return PyModuleDef_Init(&_contextvarsmodule);
Yury Selivanovf23746a2018-01-22 19:11:18 -050066}