PostgreSQL Source Code  git master
plpy_plpymodule.c File Reference
#include "postgres.h"
#include "access/xact.h"
#include "mb/pg_wchar.h"
#include "utils/builtins.h"
#include "utils/snapmgr.h"
#include "plpython.h"
#include "plpy_plpymodule.h"
#include "plpy_cursorobject.h"
#include "plpy_elog.h"
#include "plpy_main.h"
#include "plpy_planobject.h"
#include "plpy_resultobject.h"
#include "plpy_spi.h"
#include "plpy_subxactobject.h"
#include "spiexceptions.h"
Include dependency graph for plpy_plpymodule.c:

Go to the source code of this file.

Data Structures

struct  ExceptionMap
 

Typedefs

typedef struct ExceptionMap ExceptionMap
 

Functions

static void PLy_add_exceptions (PyObject *plpy)
 
static PyObject * PLy_create_exception (char *name, PyObject *base, PyObject *dict, const char *modname, PyObject *mod)
 
static void PLy_generate_spi_exceptions (PyObject *mod, PyObject *base)
 
static PyObject * PLy_debug (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_log (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_info (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_notice (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_warning (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_error (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_fatal (PyObject *self, PyObject *args, PyObject *kw)
 
static PyObject * PLy_quote_literal (PyObject *self, PyObject *args)
 
static PyObject * PLy_quote_nullable (PyObject *self, PyObject *args)
 
static PyObject * PLy_quote_ident (PyObject *self, PyObject *args)
 
static PyObject * PLy_commit (PyObject *self, PyObject *args)
 
static PyObject * PLy_rollback (PyObject *self, PyObject *args)
 
void PLy_init_plpy (void)
 
static PyObject * PLy_output (volatile int level, PyObject *self, PyObject *args, PyObject *kw)
 
static char * object_to_string (PyObject *obj)
 

Variables

HTABPLy_spi_exceptions = NULL
 
static const ExceptionMap exception_map []
 
static PyMethodDef PLy_methods []
 
static PyMethodDef PLy_exc_methods []
 

Typedef Documentation

◆ ExceptionMap

Function Documentation

◆ object_to_string()

static char* object_to_string ( PyObject *  obj)
static

Definition at line 406 of file plpy_plpymodule.c.

References pstrdup(), and generate_unaccent_rules::str.

Referenced by PLy_output().

407 {
408  if (obj)
409  {
410  PyObject *so = PyObject_Str(obj);
411 
412  if (so != NULL)
413  {
414  char *str;
415 
416  str = pstrdup(PyString_AsString(so));
417  Py_DECREF(so);
418 
419  return str;
420  }
421  }
422 
423  return NULL;
424 }
char * pstrdup(const char *in)
Definition: mcxt.c:1063

◆ PLy_add_exceptions()

static void PLy_add_exceptions ( PyObject *  plpy)
static

Definition at line 199 of file plpy_plpymodule.c.

References HASHCTL::entrysize, ERROR, HASH_BLOBS, hash_create(), HASH_ELEM, HASHCTL::keysize, PLy_create_exception(), PLy_elog, PLy_exc_error, PLy_exc_fatal, PLy_exc_methods, PLy_exc_spi_error, and PLy_generate_spi_exceptions().

Referenced by PLy_init_plpy().

200 {
201  PyObject *excmod;
202  HASHCTL hash_ctl;
203 
204 #if PY_MAJOR_VERSION < 3
205  excmod = Py_InitModule("spiexceptions", PLy_exc_methods);
206 #else
207  excmod = PyModule_Create(&PLy_exc_module);
208 #endif
209  if (excmod == NULL)
210  PLy_elog(ERROR, "could not create the spiexceptions module");
211 
212  /*
213  * PyModule_AddObject does not add a refcount to the object, for some odd
214  * reason; we must do that.
215  */
216  Py_INCREF(excmod);
217  if (PyModule_AddObject(plpy, "spiexceptions", excmod) < 0)
218  PLy_elog(ERROR, "could not add the spiexceptions module");
219 
220  PLy_exc_error = PLy_create_exception("plpy.Error", NULL, NULL,
221  "Error", plpy);
222  PLy_exc_fatal = PLy_create_exception("plpy.Fatal", NULL, NULL,
223  "Fatal", plpy);
224  PLy_exc_spi_error = PLy_create_exception("plpy.SPIError", NULL, NULL,
225  "SPIError", plpy);
226 
227  memset(&hash_ctl, 0, sizeof(hash_ctl));
228  hash_ctl.keysize = sizeof(int);
229  hash_ctl.entrysize = sizeof(PLyExceptionEntry);
230  PLy_spi_exceptions = hash_create("PL/Python SPI exceptions", 256,
231  &hash_ctl, HASH_ELEM | HASH_BLOBS);
232 
234 }
#define HASH_ELEM
Definition: hsearch.h:87
Size entrysize
Definition: hsearch.h:73
PyObject * PLy_exc_error
Definition: plpy_elog.c:19
static void PLy_generate_spi_exceptions(PyObject *mod, PyObject *base)
#define ERROR
Definition: elog.h:43
static PyObject * PLy_create_exception(char *name, PyObject *base, PyObject *dict, const char *modname, PyObject *mod)
#define PLy_elog
Definition: plpy_elog.h:36
#define HASH_BLOBS
Definition: hsearch.h:88
HTAB * hash_create(const char *tabname, long nelem, HASHCTL *info, int flags)
Definition: dynahash.c:316
Size keysize
Definition: hsearch.h:72
static PyMethodDef PLy_exc_methods[]
PyObject * PLy_exc_fatal
Definition: plpy_elog.c:20
struct PLyExceptionEntry PLyExceptionEntry
HTAB * PLy_spi_exceptions
PyObject * PLy_exc_spi_error
Definition: plpy_elog.c:21

◆ PLy_commit()

static PyObject * PLy_commit ( PyObject *  self,
PyObject *  args 
)
static

Definition at line 593 of file plpy_plpymodule.c.

References ereport, errcode(), errmsg(), ERROR, PLy_current_execution_context(), PLyExecutionContext::scratch_ctx, SPI_commit(), SPI_start_transaction(), and ThereArePinnedPortals().

594 {
596 
597  if (ThereArePinnedPortals())
598  ereport(ERROR,
599  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
600  errmsg("cannot commit transaction while a cursor is open")));
601 
602  SPI_commit();
604 
605  /* was cleared at transaction end, reset pointer */
606  exec_ctx->scratch_ctx = NULL;
607 
608  Py_RETURN_NONE;
609 }
MemoryContext scratch_ctx
Definition: plpy_main.h:21
int errcode(int sqlerrcode)
Definition: elog.c:575
PLyExecutionContext * PLy_current_execution_context(void)
Definition: plpy_main.c:414
bool ThereArePinnedPortals(void)
Definition: portalmem.c:1168
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
void SPI_commit(void)
Definition: spi.c:206
int errmsg(const char *fmt,...)
Definition: elog.c:797
void SPI_start_transaction(void)
Definition: spi.c:197

◆ PLy_create_exception()

static PyObject * PLy_create_exception ( char *  name,
PyObject *  base,
PyObject *  dict,
const char *  modname,
PyObject *  mod 
)
static

Definition at line 240 of file plpy_plpymodule.c.

References ERROR, and PLy_elog.

Referenced by PLy_add_exceptions(), and PLy_generate_spi_exceptions().

242 {
243  PyObject *exc;
244 
245  exc = PyErr_NewException(name, base, dict);
246  if (exc == NULL)
247  PLy_elog(ERROR, NULL);
248 
249  /*
250  * PyModule_AddObject does not add a refcount to the object, for some odd
251  * reason; we must do that.
252  */
253  Py_INCREF(exc);
254  PyModule_AddObject(mod, modname, exc);
255 
256  /*
257  * The caller will also store a pointer to the exception object in some
258  * permanent variable, so add another ref to account for that. This is
259  * probably excessively paranoid, but let's be sure.
260  */
261  Py_INCREF(exc);
262  return exc;
263 }
#define ERROR
Definition: elog.h:43
#define PLy_elog
Definition: plpy_elog.h:36
const char * name
Definition: encode.c:521

◆ PLy_debug()

static PyObject * PLy_debug ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 310 of file plpy_plpymodule.c.

References DEBUG2, and PLy_output().

311 {
312  return PLy_output(DEBUG2, self, args, kw);
313 }
#define DEBUG2
Definition: elog.h:24
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)

◆ PLy_error()

static PyObject * PLy_error ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 340 of file plpy_plpymodule.c.

References ERROR, and PLy_output().

341 {
342  return PLy_output(ERROR, self, args, kw);
343 }
#define ERROR
Definition: elog.h:43
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)

◆ PLy_fatal()

static PyObject * PLy_fatal ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 346 of file plpy_plpymodule.c.

References FATAL, and PLy_output().

347 {
348  return PLy_output(FATAL, self, args, kw);
349 }
#define FATAL
Definition: elog.h:52
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)

◆ PLy_generate_spi_exceptions()

static void PLy_generate_spi_exceptions ( PyObject *  mod,
PyObject *  base 
)
static

Definition at line 269 of file plpy_plpymodule.c.

References generate_unaccent_rules::args, Assert, ExceptionMap::classname, ERROR, PLyExceptionEntry::exc, HASH_ENTER, hash_search(), i, ExceptionMap::name, PLy_create_exception(), PLy_elog, PLy_output(), ExceptionMap::sqlstate, and unpack_sql_state().

Referenced by PLy_add_exceptions().

270 {
271  int i;
272 
273  for (i = 0; exception_map[i].name != NULL; i++)
274  {
275  bool found;
276  PyObject *exc;
277  PLyExceptionEntry *entry;
278  PyObject *sqlstate;
279  PyObject *dict = PyDict_New();
280 
281  if (dict == NULL)
282  PLy_elog(ERROR, NULL);
283 
284  sqlstate = PyString_FromString(unpack_sql_state(exception_map[i].sqlstate));
285  if (sqlstate == NULL)
286  PLy_elog(ERROR, "could not generate SPI exceptions");
287 
288  PyDict_SetItemString(dict, "sqlstate", sqlstate);
289  Py_DECREF(sqlstate);
290 
291  exc = PLy_create_exception(exception_map[i].name, base, dict,
292  exception_map[i].classname, mod);
293 
294  entry = hash_search(PLy_spi_exceptions, &exception_map[i].sqlstate,
295  HASH_ENTER, &found);
296  Assert(!found);
297  entry->exc = exc;
298  }
299 }
char * unpack_sql_state(int sql_state)
Definition: elog.c:2849
void * hash_search(HTAB *hashp, const void *keyPtr, HASHACTION action, bool *foundPtr)
Definition: dynahash.c:904
#define ERROR
Definition: elog.h:43
static const ExceptionMap exception_map[]
static PyObject * PLy_create_exception(char *name, PyObject *base, PyObject *dict, const char *modname, PyObject *mod)
#define PLy_elog
Definition: plpy_elog.h:36
#define Assert(condition)
Definition: c.h:688
HTAB * PLy_spi_exceptions
const char * name
Definition: encode.c:521
int i
PyObject * exc
Definition: plpy_spi.h:17

◆ PLy_info()

static PyObject * PLy_info ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 322 of file plpy_plpymodule.c.

References INFO, and PLy_output().

323 {
324  return PLy_output(INFO, self, args, kw);
325 }
#define INFO
Definition: elog.h:33
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)

◆ PLy_init_plpy()

void PLy_init_plpy ( void  )

Definition at line 157 of file plpy_plpymodule.c.

References ERROR, PLy_add_exceptions(), PLy_cursor_init_type(), PLy_elog, PLy_methods, PLy_plan_init_type(), PLy_result_init_type(), and PLy_subtransaction_init_type().

Referenced by PLy_initialize().

158 {
159  PyObject *main_mod,
160  *main_dict,
161  *plpy_mod;
162 
163 #if PY_MAJOR_VERSION < 3
164  PyObject *plpy;
165 #endif
166 
167  /*
168  * initialize plpy module
169  */
174 
175 #if PY_MAJOR_VERSION >= 3
176  PyModule_Create(&PLy_module);
177  /* for Python 3 we initialized the exceptions in PyInit_plpy */
178 #else
179  plpy = Py_InitModule("plpy", PLy_methods);
180  PLy_add_exceptions(plpy);
181 #endif
182 
183  /* PyDict_SetItemString(plpy, "PlanType", (PyObject *) &PLy_PlanType); */
184 
185  /*
186  * initialize main module, and add plpy
187  */
188  main_mod = PyImport_AddModule("__main__");
189  main_dict = PyModule_GetDict(main_mod);
190  plpy_mod = PyImport_AddModule("plpy");
191  if (plpy_mod == NULL)
192  PLy_elog(ERROR, "could not import \"plpy\" module");
193  PyDict_SetItemString(main_dict, "plpy", plpy_mod);
194  if (PyErr_Occurred())
195  PLy_elog(ERROR, "could not import \"plpy\" module");
196 }
void PLy_plan_init_type(void)
static void PLy_add_exceptions(PyObject *plpy)
#define ERROR
Definition: elog.h:43
void PLy_cursor_init_type(void)
void PLy_subtransaction_init_type(void)
#define PLy_elog
Definition: plpy_elog.h:36
void PLy_result_init_type(void)
static PyMethodDef PLy_methods[]

◆ PLy_log()

static PyObject * PLy_log ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 316 of file plpy_plpymodule.c.

References LOG, and PLy_output().

317 {
318  return PLy_output(LOG, self, args, kw);
319 }
#define LOG
Definition: elog.h:26
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)

