changeset:   105323:3aafb232f2db
branch:      3.6
parent:      105321:e0f0211d314d
user:        Raymond Hettinger <python@rcn.com>
date:        Mon Nov 21 17:24:23 2016 -0800
files:       Lib/test/test_with.py Misc/NEWS Python/ceval.c
description:
Issue #27100:  With statement reports missing __enter__ before __exit__.  (Contributed by Jonathan Ellington.)


diff -r e0f0211d314d -r 3aafb232f2db Lib/test/test_with.py
--- a/Lib/test/test_with.py	Mon Nov 21 16:59:04 2016 -0800
+++ b/Lib/test/test_with.py	Mon Nov 21 17:24:23 2016 -0800
@@ -109,7 +109,7 @@
             with foo: pass
         self.assertRaises(NameError, fooNotDeclared)
 
-    def testEnterAttributeError(self):
+    def testEnterAttributeError1(self):
         class LacksEnter(object):
             def __exit__(self, type, value, traceback):
                 pass
@@ -117,7 +117,16 @@
         def fooLacksEnter():
             foo = LacksEnter()
             with foo: pass
-        self.assertRaises(AttributeError, fooLacksEnter)
+        self.assertRaisesRegexp(AttributeError, '__enter__', fooLacksEnter)
+
+    def testEnterAttributeError2(self):
+        class LacksEnterAndExit(object):
+            pass
+
+        def fooLacksEnterAndExit():
+            foo = LacksEnterAndExit()
+            with foo: pass
+        self.assertRaisesRegexp(AttributeError, '__enter__', fooLacksEnterAndExit)
 
     def testExitAttributeError(self):
         class LacksExit(object):
@@ -127,7 +136,7 @@
         def fooLacksExit():
             foo = LacksExit()
             with foo: pass
-        self.assertRaises(AttributeError, fooLacksExit)
+        self.assertRaisesRegexp(AttributeError, '__exit__', fooLacksExit)
 
     def assertRaisesSyntaxError(self, codestr):
         def shouldRaiseSyntaxError(s):
diff -r e0f0211d314d -r 3aafb232f2db Misc/NEWS
--- a/Misc/NEWS	Mon Nov 21 16:59:04 2016 -0800
+++ b/Misc/NEWS	Mon Nov 21 17:24:23 2016 -0800
@@ -12,6 +12,10 @@
 
 - Issue #28532: Show sys.version when -V option is supplied twice.
 
+- Issue #27100: The with-statement now checks for __enter__ before it
+  checks for __exit__.  This gives less confusing error messages when
+  both methods are missing. Patch by Jonathan Ellington.
+
 - Issue #28746: Fix the set_inheritable() file descriptor method on platforms
   that do not have the ioctl FIOCLEX and FIONCLEX commands.
 
diff -r e0f0211d314d -r 3aafb232f2db Python/ceval.c
--- a/Python/ceval.c	Mon Nov 21 16:59:04 2016 -0800
+++ b/Python/ceval.c	Mon Nov 21 17:24:23 2016 -0800
@@ -3133,15 +3133,15 @@
             _Py_IDENTIFIER(__exit__);
             _Py_IDENTIFIER(__enter__);
             PyObject *mgr = TOP();
-            PyObject *exit = special_lookup(mgr, &PyId___exit__), *enter;
+            PyObject *enter = special_lookup(mgr, &PyId___enter__), *exit;
             PyObject *res;
+            if (enter == NULL)
+                goto error;
+            exit = special_lookup(mgr, &PyId___exit__);
             if (exit == NULL)
                 goto error;
             SET_TOP(exit);
-            enter = special_lookup(mgr, &PyId___enter__);
             Py_DECREF(mgr);
-            if (enter == NULL)
-                goto error;
             res = PyObject_CallFunctionObjArgs(enter, NULL);
             Py_DECREF(enter);
             if (res == NULL)