fixed crash in pdecimal_str with a few Python 2.5.x releases.

is_finite() is not available in 2.5.1, it is in 2.5.5 but is officially
supported only since 2.6.

1 files changed, 26 insertions, 14 deletions

diff --git a/psycopg/adapter_pdecimal.c b/psycopg/adapter_pdecimal.c
index 3112b8f..c11e753 100644
--- a/psycopg/adapter_pdecimal.c
+++ b/psycopg/adapter_pdecimal.c
@@ -43,29 +43,41 @@ static PyObject *
 pdecimal_str(pdecimalObject *self)
 {
     PyObject *check, *res = NULL;
-#if PY_VERSION_HEX < 0x02050000
-    check = PyObject_CallMethod(self->wrapped, "_isnan", NULL);
-    if (PyInt_AsLong(check) == 1) {
+    check = PyObject_CallMethod(self->wrapped, "is_finite", NULL);
+    if (check == Py_True) {
+        res = PyObject_Str(self->wrapped);
+        goto end;
+    }
+    else if (check) {
+        res = PyString_FromString("'NaN'::numeric");
+        goto end;
+    }
+
+    /* is_finite() was introduced 2.5.1 < somewhere <= 2.5.4.
+     * We assume we are here because we didn't find the method. */
+    PyErr_Clear();
+
+    if (!(check = PyObject_CallMethod(self->wrapped, "_isnan", NULL))) {
+        goto end;
+    }
+    if (PyObject_IsTrue(check)) {
         res = PyString_FromString("'NaN'::numeric");
         goto end;
     }
+
     Py_DECREF(check);
-    check = PyObject_CallMethod(self->wrapped, "_isinfinity", NULL);
-    if (abs(PyInt_AsLong(check)) == 1) {
+    if (!(check = PyObject_CallMethod(self->wrapped, "_isinfinity", NULL))) {
+        goto end;
+    }
+    if (PyObject_IsTrue(check)) {
         res = PyString_FromString("'NaN'::numeric");
         goto end;
     }
+
     res = PyObject_Str(self->wrapped);
-#else
-    check = PyObject_CallMethod(self->wrapped, "is_finite", NULL);
-    if (check == Py_True)
-        res = PyObject_Str(self->wrapped);
-    else
-        res = PyString_FromString("'NaN'::numeric");
-#endif
 
-   end:
-    Py_DECREF(check);
+end:
+    Py_XDECREF(check);
     return res;
 }