Fix int() and long() to repr() their argument when formatting the exception,

to avoid confusing situations like:

>>> int("")
ValueError: invalid literal for int():
>>> int("2\n\n2")
ValueError: invalid literal for int(): 2

2

Also report the base used, to avoid:

ValueError: invalid literal for int(): 2

They now report:

>>> int("")
ValueError: invalid literal for int() with base 10: ''
>>> int("2\n\n2")
ValueError: invalid literal for int() with base 10: '2\n\n2'
>>> int("2", 2)
ValueError: invalid literal for int() with base 2: '2'

(Reporting the base could be avoided when base is 10, which is the default,
but hrm.) Another effect of these changes is that the errormessage can be
longer; before, it was cut off at about 250 characters. Now, it can be up to
four times as long, as the unrepr'ed string is cut off at 200 characters,
instead.

No tests were added or changed, since testing for exact errormsgs is (pardon
the pun) somewhat errorprone, and I consider not testing the exact text
preferable. The actually changed code is tested frequent enough in the
test_builtin test as it is (120 runs for each of ints and longs.)
This commit is contained in:
Thomas Wouters 2006-04-11 23:50:33 +00:00
parent 502348d010
commit 9cb28bea04
2 changed files with 28 additions and 6 deletions

View File

@ -335,7 +335,8 @@ PyInt_FromString(char *s, char **pend, int base)
{ {
char *end; char *end;
long x; long x;
char buffer[256]; /* For errors */ Py_ssize_t slen;
PyObject *sobj, *srepr;
if ((base != 0 && base < 2) || base > 36) { if ((base != 0 && base < 2) || base > 36) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
@ -359,9 +360,18 @@ PyInt_FromString(char *s, char **pend, int base)
end++; end++;
if (*end != '\0') { if (*end != '\0') {
bad: bad:
PyOS_snprintf(buffer, sizeof(buffer), slen = strlen(s) < 200 ? strlen(s) : 200;
"invalid literal for int(): %.200s", s); sobj = PyString_FromStringAndSize(s, slen);
PyErr_SetString(PyExc_ValueError, buffer); if (sobj == NULL)
return NULL;
srepr = PyObject_Repr(sobj);
Py_DECREF(sobj);
if (srepr == NULL)
return NULL;
PyErr_Format(PyExc_ValueError,
"invalid literal for int() with base %d: %s",
base, PyString_AS_STRING(srepr));
Py_DECREF(srepr);
return NULL; return NULL;
} }
else if (errno != 0) else if (errno != 0)

View File

@ -1400,6 +1400,8 @@ PyLong_FromString(char *str, char **pend, int base)
int sign = 1; int sign = 1;
char *start, *orig_str = str; char *start, *orig_str = str;
PyLongObject *z; PyLongObject *z;
PyObject *strobj, *strrepr;
Py_ssize_t slen;
if ((base != 0 && base < 2) || base > 36) { if ((base != 0 && base < 2) || base > 36) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
@ -1465,9 +1467,19 @@ PyLong_FromString(char *str, char **pend, int base)
return (PyObject *) z; return (PyObject *) z;
onError: onError:
PyErr_Format(PyExc_ValueError,
"invalid literal for long(): %.200s", orig_str);
Py_XDECREF(z); Py_XDECREF(z);
slen = strlen(orig_str) < 200 ? strlen(orig_str) : 200;
strobj = PyString_FromStringAndSize(orig_str, slen);
if (strobj == NULL)
return NULL;
strrepr = PyObject_Repr(strobj);
Py_DECREF(strobj);
if (strrepr == NULL)
return NULL;
PyErr_Format(PyExc_ValueError,
"invalid literal for long() with base %d: %s",
base, PyString_AS_STRING(strrepr));
Py_DECREF(strrepr);
return NULL; return NULL;
} }