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.")));
385 (
errmsg(
"PL/Python trigger function returned \"MODIFY\" in a DELETE trigger -- ignored")));
394 (
errcode(ERRCODE_DATA_EXCEPTION),
395 errmsg(
"unexpected return value from trigger procedure"),
396 errdetail(
"Expected None, \"OK\", \"SKIP\", or \"MODIFY\".")));
415 PyObject *
volatile arg = NULL;
416 PyObject *
volatile args = NULL;
440 if (PyList_SetItem(
args,
i,
arg) == -1)
441 PLy_elog(
ERROR,
"PyList_SetItem() failed, while setting up arguments");
445 PLy_elog(
ERROR,
"PyDict_SetItemString() failed, while setting up arguments");
456 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
457 errmsg(
"function returning record called in context "
458 "that cannot accept type record")));
494 proc->
nargs *
sizeof(PyObject *));
498 result->
args = PyDict_GetItemString(proc->
globals,
"args");
499 Py_XINCREF(result->
args);
532 for (
i = 0;
i < savedargs->
nargs;
i++)
546 PyDict_SetItemString(proc->
globals,
"args", savedargs->
args);
547 Py_DECREF(savedargs->
args);
563 for (
i = 0;
i < savedargs->
nargs;
i++)
569 Py_XDECREF(savedargs->
args);
659 Py_XDECREF(srfstate->
iter);
660 srfstate->
iter = NULL;
692 PyObject *
volatile pltdata = NULL;
697 pltdata = PyDict_New();
702 PyDict_SetItemString(pltdata,
"name", pltname);
708 PyDict_SetItemString(pltdata,
"relid", pltrelid);
714 PyDict_SetItemString(pltdata,
"table_name", plttablename);
715 Py_DECREF(plttablename);
720 PyDict_SetItemString(pltdata,
"table_schema", plttableschema);
721 Py_DECREF(plttableschema);
735 PyDict_SetItemString(pltdata,
"when", pltwhen);
741 PyDict_SetItemString(pltdata,
"level", pltlevel);
753 PyDict_SetItemString(pltdata,
"old", Py_None);
758 PyDict_SetItemString(pltdata,
"new", pytnew);
766 PyDict_SetItemString(pltdata,
"new", Py_None);
771 PyDict_SetItemString(pltdata,
"old", pytold);
783 PyDict_SetItemString(pltdata,
"new", pytnew);
789 PyDict_SetItemString(pltdata,
"old", pytold);
799 PyDict_SetItemString(pltdata,
"event", pltevent);
805 PyDict_SetItemString(pltdata,
"level", pltlevel);
808 PyDict_SetItemString(pltdata,
"old", Py_None);
809 PyDict_SetItemString(pltdata,
"new", Py_None);
826 PyDict_SetItemString(pltdata,
"event", pltevent);
853 PyList_SetItem(pltargs,
i, pltarg);
861 PyDict_SetItemString(pltdata,
"args", pltargs);
882 PyObject *
volatile plntup;
883 PyObject *
volatile plkeys;
884 PyObject *
volatile plval;
885 Datum *
volatile modvalues;
886 bool *
volatile modnulls;
887 bool *
volatile modrepls;
894 plntup = plkeys = plval = NULL;
905 if ((plntup = PyDict_GetItemString(pltd,
"new")) == NULL)
907 (
errcode(ERRCODE_UNDEFINED_OBJECT),
908 errmsg(
"TD[\"new\"] deleted, cannot modify row")));
910 if (!PyDict_Check(plntup))
912 (
errcode(ERRCODE_DATATYPE_MISMATCH),
913 errmsg(
"TD[\"new\"] is not a dictionary")));
915 plkeys = PyDict_Keys(plntup);
916 nkeys = PyList_Size(plkeys);
921 modnulls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
922 modrepls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
924 for (
i = 0;
i < nkeys;
i++)
931 platt = PyList_GetItem(plkeys,
i);
932 if (PyUnicode_Check(platt))
937 (
errcode(ERRCODE_DATATYPE_MISMATCH),
938 errmsg(
"TD[\"new\"] dictionary key at ordinal position %d is not a string",
i)));
944 (
errcode(ERRCODE_UNDEFINED_COLUMN),
945 errmsg(
"key \"%s\" found in TD[\"new\"] does not exist as a column in the triggering row",
949 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
950 errmsg(
"cannot set system attribute \"%s\"",
954 (
errcode(ERRCODE_E_R_I_E_TRIGGER_PROTOCOL_VIOLATED),
955 errmsg(
"cannot set generated column \"%s\"",
958 plval = PyDict_GetItem(plntup, platt);
960 elog(
FATAL,
"Python interpreter is probably corrupted");
969 &modnulls[attn - 1]);
970 modrepls[attn - 1] =
true;
1020 PyObject *rv = NULL;
1023 PyDict_SetItemString(proc->
globals, kargs, vargs);
1027 #if PY_VERSION_HEX >= 0x03020000
1028 rv = PyEval_EvalCode(proc->
code,
1031 rv = PyEval_EvalCode((PyCodeObject *) proc->
code,
1062 Assert(save_subxact_level >= 0);
1071 (
errmsg(
"forcibly aborting a subtransaction that has not been exited")));
1080 pfree(subtransactiondata);
#define offsetof(type, field)
#define PG_USED_FOR_ASSERTS_ONLY
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)
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)
#define ObjectIdGetDatum(X)
#define DatumGetCString(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)