From: Raymond Hettinger <python@rcn.com>
Date: Wed, 4 Dec 2002 07:32:25 +0000 (+0000)
Subject: Replace BadInternalCall with TypeError.  Add a test case.  Fix whitespace.
X-Git-Tag: v2.3c1~3130
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=b02bb5ed0a45571c3be195cd053bb28e408a99cd;p=thirdparty%2FPython%2Fcpython.git

Replace BadInternalCall with TypeError.  Add a test case.  Fix whitespace.

Just van Rossum showed a weird, but clever way for pure python code to
trigger the BadInternalCall.  The C code had assumed that calling a class
constructor would return an instance of that class; however, classes that
abuse __new__ can invalidate that assumption.
---

diff --git a/Lib/test/test_types.py b/Lib/test/test_types.py
index 9cfc6806adb4..1e982d14f94e 100644
--- a/Lib/test/test_types.py
+++ b/Lib/test/test_types.py
@@ -558,6 +558,13 @@ if type(dictlike.fromkeys('a')) is not dictlike:
     raise TestFailed, 'dictsubclass.fromkeys created wrong type'
 if type(dictlike().fromkeys('a')) is not dictlike:
     raise TestFailed, 'dictsubclass.fromkeys created wrong type'
+from UserDict import UserDict
+class mydict(dict):
+    def __new__(cls, *args, **kwargs):
+        return UserDict(*args, **kwargs)
+try: mydict.fromkeys('a b c'.split())
+except TypeError: pass
+else: raise TestFailed, 'dict.fromkeys() failed to detect non-dict class.'
 # dict.copy()
 d = {1:1, 2:2, 3:3}
 if d.copy() != {1:1, 2:2, 3:3}: raise TestFailed, 'dict copy'
diff --git a/Objects/dictobject.c b/Objects/dictobject.c
index eca86770a657..d3603f00458b 100644
--- a/Objects/dictobject.c
+++ b/Objects/dictobject.c
@@ -973,15 +973,16 @@ dict_fromkeys(PyObject *mp, PyObject *args)
 	PyObject *cls;
 	int status;
 
-        if (!PyArg_ParseTuple(args, "OO|O:fromkeys", &cls, &seq, &value))
+	if (!PyArg_ParseTuple(args, "OO|O:fromkeys", &cls, &seq, &value))
 		return NULL;
 
 	d = PyObject_CallObject(cls, NULL);
 	if (d == NULL)
 		return NULL;
 	if (!PyDict_Check(d)) {
-		PyErr_BadInternalCall();
 		Py_DECREF(d);
+		PyErr_SetString(PyExc_TypeError,
+			"class constructor must return a subclass of dict");
 		return NULL;
 	}