◆ PLy_notice()

static PyObject * PLy_notice ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 328 of file plpy_plpymodule.c.

References NOTICE, and PLy_output().

329 {
330  return PLy_output(NOTICE, self, args, kw);
331 }
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)
#define NOTICE
Definition: elog.h:37

◆ PLy_output()

static PyObject * PLy_output ( volatile int  level,
PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 427 of file plpy_plpymodule.c.

References CopyErrorData(), CurrentMemoryContext, dgettext, ereport, err_generic_string(), errcode(), errdetail_internal(), errhint(), errmsg_internal(), ERROR, FlushErrorState(), FreeErrorData(), MAKE_SQLSTATE, MemoryContextSwitchTo(), object_to_string(), pfree(), PG_CATCH, PG_DIAG_COLUMN_NAME, PG_DIAG_CONSTRAINT_NAME, PG_DIAG_DATATYPE_NAME, PG_DIAG_SCHEMA_NAME, PG_DIAG_TABLE_NAME, PG_END_TRY, PG_TRY, pg_verifymbstr(), PLy_elog, PLy_exc_error, PLy_exception_set(), PLy_exception_set_with_details(), pstrdup(), ExceptionMap::sqlstate, TEXTDOMAIN, and value.

Referenced by PLy_debug(), PLy_error(), PLy_fatal(), PLy_generate_spi_exceptions(), PLy_info(), PLy_log(), PLy_notice(), and PLy_warning().

428 {
429  int sqlstate = 0;
430  char *volatile sqlstatestr = NULL;
431  char *volatile message = NULL;
432  char *volatile detail = NULL;
433  char *volatile hint = NULL;
434  char *volatile column_name = NULL;
435  char *volatile constraint_name = NULL;
436  char *volatile datatype_name = NULL;
437  char *volatile table_name = NULL;
438  char *volatile schema_name = NULL;
439  volatile MemoryContext oldcontext;
440  PyObject *key,
441  *value;
442  PyObject *volatile so;
443  Py_ssize_t pos = 0;
444 
445  if (PyTuple_Size(args) == 1)
446  {
447  /*
448  * Treat single argument specially to avoid undesirable ('tuple',)
449  * decoration.
450  */
451  PyObject *o;
452 
453  if (!PyArg_UnpackTuple(args, "plpy.elog", 1, 1, &o))
454  PLy_elog(ERROR, "could not unpack arguments in plpy.elog");
455  so = PyObject_Str(o);
456  }
457  else
458  so = PyObject_Str(args);
459 
460  if (so == NULL || ((message = PyString_AsString(so)) == NULL))
461  {
462  level = ERROR;
463  message = dgettext(TEXTDOMAIN, "could not parse error message in plpy.elog");
464  }
465  message = pstrdup(message);
466 
467  Py_XDECREF(so);
468 
469  if (kw != NULL)
470  {
471  while (PyDict_Next(kw, &pos, &key, &value))
472  {
473  char *keyword = PyString_AsString(key);
474 
475  if (strcmp(keyword, "message") == 0)
476  {
477  /* the message should not be overwritten */
478  if (PyTuple_Size(args) != 0)
479  {
480  PLy_exception_set(PyExc_TypeError, "argument 'message' given by name and position");
481  return NULL;
482  }
483 
484  if (message)
485  pfree(message);
486  message = object_to_string(value);
487  }
488  else if (strcmp(keyword, "detail") == 0)
489  detail = object_to_string(value);
490  else if (strcmp(keyword, "hint") == 0)
491  hint = object_to_string(value);
492  else if (strcmp(keyword, "sqlstate") == 0)
493  sqlstatestr = object_to_string(value);
494  else if (strcmp(keyword, "schema_name") == 0)
495  schema_name = object_to_string(value);
496  else if (strcmp(keyword, "table_name") == 0)
497  table_name = object_to_string(value);
498  else if (strcmp(keyword, "column_name") == 0)
499  column_name = object_to_string(value);
500  else if (strcmp(keyword, "datatype_name") == 0)
501  datatype_name = object_to_string(value);
502  else if (strcmp(keyword, "constraint_name") == 0)
503  constraint_name = object_to_string(value);
504  else
505  {
506  PLy_exception_set(PyExc_TypeError,
507  "'%s' is an invalid keyword argument for this function",
508  keyword);
509  return NULL;
510  }
511  }
512  }
513 
514  if (sqlstatestr != NULL)
515  {
516  if (strlen(sqlstatestr) != 5)
517  {
518  PLy_exception_set(PyExc_ValueError, "invalid SQLSTATE code");
519  return NULL;
520  }
521 
522  if (strspn(sqlstatestr, "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ") != 5)
523  {
524  PLy_exception_set(PyExc_ValueError, "invalid SQLSTATE code");
525  return NULL;
526  }
527 
528  sqlstate = MAKE_SQLSTATE(sqlstatestr[0],
529  sqlstatestr[1],
530  sqlstatestr[2],
531  sqlstatestr[3],
532  sqlstatestr[4]);
533  }
534 
535  oldcontext = CurrentMemoryContext;
536  PG_TRY();
537  {
538  if (message != NULL)
539  pg_verifymbstr(message, strlen(message), false);
540  if (detail != NULL)
541  pg_verifymbstr(detail, strlen(detail), false);
542  if (hint != NULL)
543  pg_verifymbstr(hint, strlen(hint), false);
544  if (schema_name != NULL)
545  pg_verifymbstr(schema_name, strlen(schema_name), false);
546  if (table_name != NULL)
547  pg_verifymbstr(table_name, strlen(table_name), false);
548  if (column_name != NULL)
549  pg_verifymbstr(column_name, strlen(column_name), false);
550  if (datatype_name != NULL)
551  pg_verifymbstr(datatype_name, strlen(datatype_name), false);
552  if (constraint_name != NULL)
553  pg_verifymbstr(constraint_name, strlen(constraint_name), false);
554 
555  ereport(level,
556  ((sqlstate != 0) ? errcode(sqlstate) : 0,
557  (message != NULL) ? errmsg_internal("%s", message) : 0,
558  (detail != NULL) ? errdetail_internal("%s", detail) : 0,
559  (hint != NULL) ? errhint("%s", hint) : 0,
560  (column_name != NULL) ?
561  err_generic_string(PG_DIAG_COLUMN_NAME, column_name) : 0,
562  (constraint_name != NULL) ?
563  err_generic_string(PG_DIAG_CONSTRAINT_NAME, constraint_name) : 0,
564  (datatype_name != NULL) ?
565  err_generic_string(PG_DIAG_DATATYPE_NAME, datatype_name) : 0,
566  (table_name != NULL) ?
567  err_generic_string(PG_DIAG_TABLE_NAME, table_name) : 0,
568  (schema_name != NULL) ?
569  err_generic_string(PG_DIAG_SCHEMA_NAME, schema_name) : 0));
570  }
571  PG_CATCH();
572  {
573  ErrorData *edata;
574 
575  MemoryContextSwitchTo(oldcontext);
576  edata = CopyErrorData();
577  FlushErrorState();
578 
580  FreeErrorData(edata);
581 
582  return NULL;
583  }
584  PG_END_TRY();
585 
586  /*
587  * return a legal object so the interpreter will continue on its merry way
588  */
589  Py_RETURN_NONE;
590 }
int errhint(const char *fmt,...)
Definition: elog.c:987
static struct @130 value
#define PG_DIAG_SCHEMA_NAME
Definition: postgres_ext.h:65
#define TEXTDOMAIN
Definition: elog.h:125
ErrorData * CopyErrorData(void)
Definition: elog.c:1497
#define PG_DIAG_COLUMN_NAME
Definition: postgres_ext.h:67
#define MAKE_SQLSTATE(ch1, ch2, ch3, ch4, ch5)
Definition: elog.h:62
char * pstrdup(const char *in)
Definition: mcxt.c:1063
#define PG_DIAG_TABLE_NAME
Definition: postgres_ext.h:66
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
int errcode(int sqlerrcode)
Definition: elog.c:575
PyObject * PLy_exc_error
Definition: plpy_elog.c:19
void PLy_exception_set(PyObject *exc, const char *fmt,...)
Definition: plpy_elog.c:500
void FlushErrorState(void)
Definition: elog.c:1587
int Py_ssize_t
Definition: plpython.h:64
int errdetail_internal(const char *fmt,...)
Definition: elog.c:900
void pfree(void *pointer)
Definition: mcxt.c:936
void FreeErrorData(ErrorData *edata)
Definition: elog.c:1551
#define ERROR
Definition: elog.h:43
void PLy_exception_set_with_details(PyObject *excclass, ErrorData *edata)
Definition: plpy_elog.c:532
MemoryContext CurrentMemoryContext
Definition: mcxt.c:37
#define ereport(elevel, rest)
Definition: elog.h:122
#define PLy_elog
Definition: plpy_elog.h:36
#define dgettext(d, x)
Definition: c.h:1010
#define PG_DIAG_DATATYPE_NAME
Definition: postgres_ext.h:68
#define PG_DIAG_CONSTRAINT_NAME
Definition: postgres_ext.h:69
int errmsg_internal(const char *fmt,...)
Definition: elog.c:827
#define PG_CATCH()
Definition: elog.h:293
bool pg_verifymbstr(const char *mbstr, int len, bool noError)
Definition: wchar.c:1866
int err_generic_string(int field, const char *str)
Definition: elog.c:1191
#define PG_TRY()
Definition: elog.h:284
static char * object_to_string(PyObject *obj)
#define PG_END_TRY()
Definition: elog.h:300

◆ PLy_quote_ident()

static PyObject * PLy_quote_ident ( PyObject *  self,
PyObject *  args 
)
static

Definition at line 389 of file plpy_plpymodule.c.

References quote_identifier(), and generate_unaccent_rules::str.

390 {
391  const char *str;
392  const char *quoted;
393  PyObject *ret;
394 
395  if (!PyArg_ParseTuple(args, "s:quote_ident", &str))
396  return NULL;
397 
398  quoted = quote_identifier(str);
399  ret = PyString_FromString(quoted);
400 
401  return ret;
402 }
const char * quote_identifier(const char *ident)
Definition: ruleutils.c:10435

◆ PLy_quote_literal()

static PyObject * PLy_quote_literal ( PyObject *  self,
PyObject *  args 
)
static

Definition at line 352 of file plpy_plpymodule.c.

References pfree(), quote_literal_cstr(), and generate_unaccent_rules::str.

353 {
354  const char *str;
355  char *quoted;
356  PyObject *ret;
357 
358  if (!PyArg_ParseTuple(args, "s:quote_literal", &str))
359  return NULL;
360 
361  quoted = quote_literal_cstr(str);
362  ret = PyString_FromString(quoted);
363  pfree(quoted);
364 
365  return ret;
366 }
char * quote_literal_cstr(const char *rawstr)
Definition: quote.c:102
void pfree(void *pointer)
Definition: mcxt.c:936

◆ PLy_quote_nullable()

static PyObject * PLy_quote_nullable ( PyObject *  self,
PyObject *  args 
)
static

Definition at line 369 of file plpy_plpymodule.c.

References pfree(), quote_literal_cstr(), and generate_unaccent_rules::str.

370 {
371  const char *str;
372  char *quoted;
373  PyObject *ret;
374 
375  if (!PyArg_ParseTuple(args, "z:quote_nullable", &str))
376  return NULL;
377 
378  if (str == NULL)
379  return PyString_FromString("NULL");
380 
381  quoted = quote_literal_cstr(str);
382  ret = PyString_FromString(quoted);
383  pfree(quoted);
384 
385  return ret;
386 }
char * quote_literal_cstr(const char *rawstr)
Definition: quote.c:102
void pfree(void *pointer)
Definition: mcxt.c:936

◆ PLy_rollback()

static PyObject * PLy_rollback ( PyObject *  self,
PyObject *  args 
)
static

Definition at line 612 of file plpy_plpymodule.c.

References ereport, errcode(), errmsg(), ERROR, PLy_current_execution_context(), PLyExecutionContext::scratch_ctx, SPI_rollback(), SPI_start_transaction(), and ThereArePinnedPortals().

613 {
615 
616  if (ThereArePinnedPortals())
617  ereport(ERROR,
618  (errcode(ERRCODE_INVALID_TRANSACTION_TERMINATION),
619  errmsg("cannot abort transaction while a cursor is open")));
620 
621  SPI_rollback();
623 
624  /* was cleared at transaction end, reset pointer */
625  exec_ctx->scratch_ctx = NULL;
626 
627  Py_RETURN_NONE;
628 }
MemoryContext scratch_ctx
Definition: plpy_main.h:21
int errcode(int sqlerrcode)
Definition: elog.c:575
PLyExecutionContext * PLy_current_execution_context(void)
Definition: plpy_main.c:414
void SPI_rollback(void)
Definition: spi.c:240
bool ThereArePinnedPortals(void)
Definition: portalmem.c:1168
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
int errmsg(const char *fmt,...)
Definition: elog.c:797
void SPI_start_transaction(void)
Definition: spi.c:197

◆ PLy_warning()

static PyObject * PLy_warning ( PyObject *  self,
PyObject *  args,
PyObject *  kw 
)
static

Definition at line 334 of file plpy_plpymodule.c.

References PLy_output(), and WARNING.

335 {
336  return PLy_output(WARNING, self, args, kw);
337 }
static PyObject * PLy_output(volatile int level, PyObject *self, PyObject *args, PyObject *kw)
#define WARNING
Definition: elog.h:40

Variable Documentation

◆ exception_map

const ExceptionMap exception_map[]
static
Initial value:
= {
{NULL, NULL, 0}
}

Definition at line 59 of file plpy_plpymodule.c.

◆ PLy_exc_methods

PyMethodDef PLy_exc_methods[]
static
Initial value:
= {
{NULL, NULL, 0, NULL}
}

Definition at line 112 of file plpy_plpymodule.c.

Referenced by PLy_add_exceptions().

◆ PLy_methods

PyMethodDef PLy_methods[]
static

Definition at line 64 of file plpy_plpymodule.c.

Referenced by PLy_init_plpy().

◆ PLy_spi_exceptions

HTAB* PLy_spi_exceptions = NULL

Definition at line 27 of file plpy_plpymodule.c.

Referenced by PLy_spi_subtransaction_abort().