psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-22 17:06:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Check for Python errors during copy (closes: #134).

Browse Source
This commit is contained in:
Federico Di Gregorio 2007-01-16 12:03:29 +00:00
parent 8c2c72a48a
commit 5c425f5294
3 changed files with 26 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ChangeLog
View File

@ -1,5 +1,8 @@
2007-01-16 Federico Di Gregorio <fog@initd.org>
* psycopg/pqpath.cs: new checks for NULL values (meaning an
exception was raised) in all method calls (fixes #134).
* psycopg/psycopgmodule.c: applied LATIN patch from #148.
* psycopg/connection_type.c: obfuscate password after using it.

28
psycopg/pqpath.c
View File

@ -609,7 +609,9 @@ _pq_copy_in_v3(cursorObject *curs)
}
Py_XDECREF(o);
Dprintf("_pq_copy_in_v3: error = %d", error);
if (error == 0 || error == 2)
/* 0 means that the copy went well, 2 that there was an error on the
backend: in both cases we'll get the error message from the
@ -626,7 +628,7 @@ _pq_copy_in_v3(cursorObject *curs)
IFCLEARPGRES(curs->pgres);
}
return 1;
return error == 0 ? 1 : -1;
}
#endif
static int
@ -639,7 +641,8 @@ _pq_copy_in(cursorObject *curs)
while (1) {
o = PyObject_CallMethod(curs->copyfile, "readline", NULL);
if (!o || o == Py_None || PyString_GET_SIZE(o) == 0) break;
if (o == NULL) return -1;
if (o == Py_None || PyString_GET_SIZE(o) == 0) break;
if (PQputline(curs->conn->pgconn, PyString_AS_STRING(o)) != 0) {
Py_DECREF(o);
return -1;
@ -666,6 +669,8 @@ _pq_copy_in(cursorObject *curs)
static int
_pq_copy_out_v3(cursorObject *curs)
{
PyObject *tmp = NULL;
char *buffer;
int len;
@ -675,8 +680,13 @@ _pq_copy_out_v3(cursorObject *curs)
Py_END_ALLOW_THREADS;
if (len > 0 && buffer) {
PyObject_CallMethod(curs->copyfile, "write", "s#", buffer, len);
tmp = PyObject_CallMethod(curs->copyfile,
"write", "s#", buffer, len);
PQfreemem(buffer);
if (tmp == NULL)
return -1;
else
Py_DECREF(tmp);
}
/* we break on len == 0 but note that that should *not* happen,
because we are not doing an async call (if it happens blame
@ -703,9 +713,11 @@ _pq_copy_out_v3(cursorObject *curs)
static int
_pq_copy_out(cursorObject *curs)
{
PyObject *tmp = NULL;
char buffer[4096];
int status, len;
while (1) {
Py_BEGIN_ALLOW_THREADS;
status = PQgetline(curs->conn->pgconn, buffer, 4096);
@ -723,7 +735,11 @@ _pq_copy_out(cursorObject *curs)
return -1;
}
PyObject_CallMethod(curs->copyfile, "write", "s#", buffer, len);
tmp = PyObject_CallMethod(curs->copyfile, "write", "s#", buffer, len);
if (tmp == NULL)
return -1;
else
Py_DECREF(tmp);
}
status = 1;

2
setup.cfg
View File

@ -1,5 +1,5 @@
[build_ext]
define=PSYCOPG_EXTENSIONS,PSYCOPG_DISPLAY_SIZE,PSYCOPG_NEW_BOOLEAN,HAVE_PQFREEMEM,HAVE_PQPROTOCOL3
define=PSYCOPG_DEBUG,PSYCOPG_EXTENSIONS,PSYCOPG_DISPLAY_SIZE,PSYCOPG_NEW_BOOLEAN,HAVE_PQFREEMEM,HAVE_PQPROTOCOL3
# PSYCOPG_EXTENSIONS enables extensions to PEP-249 (you really want this)
# PSYCOPG_DISPLAY_SIZE enable display size calculation (a little slower)
# HAVE_PQFREEMEM should be defined on PostgreSQL >= 7.4