blob: 376dddfac09c685537c7f012395897dcb918aafa [file] [log] [blame]
Antoine Pitrou1d80a562018-04-07 18:14:03 +02001#include <Python.h>
2
3typedef struct {
4 PyListObject list;
5 int state;
6} SubListObject;
7
8static PyObject *
9SubList_increment(SubListObject *self, PyObject *unused)
10{
11 self->state++;
12 return PyLong_FromLong(self->state);
13}
14
15static PyMethodDef SubList_methods[] = {
16 {"increment", (PyCFunction) SubList_increment, METH_NOARGS,
17 PyDoc_STR("increment state counter")},
18 {NULL},
19};
20
21static int
22SubList_init(SubListObject *self, PyObject *args, PyObject *kwds)
23{
24 if (PyList_Type.tp_init((PyObject *) self, args, kwds) < 0)
25 return -1;
26 self->state = 0;
27 return 0;
28}
29
30static PyTypeObject SubListType = {
31 PyVarObject_HEAD_INIT(NULL, 0)
32 .tp_name = "sublist.SubList",
33 .tp_doc = "SubList objects",
34 .tp_basicsize = sizeof(SubListObject),
35 .tp_itemsize = 0,
36 .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
37 .tp_init = (initproc) SubList_init,
38 .tp_methods = SubList_methods,
39};
40
41static PyModuleDef sublistmodule = {
42 PyModuleDef_HEAD_INIT,
43 .m_name = "sublist",
44 .m_doc = "Example module that creates an extension type.",
45 .m_size = -1,
46};
47
48PyMODINIT_FUNC
49PyInit_sublist(void)
50{
51 PyObject *m;
52 SubListType.tp_base = &PyList_Type;
53 if (PyType_Ready(&SubListType) < 0)
54 return NULL;
55
56 m = PyModule_Create(&sublistmodule);
57 if (m == NULL)
58 return NULL;
59
60 Py_INCREF(&SubListType);
61 PyModule_AddObject(m, "SubList", (PyObject *) &SubListType);
62 return m;
63}