diff options
| -rw-r--r-- | Lib/test/test_softspace.py | 14 | ||||
| -rw-r--r-- | Python/ceval.c | 22 |
2 files changed, 26 insertions, 10 deletions
diff --git a/Lib/test/test_softspace.py b/Lib/test/test_softspace.py new file mode 100644 index 0000000..ba860eb --- /dev/null +++ b/Lib/test/test_softspace.py @@ -0,0 +1,14 @@ +import test_support +import StringIO + +# SF bug 480215: softspace confused in nested print +f = StringIO.StringIO() +class C: + def __str__(self): + print >> f, 'a' + return 'b' + +print >> f, C(), 'c ', 'd\t', 'e' +print >> f, 'f', 'g' +# In 2.2 & earlier, this printed ' a\nbc d\te\nf g\n' +test_support.vereq(f.getvalue(), 'a\nb c d\te\nf g\n') diff --git a/Python/ceval.c b/Python/ceval.c index 5e67b29..97a734c 100644 --- a/Python/ceval.c +++ b/Python/ceval.c @@ -1397,29 +1397,31 @@ eval_frame(PyFrameObject *f) err = -1; } } - if (w != NULL && PyFile_SoftSpace(w, 1)) + if (w != NULL && PyFile_SoftSpace(w, 0)) err = PyFile_WriteString(" ", w); if (err == 0) err = PyFile_WriteObject(v, w, Py_PRINT_RAW); if (err == 0) { - /* XXX move into writeobject() ? */ + /* XXX move into writeobject() ? */ if (PyString_Check(v)) { char *s = PyString_AS_STRING(v); int len = PyString_GET_SIZE(v); - if (len > 0 && - isspace(Py_CHARMASK(s[len-1])) && - s[len-1] != ' ') - PyFile_SoftSpace(w, 0); + if (len == 0 || + !isspace(Py_CHARMASK(s[len-1])) || + s[len-1] == ' ') + PyFile_SoftSpace(w, 1); } #ifdef Py_USING_UNICODE else if (PyUnicode_Check(v)) { Py_UNICODE *s = PyUnicode_AS_UNICODE(v); int len = PyUnicode_GET_SIZE(v); - if (len > 0 && - Py_UNICODE_ISSPACE(s[len-1]) && - s[len-1] != ' ') - PyFile_SoftSpace(w, 0); + if (len == 0 || + !Py_UNICODE_ISSPACE(s[len-1]) || + s[len-1] == ' ') + PyFile_SoftSpace(w, 1); } + else + PyFile_SoftSpace(w, 1); #endif } Py_DECREF(v); |