59 PyObject *
volatile plargs = NULL;
60 PyObject *
volatile plrv = NULL;
97 if (srfstate == NULL || srfstate->
iter == NULL)
129 if (srfstate->
iter == NULL)
138 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
139 errmsg(
"unsupported set function return mode"),
140 errdetail(
"PL/Python set-returning functions only support returning one value per call.")));
145 srfstate->
iter = PyObject_GetIter(plrv);
150 if (srfstate->
iter == NULL)
152 (
errcode(ERRCODE_DATATYPE_MISMATCH),
153 errmsg(
"returned object cannot be iterated"),
154 errdetail(
"PL/Python set-returning functions must return an iterable object.")));
158 plrv = PyIter_Next(srfstate->
iter);
162 bool has_error = (PyErr_Occurred() != NULL);
164 Py_DECREF(srfstate->
iter);
165 srfstate->
iter = NULL;
210 (
errcode(ERRCODE_DATATYPE_MISMATCH),
211 errmsg(
"PL/Python procedure did not return None")));
214 (
errcode(ERRCODE_DATATYPE_MISMATCH),
215 errmsg(
"PL/Python function with return type \"void\" did not return None")));
221 else if (plrv == Py_None &&
222 srfstate && srfstate->
iter == NULL)
256 Py_XDECREF(srfstate->
iter);
257 srfstate->
iter = NULL;
279 if (srfstate->
iter == NULL)
308 PyObject *
volatile plargs = NULL;
309 PyObject *
volatile plrv = NULL;
363 if (PyUnicode_Check(plrv))
368 (
errcode(ERRCODE_DATA_EXCEPTION),
369 errmsg(
"unexpected return value from trigger procedure"),
370 errdetail(
"Expected None or a string.")));
383 (
errmsg(
"PL/Python trigger function returned \"MODIFY\" in a DELETE trigger -- ignored")));
392 (
errcode(ERRCODE_DATA_EXCEPTION),
393 errmsg(
"unexpected return value from trigger procedure"),
394 errdetail(
"Expected None, \"OK\", \"SKIP\", or \"MODIFY\".")));
413 PyObject *
volatile arg = NULL;
443 if (PyList_SetItem(
args,
i,
arg) == -1)
444 PLy_elog(
ERROR,
"PyList_SetItem() failed, while setting up arguments");
448 PLy_elog(
ERROR,
"PyDict_SetItemString() failed, while setting up arguments");
459 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
460 errmsg(
"function returning record called in context "
461 "that cannot accept type record")));
497 proc->
nargs *
sizeof(PyObject *));
501 result->
args = PyDict_GetItemString(proc->
globals,
"args");
502 Py_XINCREF(result->
args);
535 for (
i = 0;
i < savedargs->
nargs;
i++)
549 PyDict_SetItemString(proc->
globals,
"args", savedargs->
args);
550 Py_DECREF(savedargs->
args);
566 for (
i = 0;
i < savedargs->
nargs;
i++)
572 Py_XDECREF(savedargs->
args);
662 Py_XDECREF(srfstate->
iter);
663 srfstate->
iter = NULL;
703 pltdata = PyDict_New();
720 PyDict_SetItemString(pltdata,
"name", pltname);
726 PyDict_SetItemString(pltdata,
"relid", pltrelid);
732 PyDict_SetItemString(pltdata,
"table_name", plttablename);
733 Py_DECREF(plttablename);
738 PyDict_SetItemString(pltdata,
"table_schema", plttableschema);
739 Py_DECREF(plttableschema);
753 PyDict_SetItemString(pltdata,
"when", pltwhen);
759 PyDict_SetItemString(pltdata,
"level", pltlevel);
771 PyDict_SetItemString(pltdata,
"old", Py_None);
776 PyDict_SetItemString(pltdata,
"new", pytnew);
784 PyDict_SetItemString(pltdata,
"new", Py_None);
789 PyDict_SetItemString(pltdata,
"old", pytold);
801 PyDict_SetItemString(pltdata,
"new", pytnew);
807 PyDict_SetItemString(pltdata,
"old", pytold);
817 PyDict_SetItemString(pltdata,
"event", pltevent);
823 PyDict_SetItemString(pltdata,
"level", pltlevel);
826 PyDict_SetItemString(pltdata,
"old", Py_None);
827 PyDict_SetItemString(pltdata,
"new", Py_None);
844 PyDict_SetItemString(pltdata,
"event", pltevent);
868 PyList_SetItem(pltargs,
i, pltarg);
876 PyDict_SetItemString(pltdata,
"args", pltargs);
898 PyObject *
volatile plntup;
899 PyObject *
volatile plkeys;
900 PyObject *
volatile plval;
901 Datum *
volatile modvalues;
902 bool *
volatile modnulls;
903 bool *
volatile modrepls;
910 plntup = plkeys = plval = NULL;
921 if ((plntup = PyDict_GetItemString(pltd,
"new")) == NULL)
923 (
errcode(ERRCODE_UNDEFINED_OBJECT),
924 errmsg(
"TD[\"new\"] deleted, cannot modify row")));
926 if (!PyDict_Check(plntup))
928 (
errcode(ERRCODE_DATATYPE_MISMATCH),
929 errmsg(
"TD[\"new\"] is not a dictionary")));
931 plkeys = PyDict_Keys(plntup);
932 nkeys = PyList_Size(plkeys);
937 modnulls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
938 modrepls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
940 for (
i = 0;
i < nkeys;
i++)
947 platt = PyList_GetItem(plkeys,
i);
948 if (PyUnicode_Check(platt))
953 (
errcode(ERRCODE_DATATYPE_MISMATCH),
954 errmsg(
"TD[\"new\"] dictionary key at ordinal position %d is not a string",
i)));
960 (
errcode(ERRCODE_UNDEFINED_COLUMN),
961 errmsg(
"key \"%s\" found in TD[\"new\"] does not exist as a column in the triggering row",
965 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
966 errmsg(
"cannot set system attribute \"%s\"",
970 (
errcode(ERRCODE_E_R_I_E_TRIGGER_PROTOCOL_VIOLATED),
971 errmsg(
"cannot set generated column \"%s\"",
974 plval = PyDict_GetItem(plntup, platt);
976 elog(
FATAL,
"Python interpreter is probably corrupted");
985 &modnulls[attn - 1]);
986 modrepls[attn - 1] =
true;
1036 PyObject *rv = NULL;
1039 PyDict_SetItemString(proc->
globals, kargs, vargs);
1043 #if PY_VERSION_HEX >= 0x03020000
1044 rv = PyEval_EvalCode(proc->
code,
1047 rv = PyEval_EvalCode((PyCodeObject *) proc->
code,
1078 Assert(save_subxact_level >= 0);
1087 (
errmsg(
"forcibly aborting a subtransaction that has not been exited")));
1096 pfree(subtransactiondata);
#define PG_USED_FOR_ASSERTS_ONLY
elog(ERROR, "%s: %s", p2, msg)
int errdetail(const char *fmt,...)
ErrorContextCallback * error_context_stack
int errcode(int sqlerrcode)
int errmsg(const char *fmt,...)
#define ereport(elevel,...)
#define DirectFunctionCall1(func, arg1)
TypeFuncClass get_call_result_type(FunctionCallInfo fcinfo, Oid *resultTypeId, TupleDesc *resultTupleDesc)
#define SRF_IS_FIRSTCALL()
#define SRF_RETURN_NEXT_NULL(_funcctx)
#define SRF_PERCALL_SETUP()
#define SRF_RETURN_NEXT(_funcctx, _result)
#define SRF_FIRSTCALL_INIT()
#define SRF_RETURN_DONE(_funcctx)
HeapTuple heap_modify_tuple(HeapTuple tuple, TupleDesc tupleDesc, Datum *replValues, bool *replIsnull, bool *doReplace)
if(TABLE==NULL||TABLE_index==NULL)
Assert(fmt[strlen(fmt) - 1] !='\n')
List * list_delete_first(List *list)
void MemoryContextRegisterResetCallback(MemoryContext context, MemoryContextCallback *cb)
void pfree(void *pointer)
void * palloc0(Size size)
void * MemoryContextAllocZero(MemoryContext context, Size size)
#define IsA(nodeptr, _type_)
Datum oidout(PG_FUNCTION_ARGS)
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
static int list_length(const List *l)
static void PLy_function_drop_args(PLySavedArgs *savedargs)
static void PLy_global_args_push(PLyProcedure *proc)
struct PLySRFState PLySRFState
Datum PLy_exec_function(FunctionCallInfo fcinfo, PLyProcedure *proc)
static PyObject * PLy_function_build_args(FunctionCallInfo fcinfo, PLyProcedure *proc)
static void PLy_function_restore_args(PLyProcedure *proc, PLySavedArgs *savedargs)
static PyObject * PLy_procedure_call(PLyProcedure *proc, const char *kargs, PyObject *vargs)
static void plpython_trigger_error_callback(void *arg)
static void PLy_global_args_pop(PLyProcedure *proc)
static PyObject * PLy_trigger_build_args(FunctionCallInfo fcinfo, PLyProcedure *proc, HeapTuple *rv)
static HeapTuple PLy_modify_tuple(PLyProcedure *proc, PyObject *pltd, TriggerData *tdata, HeapTuple otup)
static void plpython_srf_cleanup_callback(void *arg)
static PLySavedArgs * PLy_function_save_args(PLyProcedure *proc)
static void PLy_abort_open_subtransactions(int save_subxact_level)
static void plpython_return_error_callback(void *arg)
HeapTuple PLy_exec_trigger(FunctionCallInfo fcinfo, PLyProcedure *proc)
PLyExecutionContext * PLy_current_execution_context(void)
List * explicit_subtransactions
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)
PyObject * PLy_input_convert(PLyDatumToOb *arg, Datum val)
void PLy_output_setup_record(PLyObToDatum *arg, TupleDesc desc, PLyProcedure *proc)
void PLy_output_setup_tuple(PLyObToDatum *arg, TupleDesc desc, PLyProcedure *proc)
Datum PLy_output_convert(PLyObToDatum *arg, PyObject *val, bool *isnull)
char * PLyUnicode_AsString(PyObject *unicode)
PyObject * PLyUnicode_FromString(const char *s)
int pg_strcasecmp(const char *s1, const char *s2)
static char * DatumGetCString(Datum X)
static Datum ObjectIdGetDatum(Oid X)
#define RelationGetDescr(relation)
ResourceOwner CurrentResourceOwner
char * SPI_getrelname(Relation rel)
int SPI_fnumber(TupleDesc tupdesc, const char *fname)
int SPI_register_trigger_data(TriggerData *tdata)
char * SPI_getnspname(Relation rel)
#define SPI_ERROR_NOATTRIBUTE
struct ErrorContextCallback * previous
void(* callback)(void *arg)
MemoryContext multi_call_memory_ctx
NullableDatum args[FLEXIBLE_ARRAY_MEMBER]
MemoryContextCallbackFunction func
union PLyObToDatum::@173 u
MemoryContextCallback callback
struct PLySavedArgs * next
PyObject * namedargs[FLEXIBLE_ARRAY_MEMBER]
SetFunctionReturnMode returnMode
#define TRIGGER_FIRED_FOR_STATEMENT(event)
#define TRIGGER_FIRED_BY_DELETE(event)
#define TRIGGER_FIRED_BEFORE(event)
#define CALLED_AS_TRIGGER(fcinfo)
#define TRIGGER_FIRED_FOR_ROW(event)
#define TRIGGER_FIRED_AFTER(event)
#define TRIGGER_FIRED_BY_TRUNCATE(event)
#define TRIGGER_FIRED_BY_INSERT(event)
#define TRIGGER_FIRED_BY_UPDATE(event)
#define TRIGGER_FIRED_INSTEAD(event)
#define TupleDescAttr(tupdesc, i)
void RollbackAndReleaseCurrentSubTransaction(void)