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();
725 PyDict_SetItemString(pltdata,
"name", pltname);
731 PyDict_SetItemString(pltdata,
"relid", pltrelid);
737 PyDict_SetItemString(pltdata,
"table_name", plttablename);
738 Py_DECREF(plttablename);
743 PyDict_SetItemString(pltdata,
"table_schema", plttableschema);
744 Py_DECREF(plttableschema);
758 PyDict_SetItemString(pltdata,
"when", pltwhen);
764 PyDict_SetItemString(pltdata,
"level", pltlevel);
776 PyDict_SetItemString(pltdata,
"old", Py_None);
781 PyDict_SetItemString(pltdata,
"new", pytnew);
789 PyDict_SetItemString(pltdata,
"new", Py_None);
794 PyDict_SetItemString(pltdata,
"old", pytold);
806 PyDict_SetItemString(pltdata,
"new", pytnew);
812 PyDict_SetItemString(pltdata,
"old", pytold);
822 PyDict_SetItemString(pltdata,
"event", pltevent);
828 PyDict_SetItemString(pltdata,
"level", pltlevel);
831 PyDict_SetItemString(pltdata,
"old", Py_None);
832 PyDict_SetItemString(pltdata,
"new", Py_None);
849 PyDict_SetItemString(pltdata,
"event", pltevent);
864 Assert(pltargs && pltargs != Py_None);
873 PyList_SetItem(pltargs,
i, pltarg);
878 Assert(pltargs == Py_None);
880 PyDict_SetItemString(pltdata,
"args", pltargs);
902 PyObject *
volatile plntup;
903 PyObject *
volatile plkeys;
904 PyObject *
volatile plval;
905 Datum *
volatile modvalues;
906 bool *
volatile modnulls;
907 bool *
volatile modrepls;
914 plntup = plkeys = plval = NULL;
925 if ((plntup = PyDict_GetItemString(pltd,
"new")) == NULL)
927 (
errcode(ERRCODE_UNDEFINED_OBJECT),
928 errmsg(
"TD[\"new\"] deleted, cannot modify row")));
930 if (!PyDict_Check(plntup))
932 (
errcode(ERRCODE_DATATYPE_MISMATCH),
933 errmsg(
"TD[\"new\"] is not a dictionary")));
935 plkeys = PyDict_Keys(plntup);
936 nkeys = PyList_Size(plkeys);
941 modnulls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
942 modrepls = (
bool *)
palloc0(tupdesc->
natts *
sizeof(
bool));
944 for (
i = 0;
i < nkeys;
i++)
951 platt = PyList_GetItem(plkeys,
i);
952 if (PyUnicode_Check(platt))
957 (
errcode(ERRCODE_DATATYPE_MISMATCH),
958 errmsg(
"TD[\"new\"] dictionary key at ordinal position %d is not a string",
i)));
964 (
errcode(ERRCODE_UNDEFINED_COLUMN),
965 errmsg(
"key \"%s\" found in TD[\"new\"] does not exist as a column in the triggering row",
969 (
errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
970 errmsg(
"cannot set system attribute \"%s\"",
974 (
errcode(ERRCODE_E_R_I_E_TRIGGER_PROTOCOL_VIOLATED),
975 errmsg(
"cannot set generated column \"%s\"",
978 plval = PyDict_GetItem(plntup, platt);
980 elog(
FATAL,
"Python interpreter is probably corrupted");
989 &modnulls[attn - 1]);
990 modrepls[attn - 1] =
true;
1040 PyObject *rv = NULL;
1043 PyDict_SetItemString(proc->
globals, kargs, vargs);
1047 #if PY_VERSION_HEX >= 0x03020000
1048 rv = PyEval_EvalCode(proc->
code,
1051 rv = PyEval_EvalCode((PyCodeObject *) proc->
code,
1082 Assert(save_subxact_level >= 0);
1091 (
errmsg(
"forcibly aborting a subtransaction that has not been exited")));
1100 pfree(subtransactiondata);
#define PG_USED_FOR_ASSERTS_ONLY
#define Assert(condition)
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, const Datum *replValues, const bool *replIsnull, const bool *doReplace)
if(TABLE==NULL||TABLE_index==NULL)
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 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)
MemoryContextSwitchTo(old_ctx)
#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::@181 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)