diff options
| -rw-r--r-- | Objects/stringobject.c | 11 | ||||
| -rw-r--r-- | Objects/unicodeobject.c | 11 |
2 files changed, 18 insertions, 4 deletions
diff --git a/Objects/stringobject.c b/Objects/stringobject.c index 1af96b1..6a0eece 100644 --- a/Objects/stringobject.c +++ b/Objects/stringobject.c @@ -2401,8 +2401,15 @@ string_zfill(PyStringObject *self, PyObject *args) return NULL; if (PyString_GET_SIZE(self) >= width) { - Py_INCREF(self); - return (PyObject*) self; + if (PyString_CheckExact(self)) { + Py_INCREF(self); + return (PyObject*) self; + } + else + return PyString_FromStringAndSize( + PyString_AS_STRING(self), + PyString_GET_SIZE(self) + ); } fill = width - PyString_GET_SIZE(self); diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c index 361612b..29ba2e4 100644 --- a/Objects/unicodeobject.c +++ b/Objects/unicodeobject.c @@ -4841,8 +4841,15 @@ unicode_zfill(PyUnicodeObject *self, PyObject *args) return NULL; if (self->length >= width) { - Py_INCREF(self); - return (PyObject*) self; + if (PyUnicode_CheckExact(self)) { + Py_INCREF(self); + return (PyObject*) self; + } + else + return PyUnicode_FromUnicode( + PyUnicode_AS_UNICODE(self), + PyUnicode_GET_SIZE(self) + ); } fill = width - self->length; |