MAINT: restore Npy__PyLong_AsInt

mattip · mattip · commit dfa6beb879a3 · 2024-07-03T17:07:15.000+03:00
diff --git a/numpy/_core/include/numpy/npy_3kcompat.h b/numpy/_core/include/numpy/npy_3kcompat.h
@@ -40,6 +40,25 @@ extern _invalid_parameter_handler _Py_silent_invalid_parameter_handler;
 
 #endif /* _MSC_VER >= 1900 */
 
+/* Python13 removes _PyLong_AsInt */
+static inline int
+Npy__PyLong_AsInt(PyObject *obj)
+{
+    int overflow;
+    long result = PyLong_AsLongAndOverflow(obj, &overflow);
+
+    /* INT_MAX and INT_MIN are defined in Python.h */
+    if (overflow || result > INT_MAX || result < INT_MIN) {
+        /* XXX: could be cute and give a different
+ *            message for overflow == -1 */
+        PyErr_SetString(PyExc_OverflowError,
+                        "Python int too large to convert to C int");
+        return -1;
+    }
+    return (int)result;
+}
+
+
 
 /*
  * PyFile_* compatibility
diff --git a/numpy/f2py/cfuncs.py b/numpy/f2py/cfuncs.py
@@ -878,13 +878,13 @@
     PyObject* tmp = NULL;
 
     if (PyLong_Check(obj)) {
-        *v = _PyLong_AsInt(obj);
+        *v = Npy__PyLong_AsInt(obj);
         return !(*v == -1 && PyErr_Occurred());
     }
 
     tmp = PyNumber_Long(obj);
     if (tmp) {
-        *v = _PyLong_AsInt(tmp);
+        *v = Npy__PyLong_AsInt(tmp);
         Py_DECREF(tmp);
         return !(*v == -1 && PyErr_Occurred());
     }
