diff options
Diffstat (limited to 'psycopg/adapter_pboolean.c')
| -rw-r--r-- | psycopg/adapter_pboolean.c | 19 |
1 files changed, 14 insertions, 5 deletions
diff --git a/psycopg/adapter_pboolean.c b/psycopg/adapter_pboolean.c index 05d1d04..1ce80dd 100644 --- a/psycopg/adapter_pboolean.c +++ b/psycopg/adapter_pboolean.c @@ -106,8 +106,8 @@ pboolean_setup(pbooleanObject *self, PyObject *obj) self, ((PyObject *)self)->ob_refcnt ); + Py_INCREF(obj); self->wrapped = obj; - Py_INCREF(self->wrapped); Dprintf("pboolean_setup: good pboolean object at %p, refcnt = " FORMAT_CODE_PY_SSIZE_T, @@ -116,12 +116,21 @@ pboolean_setup(pbooleanObject *self, PyObject *obj) return 0; } +static int +pboolean_traverse(PyObject *obj, visitproc visit, void *arg) +{ + pbooleanObject *self = (pbooleanObject *)obj; + + Py_VISIT(self->wrapped); + return 0; +} + static void pboolean_dealloc(PyObject* obj) { pbooleanObject *self = (pbooleanObject *)obj; - Py_XDECREF(self->wrapped); + Py_CLEAR(self->wrapped); Dprintf("pboolean_dealloc: deleted pboolean object at %p, refcnt = " FORMAT_CODE_PY_SSIZE_T, @@ -151,7 +160,7 @@ pboolean_new(PyTypeObject *type, PyObject *args, PyObject *kwds) static void pboolean_del(PyObject* self) { - PyObject_Del(self); + PyObject_GC_Del(self); } static PyObject * @@ -194,10 +203,10 @@ PyTypeObject pbooleanType = { 0, /*tp_setattro*/ 0, /*tp_as_buffer*/ - Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, /*tp_flags*/ + Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/ pbooleanType_doc, /*tp_doc*/ - 0, /*tp_traverse*/ + pboolean_traverse, /*tp_traverse*/ 0, /*tp_clear*/ 0, /*tp_richcompare*/ |