summaryrefslogtreecommitdiff
path: root/psycopg/adapter_binary.c
diff options
context:
space:
mode:
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>2012-03-04 17:59:51 +0000
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>2012-03-04 17:59:51 +0000
commit0e832b97ea7a84220f2148db3746e10f85eadef2 (patch)
tree747b74dfaa8bc058227ce56b965d14aad28bcf15 /psycopg/adapter_binary.c
parent76cc838a93dc2e55c6ddd101a543bda61cba4550 (diff)
downloadpsycopg2-0e832b97ea7a84220f2148db3746e10f85eadef2.tar.gz
Proper type check in prepare() methods for list, binary, qstring
Diffstat (limited to 'psycopg/adapter_binary.c')
-rw-r--r--psycopg/adapter_binary.c10
1 files changed, 4 insertions, 6 deletions
diff --git a/psycopg/adapter_binary.c b/psycopg/adapter_binary.c
index 2574c60..da3bec6 100644
--- a/psycopg/adapter_binary.c
+++ b/psycopg/adapter_binary.c
@@ -149,16 +149,14 @@ binary_str(binaryObject *self)
static PyObject *
binary_prepare(binaryObject *self, PyObject *args)
{
- connectionObject *conn;
+ PyObject *conn;
- if (!PyArg_ParseTuple(args, "O", &conn))
+ if (!PyArg_ParseTuple(args, "O!", &connectionType, &conn))
return NULL;
Py_XDECREF(self->conn);
- if (conn) {
- self->conn = (PyObject*)conn;
- Py_INCREF(self->conn);
- }
+ self->conn = conn;
+ Py_INCREF(self->conn);
Py_INCREF(Py_None);
return Py_None;