30 uint64 rows,
int status);
42 PyObject *
list = NULL;
43 PyObject *
volatile optr = NULL;
50 if (!PyArg_ParseTuple(
args,
"s|O:prepare", &query, &
list))
53 if (
list && (!PySequence_Check(
list)))
56 "second argument of plpy.prepare must be a sequence");
64 "PL/Python plan context",
68 nargs =
list ? PySequence_Length(
list) : 0;
86 for (
i = 0;
i < nargs;
i++)
92 optr = PySequence_GetItem(
list,
i);
93 if (PyUnicode_Check(optr))
98 (
errmsg(
"plpy.prepare: type name at ordinal position %d is not a string",
i)));
118 plan->types[
i] = typeId;
126 if (
plan->plan == NULL)
147 return (PyObject *)
plan;
158 PyObject *
list = NULL;
161 if (PyArg_ParseTuple(
args,
"s|l", &query, &limit))
166 if (PyArg_ParseTuple(
args,
"O|Ol", &
plan, &
list, &limit) &&
187 if (!PySequence_Check(
list) || PyUnicode_Check(
list))
189 PLy_exception_set(PyExc_TypeError,
"plpy.execute takes a sequence as its second argument");
192 nargs = PySequence_Length(
list);
199 if (nargs !=
plan->nargs)
202 PyObject *so = PyObject_Str(
list);
208 "Expected sequence of %d argument, got %d: %s",
209 "Expected sequence of %d arguments, got %d: %s",
211 plan->nargs, nargs, sv);
225 char *
volatile nulls;
229 nulls =
palloc(nargs *
sizeof(
char));
233 for (
j = 0;
j < nargs;
j++)
238 elem = PySequence_GetItem(
list,
j);
244 nulls[
j] = isnull ?
'n' :
' ';
269 for (k = 0; k < nargs; k++)
271 if (!
plan->args[k].typbyval &&
284 for (
i = 0;
i < nargs;
i++)
286 if (!
plan->args[
i].typbyval &&
297 "SPI_execute_plan failed: %s",
311 PyObject *ret = NULL;
339 "SPI_execute failed: %s",
360 Py_DECREF(result->
status);
361 result->
status = PyLong_FromLong(status);
363 if (status > 0 && tuptable == NULL)
365 Py_DECREF(result->
nrows);
366 result->
nrows = PyLong_FromUnsignedLongLong(rows);
368 else if (status > 0 && tuptable != NULL)
373 Py_DECREF(result->
nrows);
374 result->
nrows = PyLong_FromUnsignedLongLong(rows);
377 "PL/Python temp context",
398 if (rows > (uint64) PY_SSIZE_T_MAX)
400 (
errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
401 errmsg(
"query result has too many rows to fit in a Python list")));
403 Py_DECREF(result->
rows);
404 result->
rows = PyList_New(rows);
410 for (
i = 0;
i < rows;
i++)
417 PyList_SetItem(result->
rows,
i, row);
454 return (PyObject *) result;
631 PyObject *
args = NULL;
632 PyObject *spierror = NULL;
633 PyObject *spidata = NULL;
640 spierror = PyObject_CallObject(excclass,
args);
651 if (PyObject_SetAttrString(spierror,
"spidata", spidata) == -1)
654 PyErr_SetObject(excclass, spierror);
663 Py_XDECREF(spierror);
665 elog(
ERROR,
"could not convert SPI error to Python exception");
elog(ERROR, "%s: %s", p2, msg)
void * hash_search(HTAB *hashp, const void *keyPtr, HASHACTION action, bool *foundPtr)
void FreeErrorData(ErrorData *edata)
void FlushErrorState(void)
int errcode(int sqlerrcode)
int errmsg(const char *fmt,...)
ErrorData * CopyErrorData(void)
#define ereport(elevel,...)
Assert(fmt[strlen(fmt) - 1] !='\n')
bool pg_verifymbstr(const char *mbstr, int len, bool noError)
void pfree(void *pointer)
MemoryContext TopMemoryContext
void * palloc0(Size size)
MemoryContext CurrentMemoryContext
void MemoryContextDelete(MemoryContext context)
#define AllocSetContextCreate
#define ALLOCSET_DEFAULT_SIZES
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
bool parseTypeString(const char *str, Oid *typeid_p, int32 *typmod_p, Node *escontext)
PyObject * PLy_exc_spi_error
void PLy_exception_set(PyObject *exc, const char *fmt,...)
void PLy_exception_set_plural(PyObject *exc, const char *fmt_singular, const char *fmt_plural, unsigned long n,...)
PLyExecutionContext * PLy_current_execution_context(void)
PyObject * PLy_plan_new(void)
bool is_PLyPlanObject(PyObject *ob)
HTAB * PLy_spi_exceptions
PyObject * PLy_result_new(void)
PyObject * PLy_spi_prepare(PyObject *self, PyObject *args)
void PLy_spi_subtransaction_commit(MemoryContext oldcontext, ResourceOwner oldowner)
PyObject * PLy_spi_execute(PyObject *self, PyObject *args)
PyObject * PLy_rollback(PyObject *self, PyObject *args)
static PyObject * PLy_spi_execute_fetch_result(SPITupleTable *tuptable, uint64 rows, int status)
PyObject * PLy_spi_execute_plan(PyObject *ob, PyObject *list, long limit)
static PyObject * PLy_spi_execute_query(char *query, long limit)
PyObject * PLy_commit(PyObject *self, PyObject *args)
void PLy_spi_subtransaction_abort(MemoryContext oldcontext, ResourceOwner oldowner)
static void PLy_spi_exception_set(PyObject *excclass, ErrorData *edata)
void PLy_spi_subtransaction_begin(MemoryContext oldcontext, ResourceOwner oldowner)
PyObject * PLy_input_from_tuple(PLyDatumToOb *arg, HeapTuple tuple, TupleDesc desc, bool include_generated)
void PLy_output_setup_func(PLyObToDatum *arg, MemoryContext arg_mcxt, Oid typeOid, int32 typmod, PLyProcedure *proc)
void PLy_input_setup_func(PLyDatumToOb *arg, MemoryContext arg_mcxt, Oid typeOid, int32 typmod, PLyProcedure *proc)
void PLy_input_setup_tuple(PLyDatumToOb *arg, TupleDesc desc, PLyProcedure *proc)
Datum PLy_output_convert(PLyObToDatum *arg, PyObject *val, bool *isnull)
char * PLyUnicode_AsString(PyObject *unicode)
static Datum PointerGetDatum(const void *X)
static Pointer DatumGetPointer(Datum X)
ResourceOwner CurrentResourceOwner
SPITupleTable * SPI_tuptable
const char * SPI_result_code_string(int code)
int SPI_execute_plan(SPIPlanPtr plan, Datum *Values, const char *Nulls, bool read_only, long tcount)
void SPI_freetuptable(SPITupleTable *tuptable)
SPIPlanPtr SPI_prepare(const char *src, int nargs, Oid *argtypes)
int SPI_keepplan(SPIPlanPtr plan)
int SPI_execute(const char *src, bool read_only, long tcount)
MemoryContext scratch_ctx
PyObject_HEAD PyObject * nrows
TupleDesc CreateTupleDescCopy(TupleDesc tupdesc)
void BeginInternalSubTransaction(const char *name)
void RollbackAndReleaseCurrentSubTransaction(void)
void ReleaseCurrentSubTransaction(void)