diff options
author | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2012-03-04 17:59:51 +0000 |
---|---|---|
committer | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2012-03-04 17:59:51 +0000 |
commit | 0e832b97ea7a84220f2148db3746e10f85eadef2 (patch) | |
tree | 747b74dfaa8bc058227ce56b965d14aad28bcf15 /psycopg/adapter_binary.c | |
parent | 76cc838a93dc2e55c6ddd101a543bda61cba4550 (diff) | |
download | psycopg2-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.c | 10 |
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; |