mirror of
https://github.com/python/cpython
synced 2024-09-05 08:17:45 +00:00
![Erlend Egeberg Aasland](/assets/img/avatar_default.png)
Use C APIs such as PyModule_AddType instead of PyModule_AddObject. Also remove incorrect module decrefs if module fails to initialise.
413 lines
13 KiB
C
413 lines
13 KiB
C
|
|
/* Use this file as a template to start implementing a module that
|
|
also declares object types. All occurrences of 'Xxo' should be changed
|
|
to something reasonable for your objects. After that, all other
|
|
occurrences of 'xx' should be changed to something reasonable for your
|
|
module. If your module is named foo your sourcefile should be named
|
|
foomodule.c.
|
|
|
|
You will probably want to delete all references to 'x_attr' and add
|
|
your own types of attributes instead. Maybe you want to name your
|
|
local variables other than 'self'. If your object type is needed in
|
|
other files, you'll have to create a file "foobarobject.h"; see
|
|
floatobject.h for an example. */
|
|
|
|
/* Xxo objects */
|
|
|
|
#include "Python.h"
|
|
|
|
static PyObject *ErrorObject;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *x_attr; /* Attributes dictionary */
|
|
} XxoObject;
|
|
|
|
static PyTypeObject Xxo_Type;
|
|
|
|
#define XxoObject_Check(v) Py_IS_TYPE(v, &Xxo_Type)
|
|
|
|
static XxoObject *
|
|
newXxoObject(PyObject *arg)
|
|
{
|
|
XxoObject *self;
|
|
self = PyObject_New(XxoObject, &Xxo_Type);
|
|
if (self == NULL)
|
|
return NULL;
|
|
self->x_attr = NULL;
|
|
return self;
|
|
}
|
|
|
|
/* Xxo methods */
|
|
|
|
static void
|
|
Xxo_dealloc(XxoObject *self)
|
|
{
|
|
Py_XDECREF(self->x_attr);
|
|
PyObject_Free(self);
|
|
}
|
|
|
|
static PyObject *
|
|
Xxo_demo(XxoObject *self, PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, ":demo"))
|
|
return NULL;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyMethodDef Xxo_methods[] = {
|
|
{"demo", (PyCFunction)Xxo_demo, METH_VARARGS,
|
|
PyDoc_STR("demo() -> None")},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static PyObject *
|
|
Xxo_getattro(XxoObject *self, PyObject *name)
|
|
{
|
|
if (self->x_attr != NULL) {
|
|
PyObject *v = PyDict_GetItemWithError(self->x_attr, name);
|
|
if (v != NULL) {
|
|
Py_INCREF(v);
|
|
return v;
|
|
}
|
|
else if (PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
}
|
|
return PyObject_GenericGetAttr((PyObject *)self, name);
|
|
}
|
|
|
|
static int
|
|
Xxo_setattr(XxoObject *self, const char *name, PyObject *v)
|
|
{
|
|
if (self->x_attr == NULL) {
|
|
self->x_attr = PyDict_New();
|
|
if (self->x_attr == NULL)
|
|
return -1;
|
|
}
|
|
if (v == NULL) {
|
|
int rv = PyDict_DelItemString(self->x_attr, name);
|
|
if (rv < 0 && PyErr_ExceptionMatches(PyExc_KeyError))
|
|
PyErr_SetString(PyExc_AttributeError,
|
|
"delete non-existing Xxo attribute");
|
|
return rv;
|
|
}
|
|
else
|
|
return PyDict_SetItemString(self->x_attr, name, v);
|
|
}
|
|
|
|
static PyTypeObject Xxo_Type = {
|
|
/* The ob_type field must be initialized in the module init function
|
|
* to be portable to Windows without using C++. */
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"xxmodule.Xxo", /*tp_name*/
|
|
sizeof(XxoObject), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
(destructor)Xxo_dealloc, /*tp_dealloc*/
|
|
0, /*tp_vectorcall_offset*/
|
|
(getattrfunc)0, /*tp_getattr*/
|
|
(setattrfunc)Xxo_setattr, /*tp_setattr*/
|
|
0, /*tp_as_async*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
(getattrofunc)Xxo_getattro, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
Xxo_methods, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
0, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
};
|
|
/* --------------------------------------------------------------------- */
|
|
|
|
/* Function of two integers returning integer */
|
|
|
|
PyDoc_STRVAR(xx_foo_doc,
|
|
"foo(i,j)\n\
|
|
\n\
|
|
Return the sum of i and j.");
|
|
|
|
static PyObject *
|
|
xx_foo(PyObject *self, PyObject *args)
|
|
{
|
|
long i, j;
|
|
long res;
|
|
if (!PyArg_ParseTuple(args, "ll:foo", &i, &j))
|
|
return NULL;
|
|
res = i+j; /* XXX Do something here */
|
|
return PyLong_FromLong(res);
|
|
}
|
|
|
|
|
|
/* Function of no arguments returning new Xxo object */
|
|
|
|
static PyObject *
|
|
xx_new(PyObject *self, PyObject *args)
|
|
{
|
|
XxoObject *rv;
|
|
|
|
if (!PyArg_ParseTuple(args, ":new"))
|
|
return NULL;
|
|
rv = newXxoObject(args);
|
|
if (rv == NULL)
|
|
return NULL;
|
|
return (PyObject *)rv;
|
|
}
|
|
|
|
/* Example with subtle bug from extensions manual ("Thin Ice"). */
|
|
|
|
static PyObject *
|
|
xx_bug(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *list, *item;
|
|
|
|
if (!PyArg_ParseTuple(args, "O:bug", &list))
|
|
return NULL;
|
|
|
|
item = PyList_GetItem(list, 0);
|
|
/* Py_INCREF(item); */
|
|
PyList_SetItem(list, 1, PyLong_FromLong(0L));
|
|
PyObject_Print(item, stdout, 0);
|
|
printf("\n");
|
|
/* Py_DECREF(item); */
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
/* Test bad format character */
|
|
|
|
static PyObject *
|
|
xx_roj(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *a;
|
|
long b;
|
|
if (!PyArg_ParseTuple(args, "O#:roj", &a, &b))
|
|
return NULL;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
|
|
/* ---------- */
|
|
|
|
static PyTypeObject Str_Type = {
|
|
/* The ob_type field must be initialized in the module init function
|
|
* to be portable to Windows without using C++. */
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"xxmodule.Str", /*tp_name*/
|
|
0, /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
0, /*tp_dealloc*/
|
|
0, /*tp_vectorcall_offset*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_as_async*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /* see PyInit_xx */ /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
0, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
};
|
|
|
|
/* ---------- */
|
|
|
|
static PyObject *
|
|
null_richcompare(PyObject *self, PyObject *other, int op)
|
|
{
|
|
Py_INCREF(Py_NotImplemented);
|
|
return Py_NotImplemented;
|
|
}
|
|
|
|
static PyTypeObject Null_Type = {
|
|
/* The ob_type field must be initialized in the module init function
|
|
* to be portable to Windows without using C++. */
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"xxmodule.Null", /*tp_name*/
|
|
0, /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
0, /*tp_dealloc*/
|
|
0, /*tp_vectorcall_offset*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_as_async*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
null_richcompare, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /* see PyInit_xx */ /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
PyType_GenericNew, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
};
|
|
|
|
|
|
/* ---------- */
|
|
|
|
|
|
/* List of functions defined in the module */
|
|
|
|
static PyMethodDef xx_methods[] = {
|
|
{"roj", xx_roj, METH_VARARGS,
|
|
PyDoc_STR("roj(a,b) -> None")},
|
|
{"foo", xx_foo, METH_VARARGS,
|
|
xx_foo_doc},
|
|
{"new", xx_new, METH_VARARGS,
|
|
PyDoc_STR("new() -> new Xx object")},
|
|
{"bug", xx_bug, METH_VARARGS,
|
|
PyDoc_STR("bug(o) -> None")},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
PyDoc_STRVAR(module_doc,
|
|
"This is a template module just for instruction.");
|
|
|
|
|
|
static int
|
|
xx_exec(PyObject *m)
|
|
{
|
|
/* Slot initialization is subject to the rules of initializing globals.
|
|
C99 requires the initializers to be "address constants". Function
|
|
designators like 'PyType_GenericNew', with implicit conversion to
|
|
a pointer, are valid C99 address constants.
|
|
|
|
However, the unary '&' operator applied to a non-static variable
|
|
like 'PyBaseObject_Type' is not required to produce an address
|
|
constant. Compilers may support this (gcc does), MSVC does not.
|
|
|
|
Both compilers are strictly standard conforming in this particular
|
|
behavior.
|
|
*/
|
|
Null_Type.tp_base = &PyBaseObject_Type;
|
|
Str_Type.tp_base = &PyUnicode_Type;
|
|
|
|
/* Finalize the type object including setting type of the new type
|
|
* object; doing it here is required for portability, too. */
|
|
if (PyType_Ready(&Xxo_Type) < 0) {
|
|
return -1;
|
|
}
|
|
|
|
/* Add some symbolic constants to the module */
|
|
if (ErrorObject == NULL) {
|
|
ErrorObject = PyErr_NewException("xx.error", NULL, NULL);
|
|
if (ErrorObject == NULL) {
|
|
return -1;
|
|
}
|
|
}
|
|
int rc = PyModule_AddType(m, (PyTypeObject *)ErrorObject);
|
|
Py_DECREF(ErrorObject);
|
|
if (rc < 0) {
|
|
return -1;
|
|
}
|
|
|
|
/* Add Str and Null types */
|
|
if (PyModule_AddType(m, &Str_Type) < 0) {
|
|
return -1;
|
|
}
|
|
if (PyModule_AddType(m, &Null_Type) < 0) {
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static struct PyModuleDef_Slot xx_slots[] = {
|
|
{Py_mod_exec, xx_exec},
|
|
{0, NULL},
|
|
};
|
|
|
|
static struct PyModuleDef xxmodule = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"xx",
|
|
module_doc,
|
|
0,
|
|
xx_methods,
|
|
xx_slots,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
/* Export function for the module (*must* be called PyInit_xx) */
|
|
|
|
PyMODINIT_FUNC
|
|
PyInit_xx(void)
|
|
{
|
|
return PyModuleDef_Init(&xxmodule);
|
|
}
|