Georg Brandl added the comment:
Yes, that makes sense.
New patch attached, copying the dict and using PyDict_Next.
__________________________________
Tracker <[EMAIL PROTECTED]>
<http://bugs.python.org/issue1071>
__________________________________
Index: Doc/library/stdtypes.rst
===================================================================
--- Doc/library/stdtypes.rst (Revision 57875)
+++ Doc/library/stdtypes.rst (Arbeitskopie)
@@ -948,9 +948,10 @@
.. method:: str.translate(map)
Return a copy of the *s* where all characters have been mapped through the
- *map* which must be a mapping of Unicode ordinals (integers) to Unicode
- ordinals, strings or ``None``. Unmapped characters are left
- untouched. Characters mapped to ``None`` are deleted.
+ *map* which must be a dictionary of characters (strings of length 1) or
+ Unicode ordinals (integers) to Unicode ordinals, strings or ``None``.
+ Unmapped characters are left untouched. Characters mapped to ``None`` are
+ deleted.
.. note::
Index: Objects/unicodeobject.c
===================================================================
--- Objects/unicodeobject.c (Revision 57871)
+++ Objects/unicodeobject.c (Arbeitskopie)
@@ -7763,10 +7763,54 @@
static PyObject*
unicode_translate(PyUnicodeObject *self, PyObject *table)
{
- return PyUnicode_TranslateCharmap(self->str,
- self->length,
- table,
- "ignore");
+ PyObject *newtable = NULL;
+ Py_ssize_t i = 0;
+ PyObject *key, *value, *result;
+
+ if (!PyDict_Check(table)) {
+ PyErr_SetString(PyExc_TypeError, "translate argument must be a dict");
+ return NULL;
+ }
+ /* fixup the table -- allow size-1 string keys instead of only int keys */
+ newtable = PyDict_Copy(table);
+ if (!newtable) return NULL;
+ while (PyDict_Next(table, &i, &key, &value)) {
+ if (PyUnicode_Check(key)) {
+ /* convert string keys to integer keys */
+ PyObject *newkey;
+ int res;
+ if (PyUnicode_GET_SIZE(key) != 1) {
+ PyErr_SetString(PyExc_ValueError, "string items in translate "
+ "table must be 1 element long");
+ goto err;
+ }
+ newkey = PyInt_FromLong(PyUnicode_AS_UNICODE(key)[0]);
+ if (!newkey)
+ goto err;
+ res = PyDict_SetItem(newtable, newkey, value);
+ Py_DECREF(newkey);
+ if (res < 0)
+ goto err;
+ } else if (PyInt_Check(key)) {
+ /* just keep integer keys */
+ if (PyDict_SetItem(newtable, key, value) < 0)
+ goto err;
+ } else {
+ PyErr_SetString(PyExc_TypeError, "items in translate table must be
"
+ "strings or integers");
+ goto err;
+ }
+ }
+
+ result = PyUnicode_TranslateCharmap(self->str,
+ self->length,
+ newtable,
+ "ignore");
+ Py_DECREF(newtable);
+ return result;
+ err:
+ Py_DECREF(newtable);
+ return NULL;
}
PyDoc_STRVAR(upper__doc__,
_______________________________________________
Python-bugs-list mailing list
Unsubscribe:
http://mail.python.org/mailman/options/python-bugs-list/archive%40mail-archive.com
