src/pl/plpython/plpy_resultobject.c

   1 /*
   2  * the PLyResult class
   3  *
   4  * src/pl/plpython/plpy_resultobject.c
   5  */
   6
   7 #include "postgres.h"
   8
   9 #include "plpy_elog.h"
  10 #include "plpy_resultobject.h"
  11 #include "plpython.h"
  12
  13 static void PLy_result_dealloc(PyObject *arg);
  14 static PyObject *PLy_result_colnames(PyObject *self, PyObject *unused);
  15 static PyObject *PLy_result_coltypes(PyObject *self, PyObject *unused);
  16 static PyObject *PLy_result_coltypmods(PyObject *self, PyObject *unused);
  17 static PyObject *PLy_result_nrows(PyObject *self, PyObject *args);
  18 static PyObject *PLy_result_status(PyObject *self, PyObject *args);
  19 static Py_ssize_t PLy_result_length(PyObject *arg);
  20 static PyObject *PLy_result_item(PyObject *arg, Py_ssize_t idx);
  21 static PyObject *PLy_result_str(PyObject *arg);
  22 static PyObject *PLy_result_subscript(PyObject *arg, PyObject *item);
  23 static int      PLy_result_ass_subscript(PyObject *arg, PyObject *item, PyObject *value);
  24
  25 static char PLy_result_doc[] = "Results of a PostgreSQL query";
  26
  27 static PySequenceMethods PLy_result_as_sequence = {
  28         .sq_length = PLy_result_length,
  29         .sq_item = PLy_result_item,
  30 };
  31
  32 static PyMappingMethods PLy_result_as_mapping = {
  33         .mp_length = PLy_result_length,
  34         .mp_subscript = PLy_result_subscript,
  35         .mp_ass_subscript = PLy_result_ass_subscript,
  36 };
  37
  38 static PyMethodDef PLy_result_methods[] = {
  39         {"colnames", PLy_result_colnames, METH_NOARGS, NULL},
  40         {"coltypes", PLy_result_coltypes, METH_NOARGS, NULL},
  41         {"coltypmods", PLy_result_coltypmods, METH_NOARGS, NULL},
  42         {"nrows", PLy_result_nrows, METH_VARARGS, NULL},
  43         {"status", PLy_result_status, METH_VARARGS, NULL},
  44         {NULL, NULL, 0, NULL}
  45 };
  46
  47 static PyTypeObject PLy_ResultType = {
  48         PyVarObject_HEAD_INIT(NULL, 0)
  49         .tp_name = "PLyResult",
  50         .tp_basicsize = sizeof(PLyResultObject),
  51         .tp_dealloc = PLy_result_dealloc,
  52         .tp_as_sequence = &PLy_result_as_sequence,
  53         .tp_as_mapping = &PLy_result_as_mapping,
  54         .tp_str = &PLy_result_str,
  55         .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
  56         .tp_doc = PLy_result_doc,
  57         .tp_methods = PLy_result_methods,
  58 };
  59
  60 void
  61 PLy_result_init_type(void)
  62 {
  63         if (PyType_Ready(&PLy_ResultType) < 0)
  64                 elog(ERROR, "could not initialize PLy_ResultType");
  65 }
  66
  67 PyObject *
  68 PLy_result_new(void)
  69 {
  70         PLyResultObject *ob;
  71
  72         if ((ob = PyObject_New(PLyResultObject, &PLy_ResultType)) == NULL)
  73                 return NULL;
  74
  75         /* ob->tuples = NULL; */
  76
  77         Py_INCREF(Py_None);
  78         ob->status = Py_None;
  79         ob->nrows = PyLong_FromLong(-1);
  80         ob->rows = PyList_New(0);
  81         ob->tupdesc = NULL;
  82         if (!ob->rows)
  83         {
  84                 Py_DECREF(ob);
  85                 return NULL;
  86         }
  87
  88         return (PyObject *) ob;
  89 }
  90
  91 static void
  92 PLy_result_dealloc(PyObject *arg)
  93 {
  94         PLyResultObject *ob = (PLyResultObject *) arg;
  95
  96         Py_XDECREF(ob->nrows);
  97         Py_XDECREF(ob->rows);
  98         Py_XDECREF(ob->status);
  99         if (ob->tupdesc)
 100         {
 101                 FreeTupleDesc(ob->tupdesc);
 102                 ob->tupdesc = NULL;
 103         }
 104
 105         arg->ob_type->tp_free(arg);
 106 }
 107
 108 static PyObject *
 109 PLy_result_colnames(PyObject *self, PyObject *unused)
 110 {
 111         PLyResultObject *ob = (PLyResultObject *) self;
 112         PyObject   *list;
 113         int                     i;
 114
 115         if (!ob->tupdesc)
 116         {
 117                 PLy_exception_set(PLy_exc_error, "command did not produce a result set");
 118                 return NULL;
 119         }
 120
 121         list = PyList_New(ob->tupdesc->natts);
 122         if (!list)
 123                 return NULL;
 124         for (i = 0; i < ob->tupdesc->natts; i++)
 125         {
 126                 Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 127
 128                 PyList_SET_ITEM(list, i, PLyUnicode_FromString(NameStr(attr->attname)));
 129         }
 130
 131         return list;
 132 }
 133
 134 static PyObject *
 135 PLy_result_coltypes(PyObject *self, PyObject *unused)
 136 {
 137         PLyResultObject *ob = (PLyResultObject *) self;
 138         PyObject   *list;
 139         int                     i;
 140
 141         if (!ob->tupdesc)
 142         {
 143                 PLy_exception_set(PLy_exc_error, "command did not produce a result set");
 144                 return NULL;
 145         }
 146
 147         list = PyList_New(ob->tupdesc->natts);
 148         if (!list)
 149                 return NULL;
 150         for (i = 0; i < ob->tupdesc->natts; i++)
 151         {
 152                 Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 153
 154                 PyList_SET_ITEM(list, i, PyLong_FromLong(attr->atttypid));
 155         }
 156
 157         return list;
 158 }
 159
 160 static PyObject *
 161 PLy_result_coltypmods(PyObject *self, PyObject *unused)
 162 {
 163         PLyResultObject *ob = (PLyResultObject *) self;
 164         PyObject   *list;
 165         int                     i;
 166
 167         if (!ob->tupdesc)
 168         {
 169                 PLy_exception_set(PLy_exc_error, "command did not produce a result set");
 170                 return NULL;
 171         }
 172
 173         list = PyList_New(ob->tupdesc->natts);
 174         if (!list)
 175                 return NULL;
 176         for (i = 0; i < ob->tupdesc->natts; i++)
 177         {
 178                 Form_pg_attribute attr = TupleDescAttr(ob->tupdesc, i);
 179
 180                 PyList_SET_ITEM(list, i, PyLong_FromLong(attr->atttypmod));
 181         }
 182
 183         return list;
 184 }
 185
 186 static PyObject *
 187 PLy_result_nrows(PyObject *self, PyObject *args)
 188 {
 189         PLyResultObject *ob = (PLyResultObject *) self;
 190
 191         Py_INCREF(ob->nrows);
 192         return ob->nrows;
 193 }
 194
 195 static PyObject *
 196 PLy_result_status(PyObject *self, PyObject *args)
 197 {
 198         PLyResultObject *ob = (PLyResultObject *) self;
 199
 200         Py_INCREF(ob->status);
 201         return ob->status;
 202 }
 203
 204 static Py_ssize_t
 205 PLy_result_length(PyObject *arg)
 206 {
 207         PLyResultObject *ob = (PLyResultObject *) arg;
 208
 209         return PyList_Size(ob->rows);
 210 }
 211
 212 static PyObject *
 213 PLy_result_item(PyObject *arg, Py_ssize_t idx)
 214 {
 215         PyObject   *rv;
 216         PLyResultObject *ob = (PLyResultObject *) arg;
 217
 218         rv = PyList_GetItem(ob->rows, idx);
 219         if (rv != NULL)
 220                 Py_INCREF(rv);
 221         return rv;
 222 }
 223
 224 static PyObject *
 225 PLy_result_str(PyObject *arg)
 226 {
 227         PLyResultObject *ob = (PLyResultObject *) arg;
 228
 229         return PyUnicode_FromFormat("<%s status=%S nrows=%S rows=%S>",
 230                                                                 Py_TYPE(ob)->tp_name,
 231                                                                 ob->status,
 232                                                                 ob->nrows,
 233                                                                 ob->rows);
 234 }
 235
 236 static PyObject *
 237 PLy_result_subscript(PyObject *arg, PyObject *item)
 238 {
 239         PLyResultObject *ob = (PLyResultObject *) arg;
 240
 241         return PyObject_GetItem(ob->rows, item);
 242 }
 243
 244 static int
 245 PLy_result_ass_subscript(PyObject *arg, PyObject *item, PyObject *value)
 246 {
 247         PLyResultObject *ob = (PLyResultObject *) arg;
 248
 249         return PyObject_SetItem(ob->rows, item, value);
 250 }