changeset:   105387:6b8f7d1e2ba4
branch:      3.6
parent:      105382:52038705827d
user:        Serhiy Storchaka <storchaka@gmail.com>
date:        Tue Nov 29 09:54:17 2016 +0200
files:       Lib/test/test_subclassinit.py Misc/NEWS Objects/typeobject.c
description:
Issue #28797: Modifying the class __dict__ inside the __set_name__ method of
 a descriptor that is used inside that class no longer prevents calling the
 __set_name__ method of other descriptors.


diff -r 52038705827d -r 6b8f7d1e2ba4 Lib/test/test_subclassinit.py
--- a/Lib/test/test_subclassinit.py	Mon Nov 28 11:45:36 2016 -0500
+++ b/Lib/test/test_subclassinit.py	Tue Nov 29 09:54:17 2016 +0200
@@ -198,6 +198,22 @@
         self.assertIs(B.meta_owner, B)
         self.assertEqual(B.name, 'd')
 
+    def test_set_name_modifying_dict(self):
+        notified = []
+        class Descriptor:
+            def __set_name__(self, owner, name):
+                setattr(owner, name + 'x', None)
+                notified.append(name)
+
+        class A:
+            a = Descriptor()
+            b = Descriptor()
+            c = Descriptor()
+            d = Descriptor()
+            e = Descriptor()
+
+        self.assertCountEqual(notified, ['a', 'b', 'c', 'd', 'e'])
+
     def test_errors(self):
         class MyMeta(type):
             pass
diff -r 52038705827d -r 6b8f7d1e2ba4 Misc/NEWS
--- a/Misc/NEWS	Mon Nov 28 11:45:36 2016 -0500
+++ b/Misc/NEWS	Tue Nov 29 09:54:17 2016 +0200
@@ -10,6 +10,10 @@
 Core and Builtins
 -----------------
 
+- Issue #28797: Modifying the class __dict__ inside the __set_name__ method of
+  a descriptor that is used inside that class no longer prevents calling the
+  __set_name__ method of other descriptors.
+
 - Issue #28782: Fix a bug in the implementation ``yield from`` when checking
   if the next instruction is YIELD_FROM. Regression introduced by WORDCODE
   (issue #26647).
diff -r 52038705827d -r 6b8f7d1e2ba4 Objects/typeobject.c
--- a/Objects/typeobject.c	Mon Nov 28 11:45:36 2016 -0500
+++ b/Objects/typeobject.c	Tue Nov 29 09:54:17 2016 +0200
@@ -7004,10 +7004,14 @@
 static int
 set_names(PyTypeObject *type)
 {
-    PyObject *key, *value, *set_name, *tmp;
+    PyObject *names_to_set, *key, *value, *set_name, *tmp;
     Py_ssize_t i = 0;
 
-    while (PyDict_Next(type->tp_dict, &i, &key, &value)) {
+    names_to_set = PyDict_Copy(type->tp_dict);
+    if (names_to_set == NULL)
+        return -1;
+
+    while (PyDict_Next(names_to_set, &i, &key, &value)) {
         set_name = lookup_maybe(value, &PyId___set_name__);
         if (set_name != NULL) {
             tmp = PyObject_CallFunctionObjArgs(set_name, type, key, NULL);
@@ -7017,15 +7021,19 @@
                     "Error calling __set_name__ on '%.100s' instance %R "
                     "in '%.100s'",
                     value->ob_type->tp_name, key, type->tp_name);
+                Py_DECREF(names_to_set);
                 return -1;
             }
             else
                 Py_DECREF(tmp);
         }
-        else if (PyErr_Occurred())
+        else if (PyErr_Occurred()) {
+            Py_DECREF(names_to_set);
             return -1;
-    }
-
+        }
+    }
+
+    Py_DECREF(names_to_set);
     return 0;
 }