diff options
| author | Federico Di Gregorio <fog@initd.org> | 2009-01-22 11:02:38 +0100 |
|---|---|---|
| committer | Federico Di Gregorio <fog@initd.org> | 2009-01-22 11:02:38 +0100 |
| commit | e22451736a3b35bcb7da199e14a1b62e93e97663 (patch) | |
| tree | 71bd071bcd81016ea295b879d2a307884d5abcac /psycopg/psycopgmodule.c | |
| parent | 78c0765367d2a02f2ba5079ef2b600316e7d9003 (diff) | |
| download | psycopg2-e22451736a3b35bcb7da199e14a1b62e93e97663.tar.gz | |
register_type() now works on connection and cursor subclasses
Diffstat (limited to 'psycopg/psycopgmodule.c')
| -rw-r--r-- | psycopg/psycopgmodule.c | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/psycopg/psycopgmodule.c b/psycopg/psycopgmodule.c index 43e0402..eb05b84 100644 --- a/psycopg/psycopgmodule.c +++ b/psycopg/psycopgmodule.c @@ -243,10 +243,10 @@ psyco_register_type(PyObject *self, PyObject *args) } if (obj != NULL) { - if (obj->ob_type == &cursorType) { + if (PyObject_TypeCheck(obj, &cursorType)) { _psyco_register_type_set(&(((cursorObject*)obj)->string_types), type); } - else if (obj->ob_type == &connectionType) { + else if (PyObject_TypeCheck(obj, &connectionType)) { typecast_add(type, ((connectionObject*)obj)->string_types, 0); } else { |