515bca0f96
things that need to be dealt with but the majority of the code has now been transformed to use either Unicode or encoded strings with Oracle.
126 lines
4.6 KiB
C
126 lines
4.6 KiB
C
//-----------------------------------------------------------------------------
|
|
// StringUtils.c
|
|
// Defines constants and routines specific to handling strings.
|
|
//-----------------------------------------------------------------------------
|
|
|
|
// define structure for abstracting string buffers
|
|
typedef struct {
|
|
const void *ptr;
|
|
Py_ssize_t size;
|
|
#if defined(WITH_UNICODE) && defined(Py_UNICODE_WIDE)
|
|
PyObject *encodedString;
|
|
#endif
|
|
} udt_StringBuffer;
|
|
|
|
|
|
// use the bytes methods in cx_Oracle and define them as the equivalent string
|
|
// type methods as is done in Python 2.6
|
|
#ifndef PyBytes_Check
|
|
#define PyBytes_Type PyString_Type
|
|
#define PyBytes_AS_STRING PyString_AS_STRING
|
|
#define PyBytes_GET_SIZE PyString_GET_SIZE
|
|
#define PyBytes_Check PyString_Check
|
|
#define PyBytes_Format PyString_Format
|
|
#define PyBytes_FromString PyString_FromString
|
|
#define PyBytes_FromStringAndSize PyString_FromStringAndSize
|
|
#endif
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringBuffer_Init()
|
|
// Initialize the string buffer with an empty string. Returns 0 as a
|
|
// convenience to the caller.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringBuffer_Init(
|
|
udt_StringBuffer *buf) // buffer to fill
|
|
{
|
|
buf->ptr = NULL;
|
|
buf->size = 0;
|
|
#if defined(WITH_UNICODE) && defined(Py_UNICODE_WIDE)
|
|
buf->encodedString = NULL;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringBuffer_FromUnicode()
|
|
// Populate the string buffer from a unicode object.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringBuffer_FromUnicode(
|
|
udt_StringBuffer *buf, // buffer to fill
|
|
PyObject *obj) // unicode object expected
|
|
{
|
|
if (!obj)
|
|
return StringBuffer_Init(buf);
|
|
#ifdef Py_UNICODE_WIDE
|
|
int one = 1;
|
|
int byteOrder = (IS_LITTLE_ENDIAN) ? -1 : 1;
|
|
buf->encodedString = PyUnicode_EncodeUTF16(PyUnicode_AS_UNICODE(obj),
|
|
PyUnicode_GET_SIZE(obj), NULL, byteOrder);
|
|
if (!buf->encodedString)
|
|
return -1;
|
|
buf->ptr = PyBytes_AS_STRING(buf->encodedString);
|
|
buf->size = PyBytes_GET_SIZE(buf->encodedString);
|
|
#else
|
|
buf->ptr = (char*) PyUnicode_AS_UNICODE(obj);
|
|
buf->size = PyUnicode_GET_DATA_SIZE(obj);
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringBuffer_FromBytes()
|
|
// Populate the string buffer from a bytes object.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringBuffer_FromBytes(
|
|
udt_StringBuffer *buf, // buffer to fill
|
|
PyObject *obj) // bytes object expected
|
|
{
|
|
if (!obj)
|
|
return StringBuffer_Init(buf);
|
|
buf->ptr = PyBytes_AS_STRING(obj);
|
|
buf->size = PyBytes_GET_SIZE(obj);
|
|
return 0;
|
|
}
|
|
|
|
|
|
#ifdef WITH_UNICODE
|
|
#define CXORA_CHARSETID OCI_UTF16ID
|
|
#define CXORA_BYTES_PER_CHAR 2
|
|
#define cxString_Type &PyUnicode_Type
|
|
#define cxString_Format PyUnicode_Format
|
|
#define cxString_Check PyUnicode_Check
|
|
#define cxString_GetSize PyUnicode_GET_SIZE
|
|
#define cxString_FromObject PyObject_Unicode
|
|
#define cxString_FromAscii(str) \
|
|
PyUnicode_DecodeASCII(str, strlen(str), NULL)
|
|
#ifdef Py_UNICODE_WIDE
|
|
#define StringBuffer_Clear(buffer) \
|
|
Py_XDECREF((buffer)->encodedString)
|
|
#define cxString_FromEncodedString(buffer, numBytes) \
|
|
PyUnicode_DecodeUTF16(buffer, numBytes, NULL, NULL)
|
|
#else
|
|
#define StringBuffer_Clear(buffer)
|
|
#define cxString_FromEncodedString(buffer, numBytes) \
|
|
PyUnicode_FromUnicode((Py_UNICODE*) (buffer), (numBytes) / 2)
|
|
#endif
|
|
#define StringBuffer_Fill StringBuffer_FromUnicode
|
|
#else
|
|
#define CXORA_CHARSETID 0
|
|
#define CXORA_BYTES_PER_CHAR 1
|
|
#define cxString_Type &PyBytes_Type
|
|
#define cxString_Format PyBytes_Format
|
|
#define cxString_Check PyBytes_Check
|
|
#define cxString_GetSize PyBytes_GET_SIZE
|
|
#define cxString_FromObject PyObject_Str
|
|
#define StringBuffer_Clear(buffer)
|
|
#define cxString_FromAscii(str) \
|
|
PyBytes_FromString(str)
|
|
#define cxString_FromEncodedString(buffer, numBytes) \
|
|
PyBytes_FromStringAndSize(buffer, numBytes)
|
|
#define StringBuffer_Fill StringBuffer_FromBytes
|
|
#endif
|
|
|