commit bdbad71b9def0b86433de12cecca022eee91bd9f
author Serhiy Storchaka <storchaka@gmail.com> Sun Jun 02 00:05:48 2019 +0300
committer GitHub <noreply@github.com> Sun Jun 02 00:05:48 2019 +0300
tree 147c8a3e08454a95e0e4900388d88f7b65430f63
parent 1a4d9ffa1aecd7e750195f2be06d3d16c7a3a88f

bpo-20092. Use __index__ in constructors of int, float and complex. (GH-13108)

Objects/floatobject.c

diff --git a/Objects/floatobject.c b/Objects/floatobject.c
index 2bf7061..15cbe5c 100644
--- a/Objects/floatobject.c
+++ b/Objects/floatobject.c
@@ -246,6 +246,15 @@
 
     nb = Py_TYPE(op)->tp_as_number;
     if (nb == NULL || nb->nb_float == NULL) {
+        if (nb && nb->nb_index) {
+            PyObject *res = PyNumber_Index(op);
+            if (!res) {
+                return -1;
+            }
+            double val = PyLong_AsDouble(res);
+            Py_DECREF(res);
+            return val;
+        }
         PyErr_Format(PyExc_TypeError, "must be real number, not %.50s",
                      op->ob_type->tp_name);
         return -1;