plpy_resultobject.c File Reference

#include "postgres.h"
#include "plpy_elog.h"
#include "plpy_resultobject.h"
#include "plpy_util.h"

Include dependency graph for plpy_resultobject.c:

Go to the source code of this file.

Functions
static void	PLy_result_dealloc (PLyResultObject *self)
static PyObject *	PLy_result_colnames (PyObject *self, PyObject *unused)
static PyObject *	PLy_result_coltypes (PyObject *self, PyObject *unused)
static PyObject *	PLy_result_coltypmods (PyObject *self, PyObject *unused)
static PyObject *	PLy_result_nrows (PyObject *self, PyObject *args)
static PyObject *	PLy_result_status (PyObject *self, PyObject *args)
static Py_ssize_t	PLy_result_length (PyObject *arg)
static PyObject *	PLy_result_item (PyObject *arg, Py_ssize_t idx)
static PyObject *	PLy_result_str (PyObject *arg)
static PyObject *	PLy_result_subscript (PyObject *arg, PyObject *item)
static int	PLy_result_ass_subscript (PyObject *arg, PyObject *item, PyObject *value)
void	PLy_result_init_type (void)
PyObject *	PLy_result_new (void)

Variables
static char	PLy_result_doc [] = "Results of a PostgreSQL query"
static PyMethodDef	PLy_result_methods []
static PyType_Slot	PLyResult_slots []
static PyType_Spec	PLyResult_spec
static PyTypeObject *	PLy_ResultType

Function Documentation

PLy_result_ass_subscript()

static int PLy_result_ass_subscript ( PyObject *  arg, PyObject *  item, PyObject *  value  )

static

Definition at line 276 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) arg;
 
    return PyObject_SetItem(ob->rows, item, value);
}

References arg, fb(), and value.

PLy_result_colnames()

static PyObject * PLy_result_colnames ( PyObject *  self, PyObject *  unused  )

static

Definition at line 140 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) self;
    PyObject   *list;
    int         i;
 
    if (!ob->tupdesc)
    {
        PLy_exception_set(PLy_exc_error, "command did not produce a result set");
        return NULL;
    }
 
    list = PyList_New(ob->tupdesc->natts);
    if (!list)
        return NULL;
    for (i = 0; i < ob->tupdesc->natts; i++)
    {
        Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 
        PyList_SetItem(list, i, PLyUnicode_FromString(NameStr(attr->attname)));
    }
 
    return list;
}

References fb(), i, NameStr, PLy_exc_error, PLy_exception_set(), PLyUnicode_FromString(), and TupleDescAttr().

PLy_result_coltypes()

static PyObject * PLy_result_coltypes ( PyObject *  self, PyObject *  unused  )

static

Definition at line 166 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) self;
    PyObject   *list;
    int         i;
 
    if (!ob->tupdesc)
    {
        PLy_exception_set(PLy_exc_error, "command did not produce a result set");
        return NULL;
    }
 
    list = PyList_New(ob->tupdesc->natts);
    if (!list)
        return NULL;
    for (i = 0; i < ob->tupdesc->natts; i++)
    {
        Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 
        PyList_SetItem(list, i, PyLong_FromLong(attr->atttypid));
    }
 
    return list;
}

References fb(), i, PLy_exc_error, PLy_exception_set(), and TupleDescAttr().

PLy_result_coltypmods()

static PyObject * PLy_result_coltypmods ( PyObject *  self, PyObject *  unused  )

static

Definition at line 192 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) self;
    PyObject   *list;
    int         i;
 
    if (!ob->tupdesc)
    {
        PLy_exception_set(PLy_exc_error, "command did not produce a result set");
        return NULL;
    }
 
    list = PyList_New(ob->tupdesc->natts);
    if (!list)
        return NULL;
    for (i = 0; i < ob->tupdesc->natts; i++)
    {
        Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 
        PyList_SetItem(list, i, PyLong_FromLong(attr->atttypmod));
    }
 
    return list;
}

References fb(), i, PLy_exc_error, PLy_exception_set(), and TupleDescAttr().

PLy_result_dealloc()

static void PLy_result_dealloc ( PLyResultObject *  self )

static

Definition at line 117 of file plpy_resultobject.c.

{
#if PY_VERSION_HEX >= 0x03080000
    PyTypeObject *tp = Py_TYPE(self);
#endif
 
    Py_XDECREF(self->nrows);
    Py_XDECREF(self->rows);
    Py_XDECREF(self->status);
    if (self->tupdesc)
    {
        FreeTupleDesc(self->tupdesc);
        self->tupdesc = NULL;
    }
 
    PyObject_Free(self);
#if PY_VERSION_HEX >= 0x03080000
    /* This was not needed before Python 3.8 (Python issue 35810) */
    Py_DECREF(tp);
#endif
}

References fb(), FreeTupleDesc(), PLyResultObject::nrows, PLyResultObject::rows, PLyResultObject::status, and PLyResultObject::tupdesc.

PLy_result_init_type()

void PLy_result_init_type

(

void

)

Definition at line 81 of file plpy_resultobject.c.

{
    PLy_ResultType = (PyTypeObject *) PyType_FromSpec(&PLyResult_spec);
    if (!PLy_ResultType)
        elog(ERROR, "could not initialize PLy_ResultType");
}

References elog, ERROR, fb(), PLy_ResultType, and PLyResult_spec.

Referenced by PyInit_plpy().

PLy_result_item()

