correctly lookup __dir__

This commit is contained in:
Benjamin Peterson 2011-05-23 16:11:05 -05:00
parent 2cca057284
commit 7963a35b41
3 changed files with 10 additions and 5 deletions

View File

@ -1595,6 +1595,7 @@ order (MRO) for bases """
# probably not worth it.
# ("__enter__", run_context, iden),
# ("__exit__", run_context, iden),
("__dir__", dir, empty_seq, set(), {}),
]
class Checker(object):

View File

@ -10,6 +10,9 @@ What's New in Python 3.1.4?
Core and Builtins
-----------------
- Correct lookup of __dir__ on objects. Among other things, this causes errors
besides AttributeError found on lookup to be propagated.
- Issue #12060: Use sig_atomic_t type and volatile keyword in the signal
module. Patch written by Charles-François Natali.

View File

@ -1348,14 +1348,15 @@ error:
static PyObject *
_dir_object(PyObject *obj)
{
PyObject * result = NULL;
PyObject * dirfunc = PyObject_GetAttrString((PyObject*)obj->ob_type,
"__dir__");
PyObject *result = NULL;
static PyObject *dir_str = NULL;
PyObject *dirfunc = _PyObject_LookupSpecial(obj, "__dir__", &dir_str);
assert(obj);
if (dirfunc == NULL) {
if (PyErr_Occurred())
return NULL;
/* use default implementation */
PyErr_Clear();
if (PyModule_Check(obj))
result = _specialized_dir_module(obj);
else if (PyType_Check(obj))
@ -1365,7 +1366,7 @@ _dir_object(PyObject *obj)
}
else {
/* use __dir__ */
result = PyObject_CallFunctionObjArgs(dirfunc, obj, NULL);
result = PyObject_CallFunctionObjArgs(dirfunc, NULL);
Py_DECREF(dirfunc);
if (result == NULL)
return NULL;