changeset:   89713:2514a577c7cb
branch:      3.4
user:        Benjamin Peterson <benjamin@python.org>
date:        Sun Feb 16 13:49:16 2014 -0500
files:       Lib/test/test_descr.py Misc/NEWS Objects/typeobject.c
description:
look up __getnewargs__ and __getnewargs_ex__ on the object type (#16251)


diff -r cd39d4cab680 -r 2514a577c7cb Lib/test/test_descr.py
--- a/Lib/test/test_descr.py	Sat Feb 15 09:14:54 2014 +1000
+++ b/Lib/test/test_descr.py	Sun Feb 16 13:49:16 2014 -0500
@@ -4701,6 +4701,20 @@
         for proto in protocols:
             self._check_reduce(proto, obj, listitems=list(obj))
 
+    def test_special_method_lookup(self):
+        protocols = range(pickle.HIGHEST_PROTOCOL + 1)
+        class Picky:
+            def __getstate__(self):
+                return {}
+
+            def __getattr__(self, attr):
+                if attr in ("__getnewargs__", "__getnewargs_ex__"):
+                    raise AssertionError(attr)
+                return None
+        for protocol in protocols:
+            state = {} if protocol >= 2 else None
+            self._check_reduce(protocol, Picky(), state=state)
+
     def _assert_is_copy(self, obj, objcopy, msg=None):
         """Utility method to verify if two objects are copies of each others.
         """
diff -r cd39d4cab680 -r 2514a577c7cb Misc/NEWS
--- a/Misc/NEWS	Sat Feb 15 09:14:54 2014 +1000
+++ b/Misc/NEWS	Sun Feb 16 13:49:16 2014 -0500
@@ -10,6 +10,9 @@
 Core and Builtins
 -----------------
 
+- Issue #20261: In pickle, lookup __getnewargs__ and __getnewargs_ex__ on the
+  type of the object.
+
 - Issue #20619: Give the AST nodes of keyword-only arguments a column and line
   number.
 
diff -r cd39d4cab680 -r 2514a577c7cb Objects/typeobject.c
--- a/Objects/typeobject.c	Sat Feb 15 09:14:54 2014 +1000
+++ b/Objects/typeobject.c	Sun Feb 16 13:49:16 2014 -0500
@@ -3719,7 +3719,7 @@
 
     /* We first attempt to fetch the arguments for __new__ by calling
        __getnewargs_ex__ on the object. */
-    getnewargs_ex = _PyObject_GetAttrId(obj, &PyId___getnewargs_ex__);
+    getnewargs_ex = _PyObject_LookupSpecial(obj, &PyId___getnewargs_ex__);
     if (getnewargs_ex != NULL) {
         PyObject *newargs = PyObject_CallObject(getnewargs_ex, NULL);
         Py_DECREF(getnewargs_ex);
@@ -3766,16 +3766,13 @@
             return -1;
         }
         return 0;
-    } else {
-        if (!PyErr_ExceptionMatches(PyExc_AttributeError)) {
-            return -1;
-        }
-        PyErr_Clear();
+    } else if (PyErr_Occurred()) {
+        return -1;
     }
 
     /* The object does not have __getnewargs_ex__ so we fallback on using
        __getnewargs__ instead. */
-    getnewargs = _PyObject_GetAttrId(obj, &PyId___getnewargs__);
+    getnewargs = _PyObject_LookupSpecial(obj, &PyId___getnewargs__);
     if (getnewargs != NULL) {
         *args = PyObject_CallObject(getnewargs, NULL);
         Py_DECREF(getnewargs);
@@ -3791,11 +3788,8 @@
         }
         *kwargs = NULL;
         return 0;
-    } else {
-        if (!PyErr_ExceptionMatches(PyExc_AttributeError)) {
-            return -1;
-        }
-        PyErr_Clear();
+    } else if (PyErr_Occurred()) {
+        return -1;
     }
 
     /* The object does not have __getnewargs_ex__ and __getnewargs__. This may