static PyObject * PLy_result_item ( PyObject *  arg, Py_ssize_t  idx  )

static

Definition at line 244 of file plpy_resultobject.c.

{
    PyObject   *rv;
    PLyResultObject *ob = (PLyResultObject *) arg;
 
    rv = PyList_GetItem(ob->rows, idx);
    if (rv != NULL)
        Py_INCREF(rv);
    return rv;
}

References arg, fb(), and idx().

PLy_result_length()

static Py_ssize_t PLy_result_length ( PyObject *  arg )

static

Definition at line 236 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) arg;
 
    return PyList_Size(ob->rows);
}

References arg, and fb().

PLy_result_new()

PyObject * PLy_result_new

(

void

)

Definition at line 89 of file plpy_resultobject.c.

{
    PLyResultObject *ob;
 
    if ((ob = PyObject_New(PLyResultObject, PLy_ResultType)) == NULL)
        return NULL;
#if PY_VERSION_HEX < 0x03080000
    /* Workaround for Python issue 35810; no longer necessary in Python 3.8 */
    Py_INCREF(PLy_ResultType);
#endif
 
    /* ob->tuples = NULL; */
 
    Py_INCREF(Py_None);
    ob->status = Py_None;
    ob->nrows = PyLong_FromLong(-1);
    ob->rows = PyList_New(0);
    ob->tupdesc = NULL;
    if (!ob->rows)
    {
        Py_DECREF(ob);
        return NULL;
    }
 
    return (PyObject *) ob;
}

References fb(), and PLy_ResultType.

Referenced by PLy_cursor_fetch(), and PLy_spi_execute_fetch_result().

PLy_result_nrows()

static PyObject * PLy_result_nrows ( PyObject *  self, PyObject *  args  )

static

Definition at line 218 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) self;
 
    Py_INCREF(ob->nrows);
    return ob->nrows;
}

References fb().

PLy_result_status()

static PyObject * PLy_result_status ( PyObject *  self, PyObject *  args  )

static

Definition at line 227 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) self;
 
    Py_INCREF(ob->status);
    return ob->status;
}

References fb().

PLy_result_str()

static PyObject * PLy_result_str ( PyObject *  arg )

static

Definition at line 256 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) arg;
 
    return PyUnicode_FromFormat("<%s status=%S nrows=%S rows=%S>",
                                "PLyResult",
                                ob->status,
                                ob->nrows,
                                ob->rows);
}

References arg, and fb().

PLy_result_subscript()

static PyObject * PLy_result_subscript ( PyObject *  arg, PyObject *  item  )

static

Definition at line 268 of file plpy_resultobject.c.

{
    PLyResultObject *ob = (PLyResultObject *) arg;
 
    return PyObject_GetItem(ob->rows, item);
}

References arg, and fb().

Variable Documentation

PLy_result_doc

char PLy_result_doc[] = "Results of a PostgreSQL query"

static

Definition at line 25 of file plpy_resultobject.c.

PLy_result_methods

PyMethodDef PLy_result_methods[]

static

Initial value:

= {
    {"colnames", PLy_result_colnames, METH_NOARGS, NULL},
    {"coltypes", PLy_result_coltypes, METH_NOARGS, NULL},
    {"coltypmods", PLy_result_coltypmods, METH_NOARGS, NULL},
    {"nrows", PLy_result_nrows, METH_VARARGS, NULL},
    {"status", PLy_result_status, METH_VARARGS, NULL},
    {NULL, NULL, 0, NULL}
}

Definition at line 27 of file plpy_resultobject.c.

                                          {
    {"colnames", PLy_result_colnames, METH_NOARGS, NULL},
    {"coltypes", PLy_result_coltypes, METH_NOARGS, NULL},
    {"coltypmods", PLy_result_coltypmods, METH_NOARGS, NULL},
    {"nrows", PLy_result_nrows, METH_VARARGS, NULL},
    {"status", PLy_result_status, METH_VARARGS, NULL},
    {NULL, NULL, 0, NULL}
};

PLy_ResultType

PyTypeObject* PLy_ResultType

static

Definition at line 78 of file plpy_resultobject.c.

Referenced by PLy_result_init_type(), and PLy_result_new().

PLyResult_slots

PyType_Slot PLyResult_slots[]

static

Definition at line 36 of file plpy_resultobject.c.

{
    {
        Py_tp_dealloc, PLy_result_dealloc
    },
    {
        Py_sq_length, PLy_result_length
    },
    {
        Py_sq_item, PLy_result_item
    },
    {
        Py_mp_length, PLy_result_length
    },
    {
        Py_mp_subscript, PLy_result_subscript
    },
    {
        Py_mp_ass_subscript, PLy_result_ass_subscript
    },
    {
        Py_tp_str, PLy_result_str
    },
    {
        Py_tp_doc, (char *) PLy_result_doc
    },
    {
        Py_tp_methods, PLy_result_methods
    },
    {
        0, NULL
    }
};

PLyResult_spec

PyType_Spec PLyResult_spec

static

Initial value:

=
{
    .name = "PLyResult",
    .basicsize = sizeof(PLyResultObject),
    .flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
    .slots = PLyResult_slots,
}

Definition at line 70 of file plpy_resultobject.c.

{
    .name = "PLyResult",
    .basicsize = sizeof(PLyResultObject),
    .flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
    .slots = PLyResult_slots,
};

Referenced by PLy_result_init_type().