psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-01-31 09:24:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added adaptation for objects supporting the new-style buffer interface

Browse Source 
Supporting this interface is required to adapt memoryview on Python 2.7 as they
don't support the old style. But because the old style is long deprecated it
makes sense to start supporting the new one.
This commit is contained in:
Daniele Varrazzo 2011-02-16 01:23:53 +00:00
parent 3ae2f221b3
commit bccbcf42d0
1 changed files with 59 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
psycopg/adapter_binary.c
View File

@ -47,57 +47,79 @@ binary_escape(unsigned char *from, size_t from_length,
return PQescapeBytea(from, from_length, to_length); return PQescapeBytea(from, from_length, to_length);
} }
#define HAS_BUFFER (PY_MAJOR_VERSION < 3)
#define HAS_MEMORYVIEW (PY_MAJOR_VERSION > 2 || PY_MINOR_VERSION >= 6)
/* binary_quote - do the quote process on plain and unicode strings */ /* binary_quote - do the quote process on plain and unicode strings */
static PyObject * static PyObject *
binary_quote(binaryObject *self) binary_quote(binaryObject *self)
{ {
char *to; char *to = NULL;
const char *buffer; const char *buffer = NULL;
Py_ssize_t buffer_len; Py_ssize_t buffer_len;
size_t len = 0; size_t len = 0;
PyObject *rv = NULL;
#if HAS_MEMORYVIEW
Py_buffer view;
int got_view = 0;
#endif
/* if we got a plain string or a buffer we escape it and save the buffer */ /* if we got a plain string or a buffer we escape it and save the buffer */
if (Bytes_Check(self->wrapped)
#if PY_MAJOR_VERSION < 3
|| PyBuffer_Check(self->wrapped)
#endif
#if PY_MAJOR_VERSION >= 3 || PY_MINOR_VERSION >= 6
|| PyByteArray_Check(self->wrapped)
#endif
#if PY_MAJOR_VERSION >= 3 || PY_MINOR_VERSION >= 7
|| PyMemoryView_Check(self->wrapped)
#endif
) {
/* escape and build quoted buffer */
if (PyObject_AsReadBuffer(self->wrapped, (const void **)&buffer,
&buffer_len) < 0)
return NULL;
to = (char *)binary_escape((unsigned char*)buffer, (size_t) buffer_len, #if HAS_MEMORYVIEW
&len, self->conn ? ((connectionObject*)self->conn)->pgconn : NULL); if (PyObject_CheckBuffer(self->wrapped)) {
if (to == NULL) { if (0 > PyObject_GetBuffer(self->wrapped, &view, PyBUF_CONTIG_RO)) {
PyErr_NoMemory(); goto exit;
return NULL;
} }
got_view = 1;
if (len > 0) buffer = (const char *)(view.buf);
self->buffer = Bytes_FromFormat( buffer_len = view.len;
(self->conn && ((connectionObject*)self->conn)->equote)
? "E'%s'::bytea" : "'%s'::bytea" , to);
else
self->buffer = Bytes_FromString("''::bytea");
PQfreemem(to);
} }
#endif
#if HAS_BUFFER
if (!buffer && (Bytes_Check(self->wrapped) || PyBuffer_Check(self->wrapped))) {
if (PyObject_AsReadBuffer(self->wrapped, (const void **)&buffer,
&buffer_len) < 0) {
goto exit;
}
}
#endif
if (!buffer) {
goto exit;
}
/* escape and build quoted buffer */
to = (char *)binary_escape((unsigned char*)buffer, (size_t) buffer_len,
&len, self->conn ? ((connectionObject*)self->conn)->pgconn : NULL);
if (to == NULL) {
PyErr_NoMemory();
goto exit;
}
if (len > 0)
rv = Bytes_FromFormat(
(self->conn && ((connectionObject*)self->conn)->equote)
? "E'%s'::bytea" : "'%s'::bytea" , to);
else
rv = Bytes_FromString("''::bytea");
exit:
if (to) { PQfreemem(to); }
#if HAS_MEMORYVIEW
if (got_view) { PyBuffer_Release(&view); }
#endif
/* if the wrapped object is not bytes or a buffer, this is an error */ /* if the wrapped object is not bytes or a buffer, this is an error */
else { if (!rv && !PyErr_Occurred()) {
PyErr_SetString(PyExc_TypeError, "can't escape non-bytes object"); PyErr_Format(PyExc_TypeError, "can't escape %s to binary",
return NULL; Py_TYPE(self->wrapped)->tp_name);
} }
return self->buffer; return rv;
} }
/* binary_str, binary_getquoted - return result of quoting */ /* binary_str, binary_getquoted - return result of quoting */
@ -106,11 +128,9 @@ static PyObject *
binary_getquoted(binaryObject *self, PyObject *args) binary_getquoted(binaryObject *self, PyObject *args)
{ {
if (self->buffer == NULL) { if (self->buffer == NULL) {
if (!(binary_quote(self))) { self->buffer = binary_quote(self);
return NULL;
}
} }
Py_INCREF(self->buffer); Py_XINCREF(self->buffer);
return self->buffer; return self->buffer;
} }