diff options
author | Walter Dörwald <walter@livinglogic.de> | 2007-06-05 20:22:04 +0000 |
---|---|---|
committer | Walter Dörwald <walter@livinglogic.de> | 2007-06-05 20:22:04 +0000 |
commit | 446b194d475c6cf010eeb337dba01619d0588c30 (patch) | |
tree | bc5d0e6b4e334cfcdc71054076ab52baa440e214 /Python/sysmodule.c | |
parent | cf22627fff4555981400f83d0bd52f3518b524e3 (diff) | |
download | cpython-446b194d475c6cf010eeb337dba01619d0588c30.tar.gz |
Change sys.intern() so that unicode strings can be
interned too. Add a test for this.
Diffstat (limited to 'Python/sysmodule.c')
-rw-r--r-- | Python/sysmodule.c | 19 |
1 files changed, 13 insertions, 6 deletions
diff --git a/Python/sysmodule.c b/Python/sysmodule.c index 17568349d5..a8bb918f5a 100644 --- a/Python/sysmodule.c +++ b/Python/sysmodule.c @@ -266,14 +266,21 @@ sys_intern(PyObject *self, PyObject *args) PyObject *s; if (!PyArg_ParseTuple(args, "S:intern", &s)) return NULL; - if (!PyString_CheckExact(s)) { - PyErr_SetString(PyExc_TypeError, - "can't intern subclass of string"); + if (PyString_CheckExact(s)) { + Py_INCREF(s); + PyString_InternInPlace(&s); + return s; + } + else if (PyUnicode_CheckExact(s)) { + Py_INCREF(s); + PyUnicode_InternInPlace(&s); + return s; + } + else { + PyErr_Format(PyExc_TypeError, + "can't intern %.400s", s->ob_type->tp_name); return NULL; } - Py_INCREF(s); - PyString_InternInPlace(&s); - return s; } PyDoc_STRVAR(intern_doc, |