plpy_util.c File Reference

#include "postgres.h"
#include "mb/pg_wchar.h"
#include "plpy_elog.h"
#include "plpy_util.h"
#include "plpython.h"
#include "utils/memutils.h"

Include dependency graph for plpy_util.c:

Go to the source code of this file.

Functions
PyObject *	PLyUnicode_Bytes (PyObject *unicode)
char *	PLyUnicode_AsString (PyObject *unicode)
PyObject *	PLyUnicode_FromStringAndSize (const char *s, Py_ssize_t size)
PyObject *	PLyUnicode_FromString (const char *s)

Function Documentation

PLyUnicode_AsString()

char* PLyUnicode_AsString

(

PyObject *

unicode

)

Definition at line 83 of file plpy_util.c.

{
    PyObject   *o = PLyUnicode_Bytes(unicode);
    char       *rv = pstrdup(PyBytes_AsString(o));
 
    Py_XDECREF(o);
    return rv;
}

References PLyUnicode_Bytes(), and pstrdup().

Referenced by get_string_attr(), object_to_string(), PLy_cursor_plan(), PLy_exec_trigger(), PLy_get_sqlerrcode(), PLy_modify_tuple(), PLy_output(), PLy_spi_execute_plan(), PLy_spi_prepare(), and PLy_traceback().

PLyUnicode_Bytes()

PyObject* PLyUnicode_Bytes

(

PyObject *

unicode

)

Definition at line 21 of file plpy_util.c.

{
    PyObject   *bytes,
               *rv;
    char       *utf8string,
               *encoded;
 
    /* First encode the Python unicode object with UTF-8. */
    bytes = PyUnicode_AsUTF8String(unicode);
    if (bytes == NULL)
        PLy_elog(ERROR, "could not convert Python Unicode object to bytes");
 
    utf8string = PyBytes_AsString(bytes);
    if (utf8string == NULL)
    {
        Py_DECREF(bytes);
        PLy_elog(ERROR, "could not extract bytes from encoded string");
    }
 
    /*
     * Then convert to server encoding if necessary.
     *
     * PyUnicode_AsEncodedString could be used to encode the object directly
     * in the server encoding, but Python doesn't support all the encodings
     * that PostgreSQL does (EUC_TW and MULE_INTERNAL). UTF-8 is used as an
     * intermediary in PLyUnicode_FromString as well.
     */
    if (GetDatabaseEncoding() != PG_UTF8)
    {
        PG_TRY();
        {
            encoded = pg_any_to_server(utf8string,
                                       strlen(utf8string),
                                       PG_UTF8);
        }
        PG_CATCH();
        {
            Py_DECREF(bytes);
            PG_RE_THROW();
        }
        PG_END_TRY();
    }
    else
        encoded = utf8string;
 
    /* finally, build a bytes object in the server encoding */
    rv = PyBytes_FromStringAndSize(encoded, strlen(encoded));
 
    /* if pg_any_to_server allocated memory, free it now */
    if (utf8string != encoded)
        pfree(encoded);
 
    Py_DECREF(bytes);
    return rv;
}

References ERROR, GetDatabaseEncoding(), pfree(), pg_any_to_server(), PG_CATCH, PG_END_TRY, PG_RE_THROW, PG_TRY, PG_UTF8, and PLy_elog.

Referenced by PLyObject_AsString(), and PLyUnicode_AsString().

PLyUnicode_FromString()

PyObject* PLyUnicode_FromString

(

const char *

s

)

Definition at line 118 of file plpy_util.c.

{
    return PLyUnicode_FromStringAndSize(s, strlen(s));
}

References PLyUnicode_FromStringAndSize().

Referenced by PLy_generate_spi_exceptions(), PLy_quote_ident(), PLy_quote_literal(), PLy_quote_nullable(), PLy_result_colnames(), PLy_trigger_build_args(), PLyUnicode_FromScalar(), and set_string_attr().

PLyUnicode_FromStringAndSize()

PyObject* PLyUnicode_FromStringAndSize	(	const char *	s,
		Py_ssize_t	size
	)

Definition at line 97 of file plpy_util.c.

{
    char       *utf8string;
    PyObject   *o;
 
    utf8string = pg_server_to_any(s, size, PG_UTF8);
 
    if (utf8string == s)
    {
        o = PyUnicode_FromStringAndSize(s, size);
    }
    else
    {
        o = PyUnicode_FromString(utf8string);
        pfree(utf8string);
    }
 
    return o;
}

References pfree(), pg_server_to_any(), and PG_UTF8.

Referenced by PLyUnicode_FromString().