From: Benjamin Peterson <benjamin@python.org>
Date: Sat, 9 May 2009 17:23:03 +0000 (+0000)
Subject: *sigh* deal with instances correctly
X-Git-Tag: v2.7a1~1227
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=784d45506670ce86d857ee118f0f64a2b667d386;p=python

*sigh* deal with instances correctly
---

diff --git a/Lib/test/test_enumerate.py b/Lib/test/test_enumerate.py
index 074179cfe0..d8a3251885 100644
--- a/Lib/test/test_enumerate.py
+++ b/Lib/test/test_enumerate.py
@@ -141,6 +141,12 @@ class TestReversed(unittest.TestCase):
         # don't allow keyword arguments
         self.assertRaises(TypeError, reversed, [], a=1)
 
+    def test_class_class(self):
+        class A:
+            def __reversed__(self):
+                return [2, 1]
+        self.assertEqual(list(reversed(A())), [2, 1])
+
     def test_xrange_optimization(self):
         x = xrange(1)
         self.assertEqual(type(reversed(x)), type(iter(x)))
diff --git a/Objects/enumobject.c b/Objects/enumobject.c
index 12a226c748..65d46334cf 100644
--- a/Objects/enumobject.c
+++ b/Objects/enumobject.c
@@ -232,7 +232,18 @@ reversed_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 	if (!PyArg_UnpackTuple(args, "reversed", 1, 1, &seq) )
 		return NULL;
 
-	reversed_meth = _PyObject_LookupSpecial(seq, "__reversed__", &reversed_cache);
+	if (PyInstance_Check(seq)) {
+		reversed_meth = PyObject_GetAttrString(seq, "__reversed__");
+		if (reversed_meth == NULL) {
+			if (PyErr_ExceptionMatches(PyExc_AttributeError))
+				PyErr_Clear();
+			else
+				return NULL;
+		}
+	}
+	else
+		reversed_meth = _PyObject_LookupSpecial(seq, "__reversed__",
+							&reversed_cache);
 	if (reversed_meth != NULL) {
 		PyObject *res = PyObject_CallFunctionObjArgs(reversed_meth, NULL);
 		Py_DECREF(reversed_meth);