1990-10-14 09:07:46 -03:00
|
|
|
|
/* String object implementation */
|
|
|
|
|
|
2006-04-14 06:08:42 -03:00
|
|
|
|
#define PY_SSIZE_T_CLEAN
|
2006-05-26 13:32:42 -03:00
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
#include "Python.h"
|
1990-10-14 09:07:46 -03:00
|
|
|
|
|
1994-08-30 05:19:36 -03:00
|
|
|
|
#include <ctype.h>
|
|
|
|
|
|
1993-10-22 09:04:32 -03:00
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
int null_strings, one_strings;
|
|
|
|
|
#endif
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PyStringObject *characters[UCHAR_MAX + 1];
|
|
|
|
|
static PyStringObject *nullstring;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
|
2002-08-19 18:43:18 -03:00
|
|
|
|
/* This dictionary holds all interned strings. Note that references to
|
|
|
|
|
strings in this dictionary are *not* counted in the string's ob_refcnt.
|
|
|
|
|
When the interned string reaches a refcnt of 0 the string deallocation
|
|
|
|
|
function will delete the reference from this dictionary.
|
|
|
|
|
|
2006-03-16 23:29:34 -04:00
|
|
|
|
Another way to look at this is that to say that the actual reference
|
2002-08-19 18:43:18 -03:00
|
|
|
|
count of a string is: s->ob_refcnt + (s->ob_sstate?2:0)
|
|
|
|
|
*/
|
|
|
|
|
static PyObject *interned;
|
|
|
|
|
|
1993-10-22 09:04:32 -03:00
|
|
|
|
/*
|
2002-04-15 10:48:52 -03:00
|
|
|
|
For both PyString_FromString() and PyString_FromStringAndSize(), the
|
|
|
|
|
parameter `size' denotes number of characters to allocate, not counting any
|
2002-01-16 06:53:24 -04:00
|
|
|
|
null terminating character.
|
|
|
|
|
|
2002-04-15 10:48:52 -03:00
|
|
|
|
For PyString_FromString(), the parameter `str' points to a null-terminated
|
2002-01-16 06:53:24 -04:00
|
|
|
|
string containing exactly `size' bytes.
|
|
|
|
|
|
2002-04-15 10:48:52 -03:00
|
|
|
|
For PyString_FromStringAndSize(), the parameter the parameter `str' is
|
|
|
|
|
either NULL or else points to a string containing at least `size' bytes.
|
|
|
|
|
For PyString_FromStringAndSize(), the string in the `str' parameter does
|
|
|
|
|
not have to be null-terminated. (Therefore it is safe to construct a
|
|
|
|
|
substring by calling `PyString_FromStringAndSize(origstring, substrlen)'.)
|
|
|
|
|
If `str' is NULL then PyString_FromStringAndSize() will allocate `size+1'
|
|
|
|
|
bytes (setting the last byte to the null terminating character) and you can
|
|
|
|
|
fill in the data yourself. If `str' is non-NULL then the resulting
|
|
|
|
|
PyString object must be treated as immutable and you must not fill in nor
|
|
|
|
|
alter the data yourself, since the strings may be shared.
|
|
|
|
|
|
|
|
|
|
The PyObject member `op->ob_size', which denotes the number of "extra
|
|
|
|
|
items" in a variable-size object, will contain the number of bytes
|
|
|
|
|
allocated for string data, not counting the null terminating character. It
|
|
|
|
|
is therefore equal to the equal to the `size' parameter (for
|
|
|
|
|
PyString_FromStringAndSize()) or the length of the string in the `str'
|
|
|
|
|
parameter (for PyString_FromString()).
|
1993-10-22 09:04:32 -03:00
|
|
|
|
*/
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *
|
2006-02-15 13:27:45 -04:00
|
|
|
|
PyString_FromStringAndSize(const char *str, Py_ssize_t size)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2001-05-09 04:37:07 -03:00
|
|
|
|
register PyStringObject *op;
|
2005-01-31 13:09:25 -04:00
|
|
|
|
assert(size >= 0);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
if (size == 0 && (op = nullstring) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
null_strings++;
|
|
|
|
|
#endif
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (size == 1 && str != NULL &&
|
|
|
|
|
(op = characters[*str & UCHAR_MAX]) != NULL)
|
|
|
|
|
{
|
1993-10-22 09:04:32 -03:00
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
2000-05-03 20:44:39 -03:00
|
|
|
|
|
2002-08-19 16:26:42 -03:00
|
|
|
|
/* Inline PyObject_NewVar */
|
2004-06-27 14:24:49 -03:00
|
|
|
|
op = (PyStringObject *)PyObject_MALLOC(sizeof(PyStringObject) + size);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
if (op == NULL)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-03 20:44:39 -03:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
op->ob_shash = -1;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
op->ob_sstate = SSTATE_NOT_INTERNED;
|
1990-10-21 19:15:08 -03:00
|
|
|
|
if (str != NULL)
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(op->ob_sval, str, size);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
op->ob_sval[size] = '\0';
|
2002-03-30 06:06:07 -04:00
|
|
|
|
/* share short strings */
|
1993-10-22 09:04:32 -03:00
|
|
|
|
if (size == 0) {
|
2001-05-09 04:37:07 -03:00
|
|
|
|
PyObject *t = (PyObject *)op;
|
|
|
|
|
PyString_InternInPlace(&t);
|
2001-05-09 05:43:21 -03:00
|
|
|
|
op = (PyStringObject *)t;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
nullstring = op;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
} else if (size == 1 && str != NULL) {
|
2001-05-09 04:37:07 -03:00
|
|
|
|
PyObject *t = (PyObject *)op;
|
|
|
|
|
PyString_InternInPlace(&t);
|
2001-05-09 05:43:21 -03:00
|
|
|
|
op = (PyStringObject *)t;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
characters[*str & UCHAR_MAX] = op;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_FromString(const char *str)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2001-12-06 16:29:32 -04:00
|
|
|
|
register size_t size;
|
2001-05-09 04:37:07 -03:00
|
|
|
|
register PyStringObject *op;
|
2001-12-06 16:29:32 -04:00
|
|
|
|
|
|
|
|
|
assert(str != NULL);
|
|
|
|
|
size = strlen(str);
|
2006-04-13 04:22:51 -03:00
|
|
|
|
if (size > PY_SSIZE_T_MAX) {
|
2000-06-30 07:29:57 -03:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"string is too long for a Python string");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1993-10-22 09:04:32 -03:00
|
|
|
|
if (size == 0 && (op = nullstring) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
null_strings++;
|
|
|
|
|
#endif
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
|
|
|
|
if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
2000-05-03 20:44:39 -03:00
|
|
|
|
|
2002-08-19 16:26:42 -03:00
|
|
|
|
/* Inline PyObject_NewVar */
|
2004-06-27 14:24:49 -03:00
|
|
|
|
op = (PyStringObject *)PyObject_MALLOC(sizeof(PyStringObject) + size);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
if (op == NULL)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-03 20:44:39 -03:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
op->ob_shash = -1;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
op->ob_sstate = SSTATE_NOT_INTERNED;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(op->ob_sval, str, size+1);
|
2002-03-30 06:06:07 -04:00
|
|
|
|
/* share short strings */
|
1993-10-22 09:04:32 -03:00
|
|
|
|
if (size == 0) {
|
2001-05-09 04:37:07 -03:00
|
|
|
|
PyObject *t = (PyObject *)op;
|
|
|
|
|
PyString_InternInPlace(&t);
|
2001-05-09 05:43:21 -03:00
|
|
|
|
op = (PyStringObject *)t;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
nullstring = op;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
} else if (size == 1) {
|
2001-05-09 04:37:07 -03:00
|
|
|
|
PyObject *t = (PyObject *)op;
|
|
|
|
|
PyString_InternInPlace(&t);
|
2001-05-09 05:43:21 -03:00
|
|
|
|
op = (PyStringObject *)t;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
characters[*str & UCHAR_MAX] = op;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
PyObject *
|
|
|
|
|
PyString_FromFormatV(const char *format, va_list vargs)
|
|
|
|
|
{
|
2001-10-02 18:32:07 -03:00
|
|
|
|
va_list count;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t n = 0;
|
2001-08-24 15:32:06 -03:00
|
|
|
|
const char* f;
|
|
|
|
|
char *s;
|
|
|
|
|
PyObject* string;
|
|
|
|
|
|
2001-10-02 18:32:07 -03:00
|
|
|
|
#ifdef VA_LIST_IS_ARRAY
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(count, vargs, sizeof(va_list));
|
2002-07-28 07:23:27 -03:00
|
|
|
|
#else
|
|
|
|
|
#ifdef __va_copy
|
|
|
|
|
__va_copy(count, vargs);
|
2001-10-02 18:32:07 -03:00
|
|
|
|
#else
|
|
|
|
|
count = vargs;
|
2002-07-28 07:23:27 -03:00
|
|
|
|
#endif
|
2001-10-02 18:32:07 -03:00
|
|
|
|
#endif
|
2001-08-24 15:32:06 -03:00
|
|
|
|
/* step 1: figure out how large a buffer we need */
|
|
|
|
|
for (f = format; *f; f++) {
|
|
|
|
|
if (*f == '%') {
|
|
|
|
|
const char* p = f;
|
|
|
|
|
while (*++f && *f != '%' && !isalpha(Py_CHARMASK(*f)))
|
|
|
|
|
;
|
|
|
|
|
|
2006-05-13 20:28:20 -03:00
|
|
|
|
/* skip the 'l' or 'z' in {%ld, %zd, %lu, %zu} since
|
|
|
|
|
* they don't affect the amount of space we reserve.
|
|
|
|
|
*/
|
|
|
|
|
if ((*f == 'l' || *f == 'z') &&
|
|
|
|
|
(f[1] == 'd' || f[1] == 'u'))
|
2006-03-16 23:29:34 -04:00
|
|
|
|
++f;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
switch (*f) {
|
|
|
|
|
case 'c':
|
|
|
|
|
(void)va_arg(count, int);
|
|
|
|
|
/* fall through... */
|
|
|
|
|
case '%':
|
|
|
|
|
n++;
|
|
|
|
|
break;
|
2006-05-13 20:28:20 -03:00
|
|
|
|
case 'd': case 'u': case 'i': case 'x':
|
2001-08-24 15:32:06 -03:00
|
|
|
|
(void) va_arg(count, int);
|
2001-12-02 21:55:38 -04:00
|
|
|
|
/* 20 bytes is enough to hold a 64-bit
|
|
|
|
|
integer. Decimal takes the most space.
|
|
|
|
|
This isn't enough for octal. */
|
2001-08-24 15:32:06 -03:00
|
|
|
|
n += 20;
|
|
|
|
|
break;
|
|
|
|
|
case 's':
|
|
|
|
|
s = va_arg(count, char*);
|
|
|
|
|
n += strlen(s);
|
|
|
|
|
break;
|
|
|
|
|
case 'p':
|
|
|
|
|
(void) va_arg(count, int);
|
|
|
|
|
/* maximum 64-bit pointer representation:
|
|
|
|
|
* 0xffffffffffffffff
|
|
|
|
|
* so 19 characters is enough.
|
2001-12-02 21:55:38 -04:00
|
|
|
|
* XXX I count 18 -- what's the extra for?
|
2001-08-24 15:32:06 -03:00
|
|
|
|
*/
|
|
|
|
|
n += 19;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
/* if we stumble upon an unknown
|
|
|
|
|
formatting code, copy the rest of
|
|
|
|
|
the format string to the output
|
|
|
|
|
string. (we cannot just skip the
|
|
|
|
|
code, since there's no way to know
|
2002-04-15 10:48:52 -03:00
|
|
|
|
what's in the argument list) */
|
2001-08-24 15:32:06 -03:00
|
|
|
|
n += strlen(p);
|
|
|
|
|
goto expand;
|
|
|
|
|
}
|
|
|
|
|
} else
|
|
|
|
|
n++;
|
|
|
|
|
}
|
|
|
|
|
expand:
|
|
|
|
|
/* step 2: fill the buffer */
|
2001-12-02 21:55:38 -04:00
|
|
|
|
/* Since we've analyzed how much space we need for the worst case,
|
|
|
|
|
use sprintf directly instead of the slower PyOS_snprintf. */
|
2001-08-24 15:32:06 -03:00
|
|
|
|
string = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (!string)
|
|
|
|
|
return NULL;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
s = PyString_AsString(string);
|
|
|
|
|
|
|
|
|
|
for (f = format; *f; f++) {
|
|
|
|
|
if (*f == '%') {
|
|
|
|
|
const char* p = f++;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i;
|
|
|
|
|
int longflag = 0;
|
2006-02-16 02:54:25 -04:00
|
|
|
|
int size_tflag = 0;
|
2001-08-24 15:32:06 -03:00
|
|
|
|
/* parse the width.precision part (we're only
|
|
|
|
|
interested in the precision value, if any) */
|
|
|
|
|
n = 0;
|
|
|
|
|
while (isdigit(Py_CHARMASK(*f)))
|
|
|
|
|
n = (n*10) + *f++ - '0';
|
|
|
|
|
if (*f == '.') {
|
|
|
|
|
f++;
|
|
|
|
|
n = 0;
|
|
|
|
|
while (isdigit(Py_CHARMASK(*f)))
|
|
|
|
|
n = (n*10) + *f++ - '0';
|
|
|
|
|
}
|
|
|
|
|
while (*f && *f != '%' && !isalpha(Py_CHARMASK(*f)))
|
|
|
|
|
f++;
|
2006-05-13 20:28:20 -03:00
|
|
|
|
/* handle the long flag, but only for %ld and %lu.
|
|
|
|
|
others can be added when necessary. */
|
|
|
|
|
if (*f == 'l' && (f[1] == 'd' || f[1] == 'u')) {
|
2001-08-24 15:32:06 -03:00
|
|
|
|
longflag = 1;
|
|
|
|
|
++f;
|
|
|
|
|
}
|
2006-02-16 02:54:25 -04:00
|
|
|
|
/* handle the size_t flag. */
|
2006-05-13 20:28:20 -03:00
|
|
|
|
if (*f == 'z' && (f[1] == 'd' || f[1] == 'u')) {
|
2006-02-16 02:54:25 -04:00
|
|
|
|
size_tflag = 1;
|
|
|
|
|
++f;
|
|
|
|
|
}
|
2002-04-15 10:48:52 -03:00
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
switch (*f) {
|
|
|
|
|
case 'c':
|
|
|
|
|
*s++ = va_arg(vargs, int);
|
|
|
|
|
break;
|
|
|
|
|
case 'd':
|
|
|
|
|
if (longflag)
|
|
|
|
|
sprintf(s, "%ld", va_arg(vargs, long));
|
2006-03-16 23:29:34 -04:00
|
|
|
|
else if (size_tflag)
|
2006-05-13 10:34:04 -03:00
|
|
|
|
sprintf(s, "%" PY_FORMAT_SIZE_T "d",
|
|
|
|
|
va_arg(vargs, Py_ssize_t));
|
2001-08-24 15:32:06 -03:00
|
|
|
|
else
|
|
|
|
|
sprintf(s, "%d", va_arg(vargs, int));
|
|
|
|
|
s += strlen(s);
|
|
|
|
|
break;
|
2006-05-13 20:28:20 -03:00
|
|
|
|
case 'u':
|
|
|
|
|
if (longflag)
|
|
|
|
|
sprintf(s, "%lu",
|
|
|
|
|
va_arg(vargs, unsigned long));
|
|
|
|
|
else if (size_tflag)
|
|
|
|
|
sprintf(s, "%" PY_FORMAT_SIZE_T "u",
|
|
|
|
|
va_arg(vargs, size_t));
|
|
|
|
|
else
|
|
|
|
|
sprintf(s, "%u",
|
|
|
|
|
va_arg(vargs, unsigned int));
|
|
|
|
|
s += strlen(s);
|
|
|
|
|
break;
|
2001-08-24 15:32:06 -03:00
|
|
|
|
case 'i':
|
|
|
|
|
sprintf(s, "%i", va_arg(vargs, int));
|
|
|
|
|
s += strlen(s);
|
|
|
|
|
break;
|
|
|
|
|
case 'x':
|
|
|
|
|
sprintf(s, "%x", va_arg(vargs, int));
|
|
|
|
|
s += strlen(s);
|
|
|
|
|
break;
|
|
|
|
|
case 's':
|
|
|
|
|
p = va_arg(vargs, char*);
|
|
|
|
|
i = strlen(p);
|
|
|
|
|
if (n > 0 && i > n)
|
|
|
|
|
i = n;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(s, p, i);
|
2001-08-24 15:32:06 -03:00
|
|
|
|
s += i;
|
|
|
|
|
break;
|
|
|
|
|
case 'p':
|
|
|
|
|
sprintf(s, "%p", va_arg(vargs, void*));
|
2001-08-25 00:02:28 -03:00
|
|
|
|
/* %p is ill-defined: ensure leading 0x. */
|
|
|
|
|
if (s[1] == 'X')
|
|
|
|
|
s[1] = 'x';
|
|
|
|
|
else if (s[1] != 'x') {
|
|
|
|
|
memmove(s+2, s, strlen(s)+1);
|
|
|
|
|
s[0] = '0';
|
|
|
|
|
s[1] = 'x';
|
|
|
|
|
}
|
2001-08-24 15:32:06 -03:00
|
|
|
|
s += strlen(s);
|
|
|
|
|
break;
|
|
|
|
|
case '%':
|
|
|
|
|
*s++ = '%';
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
strcpy(s, p);
|
|
|
|
|
s += strlen(s);
|
|
|
|
|
goto end;
|
|
|
|
|
}
|
|
|
|
|
} else
|
|
|
|
|
*s++ = *f;
|
|
|
|
|
}
|
2002-04-15 10:48:52 -03:00
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
end:
|
2001-08-27 00:11:09 -03:00
|
|
|
|
_PyString_Resize(&string, s - PyString_AS_STRING(string));
|
2001-08-24 15:32:06 -03:00
|
|
|
|
return string;
|
|
|
|
|
}
|
2002-04-15 10:48:52 -03:00
|
|
|
|
|
2001-08-24 15:32:06 -03:00
|
|
|
|
PyObject *
|
2002-04-15 10:48:52 -03:00
|
|
|
|
PyString_FromFormat(const char *format, ...)
|
2001-08-24 15:32:06 -03:00
|
|
|
|
{
|
2001-08-27 00:11:09 -03:00
|
|
|
|
PyObject* ret;
|
2001-08-24 15:32:06 -03:00
|
|
|
|
va_list vargs;
|
|
|
|
|
|
|
|
|
|
#ifdef HAVE_STDARG_PROTOTYPES
|
|
|
|
|
va_start(vargs, format);
|
|
|
|
|
#else
|
|
|
|
|
va_start(vargs);
|
|
|
|
|
#endif
|
2001-08-27 00:11:09 -03:00
|
|
|
|
ret = PyString_FromFormatV(format, vargs);
|
|
|
|
|
va_end(vargs);
|
|
|
|
|
return ret;
|
2001-08-24 15:32:06 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
PyObject *PyString_Decode(const char *s,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t size,
|
2000-07-06 08:29:01 -03:00
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
2001-05-15 09:00:02 -03:00
|
|
|
|
PyObject *v, *str;
|
|
|
|
|
|
|
|
|
|
str = PyString_FromStringAndSize(s, size);
|
|
|
|
|
if (str == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
v = PyString_AsDecodedString(str, encoding, errors);
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
return v;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_AsDecodedObject(PyObject *str,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
|
|
if (!PyString_Check(str)) {
|
|
|
|
|
PyErr_BadArgument();
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
2001-08-17 15:39:25 -03:00
|
|
|
|
if (encoding == NULL) {
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-07-06 08:29:01 -03:00
|
|
|
|
encoding = PyUnicode_GetDefaultEncoding();
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#else
|
|
|
|
|
PyErr_SetString(PyExc_ValueError, "no encoding specified");
|
|
|
|
|
goto onError;
|
|
|
|
|
#endif
|
|
|
|
|
}
|
2000-07-06 08:29:01 -03:00
|
|
|
|
|
|
|
|
|
/* Decode via the codec registry */
|
2001-05-15 09:00:02 -03:00
|
|
|
|
v = PyCodec_Decode(str, encoding, errors);
|
|
|
|
|
if (v == NULL)
|
2000-07-06 08:29:01 -03:00
|
|
|
|
goto onError;
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
|
|
|
|
return v;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_AsDecodedString(PyObject *str,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
|
|
v = PyString_AsDecodedObject(str, encoding, errors);
|
|
|
|
|
if (v == NULL)
|
2000-07-06 08:29:01 -03:00
|
|
|
|
goto onError;
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-07-06 08:29:01 -03:00
|
|
|
|
/* Convert Unicode to a string using the default encoding */
|
2001-05-15 09:00:02 -03:00
|
|
|
|
if (PyUnicode_Check(v)) {
|
|
|
|
|
PyObject *temp = v;
|
|
|
|
|
v = PyUnicode_AsEncodedString(v, NULL, NULL);
|
2000-07-06 08:29:01 -03:00
|
|
|
|
Py_DECREF(temp);
|
2001-05-15 09:00:02 -03:00
|
|
|
|
if (v == NULL)
|
2000-07-06 08:29:01 -03:00
|
|
|
|
goto onError;
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2001-05-15 09:00:02 -03:00
|
|
|
|
if (!PyString_Check(v)) {
|
2000-07-06 08:29:01 -03:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2000-07-11 23:58:28 -03:00
|
|
|
|
"decoder did not return a string object (type=%.400s)",
|
2001-05-15 09:00:02 -03:00
|
|
|
|
v->ob_type->tp_name);
|
|
|
|
|
Py_DECREF(v);
|
2000-07-06 08:29:01 -03:00
|
|
|
|
goto onError;
|
|
|
|
|
}
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
|
|
|
|
return v;
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_Encode(const char *s,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t size,
|
2000-07-06 08:29:01 -03:00
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v, *str;
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
str = PyString_FromStringAndSize(s, size);
|
|
|
|
|
if (str == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
v = PyString_AsEncodedString(str, encoding, errors);
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
return v;
|
|
|
|
|
}
|
|
|
|
|
|
2001-05-15 09:00:02 -03:00
|
|
|
|
PyObject *PyString_AsEncodedObject(PyObject *str,
|
2000-07-06 08:29:01 -03:00
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v;
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
if (!PyString_Check(str)) {
|
|
|
|
|
PyErr_BadArgument();
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
|
|
|
|
|
2001-08-17 15:39:25 -03:00
|
|
|
|
if (encoding == NULL) {
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-07-06 08:29:01 -03:00
|
|
|
|
encoding = PyUnicode_GetDefaultEncoding();
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#else
|
|
|
|
|
PyErr_SetString(PyExc_ValueError, "no encoding specified");
|
|
|
|
|
goto onError;
|
|
|
|
|
#endif
|
|
|
|
|
}
|
2000-07-06 08:29:01 -03:00
|
|
|
|
|
|
|
|
|
/* Encode via the codec registry */
|
|
|
|
|
v = PyCodec_Encode(str, encoding, errors);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
|
|
|
|
return v;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_AsEncodedString(PyObject *str,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v;
|
|
|
|
|
|
2001-06-12 10:14:10 -03:00
|
|
|
|
v = PyString_AsEncodedObject(str, encoding, errors);
|
2001-05-15 09:00:02 -03:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-07-06 08:29:01 -03:00
|
|
|
|
/* Convert Unicode to a string using the default encoding */
|
|
|
|
|
if (PyUnicode_Check(v)) {
|
|
|
|
|
PyObject *temp = v;
|
|
|
|
|
v = PyUnicode_AsEncodedString(v, NULL, NULL);
|
|
|
|
|
Py_DECREF(temp);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2000-07-06 08:29:01 -03:00
|
|
|
|
if (!PyString_Check(v)) {
|
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"encoder did not return a string object (type=%.400s)",
|
|
|
|
|
v->ob_type->tp_name);
|
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
return v;
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
1993-06-17 09:35:49 -03:00
|
|
|
|
static void
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_dealloc(PyObject *op)
|
1992-03-27 13:31:02 -04:00
|
|
|
|
{
|
2002-08-19 18:43:18 -03:00
|
|
|
|
switch (PyString_CHECK_INTERNED(op)) {
|
|
|
|
|
case SSTATE_NOT_INTERNED:
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case SSTATE_INTERNED_MORTAL:
|
|
|
|
|
/* revive dead object temporarily for DelItem */
|
|
|
|
|
op->ob_refcnt = 3;
|
|
|
|
|
if (PyDict_DelItem(interned, op) != 0)
|
|
|
|
|
Py_FatalError(
|
|
|
|
|
"deletion of interned string failed");
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case SSTATE_INTERNED_IMMORTAL:
|
|
|
|
|
Py_FatalError("Immortal interned string died.");
|
|
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
Py_FatalError("Inconsistent interned string state.");
|
|
|
|
|
}
|
2001-10-05 17:51:39 -03:00
|
|
|
|
op->ob_type->tp_free(op);
|
1992-03-27 13:31:02 -04:00
|
|
|
|
}
|
|
|
|
|
|
2002-08-14 04:46:28 -03:00
|
|
|
|
/* Unescape a backslash-escaped string. If unicode is non-zero,
|
|
|
|
|
the string is a u-literal. If recode_encoding is non-zero,
|
|
|
|
|
the string is UTF-8 encoded and should be re-encoded in the
|
|
|
|
|
specified encoding. */
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_DecodeEscape(const char *s,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len,
|
2002-08-14 04:46:28 -03:00
|
|
|
|
const char *errors,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t unicode,
|
2002-08-14 04:46:28 -03:00
|
|
|
|
const char *recode_encoding)
|
|
|
|
|
{
|
|
|
|
|
int c;
|
|
|
|
|
char *p, *buf;
|
|
|
|
|
const char *end;
|
|
|
|
|
PyObject *v;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t newlen = recode_encoding ? 4*len:len;
|
2002-09-03 10:53:40 -03:00
|
|
|
|
v = PyString_FromStringAndSize((char *)NULL, newlen);
|
2002-08-14 04:46:28 -03:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
p = buf = PyString_AsString(v);
|
|
|
|
|
end = s + len;
|
|
|
|
|
while (s < end) {
|
|
|
|
|
if (*s != '\\') {
|
2002-09-09 03:17:05 -03:00
|
|
|
|
non_esc:
|
2002-08-14 04:46:28 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
if (recode_encoding && (*s & 0x80)) {
|
|
|
|
|
PyObject *u, *w;
|
|
|
|
|
char *r;
|
|
|
|
|
const char* t;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t rn;
|
2002-08-14 04:46:28 -03:00
|
|
|
|
t = s;
|
|
|
|
|
/* Decode non-ASCII bytes as UTF-8. */
|
|
|
|
|
while (t < end && (*t & 0x80)) t++;
|
|
|
|
|
u = PyUnicode_DecodeUTF8(s, t - s, errors);
|
|
|
|
|
if(!u) goto failed;
|
|
|
|
|
|
|
|
|
|
/* Recode them in target encoding. */
|
|
|
|
|
w = PyUnicode_AsEncodedString(
|
|
|
|
|
u, recode_encoding, errors);
|
|
|
|
|
Py_DECREF(u);
|
|
|
|
|
if (!w) goto failed;
|
|
|
|
|
|
|
|
|
|
/* Append bytes to output buffer. */
|
2006-03-19 21:53:23 -04:00
|
|
|
|
assert(PyString_Check(w));
|
|
|
|
|
r = PyString_AS_STRING(w);
|
|
|
|
|
rn = PyString_GET_SIZE(w);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(p, r, rn);
|
2002-08-14 04:46:28 -03:00
|
|
|
|
p += rn;
|
|
|
|
|
Py_DECREF(w);
|
|
|
|
|
s = t;
|
|
|
|
|
} else {
|
|
|
|
|
*p++ = *s++;
|
|
|
|
|
}
|
|
|
|
|
#else
|
|
|
|
|
*p++ = *s++;
|
|
|
|
|
#endif
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
s++;
|
2002-08-14 05:22:50 -03:00
|
|
|
|
if (s==end) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"Trailing \\ in string");
|
|
|
|
|
goto failed;
|
|
|
|
|
}
|
2002-08-14 04:46:28 -03:00
|
|
|
|
switch (*s++) {
|
|
|
|
|
/* XXX This assumes ASCII! */
|
|
|
|
|
case '\n': break;
|
|
|
|
|
case '\\': *p++ = '\\'; break;
|
|
|
|
|
case '\'': *p++ = '\''; break;
|
|
|
|
|
case '\"': *p++ = '\"'; break;
|
|
|
|
|
case 'b': *p++ = '\b'; break;
|
|
|
|
|
case 'f': *p++ = '\014'; break; /* FF */
|
|
|
|
|
case 't': *p++ = '\t'; break;
|
|
|
|
|
case 'n': *p++ = '\n'; break;
|
|
|
|
|
case 'r': *p++ = '\r'; break;
|
|
|
|
|
case 'v': *p++ = '\013'; break; /* VT */
|
|
|
|
|
case 'a': *p++ = '\007'; break; /* BEL, not classic C */
|
|
|
|
|
case '0': case '1': case '2': case '3':
|
|
|
|
|
case '4': case '5': case '6': case '7':
|
|
|
|
|
c = s[-1] - '0';
|
|
|
|
|
if ('0' <= *s && *s <= '7') {
|
|
|
|
|
c = (c<<3) + *s++ - '0';
|
|
|
|
|
if ('0' <= *s && *s <= '7')
|
|
|
|
|
c = (c<<3) + *s++ - '0';
|
|
|
|
|
}
|
|
|
|
|
*p++ = c;
|
|
|
|
|
break;
|
|
|
|
|
case 'x':
|
2006-03-16 23:29:34 -04:00
|
|
|
|
if (isxdigit(Py_CHARMASK(s[0]))
|
2002-08-14 04:46:28 -03:00
|
|
|
|
&& isxdigit(Py_CHARMASK(s[1]))) {
|
|
|
|
|
unsigned int x = 0;
|
|
|
|
|
c = Py_CHARMASK(*s);
|
|
|
|
|
s++;
|
|
|
|
|
if (isdigit(c))
|
|
|
|
|
x = c - '0';
|
|
|
|
|
else if (islower(c))
|
|
|
|
|
x = 10 + c - 'a';
|
|
|
|
|
else
|
|
|
|
|
x = 10 + c - 'A';
|
|
|
|
|
x = x << 4;
|
|
|
|
|
c = Py_CHARMASK(*s);
|
|
|
|
|
s++;
|
|
|
|
|
if (isdigit(c))
|
|
|
|
|
x += c - '0';
|
|
|
|
|
else if (islower(c))
|
|
|
|
|
x += 10 + c - 'a';
|
|
|
|
|
else
|
|
|
|
|
x += 10 + c - 'A';
|
|
|
|
|
*p++ = x;
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
if (!errors || strcmp(errors, "strict") == 0) {
|
2006-03-16 23:29:34 -04:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
2002-08-14 04:46:28 -03:00
|
|
|
|
"invalid \\x escape");
|
2002-08-14 05:22:50 -03:00
|
|
|
|
goto failed;
|
2002-08-14 04:46:28 -03:00
|
|
|
|
}
|
|
|
|
|
if (strcmp(errors, "replace") == 0) {
|
|
|
|
|
*p++ = '?';
|
|
|
|
|
} else if (strcmp(errors, "ignore") == 0)
|
|
|
|
|
/* do nothing */;
|
|
|
|
|
else {
|
|
|
|
|
PyErr_Format(PyExc_ValueError,
|
|
|
|
|
"decoding error; "
|
|
|
|
|
"unknown error handling code: %.400s",
|
|
|
|
|
errors);
|
2002-08-14 05:22:50 -03:00
|
|
|
|
goto failed;
|
2002-08-14 04:46:28 -03:00
|
|
|
|
}
|
|
|
|
|
#ifndef Py_USING_UNICODE
|
|
|
|
|
case 'u':
|
|
|
|
|
case 'U':
|
|
|
|
|
case 'N':
|
|
|
|
|
if (unicode) {
|
2002-08-16 20:20:39 -03:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
2002-08-14 04:46:28 -03:00
|
|
|
|
"Unicode escapes not legal "
|
|
|
|
|
"when Unicode disabled");
|
2002-08-14 05:22:50 -03:00
|
|
|
|
goto failed;
|
2002-08-14 04:46:28 -03:00
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
default:
|
|
|
|
|
*p++ = '\\';
|
2002-09-09 03:17:05 -03:00
|
|
|
|
s--;
|
|
|
|
|
goto non_esc; /* an arbitry number of unescaped
|
|
|
|
|
UTF-8 bytes may follow. */
|
2002-08-14 04:46:28 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
2002-09-03 10:53:40 -03:00
|
|
|
|
if (p-buf < newlen)
|
2006-02-15 13:27:45 -04:00
|
|
|
|
_PyString_Resize(&v, p - buf);
|
2002-08-14 04:46:28 -03:00
|
|
|
|
return v;
|
|
|
|
|
failed:
|
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
|
/* object api */
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
2000-09-19 18:04:18 -03:00
|
|
|
|
string_getsize(register PyObject *op)
|
|
|
|
|
{
|
|
|
|
|
char *s;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len;
|
2000-09-19 18:04:18 -03:00
|
|
|
|
if (PyString_AsStringAndSize(op, &s, &len))
|
|
|
|
|
return -1;
|
|
|
|
|
return len;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static /*const*/ char *
|
|
|
|
|
string_getbuffer(register PyObject *op)
|
|
|
|
|
{
|
|
|
|
|
char *s;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len;
|
2000-09-19 18:04:18 -03:00
|
|
|
|
if (PyString_AsStringAndSize(op, &s, &len))
|
|
|
|
|
return NULL;
|
|
|
|
|
return s;
|
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_Size(register PyObject *op)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2000-09-19 18:04:18 -03:00
|
|
|
|
if (!PyString_Check(op))
|
|
|
|
|
return string_getsize(op);
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return ((PyStringObject *)op) -> ob_size;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*const*/ char *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_AsString(register PyObject *op)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2000-09-19 18:04:18 -03:00
|
|
|
|
if (!PyString_Check(op))
|
|
|
|
|
return string_getbuffer(op);
|
|
|
|
|
return ((PyStringObject *)op) -> ob_sval;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
PyString_AsStringAndSize(register PyObject *obj,
|
|
|
|
|
register char **s,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t *len)
|
2000-09-19 18:04:18 -03:00
|
|
|
|
{
|
|
|
|
|
if (s == NULL) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_BadInternalCall();
|
2000-09-19 18:04:18 -03:00
|
|
|
|
return -1;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
2000-09-19 18:04:18 -03:00
|
|
|
|
|
|
|
|
|
if (!PyString_Check(obj)) {
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-09-19 18:04:18 -03:00
|
|
|
|
if (PyUnicode_Check(obj)) {
|
|
|
|
|
obj = _PyUnicode_AsDefaultEncodedString(obj, NULL);
|
|
|
|
|
if (obj == NULL)
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2002-04-15 10:48:52 -03:00
|
|
|
|
else
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
|
|
|
|
{
|
2000-09-19 18:04:18 -03:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"expected string or Unicode object, "
|
|
|
|
|
"%.200s found", obj->ob_type->tp_name);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*s = PyString_AS_STRING(obj);
|
|
|
|
|
if (len != NULL)
|
|
|
|
|
*len = PyString_GET_SIZE(obj);
|
2006-04-17 21:35:43 -03:00
|
|
|
|
else if (strlen(*s) != (size_t)PyString_GET_SIZE(obj)) {
|
2000-09-19 18:04:18 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"expected string without null bytes");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
return 0;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-25 14:55:31 -03:00
|
|
|
|
/* -------------------------------------------------------------------- */
|
2006-05-27 07:05:10 -03:00
|
|
|
|
/* Methods */
|
2006-05-25 14:55:31 -03:00
|
|
|
|
|
2006-05-26 14:04:58 -03:00
|
|
|
|
#define STRINGLIB_CHAR char
|
2006-05-26 15:15:38 -03:00
|
|
|
|
|
|
|
|
|
#define STRINGLIB_CMP memcmp
|
2006-05-27 07:05:10 -03:00
|
|
|
|
#define STRINGLIB_LEN PyString_GET_SIZE
|
|
|
|
|
#define STRINGLIB_NEW PyString_FromStringAndSize
|
|
|
|
|
#define STRINGLIB_STR PyString_AS_STRING
|
2006-05-26 15:15:38 -03:00
|
|
|
|
|
2006-05-26 14:22:38 -03:00
|
|
|
|
#define STRINGLIB_EMPTY nullstring
|
2006-05-25 14:55:31 -03:00
|
|
|
|
|
2006-05-26 14:04:58 -03:00
|
|
|
|
#include "stringlib/fastsearch.h"
|
2006-05-26 16:48:07 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
#include "stringlib/count.h"
|
2006-05-26 16:48:07 -03:00
|
|
|
|
#include "stringlib/find.h"
|
2006-05-26 14:22:38 -03:00
|
|
|
|
#include "stringlib/partition.h"
|
2006-05-25 14:55:31 -03:00
|
|
|
|
|
1990-10-14 09:07:46 -03:00
|
|
|
|
|
1991-06-07 19:58:57 -03:00
|
|
|
|
static int
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_print(PyStringObject *op, FILE *fp, int flags)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
char c;
|
1993-10-26 12:25:16 -03:00
|
|
|
|
int quote;
|
2001-10-16 17:18:24 -03:00
|
|
|
|
|
1991-06-07 19:58:57 -03:00
|
|
|
|
/* XXX Ought to check for interrupts when writing long strings */
|
2001-10-16 17:18:24 -03:00
|
|
|
|
if (! PyString_CheckExact(op)) {
|
|
|
|
|
int ret;
|
|
|
|
|
/* A str subclass may have its own __str__ method. */
|
|
|
|
|
op = (PyStringObject *) PyObject_Str((PyObject *)op);
|
|
|
|
|
if (op == NULL)
|
|
|
|
|
return -1;
|
|
|
|
|
ret = string_print(op, fp, flags);
|
|
|
|
|
Py_DECREF(op);
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (flags & Py_PRINT_RAW) {
|
2006-10-04 08:44:06 -03:00
|
|
|
|
char *data = op->ob_sval;
|
|
|
|
|
Py_ssize_t size = op->ob_size;
|
|
|
|
|
while (size > INT_MAX) {
|
|
|
|
|
/* Very long strings cannot be written atomically.
|
|
|
|
|
* But don't write exactly INT_MAX bytes at a time
|
|
|
|
|
* to avoid memory aligment issues.
|
|
|
|
|
*/
|
|
|
|
|
const int chunk_size = INT_MAX & ~0x3FFF;
|
|
|
|
|
fwrite(data, 1, chunk_size, fp);
|
|
|
|
|
data += chunk_size;
|
|
|
|
|
size -= chunk_size;
|
|
|
|
|
}
|
2002-12-06 08:48:53 -04:00
|
|
|
|
#ifdef __VMS
|
2006-10-04 08:44:06 -03:00
|
|
|
|
if (size) fwrite(data, (int)size, 1, fp);
|
2002-12-06 08:48:53 -04:00
|
|
|
|
#else
|
2006-10-04 08:44:06 -03:00
|
|
|
|
fwrite(data, 1, (int)size, fp);
|
2002-12-06 08:48:53 -04:00
|
|
|
|
#endif
|
1991-06-07 19:58:57 -03:00
|
|
|
|
return 0;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
1993-10-26 12:25:16 -03:00
|
|
|
|
|
2000-07-16 09:04:32 -03:00
|
|
|
|
/* figure out which quote to use; single is preferred */
|
1993-10-26 12:25:16 -03:00
|
|
|
|
quote = '\'';
|
2002-04-15 10:48:52 -03:00
|
|
|
|
if (memchr(op->ob_sval, '\'', op->ob_size) &&
|
|
|
|
|
!memchr(op->ob_sval, '"', op->ob_size))
|
1993-10-26 12:25:16 -03:00
|
|
|
|
quote = '"';
|
|
|
|
|
|
|
|
|
|
fputc(quote, fp);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
for (i = 0; i < op->ob_size; i++) {
|
|
|
|
|
c = op->ob_sval[i];
|
1993-10-26 12:25:16 -03:00
|
|
|
|
if (c == quote || c == '\\')
|
2002-10-11 02:37:59 -03:00
|
|
|
|
fprintf(fp, "\\%c", c);
|
2001-01-24 13:19:08 -04:00
|
|
|
|
else if (c == '\t')
|
2002-10-11 02:37:59 -03:00
|
|
|
|
fprintf(fp, "\\t");
|
2001-01-24 13:19:08 -04:00
|
|
|
|
else if (c == '\n')
|
2002-10-11 02:37:59 -03:00
|
|
|
|
fprintf(fp, "\\n");
|
2001-01-24 13:19:08 -04:00
|
|
|
|
else if (c == '\r')
|
2002-10-11 02:37:59 -03:00
|
|
|
|
fprintf(fp, "\\r");
|
|
|
|
|
else if (c < ' ' || c >= 0x7f)
|
|
|
|
|
fprintf(fp, "\\x%02x", c & 0xff);
|
2002-10-07 10:55:50 -03:00
|
|
|
|
else
|
2002-10-11 02:37:59 -03:00
|
|
|
|
fputc(c, fp);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
1993-10-26 12:25:16 -03:00
|
|
|
|
fputc(quote, fp);
|
1991-06-07 19:58:57 -03:00
|
|
|
|
return 0;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
2002-08-14 04:46:28 -03:00
|
|
|
|
PyObject *
|
|
|
|
|
PyString_Repr(PyObject *obj, int smartquotes)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2002-08-14 04:46:28 -03:00
|
|
|
|
register PyStringObject* op = (PyStringObject*) obj;
|
2004-06-27 14:24:49 -03:00
|
|
|
|
size_t newsize = 2 + 4 * op->ob_size;
|
2000-06-30 07:29:57 -03:00
|
|
|
|
PyObject *v;
|
2006-10-04 08:44:06 -03:00
|
|
|
|
if (newsize > PY_SSIZE_T_MAX || newsize / 4 != op->ob_size) {
|
2000-06-30 07:29:57 -03:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"string is too large to make repr");
|
|
|
|
|
}
|
|
|
|
|
v = PyString_FromStringAndSize((char *)NULL, newsize);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
if (v == NULL) {
|
1991-06-07 19:58:57 -03:00
|
|
|
|
return NULL;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
else {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t i;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
register char c;
|
|
|
|
|
register char *p;
|
1993-10-26 12:25:16 -03:00
|
|
|
|
int quote;
|
|
|
|
|
|
2000-07-16 09:04:32 -03:00
|
|
|
|
/* figure out which quote to use; single is preferred */
|
1993-10-26 12:25:16 -03:00
|
|
|
|
quote = '\'';
|
2006-03-16 23:29:34 -04:00
|
|
|
|
if (smartquotes &&
|
2002-08-14 04:46:28 -03:00
|
|
|
|
memchr(op->ob_sval, '\'', op->ob_size) &&
|
2002-04-15 10:48:52 -03:00
|
|
|
|
!memchr(op->ob_sval, '"', op->ob_size))
|
1993-10-26 12:25:16 -03:00
|
|
|
|
quote = '"';
|
|
|
|
|
|
2001-12-02 21:55:38 -04:00
|
|
|
|
p = PyString_AS_STRING(v);
|
1993-10-26 12:25:16 -03:00
|
|
|
|
*p++ = quote;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
for (i = 0; i < op->ob_size; i++) {
|
2001-12-02 21:55:38 -04:00
|
|
|
|
/* There's at least enough room for a hex escape
|
|
|
|
|
and a closing quote. */
|
|
|
|
|
assert(newsize - (p - PyString_AS_STRING(v)) >= 5);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
c = op->ob_sval[i];
|
1993-10-26 12:25:16 -03:00
|
|
|
|
if (c == quote || c == '\\')
|
1990-10-14 09:07:46 -03:00
|
|
|
|
*p++ = '\\', *p++ = c;
|
2001-01-24 13:19:08 -04:00
|
|
|
|
else if (c == '\t')
|
|
|
|
|
*p++ = '\\', *p++ = 't';
|
|
|
|
|
else if (c == '\n')
|
|
|
|
|
*p++ = '\\', *p++ = 'n';
|
|
|
|
|
else if (c == '\r')
|
|
|
|
|
*p++ = '\\', *p++ = 'r';
|
2002-10-11 02:37:59 -03:00
|
|
|
|
else if (c < ' ' || c >= 0x7f) {
|
|
|
|
|
/* For performance, we don't want to call
|
|
|
|
|
PyOS_snprintf here (extra layers of
|
|
|
|
|
function call). */
|
|
|
|
|
sprintf(p, "\\x%02x", c & 0xff);
|
|
|
|
|
p += 4;
|
2002-10-07 10:55:50 -03:00
|
|
|
|
}
|
2002-10-11 02:37:59 -03:00
|
|
|
|
else
|
|
|
|
|
*p++ = c;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
2001-12-02 21:55:38 -04:00
|
|
|
|
assert(newsize - (p - PyString_AS_STRING(v)) >= 1);
|
1993-10-26 12:25:16 -03:00
|
|
|
|
*p++ = quote;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
*p = '\0';
|
1997-05-02 00:12:38 -03:00
|
|
|
|
_PyString_Resize(
|
2006-04-21 10:54:43 -03:00
|
|
|
|
&v, (p - PyString_AS_STRING(v)));
|
1990-10-21 19:15:08 -03:00
|
|
|
|
return v;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2002-08-14 04:46:28 -03:00
|
|
|
|
static PyObject *
|
|
|
|
|
string_repr(PyObject *op)
|
|
|
|
|
{
|
|
|
|
|
return PyString_Repr(op, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2001-05-01 13:51:53 -03:00
|
|
|
|
static PyObject *
|
|
|
|
|
string_str(PyObject *s)
|
|
|
|
|
{
|
2001-10-16 17:18:24 -03:00
|
|
|
|
assert(PyString_Check(s));
|
|
|
|
|
if (PyString_CheckExact(s)) {
|
|
|
|
|
Py_INCREF(s);
|
|
|
|
|
return s;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
/* Subtype -- return genuine string with the same value. */
|
|
|
|
|
PyStringObject *t = (PyStringObject *) s;
|
|
|
|
|
return PyString_FromStringAndSize(t->ob_sval, t->ob_size);
|
|
|
|
|
}
|
2001-05-01 13:51:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_length(PyStringObject *a)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
|
|
|
|
return a->ob_size;
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_concat(register PyStringObject *a, register PyObject *bb)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2006-05-25 15:18:39 -03:00
|
|
|
|
register Py_ssize_t size;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
register PyStringObject *op;
|
|
|
|
|
if (!PyString_Check(bb)) {
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (PyUnicode_Check(bb))
|
|
|
|
|
return PyUnicode_Concat((PyObject *)a, bb);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2001-04-28 02:38:26 -03:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2001-10-22 01:12:44 -03:00
|
|
|
|
"cannot concatenate 'str' and '%.200s' objects",
|
2000-06-01 00:12:13 -03:00
|
|
|
|
bb->ob_type->tp_name);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
#define b ((PyStringObject *)bb)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
/* Optimize cases with empty left or right operand */
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if ((a->ob_size == 0 || b->ob_size == 0) &&
|
|
|
|
|
PyString_CheckExact(a) && PyString_CheckExact(b)) {
|
|
|
|
|
if (a->ob_size == 0) {
|
|
|
|
|
Py_INCREF(bb);
|
|
|
|
|
return bb;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
size = a->ob_size + b->ob_size;
|
2006-05-25 15:18:39 -03:00
|
|
|
|
if (size < 0) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"strings are too large to concat");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2002-08-19 16:26:42 -03:00
|
|
|
|
/* Inline PyObject_NewVar */
|
2004-06-27 14:24:49 -03:00
|
|
|
|
op = (PyStringObject *)PyObject_MALLOC(sizeof(PyStringObject) + size);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
if (op == NULL)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-03 20:44:39 -03:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
op->ob_shash = -1;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
op->ob_sstate = SSTATE_NOT_INTERNED;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(op->ob_sval, a->ob_sval, a->ob_size);
|
|
|
|
|
Py_MEMCPY(op->ob_sval + a->ob_size, b->ob_sval, b->ob_size);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
op->ob_sval[size] = '\0';
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
#undef b
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PyObject *
|
2006-02-15 13:27:45 -04:00
|
|
|
|
string_repeat(register PyStringObject *a, register Py_ssize_t n)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t i;
|
|
|
|
|
register Py_ssize_t j;
|
|
|
|
|
register Py_ssize_t size;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
register PyStringObject *op;
|
2000-09-09 03:13:41 -03:00
|
|
|
|
size_t nbytes;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
if (n < 0)
|
|
|
|
|
n = 0;
|
2000-09-09 03:13:41 -03:00
|
|
|
|
/* watch out for overflows: the size can overflow int,
|
|
|
|
|
* and the # of bytes needed can overflow size_t
|
|
|
|
|
*/
|
1990-10-14 09:07:46 -03:00
|
|
|
|
size = a->ob_size * n;
|
2000-09-09 03:13:41 -03:00
|
|
|
|
if (n && size / n != a->ob_size) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"repeated string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (size == a->ob_size && PyString_CheckExact(a)) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
2004-06-27 14:24:49 -03:00
|
|
|
|
nbytes = (size_t)size;
|
|
|
|
|
if (nbytes + sizeof(PyStringObject) <= nbytes) {
|
2000-09-09 03:13:41 -03:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"repeated string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
op = (PyStringObject *)
|
2002-04-12 00:05:19 -03:00
|
|
|
|
PyObject_MALLOC(sizeof(PyStringObject) + nbytes);
|
1990-10-21 19:15:08 -03:00
|
|
|
|
if (op == NULL)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-03 20:44:39 -03:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 09:04:32 -03:00
|
|
|
|
op->ob_shash = -1;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
op->ob_sstate = SSTATE_NOT_INTERNED;
|
2003-01-06 18:42:41 -04:00
|
|
|
|
op->ob_sval[size] = '\0';
|
|
|
|
|
if (a->ob_size == 1 && n > 0) {
|
|
|
|
|
memset(op->ob_sval, a->ob_sval[0] , n);
|
|
|
|
|
return (PyObject *) op;
|
|
|
|
|
}
|
2003-01-06 06:33:56 -04:00
|
|
|
|
i = 0;
|
|
|
|
|
if (i < size) {
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(op->ob_sval, a->ob_sval, a->ob_size);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
i = a->ob_size;
|
2003-01-06 06:33:56 -04:00
|
|
|
|
}
|
|
|
|
|
while (i < size) {
|
|
|
|
|
j = (i <= size-i) ? i : size-i;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(op->ob_sval+i, op->ob_sval, j);
|
2003-01-06 06:33:56 -04:00
|
|
|
|
i += j;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* String slice a[i:j] consists of characters a[i] ... a[j-1] */
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PyObject *
|
2006-03-16 23:29:34 -04:00
|
|
|
|
string_slice(register PyStringObject *a, register Py_ssize_t i,
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t j)
|
2000-07-09 04:04:36 -03:00
|
|
|
|
/* j -- may be negative! */
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i = 0;
|
|
|
|
|
if (j < 0)
|
|
|
|
|
j = 0; /* Avoid signed/unsigned bug in next line */
|
|
|
|
|
if (j > a->ob_size)
|
|
|
|
|
j = a->ob_size;
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (i == 0 && j == a->ob_size && PyString_CheckExact(a)) {
|
|
|
|
|
/* It's the same as a */
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
if (j < i)
|
|
|
|
|
j = i;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return PyString_FromStringAndSize(a->ob_sval + i, j-i);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
2000-03-07 11:53:43 -04:00
|
|
|
|
static int
|
2006-05-27 07:05:10 -03:00
|
|
|
|
string_contains(PyObject *str_obj, PyObject *sub_obj)
|
2000-03-07 11:53:43 -04:00
|
|
|
|
{
|
2006-05-27 07:05:10 -03:00
|
|
|
|
if (!PyString_CheckExact(sub_obj)) {
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2006-05-27 07:05:10 -03:00
|
|
|
|
if (PyUnicode_Check(sub_obj))
|
|
|
|
|
return PyUnicode_Contains(str_obj, sub_obj);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2006-05-27 07:05:10 -03:00
|
|
|
|
if (!PyString_Check(sub_obj)) {
|
2002-08-24 03:57:49 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"'in <string>' requires string as left operand");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2000-03-07 11:53:43 -04:00
|
|
|
|
}
|
2002-08-06 13:58:21 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
return stringlib_contains_obj(str_obj, sub_obj);
|
2000-03-07 11:53:43 -04:00
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PyObject *
|
2006-02-15 13:27:45 -04:00
|
|
|
|
string_item(PyStringObject *a, register Py_ssize_t i)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2006-05-27 07:05:10 -03:00
|
|
|
|
char pchar;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *v;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_IndexError, "string index out of range");
|
1990-10-14 09:07:46 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-05-27 07:05:10 -03:00
|
|
|
|
pchar = a->ob_sval[i];
|
|
|
|
|
v = (PyObject *)characters[pchar & UCHAR_MAX];
|
2001-05-08 19:33:50 -03:00
|
|
|
|
if (v == NULL)
|
2006-05-27 07:05:10 -03:00
|
|
|
|
v = PyString_FromStringAndSize(&pchar, 1);
|
2001-05-08 21:31:40 -03:00
|
|
|
|
else {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
2001-05-08 21:24:55 -03:00
|
|
|
|
Py_INCREF(v);
|
2001-05-08 21:31:40 -03:00
|
|
|
|
}
|
1991-04-04 06:48:33 -04:00
|
|
|
|
return v;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
2001-05-24 13:56:35 -03:00
|
|
|
|
static PyObject*
|
|
|
|
|
string_richcompare(PyStringObject *a, PyStringObject *b, int op)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
2001-05-24 13:56:35 -03:00
|
|
|
|
int c;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len_a, len_b;
|
|
|
|
|
Py_ssize_t min_len;
|
2001-05-24 13:56:35 -03:00
|
|
|
|
PyObject *result;
|
|
|
|
|
|
2001-09-27 17:30:07 -03:00
|
|
|
|
/* Make sure both arguments are strings. */
|
|
|
|
|
if (!(PyString_Check(a) && PyString_Check(b))) {
|
2001-05-24 13:56:35 -03:00
|
|
|
|
result = Py_NotImplemented;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
if (a == b) {
|
|
|
|
|
switch (op) {
|
|
|
|
|
case Py_EQ:case Py_LE:case Py_GE:
|
|
|
|
|
result = Py_True;
|
|
|
|
|
goto out;
|
|
|
|
|
case Py_NE:case Py_LT:case Py_GT:
|
|
|
|
|
result = Py_False;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
1993-10-22 09:04:32 -03:00
|
|
|
|
}
|
2001-05-24 13:56:35 -03:00
|
|
|
|
if (op == Py_EQ) {
|
|
|
|
|
/* Supporting Py_NE here as well does not save
|
|
|
|
|
much time, since Py_NE is rarely used. */
|
|
|
|
|
if (a->ob_size == b->ob_size
|
|
|
|
|
&& (a->ob_sval[0] == b->ob_sval[0]
|
2002-04-15 10:48:52 -03:00
|
|
|
|
&& memcmp(a->ob_sval, b->ob_sval,
|
2001-05-24 13:56:35 -03:00
|
|
|
|
a->ob_size) == 0)) {
|
|
|
|
|
result = Py_True;
|
|
|
|
|
} else {
|
|
|
|
|
result = Py_False;
|
|
|
|
|
}
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
len_a = a->ob_size; len_b = b->ob_size;
|
|
|
|
|
min_len = (len_a < len_b) ? len_a : len_b;
|
|
|
|
|
if (min_len > 0) {
|
|
|
|
|
c = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval);
|
|
|
|
|
if (c==0)
|
|
|
|
|
c = memcmp(a->ob_sval, b->ob_sval, min_len);
|
|
|
|
|
}else
|
|
|
|
|
c = 0;
|
|
|
|
|
if (c == 0)
|
|
|
|
|
c = (len_a < len_b) ? -1 : (len_a > len_b) ? 1 : 0;
|
|
|
|
|
switch (op) {
|
|
|
|
|
case Py_LT: c = c < 0; break;
|
|
|
|
|
case Py_LE: c = c <= 0; break;
|
|
|
|
|
case Py_EQ: assert(0); break; /* unreachable */
|
|
|
|
|
case Py_NE: c = c != 0; break;
|
|
|
|
|
case Py_GT: c = c > 0; break;
|
|
|
|
|
case Py_GE: c = c >= 0; break;
|
|
|
|
|
default:
|
|
|
|
|
result = Py_NotImplemented;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
result = c ? Py_True : Py_False;
|
|
|
|
|
out:
|
|
|
|
|
Py_INCREF(result);
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
_PyString_Eq(PyObject *o1, PyObject *o2)
|
|
|
|
|
{
|
2006-05-27 07:05:10 -03:00
|
|
|
|
PyStringObject *a = (PyStringObject*) o1;
|
|
|
|
|
PyStringObject *b = (PyStringObject*) o2;
|
2001-05-24 13:56:35 -03:00
|
|
|
|
return a->ob_size == b->ob_size
|
|
|
|
|
&& *a->ob_sval == *b->ob_sval
|
|
|
|
|
&& memcmp(a->ob_sval, b->ob_sval, a->ob_size) == 0;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
1993-03-29 06:43:31 -04:00
|
|
|
|
static long
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_hash(PyStringObject *a)
|
1993-03-29 06:43:31 -04:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t len;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
register unsigned char *p;
|
|
|
|
|
register long x;
|
|
|
|
|
|
|
|
|
|
if (a->ob_shash != -1)
|
|
|
|
|
return a->ob_shash;
|
|
|
|
|
len = a->ob_size;
|
|
|
|
|
p = (unsigned char *) a->ob_sval;
|
|
|
|
|
x = *p << 7;
|
1993-03-29 06:43:31 -04:00
|
|
|
|
while (--len >= 0)
|
1996-09-11 17:22:48 -03:00
|
|
|
|
x = (1000003*x) ^ *p++;
|
1993-03-29 06:43:31 -04:00
|
|
|
|
x ^= a->ob_size;
|
|
|
|
|
if (x == -1)
|
|
|
|
|
x = -2;
|
1993-10-22 09:04:32 -03:00
|
|
|
|
a->ob_shash = x;
|
1993-03-29 06:43:31 -04:00
|
|
|
|
return x;
|
|
|
|
|
}
|
|
|
|
|
|
2002-06-11 07:55:12 -03:00
|
|
|
|
static PyObject*
|
|
|
|
|
string_subscript(PyStringObject* self, PyObject* item)
|
|
|
|
|
{
|
2006-08-12 14:03:09 -03:00
|
|
|
|
if (PyIndex_Check(item)) {
|
|
|
|
|
Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
|
2002-06-11 07:55:12 -03:00
|
|
|
|
if (i == -1 && PyErr_Occurred())
|
|
|
|
|
return NULL;
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i += PyString_GET_SIZE(self);
|
2006-03-07 14:50:55 -04:00
|
|
|
|
return string_item(self, i);
|
2002-06-11 07:55:12 -03:00
|
|
|
|
}
|
|
|
|
|
else if (PySlice_Check(item)) {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t start, stop, step, slicelength, cur, i;
|
2002-06-11 07:55:12 -03:00
|
|
|
|
char* source_buf;
|
|
|
|
|
char* result_buf;
|
|
|
|
|
PyObject* result;
|
|
|
|
|
|
2006-03-16 23:29:34 -04:00
|
|
|
|
if (PySlice_GetIndicesEx((PySliceObject*)item,
|
2002-06-11 07:55:12 -03:00
|
|
|
|
PyString_GET_SIZE(self),
|
|
|
|
|
&start, &stop, &step, &slicelength) < 0) {
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (slicelength <= 0) {
|
|
|
|
|
return PyString_FromStringAndSize("", 0);
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
source_buf = PyString_AsString((PyObject*)self);
|
2006-04-11 04:42:36 -03:00
|
|
|
|
result_buf = (char *)PyMem_Malloc(slicelength);
|
2005-10-20 01:15:52 -03:00
|
|
|
|
if (result_buf == NULL)
|
|
|
|
|
return PyErr_NoMemory();
|
2002-06-11 07:55:12 -03:00
|
|
|
|
|
2006-03-16 23:29:34 -04:00
|
|
|
|
for (cur = start, i = 0; i < slicelength;
|
2002-06-11 07:55:12 -03:00
|
|
|
|
cur += step, i++) {
|
|
|
|
|
result_buf[i] = source_buf[cur];
|
|
|
|
|
}
|
2006-03-16 23:29:34 -04:00
|
|
|
|
|
|
|
|
|
result = PyString_FromStringAndSize(result_buf,
|
2002-06-11 07:55:12 -03:00
|
|
|
|
slicelength);
|
|
|
|
|
PyMem_Free(result_buf);
|
|
|
|
|
return result;
|
|
|
|
|
}
|
2006-03-16 23:29:34 -04:00
|
|
|
|
}
|
2002-06-11 07:55:12 -03:00
|
|
|
|
else {
|
2006-03-16 23:29:34 -04:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-06-11 07:55:12 -03:00
|
|
|
|
"string indices must be integers");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
|
|
|
|
string_buffer_getreadbuf(PyStringObject *self, Py_ssize_t index, const void **ptr)
|
1997-05-05 19:15:02 -03:00
|
|
|
|
{
|
|
|
|
|
if ( index != 0 ) {
|
1997-09-08 15:30:11 -03:00
|
|
|
|
PyErr_SetString(PyExc_SystemError,
|
1998-10-07 23:18:52 -03:00
|
|
|
|
"accessing non-existent string segment");
|
1997-05-05 19:15:02 -03:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
*ptr = (void *)self->ob_sval;
|
|
|
|
|
return self->ob_size;
|
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
|
|
|
|
string_buffer_getwritebuf(PyStringObject *self, Py_ssize_t index, const void **ptr)
|
1997-05-05 19:15:02 -03:00
|
|
|
|
{
|
1997-09-08 15:30:11 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
1998-10-01 12:59:48 -03:00
|
|
|
|
"Cannot use string as modifiable buffer");
|
1997-05-05 19:15:02 -03:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
|
|
|
|
string_buffer_getsegcount(PyStringObject *self, Py_ssize_t *lenp)
|
1997-05-05 19:15:02 -03:00
|
|
|
|
{
|
|
|
|
|
if ( lenp )
|
|
|
|
|
*lenp = self->ob_size;
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
2006-02-15 13:27:45 -04:00
|
|
|
|
static Py_ssize_t
|
|
|
|
|
string_buffer_getcharbuf(PyStringObject *self, Py_ssize_t index, const char **ptr)
|
1998-10-07 23:18:52 -03:00
|
|
|
|
{
|
|
|
|
|
if ( index != 0 ) {
|
|
|
|
|
PyErr_SetString(PyExc_SystemError,
|
|
|
|
|
"accessing non-existent string segment");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
*ptr = self->ob_sval;
|
|
|
|
|
return self->ob_size;
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
static PySequenceMethods string_as_sequence = {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
(lenfunc)string_length, /*sq_length*/
|
1994-08-30 05:19:36 -03:00
|
|
|
|
(binaryfunc)string_concat, /*sq_concat*/
|
2006-02-15 13:27:45 -04:00
|
|
|
|
(ssizeargfunc)string_repeat, /*sq_repeat*/
|
|
|
|
|
(ssizeargfunc)string_item, /*sq_item*/
|
|
|
|
|
(ssizessizeargfunc)string_slice, /*sq_slice*/
|
1991-06-04 16:36:32 -03:00
|
|
|
|
0, /*sq_ass_item*/
|
|
|
|
|
0, /*sq_ass_slice*/
|
2000-03-07 11:53:43 -04:00
|
|
|
|
(objobjproc)string_contains /*sq_contains*/
|
1990-10-14 09:07:46 -03:00
|
|
|
|
};
|
|
|
|
|
|
2002-06-11 07:55:12 -03:00
|
|
|
|
static PyMappingMethods string_as_mapping = {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
(lenfunc)string_length,
|
2002-06-11 07:55:12 -03:00
|
|
|
|
(binaryfunc)string_subscript,
|
|
|
|
|
0,
|
|
|
|
|
};
|
|
|
|
|
|
1997-05-05 19:15:02 -03:00
|
|
|
|
static PyBufferProcs string_as_buffer = {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
(readbufferproc)string_buffer_getreadbuf,
|
|
|
|
|
(writebufferproc)string_buffer_getwritebuf,
|
|
|
|
|
(segcountproc)string_buffer_getsegcount,
|
|
|
|
|
(charbufferproc)string_buffer_getcharbuf,
|
1997-05-05 19:15:02 -03:00
|
|
|
|
};
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define LEFTSTRIP 0
|
|
|
|
|
#define RIGHTSTRIP 1
|
|
|
|
|
#define BOTHSTRIP 2
|
|
|
|
|
|
2002-04-12 21:56:08 -03:00
|
|
|
|
/* Arrays indexed by above */
|
2002-04-22 14:42:37 -03:00
|
|
|
|
static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
|
|
|
|
|
|
|
|
|
|
#define STRIPNAME(i) (stripformat[i]+3)
|
2002-04-12 21:56:08 -03:00
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
|
2006-05-26 16:02:09 -03:00
|
|
|
|
/* Don't call if length < 2 */
|
|
|
|
|
#define Py_STRING_MATCH(target, offset, pattern, length) \
|
|
|
|
|
(target[offset] == pattern[0] && \
|
|
|
|
|
target[offset+length-1] == pattern[length-1] && \
|
|
|
|
|
!memcmp(target+offset+1, pattern+1, length-2) )
|
|
|
|
|
|
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
/* Overallocate the initial list to reduce the number of reallocs for small
|
|
|
|
|
split sizes. Eg, "A A A A A A A A A A".split() (10 elements) has three
|
|
|
|
|
resizes, to sizes 4, 8, then 16. Most observed string splits are for human
|
|
|
|
|
text (roughly 11 words per line) and field delimited data (usually 1-10
|
|
|
|
|
fields). For large strings the split algorithms are bandwidth limited
|
|
|
|
|
so increasing the preallocation likely will not improve things.*/
|
|
|
|
|
|
|
|
|
|
#define MAX_PREALLOC 12
|
|
|
|
|
|
|
|
|
|
/* 5 splits gives 6 elements */
|
|
|
|
|
#define PREALLOC_SIZE(maxsplit) \
|
|
|
|
|
(maxsplit >= MAX_PREALLOC ? MAX_PREALLOC : maxsplit+1)
|
|
|
|
|
|
2004-01-04 20:29:51 -04:00
|
|
|
|
#define SPLIT_APPEND(data, left, right) \
|
|
|
|
|
str = PyString_FromStringAndSize((data) + (left), \
|
|
|
|
|
(right) - (left)); \
|
|
|
|
|
if (str == NULL) \
|
|
|
|
|
goto onError; \
|
|
|
|
|
if (PyList_Append(list, str)) { \
|
|
|
|
|
Py_DECREF(str); \
|
|
|
|
|
goto onError; \
|
|
|
|
|
} \
|
|
|
|
|
else \
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
|
2006-05-26 12:21:01 -03:00
|
|
|
|
#define SPLIT_ADD(data, left, right) { \
|
2006-05-26 11:00:45 -03:00
|
|
|
|
str = PyString_FromStringAndSize((data) + (left), \
|
|
|
|
|
(right) - (left)); \
|
|
|
|
|
if (str == NULL) \
|
|
|
|
|
goto onError; \
|
|
|
|
|
if (count < MAX_PREALLOC) { \
|
|
|
|
|
PyList_SET_ITEM(list, count, str); \
|
|
|
|
|
} else { \
|
|
|
|
|
if (PyList_Append(list, str)) { \
|
|
|
|
|
Py_DECREF(str); \
|
|
|
|
|
goto onError; \
|
|
|
|
|
} \
|
|
|
|
|
else \
|
|
|
|
|
Py_DECREF(str); \
|
|
|
|
|
} \
|
2006-05-26 12:21:01 -03:00
|
|
|
|
count++; }
|
2006-05-26 11:00:45 -03:00
|
|
|
|
|
|
|
|
|
/* Always force the list to the expected size. */
|
2006-06-01 02:32:49 -03:00
|
|
|
|
#define FIX_PREALLOC_SIZE(list) ((PyListObject *)list)->ob_size = count
|
2006-05-26 11:00:45 -03:00
|
|
|
|
|
2006-05-26 12:21:01 -03:00
|
|
|
|
#define SKIP_SPACE(s, i, len) { while (i<len && isspace(Py_CHARMASK(s[i]))) i++; }
|
|
|
|
|
#define SKIP_NONSPACE(s, i, len) { while (i<len && !isspace(Py_CHARMASK(s[i]))) i++; }
|
|
|
|
|
#define RSKIP_SPACE(s, i) { while (i>=0 && isspace(Py_CHARMASK(s[i]))) i--; }
|
|
|
|
|
#define RSKIP_NONSPACE(s, i) { while (i>=0 && !isspace(Py_CHARMASK(s[i]))) i--; }
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-04-13 05:52:56 -03:00
|
|
|
|
split_whitespace(const char *s, Py_ssize_t len, Py_ssize_t maxsplit)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-05-26 11:00:45 -03:00
|
|
|
|
Py_ssize_t i, j, count=0;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
PyObject *str;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list = PyList_New(PREALLOC_SIZE(maxsplit));
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 12:21:01 -03:00
|
|
|
|
i = j = 0;
|
|
|
|
|
|
|
|
|
|
while (maxsplit-- > 0) {
|
|
|
|
|
SKIP_SPACE(s, i, len);
|
|
|
|
|
if (i==len) break;
|
|
|
|
|
j = i; i++;
|
|
|
|
|
SKIP_NONSPACE(s, i, len);
|
|
|
|
|
SPLIT_ADD(s, j, i);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2006-05-26 12:21:01 -03:00
|
|
|
|
|
|
|
|
|
if (i < len) {
|
|
|
|
|
/* Only occurs when maxsplit was reached */
|
|
|
|
|
/* Skip any remaining whitespace and copy to end of string */
|
|
|
|
|
SKIP_SPACE(s, i, len);
|
|
|
|
|
if (i != len)
|
|
|
|
|
SPLIT_ADD(s, i, len);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2006-05-26 11:00:45 -03:00
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return list;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
onError:
|
1999-10-12 16:54:53 -03:00
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-04-13 05:52:56 -03:00
|
|
|
|
split_char(const char *s, Py_ssize_t len, char ch, Py_ssize_t maxcount)
|
2004-01-04 20:29:51 -04:00
|
|
|
|
{
|
2006-05-26 11:00:45 -03:00
|
|
|
|
register Py_ssize_t i, j, count=0;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
PyObject *str;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list = PyList_New(PREALLOC_SIZE(maxcount));
|
2004-01-04 20:29:51 -04:00
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 16:02:09 -03:00
|
|
|
|
i = j = 0;
|
|
|
|
|
while ((j < len) && (maxcount-- > 0)) {
|
|
|
|
|
for(; j<len; j++) {
|
|
|
|
|
/* I found that using memchr makes no difference */
|
|
|
|
|
if (s[j] == ch) {
|
|
|
|
|
SPLIT_ADD(s, i, j);
|
|
|
|
|
i = j = j + 1;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
break;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
2004-01-04 20:29:51 -04:00
|
|
|
|
}
|
2006-05-26 16:02:09 -03:00
|
|
|
|
if (i <= len) {
|
|
|
|
|
SPLIT_ADD(s, i, len);
|
2004-01-04 20:29:51 -04:00
|
|
|
|
}
|
2006-05-26 11:00:45 -03:00
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
2004-01-04 20:29:51 -04:00
|
|
|
|
return list;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(split__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.split([sep [,maxsplit]]) -> list of strings\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a list of the words in the string S, using sep as the\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
delimiter string. If maxsplit is given, at most maxsplit\n\
|
2002-08-05 03:28:21 -03:00
|
|
|
|
splits are done. If sep is not specified or is None, any\n\
|
|
|
|
|
whitespace string is a separator.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_split(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len = PyString_GET_SIZE(self), n, i, j;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
Py_ssize_t maxsplit = -1, count=0;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
const char *s = PyString_AS_STRING(self), *sub;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list, *str, *subobj = Py_None;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
#ifdef USE_FAST
|
|
|
|
|
Py_ssize_t pos;
|
|
|
|
|
#endif
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-04-13 05:37:17 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|On:split", &subobj, &maxsplit))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (maxsplit < 0)
|
2006-04-13 04:22:51 -03:00
|
|
|
|
maxsplit = PY_SSIZE_T_MAX;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (subobj == Py_None)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return split_whitespace(s, len, maxsplit);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
n = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyUnicode_Split((PyObject *)self, subobj, maxsplit);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &n))
|
|
|
|
|
return NULL;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (n == 0) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError, "empty separator");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2004-01-04 20:29:51 -04:00
|
|
|
|
else if (n == 1)
|
|
|
|
|
return split_char(s, len, sub[0], maxsplit);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
list = PyList_New(PREALLOC_SIZE(maxsplit));
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 16:02:09 -03:00
|
|
|
|
#ifdef USE_FAST
|
|
|
|
|
i = j = 0;
|
|
|
|
|
while (maxsplit-- > 0) {
|
|
|
|
|
pos = fastsearch(s+i, len-i, sub, n, FAST_SEARCH);
|
|
|
|
|
if (pos < 0)
|
|
|
|
|
break;
|
|
|
|
|
j = i+pos;
|
|
|
|
|
SPLIT_ADD(s, i, j);
|
|
|
|
|
i = j + n;
|
|
|
|
|
}
|
|
|
|
|
#else
|
1999-10-12 16:54:53 -03:00
|
|
|
|
i = j = 0;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
while ((j+n <= len) && (maxsplit-- > 0)) {
|
|
|
|
|
for (; j+n <= len; j++) {
|
|
|
|
|
if (Py_STRING_MATCH(s, j, sub, n)) {
|
|
|
|
|
SPLIT_ADD(s, i, j);
|
|
|
|
|
i = j = j + n;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
break;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
2006-05-26 16:02:09 -03:00
|
|
|
|
#endif
|
|
|
|
|
SPLIT_ADD(s, i, len);
|
2006-05-26 11:00:45 -03:00
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return list;
|
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
onError:
|
1999-10-12 16:54:53 -03:00
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-25 20:27:53 -03:00
|
|
|
|
PyDoc_STRVAR(partition__doc__,
|
|
|
|
|
"S.partition(sep) -> (head, sep, tail)\n\
|
|
|
|
|
\n\
|
|
|
|
|
Searches for the separator sep in S, and returns the part before it,\n\
|
|
|
|
|
the separator itself, and the part after it. If the separator is not\n\
|
|
|
|
|
found, returns S and two empty strings.");
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2006-05-26 06:46:59 -03:00
|
|
|
|
string_partition(PyStringObject *self, PyObject *sep_obj)
|
2006-05-25 20:27:53 -03:00
|
|
|
|
{
|
2006-05-26 14:26:39 -03:00
|
|
|
|
const char *sep;
|
|
|
|
|
Py_ssize_t sep_len;
|
2006-05-25 20:27:53 -03:00
|
|
|
|
|
2006-05-26 05:54:28 -03:00
|
|
|
|
if (PyString_Check(sep_obj)) {
|
|
|
|
|
sep = PyString_AS_STRING(sep_obj);
|
|
|
|
|
sep_len = PyString_GET_SIZE(sep_obj);
|
2006-05-25 20:27:53 -03:00
|
|
|
|
}
|
2006-05-26 05:54:28 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
else if (PyUnicode_Check(sep_obj))
|
2006-05-26 14:26:39 -03:00
|
|
|
|
return PyUnicode_Partition((PyObject *) self, sep_obj);
|
2006-05-25 20:27:53 -03:00
|
|
|
|
#endif
|
2006-05-26 05:54:28 -03:00
|
|
|
|
else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len))
|
2006-05-25 20:27:53 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 16:24:53 -03:00
|
|
|
|
return stringlib_partition(
|
2006-05-26 14:26:39 -03:00
|
|
|
|
(PyObject*) self,
|
|
|
|
|
PyString_AS_STRING(self), PyString_GET_SIZE(self),
|
|
|
|
|
sep_obj, sep, sep_len
|
|
|
|
|
);
|
2006-05-25 20:27:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-26 15:15:38 -03:00
|
|
|
|
PyDoc_STRVAR(rpartition__doc__,
|
2006-09-04 23:21:38 -03:00
|
|
|
|
"S.rpartition(sep) -> (tail, sep, head)\n\
|
2006-05-26 15:15:38 -03:00
|
|
|
|
\n\
|
|
|
|
|
Searches for the separator sep in S, starting at the end of S, and returns\n\
|
|
|
|
|
the part before it, the separator itself, and the part after it. If the\n\
|
2006-09-04 23:21:38 -03:00
|
|
|
|
separator is not found, returns two empty strings and S.");
|
2006-05-26 15:15:38 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_rpartition(PyStringObject *self, PyObject *sep_obj)
|
|
|
|
|
{
|
|
|
|
|
const char *sep;
|
|
|
|
|
Py_ssize_t sep_len;
|
|
|
|
|
|
|
|
|
|
if (PyString_Check(sep_obj)) {
|
|
|
|
|
sep = PyString_AS_STRING(sep_obj);
|
|
|
|
|
sep_len = PyString_GET_SIZE(sep_obj);
|
|
|
|
|
}
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
else if (PyUnicode_Check(sep_obj))
|
|
|
|
|
return PyUnicode_Partition((PyObject *) self, sep_obj);
|
|
|
|
|
#endif
|
|
|
|
|
else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 16:24:53 -03:00
|
|
|
|
return stringlib_rpartition(
|
2006-05-26 15:15:38 -03:00
|
|
|
|
(PyObject*) self,
|
|
|
|
|
PyString_AS_STRING(self), PyString_GET_SIZE(self),
|
|
|
|
|
sep_obj, sep, sep_len
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-04-13 05:52:56 -03:00
|
|
|
|
rsplit_whitespace(const char *s, Py_ssize_t len, Py_ssize_t maxsplit)
|
2003-12-15 14:49:53 -04:00
|
|
|
|
{
|
2006-05-26 11:00:45 -03:00
|
|
|
|
Py_ssize_t i, j, count=0;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
PyObject *str;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list = PyList_New(PREALLOC_SIZE(maxsplit));
|
2003-12-15 14:49:53 -04:00
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 12:21:01 -03:00
|
|
|
|
i = j = len-1;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-26 12:21:01 -03:00
|
|
|
|
while (maxsplit-- > 0) {
|
|
|
|
|
RSKIP_SPACE(s, i);
|
|
|
|
|
if (i<0) break;
|
|
|
|
|
j = i; i--;
|
|
|
|
|
RSKIP_NONSPACE(s, i);
|
|
|
|
|
SPLIT_ADD(s, i + 1, j + 1);
|
2003-12-15 14:49:53 -04:00
|
|
|
|
}
|
2006-05-26 12:21:01 -03:00
|
|
|
|
if (i >= 0) {
|
|
|
|
|
/* Only occurs when maxsplit was reached */
|
|
|
|
|
/* Skip any remaining whitespace and copy to beginning of string */
|
|
|
|
|
RSKIP_SPACE(s, i);
|
|
|
|
|
if (i >= 0)
|
|
|
|
|
SPLIT_ADD(s, 0, i + 1);
|
|
|
|
|
|
2003-12-15 14:49:53 -04:00
|
|
|
|
}
|
2006-05-26 11:00:45 -03:00
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
2006-05-25 16:19:05 -03:00
|
|
|
|
if (PyList_Reverse(list) < 0)
|
|
|
|
|
goto onError;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
return list;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
onError:
|
2003-12-15 14:49:53 -04:00
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-04-13 05:52:56 -03:00
|
|
|
|
rsplit_char(const char *s, Py_ssize_t len, char ch, Py_ssize_t maxcount)
|
2004-01-04 20:29:51 -04:00
|
|
|
|
{
|
2006-05-26 11:00:45 -03:00
|
|
|
|
register Py_ssize_t i, j, count=0;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
PyObject *str;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list = PyList_New(PREALLOC_SIZE(maxcount));
|
2004-01-04 20:29:51 -04:00
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-26 16:02:09 -03:00
|
|
|
|
i = j = len - 1;
|
|
|
|
|
while ((i >= 0) && (maxcount-- > 0)) {
|
|
|
|
|
for (; i >= 0; i--) {
|
|
|
|
|
if (s[i] == ch) {
|
|
|
|
|
SPLIT_ADD(s, i + 1, j + 1);
|
|
|
|
|
j = i = i - 1;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
break;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
2004-01-04 20:29:51 -04:00
|
|
|
|
}
|
|
|
|
|
if (j >= -1) {
|
2006-05-26 11:00:45 -03:00
|
|
|
|
SPLIT_ADD(s, 0, j + 1);
|
2004-01-04 20:29:51 -04:00
|
|
|
|
}
|
2006-05-26 11:00:45 -03:00
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
2006-05-25 16:19:05 -03:00
|
|
|
|
if (PyList_Reverse(list) < 0)
|
|
|
|
|
goto onError;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
return list;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2003-12-15 14:49:53 -04:00
|
|
|
|
|
|
|
|
|
PyDoc_STRVAR(rsplit__doc__,
|
|
|
|
|
"S.rsplit([sep [,maxsplit]]) -> list of strings\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a list of the words in the string S, using sep as the\n\
|
|
|
|
|
delimiter string, starting at the end of the string and working\n\
|
|
|
|
|
to the front. If maxsplit is given, at most maxsplit splits are\n\
|
|
|
|
|
done. If sep is not specified or is None, any whitespace string\n\
|
|
|
|
|
is a separator.");
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_rsplit(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len = PyString_GET_SIZE(self), n, i, j;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
Py_ssize_t maxsplit = -1, count=0;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
const char *s = PyString_AS_STRING(self), *sub;
|
2006-05-26 11:00:45 -03:00
|
|
|
|
PyObject *list, *str, *subobj = Py_None;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
|
2006-04-13 05:37:17 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|On:rsplit", &subobj, &maxsplit))
|
2003-12-15 14:49:53 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
if (maxsplit < 0)
|
2006-04-13 04:22:51 -03:00
|
|
|
|
maxsplit = PY_SSIZE_T_MAX;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
if (subobj == Py_None)
|
|
|
|
|
return rsplit_whitespace(s, len, maxsplit);
|
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
n = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyUnicode_RSplit((PyObject *)self, subobj, maxsplit);
|
|
|
|
|
#endif
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &n))
|
|
|
|
|
return NULL;
|
2004-01-04 20:29:51 -04:00
|
|
|
|
|
2003-12-15 14:49:53 -04:00
|
|
|
|
if (n == 0) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError, "empty separator");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2004-01-04 20:29:51 -04:00
|
|
|
|
else if (n == 1)
|
|
|
|
|
return rsplit_char(s, len, sub[0], maxsplit);
|
2003-12-15 14:49:53 -04:00
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
list = PyList_New(PREALLOC_SIZE(maxsplit));
|
2003-12-15 14:49:53 -04:00
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
j = len;
|
|
|
|
|
i = j - n;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
|
|
|
|
|
while ( (i >= 0) && (maxsplit-- > 0) ) {
|
|
|
|
|
for (; i>=0; i--) {
|
|
|
|
|
if (Py_STRING_MATCH(s, i, sub, n)) {
|
|
|
|
|
SPLIT_ADD(s, i + n, j);
|
|
|
|
|
j = i;
|
|
|
|
|
i -= n;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
break;
|
2006-05-26 16:02:09 -03:00
|
|
|
|
}
|
2003-12-15 14:49:53 -04:00
|
|
|
|
}
|
|
|
|
|
}
|
2006-05-26 11:00:45 -03:00
|
|
|
|
SPLIT_ADD(s, 0, j);
|
|
|
|
|
FIX_PREALLOC_SIZE(list);
|
|
|
|
|
if (PyList_Reverse(list) < 0)
|
|
|
|
|
goto onError;
|
2003-12-15 14:49:53 -04:00
|
|
|
|
return list;
|
|
|
|
|
|
2006-05-26 11:00:45 -03:00
|
|
|
|
onError:
|
2003-12-15 14:49:53 -04:00
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(join__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.join(sequence) -> string\n\
|
|
|
|
|
\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Return a string which is the concatenation of the strings in the\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
sequence. The separator between elements is S.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_join(PyStringObject *self, PyObject *orig)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
|
|
|
|
char *sep = PyString_AS_STRING(self);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
const Py_ssize_t seplen = PyString_GET_SIZE(self);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
PyObject *res = NULL;
|
|
|
|
|
char *p;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t seqlen = 0;
|
2001-01-18 23:03:47 -04:00
|
|
|
|
size_t sz = 0;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i;
|
2001-08-16 10:15:00 -03:00
|
|
|
|
PyObject *seq, *item;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2001-01-18 23:03:47 -04:00
|
|
|
|
seq = PySequence_Fast(orig, "");
|
|
|
|
|
if (seq == NULL) {
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-07-10 18:30:28 -03:00
|
|
|
|
}
|
2001-01-18 23:03:47 -04:00
|
|
|
|
|
2000-07-12 10:05:33 -03:00
|
|
|
|
seqlen = PySequence_Size(seq);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
if (seqlen == 0) {
|
|
|
|
|
Py_DECREF(seq);
|
|
|
|
|
return PyString_FromString("");
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (seqlen == 1) {
|
2000-07-10 18:30:28 -03:00
|
|
|
|
item = PySequence_Fast_GET_ITEM(seq, 0);
|
2004-08-23 20:23:54 -03:00
|
|
|
|
if (PyString_CheckExact(item) || PyUnicode_CheckExact(item)) {
|
|
|
|
|
Py_INCREF(item);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
Py_DECREF(seq);
|
2004-08-23 20:23:54 -03:00
|
|
|
|
return item;
|
2001-01-18 23:03:47 -04:00
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2000-07-10 18:30:28 -03:00
|
|
|
|
|
2004-08-23 20:23:54 -03:00
|
|
|
|
/* There are at least two things to join, or else we have a subclass
|
2006-03-16 23:29:34 -04:00
|
|
|
|
* of the builtin types in the sequence.
|
2004-08-23 20:23:54 -03:00
|
|
|
|
* Do a pre-pass to figure out the total amount of space we'll
|
|
|
|
|
* need (sz), see whether any argument is absurd, and defer to
|
|
|
|
|
* the Unicode join if appropriate.
|
2001-01-18 23:03:47 -04:00
|
|
|
|
*/
|
2000-07-10 18:30:28 -03:00
|
|
|
|
for (i = 0; i < seqlen; i++) {
|
2001-01-18 23:03:47 -04:00
|
|
|
|
const size_t old_sz = sz;
|
2000-07-10 18:30:28 -03:00
|
|
|
|
item = PySequence_Fast_GET_ITEM(seq, i);
|
|
|
|
|
if (!PyString_Check(item)){
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-07-10 18:30:28 -03:00
|
|
|
|
if (PyUnicode_Check(item)) {
|
2001-05-05 02:36:48 -03:00
|
|
|
|
/* Defer to Unicode join.
|
|
|
|
|
* CAUTION: There's no gurantee that the
|
|
|
|
|
* original sequence can be iterated over
|
|
|
|
|
* again, so we must pass seq here.
|
|
|
|
|
*/
|
|
|
|
|
PyObject *result;
|
|
|
|
|
result = PyUnicode_Join((PyObject *)self, seq);
|
2000-07-11 01:58:12 -03:00
|
|
|
|
Py_DECREF(seq);
|
2001-05-05 02:36:48 -03:00
|
|
|
|
return result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2000-07-10 18:30:28 -03:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2006-04-17 02:56:32 -03:00
|
|
|
|
"sequence item %zd: expected string,"
|
2000-07-11 17:55:38 -03:00
|
|
|
|
" %.80s found",
|
2006-04-17 02:56:32 -03:00
|
|
|
|
i, item->ob_type->tp_name);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
Py_DECREF(seq);
|
|
|
|
|
return NULL;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2001-01-18 23:03:47 -04:00
|
|
|
|
sz += PyString_GET_SIZE(item);
|
|
|
|
|
if (i != 0)
|
|
|
|
|
sz += seplen;
|
2006-04-13 04:22:51 -03:00
|
|
|
|
if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
|
2001-01-18 23:03:47 -04:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
2006-06-10 03:40:50 -03:00
|
|
|
|
"join() result is too long for a Python string");
|
2001-01-18 23:03:47 -04:00
|
|
|
|
Py_DECREF(seq);
|
|
|
|
|
return NULL;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2001-01-18 23:03:47 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Allocate result space. */
|
2006-02-15 13:27:45 -04:00
|
|
|
|
res = PyString_FromStringAndSize((char*)NULL, sz);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
if (res == NULL) {
|
|
|
|
|
Py_DECREF(seq);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Catenate everything. */
|
|
|
|
|
p = PyString_AS_STRING(res);
|
|
|
|
|
for (i = 0; i < seqlen; ++i) {
|
|
|
|
|
size_t n;
|
|
|
|
|
item = PySequence_Fast_GET_ITEM(seq, i);
|
|
|
|
|
n = PyString_GET_SIZE(item);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(p, PyString_AS_STRING(item), n);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
p += n;
|
|
|
|
|
if (i < seqlen - 1) {
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(p, sep, seplen);
|
2000-07-10 18:30:28 -03:00
|
|
|
|
p += seplen;
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2000-07-11 00:28:17 -03:00
|
|
|
|
Py_DECREF(seq);
|
2001-01-18 23:03:47 -04:00
|
|
|
|
return res;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2001-06-16 02:42:57 -03:00
|
|
|
|
PyObject *
|
|
|
|
|
_PyString_Join(PyObject *sep, PyObject *x)
|
2001-06-16 02:11:17 -03:00
|
|
|
|
{
|
|
|
|
|
assert(sep != NULL && PyString_Check(sep));
|
|
|
|
|
assert(x != NULL);
|
2001-08-16 10:15:00 -03:00
|
|
|
|
return string_join((PyStringObject *)sep, x);
|
2001-06-16 02:11:17 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(void)
|
2006-02-15 13:27:45 -04:00
|
|
|
|
string_adjust_indices(Py_ssize_t *start, Py_ssize_t *end, Py_ssize_t len)
|
2002-06-13 21:50:42 -03:00
|
|
|
|
{
|
|
|
|
|
if (*end > len)
|
|
|
|
|
*end = len;
|
|
|
|
|
else if (*end < 0)
|
|
|
|
|
*end += len;
|
|
|
|
|
if (*end < 0)
|
|
|
|
|
*end = 0;
|
|
|
|
|
if (*start < 0)
|
|
|
|
|
*start += len;
|
|
|
|
|
if (*start < 0)
|
|
|
|
|
*start = 0;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(Py_ssize_t)
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_find_internal(PyStringObject *self, PyObject *args, int dir)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *subobj;
|
2006-05-27 12:26:19 -03:00
|
|
|
|
const char *sub;
|
|
|
|
|
Py_ssize_t sub_len;
|
|
|
|
|
Py_ssize_t start=0, end=PY_SSIZE_T_MAX;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-27 12:26:19 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:find/rfind/index/rindex", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return -2;
|
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
2006-05-27 12:26:19 -03:00
|
|
|
|
sub_len = PyString_GET_SIZE(subobj);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyUnicode_Check(subobj))
|
2006-05-27 12:26:19 -03:00
|
|
|
|
return PyUnicode_Find(
|
|
|
|
|
(PyObject *)self, subobj, start, end, dir);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2006-05-27 12:26:19 -03:00
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &sub_len))
|
2006-05-28 09:06:46 -03:00
|
|
|
|
/* XXX - the "expected a character buffer object" is pretty
|
|
|
|
|
confusing for a non-expert. remap to something else ? */
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return -2;
|
|
|
|
|
|
2006-05-26 16:48:07 -03:00
|
|
|
|
if (dir > 0)
|
2006-05-27 12:26:19 -03:00
|
|
|
|
return stringlib_find_slice(
|
|
|
|
|
PyString_AS_STRING(self), PyString_GET_SIZE(self),
|
|
|
|
|
sub, sub_len, start, end);
|
2006-05-26 16:48:07 -03:00
|
|
|
|
else
|
2006-05-27 12:26:19 -03:00
|
|
|
|
return stringlib_rfind_slice(
|
|
|
|
|
PyString_AS_STRING(self), PyString_GET_SIZE(self),
|
|
|
|
|
sub, sub_len, start, end);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(find__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.find(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return the lowest index in S where substring sub is found,\n\
|
|
|
|
|
such that sub is contained within s[start,end]. Optional\n\
|
|
|
|
|
arguments start and end are interpreted as in slice notation.\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Return -1 on failure.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_find(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t result = string_find_internal(self, args, +1);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return PyInt_FromSsize_t(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(index__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.index(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Like S.find() but raise ValueError when the substring is not found.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_index(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t result = string_find_internal(self, args, +1);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
if (result == -1) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
2003-01-15 01:32:57 -04:00
|
|
|
|
"substring not found");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return PyInt_FromSsize_t(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(rfind__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.rfind(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return the highest index in S where substring sub is found,\n\
|
|
|
|
|
such that sub is contained within s[start,end]. Optional\n\
|
|
|
|
|
arguments start and end are interpreted as in slice notation.\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Return -1 on failure.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_rfind(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t result = string_find_internal(self, args, -1);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return PyInt_FromSsize_t(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(rindex__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.rindex(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Like S.rfind() but raise ValueError when the substring is not found.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_rindex(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t result = string_find_internal(self, args, -1);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
if (result == -1) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
2003-01-15 01:32:57 -04:00
|
|
|
|
"substring not found");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return PyInt_FromSsize_t(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2002-04-12 21:56:08 -03:00
|
|
|
|
do_xstrip(PyStringObject *self, int striptype, PyObject *sepobj)
|
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len = PyString_GET_SIZE(self);
|
2002-04-12 21:56:08 -03:00
|
|
|
|
char *sep = PyString_AS_STRING(sepobj);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t seplen = PyString_GET_SIZE(sepobj);
|
|
|
|
|
Py_ssize_t i, j;
|
2002-04-12 21:56:08 -03:00
|
|
|
|
|
|
|
|
|
i = 0;
|
|
|
|
|
if (striptype != RIGHTSTRIP) {
|
|
|
|
|
while (i < len && memchr(sep, Py_CHARMASK(s[i]), seplen)) {
|
|
|
|
|
i++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
j = len;
|
|
|
|
|
if (striptype != LEFTSTRIP) {
|
|
|
|
|
do {
|
|
|
|
|
j--;
|
|
|
|
|
} while (j >= i && memchr(sep, Py_CHARMASK(s[j]), seplen));
|
|
|
|
|
j++;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (i == 0 && j == len && PyString_CheckExact(self)) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*)self;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
return PyString_FromStringAndSize(s+i, j-i);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2001-08-16 10:15:00 -03:00
|
|
|
|
do_strip(PyStringObject *self, int striptype)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t len = PyString_GET_SIZE(self), i, j;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
i = 0;
|
|
|
|
|
if (striptype != RIGHTSTRIP) {
|
|
|
|
|
while (i < len && isspace(Py_CHARMASK(s[i]))) {
|
|
|
|
|
i++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
j = len;
|
|
|
|
|
if (striptype != LEFTSTRIP) {
|
|
|
|
|
do {
|
|
|
|
|
j--;
|
|
|
|
|
} while (j >= i && isspace(Py_CHARMASK(s[j])));
|
|
|
|
|
j++;
|
|
|
|
|
}
|
|
|
|
|
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (i == 0 && j == len && PyString_CheckExact(self)) {
|
1999-10-12 16:54:53 -03:00
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*)self;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
return PyString_FromStringAndSize(s+i, j-i);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2002-04-12 21:56:08 -03:00
|
|
|
|
do_argstrip(PyStringObject *self, int striptype, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
PyObject *sep = NULL;
|
|
|
|
|
|
2002-04-22 14:42:37 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
|
2002-04-12 21:56:08 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (sep != NULL && sep != Py_None) {
|
2002-04-22 14:42:37 -03:00
|
|
|
|
if (PyString_Check(sep))
|
|
|
|
|
return do_xstrip(self, striptype, sep);
|
2002-05-13 06:00:41 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2002-04-22 14:42:37 -03:00
|
|
|
|
else if (PyUnicode_Check(sep)) {
|
|
|
|
|
PyObject *uniself = PyUnicode_FromObject((PyObject *)self);
|
|
|
|
|
PyObject *res;
|
|
|
|
|
if (uniself==NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
res = _PyUnicode_XStrip((PyUnicodeObject *)uniself,
|
|
|
|
|
striptype, sep);
|
|
|
|
|
Py_DECREF(uniself);
|
|
|
|
|
return res;
|
|
|
|
|
}
|
2002-05-13 06:00:41 -03:00
|
|
|
|
#endif
|
2006-04-06 05:17:41 -03:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2002-05-13 06:00:41 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2006-04-06 05:17:41 -03:00
|
|
|
|
"%s arg must be None, str or unicode",
|
2002-05-13 06:00:41 -03:00
|
|
|
|
#else
|
2006-04-06 05:17:41 -03:00
|
|
|
|
"%s arg must be None or str",
|
2002-05-13 06:00:41 -03:00
|
|
|
|
#endif
|
2006-04-06 05:17:41 -03:00
|
|
|
|
STRIPNAME(striptype));
|
|
|
|
|
return NULL;
|
2002-04-12 21:56:08 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return do_strip(self, striptype);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(strip__doc__,
|
2003-04-10 19:35:32 -03:00
|
|
|
|
"S.strip([chars]) -> string or unicode\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with leading and trailing\n\
|
2002-04-12 21:56:08 -03:00
|
|
|
|
whitespace removed.\n\
|
2003-04-10 19:35:32 -03:00
|
|
|
|
If chars is given and not None, remove characters in chars instead.\n\
|
|
|
|
|
If chars is unicode, S will be converted to unicode before stripping");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2002-04-12 21:56:08 -03:00
|
|
|
|
string_strip(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2002-04-12 21:56:08 -03:00
|
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
|
return do_strip(self, BOTHSTRIP); /* Common case */
|
|
|
|
|
else
|
|
|
|
|
return do_argstrip(self, BOTHSTRIP, args);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(lstrip__doc__,
|
2003-04-10 19:35:32 -03:00
|
|
|
|
"S.lstrip([chars]) -> string or unicode\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
2002-04-12 21:56:08 -03:00
|
|
|
|
Return a copy of the string S with leading whitespace removed.\n\
|
2003-04-10 19:35:32 -03:00
|
|
|
|
If chars is given and not None, remove characters in chars instead.\n\
|
|
|
|
|
If chars is unicode, S will be converted to unicode before stripping");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2002-04-12 21:56:08 -03:00
|
|
|
|
string_lstrip(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2002-04-12 21:56:08 -03:00
|
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
|
return do_strip(self, LEFTSTRIP); /* Common case */
|
|
|
|
|
else
|
|
|
|
|
return do_argstrip(self, LEFTSTRIP, args);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(rstrip__doc__,
|
2003-04-10 19:35:32 -03:00
|
|
|
|
"S.rstrip([chars]) -> string or unicode\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
2002-04-12 21:56:08 -03:00
|
|
|
|
Return a copy of the string S with trailing whitespace removed.\n\
|
2003-04-10 19:35:32 -03:00
|
|
|
|
If chars is given and not None, remove characters in chars instead.\n\
|
|
|
|
|
If chars is unicode, S will be converted to unicode before stripping");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2002-04-12 21:56:08 -03:00
|
|
|
|
string_rstrip(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2002-04-12 21:56:08 -03:00
|
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
|
return do_strip(self, RIGHTSTRIP); /* Common case */
|
|
|
|
|
else
|
|
|
|
|
return do_argstrip(self, RIGHTSTRIP, args);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(lower__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.lower() -> string\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Return a copy of the string S converted to lowercase.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 13:10:12 -03:00
|
|
|
|
/* _tolower and _toupper are defined by SUSv2, but they're not ISO C */
|
|
|
|
|
#ifndef _tolower
|
|
|
|
|
#define _tolower tolower
|
|
|
|
|
#endif
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
static PyObject *
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_lower(PyStringObject *self)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-05-25 12:22:03 -03:00
|
|
|
|
char *s;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n = PyString_GET_SIZE(self);
|
2006-04-11 04:42:36 -03:00
|
|
|
|
PyObject *newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 12:49:45 -03:00
|
|
|
|
newobj = PyString_FromStringAndSize(NULL, n);
|
2006-05-25 12:22:03 -03:00
|
|
|
|
if (!newobj)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2006-05-25 12:22:03 -03:00
|
|
|
|
|
|
|
|
|
s = PyString_AS_STRING(newobj);
|
|
|
|
|
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(s, PyString_AS_STRING(self), n);
|
2006-05-25 12:49:45 -03:00
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
for (i = 0; i < n; i++) {
|
2006-05-25 12:49:45 -03:00
|
|
|
|
int c = Py_CHARMASK(s[i]);
|
2006-05-25 12:22:03 -03:00
|
|
|
|
if (isupper(c))
|
|
|
|
|
s[i] = _tolower(c);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2006-05-25 12:22:03 -03:00
|
|
|
|
|
2006-04-11 04:42:36 -03:00
|
|
|
|
return newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(upper__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.upper() -> string\n\
|
|
|
|
|
\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
Return a copy of the string S converted to uppercase.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 13:10:12 -03:00
|
|
|
|
#ifndef _toupper
|
|
|
|
|
#define _toupper toupper
|
|
|
|
|
#endif
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
static PyObject *
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_upper(PyStringObject *self)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-05-25 12:22:03 -03:00
|
|
|
|
char *s;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n = PyString_GET_SIZE(self);
|
2006-04-11 04:42:36 -03:00
|
|
|
|
PyObject *newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 12:49:45 -03:00
|
|
|
|
newobj = PyString_FromStringAndSize(NULL, n);
|
2006-05-25 12:22:03 -03:00
|
|
|
|
if (!newobj)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2006-05-25 12:22:03 -03:00
|
|
|
|
|
|
|
|
|
s = PyString_AS_STRING(newobj);
|
|
|
|
|
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(s, PyString_AS_STRING(self), n);
|
2006-05-25 12:49:45 -03:00
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
for (i = 0; i < n; i++) {
|
2006-05-25 12:49:45 -03:00
|
|
|
|
int c = Py_CHARMASK(s[i]);
|
2006-05-25 12:22:03 -03:00
|
|
|
|
if (islower(c))
|
|
|
|
|
s[i] = _toupper(c);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2006-05-25 12:22:03 -03:00
|
|
|
|
|
2006-04-11 04:42:36 -03:00
|
|
|
|
return newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(title__doc__,
|
2000-03-10 18:55:18 -04:00
|
|
|
|
"S.title() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a titlecased version of S, i.e. words start with uppercase\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
characters, all remaining cased characters have lowercase.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_title(PyStringObject *self)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n = PyString_GET_SIZE(self);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
int previous_is_cased = 0;
|
2006-04-11 04:42:36 -03:00
|
|
|
|
PyObject *newobj;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-04-11 04:42:36 -03:00
|
|
|
|
newobj = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (newobj == NULL)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
2006-04-11 04:42:36 -03:00
|
|
|
|
s_new = PyString_AsString(newobj);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (islower(c)) {
|
|
|
|
|
if (!previous_is_cased)
|
|
|
|
|
c = toupper(c);
|
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
} else if (isupper(c)) {
|
|
|
|
|
if (previous_is_cased)
|
|
|
|
|
c = tolower(c);
|
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
} else
|
|
|
|
|
previous_is_cased = 0;
|
|
|
|
|
*s_new++ = c;
|
|
|
|
|
}
|
2006-04-11 04:42:36 -03:00
|
|
|
|
return newobj;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(capitalize__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.capitalize() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with only its first character\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
capitalized.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_capitalize(PyStringObject *self)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n = PyString_GET_SIZE(self);
|
2006-04-11 04:42:36 -03:00
|
|
|
|
PyObject *newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-04-11 04:42:36 -03:00
|
|
|
|
newobj = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (newobj == NULL)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2006-04-11 04:42:36 -03:00
|
|
|
|
s_new = PyString_AsString(newobj);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
if (0 < n) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (islower(c))
|
|
|
|
|
*s_new = toupper(c);
|
|
|
|
|
else
|
|
|
|
|
*s_new = c;
|
|
|
|
|
s_new++;
|
|
|
|
|
}
|
|
|
|
|
for (i = 1; i < n; i++) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (isupper(c))
|
|
|
|
|
*s_new = tolower(c);
|
|
|
|
|
else
|
|
|
|
|
*s_new = c;
|
|
|
|
|
s_new++;
|
|
|
|
|
}
|
2006-04-11 04:42:36 -03:00
|
|
|
|
return newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(count__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.count(sub[, start[, end]]) -> int\n\
|
|
|
|
|
\n\
|
2006-05-22 12:35:12 -03:00
|
|
|
|
Return the number of non-overlapping occurrences of substring sub in\n\
|
|
|
|
|
string S[start:end]. Optional arguments start and end are interpreted\n\
|
|
|
|
|
as in slice notation.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_count(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-05-27 07:05:10 -03:00
|
|
|
|
PyObject *sub_obj;
|
|
|
|
|
const char *str = PyString_AS_STRING(self), *sub;
|
|
|
|
|
Py_ssize_t sub_len;
|
|
|
|
|
Py_ssize_t start = 0, end = PY_SSIZE_T_MAX;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:count", &sub_obj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
2000-05-08 11:08:05 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
if (PyString_Check(sub_obj)) {
|
|
|
|
|
sub = PyString_AS_STRING(sub_obj);
|
|
|
|
|
sub_len = PyString_GET_SIZE(sub_obj);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2006-05-27 07:05:10 -03:00
|
|
|
|
else if (PyUnicode_Check(sub_obj)) {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t count;
|
2006-05-27 07:05:10 -03:00
|
|
|
|
count = PyUnicode_Count((PyObject *)self, sub_obj, start, end);
|
2001-01-16 07:54:12 -04:00
|
|
|
|
if (count == -1)
|
|
|
|
|
return NULL;
|
|
|
|
|
else
|
2006-05-27 07:05:10 -03:00
|
|
|
|
return PyInt_FromSsize_t(count);
|
2001-01-16 07:54:12 -04:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2006-05-27 07:05:10 -03:00
|
|
|
|
else if (PyObject_AsCharBuffer(sub_obj, &sub, &sub_len))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
string_adjust_indices(&start, &end, PyString_GET_SIZE(self));
|
2002-06-13 21:50:42 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
return PyInt_FromSsize_t(
|
|
|
|
|
stringlib_count(str + start, end - start, sub, sub_len)
|
|
|
|
|
);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(swapcase__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.swapcase() -> string\n\
|
|
|
|
|
\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Return a copy of the string S with uppercase characters\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
converted to lowercase and vice versa.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_swapcase(PyStringObject *self)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n = PyString_GET_SIZE(self);
|
2006-04-11 04:42:36 -03:00
|
|
|
|
PyObject *newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-04-11 04:42:36 -03:00
|
|
|
|
newobj = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (newobj == NULL)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2006-04-11 04:42:36 -03:00
|
|
|
|
s_new = PyString_AsString(newobj);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (islower(c)) {
|
|
|
|
|
*s_new = toupper(c);
|
|
|
|
|
}
|
|
|
|
|
else if (isupper(c)) {
|
|
|
|
|
*s_new = tolower(c);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
*s_new = c;
|
|
|
|
|
s_new++;
|
|
|
|
|
}
|
2006-04-11 04:42:36 -03:00
|
|
|
|
return newobj;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(translate__doc__,
|
1999-10-12 16:54:53 -03:00
|
|
|
|
"S.translate(table [,deletechars]) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S, where all characters occurring\n\
|
|
|
|
|
in the optional argument deletechars are removed, and the\n\
|
|
|
|
|
remaining characters have been mapped through the given\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
translation table, which must be a string of length 256.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_translate(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2000-03-10 18:55:18 -04:00
|
|
|
|
register char *input, *output;
|
|
|
|
|
register const char *table;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t i, c, changed = 0;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
PyObject *input_obj = (PyObject*)self;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
const char *table1, *output_start, *del_table=NULL;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t inlen, tablen, dellen = 0;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
PyObject *result;
|
|
|
|
|
int trans_table[256];
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *tableobj, *delobj = NULL;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2002-12-29 12:33:45 -04:00
|
|
|
|
if (!PyArg_UnpackTuple(args, "translate", 1, 2,
|
2000-03-10 18:55:18 -04:00
|
|
|
|
&tableobj, &delobj))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
if (PyString_Check(tableobj)) {
|
|
|
|
|
table1 = PyString_AS_STRING(tableobj);
|
|
|
|
|
tablen = PyString_GET_SIZE(tableobj);
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyUnicode_Check(tableobj)) {
|
2001-04-28 02:38:26 -03:00
|
|
|
|
/* Unicode .translate() does not support the deletechars
|
2000-03-10 18:55:18 -04:00
|
|
|
|
parameter; instead a mapping to None will cause characters
|
|
|
|
|
to be deleted. */
|
|
|
|
|
if (delobj != NULL) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"deletions are implemented differently for unicode");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
return PyUnicode_Translate((PyObject *)self, tableobj, NULL);
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyObject_AsCharBuffer(tableobj, &table1, &tablen))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2002-12-12 16:03:19 -04:00
|
|
|
|
if (tablen != 256) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"translation table must be 256 characters long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (delobj != NULL) {
|
|
|
|
|
if (PyString_Check(delobj)) {
|
|
|
|
|
del_table = PyString_AS_STRING(delobj);
|
|
|
|
|
dellen = PyString_GET_SIZE(delobj);
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyUnicode_Check(delobj)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"deletions are implemented differently for unicode");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (PyObject_AsCharBuffer(delobj, &del_table, &dellen))
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
del_table = NULL;
|
|
|
|
|
dellen = 0;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
table = table1;
|
2006-03-19 21:53:23 -04:00
|
|
|
|
inlen = PyString_GET_SIZE(input_obj);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
result = PyString_FromStringAndSize((char *)NULL, inlen);
|
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
output_start = output = PyString_AsString(result);
|
2006-03-19 21:53:23 -04:00
|
|
|
|
input = PyString_AS_STRING(input_obj);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
if (dellen == 0) {
|
|
|
|
|
/* If no deletions are required, use faster code */
|
|
|
|
|
for (i = inlen; --i >= 0; ) {
|
|
|
|
|
c = Py_CHARMASK(*input++);
|
|
|
|
|
if (Py_CHARMASK((*output++ = table[c])) != c)
|
|
|
|
|
changed = 1;
|
|
|
|
|
}
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (changed || !PyString_CheckExact(input_obj))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return result;
|
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
Py_INCREF(input_obj);
|
|
|
|
|
return input_obj;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (i = 0; i < 256; i++)
|
|
|
|
|
trans_table[i] = Py_CHARMASK(table[i]);
|
|
|
|
|
|
|
|
|
|
for (i = 0; i < dellen; i++)
|
|
|
|
|
trans_table[(int) Py_CHARMASK(del_table[i])] = -1;
|
|
|
|
|
|
|
|
|
|
for (i = inlen; --i >= 0; ) {
|
|
|
|
|
c = Py_CHARMASK(*input++);
|
|
|
|
|
if (trans_table[c] != -1)
|
|
|
|
|
if (Py_CHARMASK(*output++ = (char)trans_table[c]) == c)
|
|
|
|
|
continue;
|
|
|
|
|
changed = 1;
|
|
|
|
|
}
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (!changed && PyString_CheckExact(input_obj)) {
|
1999-10-12 16:54:53 -03:00
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
Py_INCREF(input_obj);
|
|
|
|
|
return input_obj;
|
|
|
|
|
}
|
|
|
|
|
/* Fix the size of the resulting string */
|
2002-04-27 15:44:32 -03:00
|
|
|
|
if (inlen > 0)
|
|
|
|
|
_PyString_Resize(&result, output - output_start);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
#define FORWARD 1
|
|
|
|
|
#define REVERSE -1
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* find and count characters and substrings */
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
#define findchar(target, target_len, c) \
|
|
|
|
|
((char *)memchr((const void *)(target), c, target_len))
|
|
|
|
|
|
|
|
|
|
/* String ops must return a string. */
|
|
|
|
|
/* If the object is subclass of string, create a copy */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return_self(PyStringObject *self)
|
|
|
|
|
{
|
|
|
|
|
if (PyString_CheckExact(self)) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return self;
|
|
|
|
|
}
|
|
|
|
|
return (PyStringObject *)PyString_FromStringAndSize(
|
|
|
|
|
PyString_AS_STRING(self),
|
|
|
|
|
PyString_GET_SIZE(self));
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(Py_ssize_t)
|
2006-07-30 03:57:04 -03:00
|
|
|
|
countchar(const char *target, int target_len, char c, Py_ssize_t maxcount)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t count=0;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *start=target;
|
|
|
|
|
const char *end=target+target_len;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
while ( (start=findchar(start, end-start, c)) != NULL ) {
|
|
|
|
|
count++;
|
2006-05-26 17:25:22 -03:00
|
|
|
|
if (count >= maxcount)
|
|
|
|
|
break;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start += 1;
|
|
|
|
|
}
|
|
|
|
|
return count;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(Py_ssize_t)
|
2006-07-30 03:57:04 -03:00
|
|
|
|
findstring(const char *target, Py_ssize_t target_len,
|
|
|
|
|
const char *pattern, Py_ssize_t pattern_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t start,
|
|
|
|
|
Py_ssize_t end,
|
|
|
|
|
int direction)
|
|
|
|
|
{
|
|
|
|
|
if (start < 0) {
|
|
|
|
|
start += target_len;
|
|
|
|
|
if (start < 0)
|
|
|
|
|
start = 0;
|
|
|
|
|
}
|
|
|
|
|
if (end > target_len) {
|
|
|
|
|
end = target_len;
|
|
|
|
|
} else if (end < 0) {
|
|
|
|
|
end += target_len;
|
|
|
|
|
if (end < 0)
|
|
|
|
|
end = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* zero-length substrings always match at the first attempt */
|
|
|
|
|
if (pattern_len == 0)
|
|
|
|
|
return (direction > 0) ? start : end;
|
|
|
|
|
|
|
|
|
|
end -= pattern_len;
|
|
|
|
|
|
|
|
|
|
if (direction < 0) {
|
|
|
|
|
for (; end >= start; end--)
|
|
|
|
|
if (Py_STRING_MATCH(target, end, pattern, pattern_len))
|
|
|
|
|
return end;
|
|
|
|
|
} else {
|
|
|
|
|
for (; start <= end; start++)
|
|
|
|
|
if (Py_STRING_MATCH(target, start, pattern, pattern_len))
|
|
|
|
|
return start;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(Py_ssize_t)
|
2006-07-30 03:57:04 -03:00
|
|
|
|
countstring(const char *target, Py_ssize_t target_len,
|
|
|
|
|
const char *pattern, Py_ssize_t pattern_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t start,
|
|
|
|
|
Py_ssize_t end,
|
2006-05-26 17:25:22 -03:00
|
|
|
|
int direction, Py_ssize_t maxcount)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
{
|
|
|
|
|
Py_ssize_t count=0;
|
|
|
|
|
|
|
|
|
|
if (start < 0) {
|
|
|
|
|
start += target_len;
|
|
|
|
|
if (start < 0)
|
|
|
|
|
start = 0;
|
|
|
|
|
}
|
|
|
|
|
if (end > target_len) {
|
|
|
|
|
end = target_len;
|
|
|
|
|
} else if (end < 0) {
|
|
|
|
|
end += target_len;
|
|
|
|
|
if (end < 0)
|
|
|
|
|
end = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* zero-length substrings match everywhere */
|
2006-05-26 17:25:22 -03:00
|
|
|
|
if (pattern_len == 0 || maxcount == 0) {
|
|
|
|
|
if (target_len+1 < maxcount)
|
|
|
|
|
return target_len+1;
|
|
|
|
|
return maxcount;
|
|
|
|
|
}
|
2006-05-25 14:08:14 -03:00
|
|
|
|
|
|
|
|
|
end -= pattern_len;
|
|
|
|
|
if (direction < 0) {
|
2006-05-26 17:25:22 -03:00
|
|
|
|
for (; (end >= start); end--)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (Py_STRING_MATCH(target, end, pattern, pattern_len)) {
|
|
|
|
|
count++;
|
2006-05-26 17:25:22 -03:00
|
|
|
|
if (--maxcount <= 0) break;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
end -= pattern_len-1;
|
|
|
|
|
}
|
|
|
|
|
} else {
|
2006-05-26 17:25:22 -03:00
|
|
|
|
for (; (start <= end); start++)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (Py_STRING_MATCH(target, start, pattern, pattern_len)) {
|
|
|
|
|
count++;
|
2006-05-26 17:25:22 -03:00
|
|
|
|
if (--maxcount <= 0)
|
|
|
|
|
break;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start += pattern_len-1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return count;
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
|
2006-05-27 07:05:10 -03:00
|
|
|
|
/* Algorithms for different cases of string replacement */
|
2006-05-25 14:08:14 -03:00
|
|
|
|
|
|
|
|
|
/* len(self)>=1, from="", len(to)>=1, maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_interleave(PyStringObject *self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *to_s, Py_ssize_t to_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-07-30 03:57:04 -03:00
|
|
|
|
char *self_s, *result_s;
|
|
|
|
|
Py_ssize_t self_len, result_len;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t count, i, product;
|
|
|
|
|
PyStringObject *result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* 1 at the end plus 1 after every character */
|
|
|
|
|
count = self_len+1;
|
|
|
|
|
if (maxcount < count)
|
|
|
|
|
count = maxcount;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* Check for overflow */
|
|
|
|
|
/* result_len = count * to_len + self_len; */
|
|
|
|
|
product = count * to_len;
|
|
|
|
|
if (product / to_len != count) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
result_len = product + self_len;
|
|
|
|
|
if (result_len < 0) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (! (result = (PyStringObject *)
|
|
|
|
|
PyString_FromStringAndSize(NULL, result_len)) )
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
|
|
|
|
|
|
|
|
|
/* TODO: special case single character, which doesn't need memcpy */
|
|
|
|
|
|
|
|
|
|
/* Lay the first one down (guaranteed this will occur) */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
count -= 1;
|
|
|
|
|
|
|
|
|
|
for (i=0; i<count; i++) {
|
|
|
|
|
*result_s++ = *self_s++;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Copy the rest of the original string */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, self_s, self_len-i);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Special case for deleting a single character */
|
|
|
|
|
/* len(self)>=1, len(from)==1, to="", maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_delete_single_character(PyStringObject *self,
|
|
|
|
|
char from_c, Py_ssize_t maxcount)
|
|
|
|
|
{
|
|
|
|
|
char *self_s, *result_s;
|
|
|
|
|
char *start, *next, *end;
|
|
|
|
|
Py_ssize_t self_len, result_len;
|
|
|
|
|
Py_ssize_t count;
|
|
|
|
|
PyStringObject *result;
|
|
|
|
|
|
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
|
2006-05-26 17:25:22 -03:00
|
|
|
|
count = countchar(self_s, self_len, from_c, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (count == 0) {
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
result_len = self_len - count; /* from_len == 1 */
|
|
|
|
|
assert(result_len>=0);
|
|
|
|
|
|
|
|
|
|
if ( (result = (PyStringObject *)
|
|
|
|
|
PyString_FromStringAndSize(NULL, result_len)) == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
|
|
|
|
|
|
|
|
|
start = self_s;
|
|
|
|
|
end = self_s + self_len;
|
|
|
|
|
while (count-- > 0) {
|
|
|
|
|
next = findchar(start, end-start, from_c);
|
|
|
|
|
if (next == NULL)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
break;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, next-start);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += (next-start);
|
|
|
|
|
start = next+1;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, end-start);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* len(self)>=1, len(from)>=2, to="", maxcount>=1 */
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-07-30 03:57:04 -03:00
|
|
|
|
replace_delete_substring(PyStringObject *self,
|
|
|
|
|
const char *from_s, Py_ssize_t from_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount) {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
char *self_s, *result_s;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
char *start, *next, *end;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
Py_ssize_t self_len, result_len;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t count, offset;
|
|
|
|
|
PyStringObject *result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
|
|
|
|
self_s = PyString_AS_STRING(self);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
count = countstring(self_s, self_len,
|
|
|
|
|
from_s, from_len,
|
2006-05-26 17:25:22 -03:00
|
|
|
|
0, self_len, 1,
|
|
|
|
|
maxcount);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (count == 0) {
|
|
|
|
|
/* no matches */
|
|
|
|
|
return return_self(self);
|
2001-05-09 21:05:33 -03:00
|
|
|
|
}
|
2006-05-25 14:08:14 -03:00
|
|
|
|
|
|
|
|
|
result_len = self_len - (count * from_len);
|
|
|
|
|
assert (result_len>=0);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if ( (result = (PyStringObject *)
|
|
|
|
|
PyString_FromStringAndSize(NULL, result_len)) == NULL )
|
|
|
|
|
return NULL;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s = PyString_AS_STRING(result);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start = self_s;
|
|
|
|
|
end = self_s + self_len;
|
|
|
|
|
while (count-- > 0) {
|
|
|
|
|
offset = findstring(start, end-start,
|
|
|
|
|
from_s, from_len,
|
|
|
|
|
0, end-start, FORWARD);
|
|
|
|
|
if (offset == -1)
|
|
|
|
|
break;
|
|
|
|
|
next = start + offset;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, next-start);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += (next-start);
|
|
|
|
|
start = next+from_len;
|
|
|
|
|
}
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, end-start);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* len(self)>=1, len(from)==len(to)==1, maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_single_character_in_place(PyStringObject *self,
|
|
|
|
|
char from_c, char to_c,
|
|
|
|
|
Py_ssize_t maxcount)
|
|
|
|
|
{
|
|
|
|
|
char *self_s, *result_s, *start, *end, *next;
|
|
|
|
|
Py_ssize_t self_len;
|
|
|
|
|
PyStringObject *result;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* The result string will be the same size */
|
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
next = findchar(self_s, self_len, from_c);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (next == NULL) {
|
|
|
|
|
/* No matches; return the original string */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* Need to make a new string */
|
2006-05-25 14:53:00 -03:00
|
|
|
|
result = (PyStringObject *) PyString_FromStringAndSize(NULL, self_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, self_s, self_len);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* change everything in-place, starting with this one */
|
|
|
|
|
start = result_s + (next-self_s);
|
|
|
|
|
*start = to_c;
|
|
|
|
|
start++;
|
|
|
|
|
end = result_s + self_len;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
while (--maxcount > 0) {
|
|
|
|
|
next = findchar(start, end-start, from_c);
|
|
|
|
|
if (next == NULL)
|
|
|
|
|
break;
|
|
|
|
|
*next = to_c;
|
|
|
|
|
start = next+1;
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* len(self)>=1, len(from)==len(to)>=2, maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_substring_in_place(PyStringObject *self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *from_s, Py_ssize_t from_len,
|
|
|
|
|
const char *to_s, Py_ssize_t to_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount)
|
|
|
|
|
{
|
|
|
|
|
char *result_s, *start, *end;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
char *self_s;
|
|
|
|
|
Py_ssize_t self_len, offset;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
PyStringObject *result;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* The result string will be the same size */
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
2006-07-30 03:57:04 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
offset = findstring(self_s, self_len,
|
|
|
|
|
from_s, from_len,
|
|
|
|
|
0, self_len, FORWARD);
|
|
|
|
|
if (offset == -1) {
|
|
|
|
|
/* No matches; return the original string */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* Need to make a new string */
|
2006-05-25 14:53:00 -03:00
|
|
|
|
result = (PyStringObject *) PyString_FromStringAndSize(NULL, self_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, self_s, self_len);
|
2006-05-25 14:53:00 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* change everything in-place, starting with this one */
|
|
|
|
|
start = result_s + offset;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(start, to_s, from_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start += from_len;
|
|
|
|
|
end = result_s + self_len;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
while ( --maxcount > 0) {
|
|
|
|
|
offset = findstring(start, end-start,
|
|
|
|
|
from_s, from_len,
|
|
|
|
|
0, end-start, FORWARD);
|
|
|
|
|
if (offset==-1)
|
|
|
|
|
break;
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(start+offset, to_s, from_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start += offset+from_len;
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* len(self)>=1, len(from)==1, len(to)>=2, maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_single_character(PyStringObject *self,
|
|
|
|
|
char from_c,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *to_s, Py_ssize_t to_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount)
|
|
|
|
|
{
|
2006-07-30 03:57:04 -03:00
|
|
|
|
char *self_s, *result_s;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
char *start, *next, *end;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
Py_ssize_t self_len, result_len;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t count, product;
|
|
|
|
|
PyStringObject *result;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-26 17:25:22 -03:00
|
|
|
|
count = countchar(self_s, self_len, from_c, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (count == 0) {
|
|
|
|
|
/* no matches, return unchanged */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
2006-07-30 03:57:04 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* use the difference between current and new, hence the "-1" */
|
|
|
|
|
/* result_len = self_len + count * (to_len-1) */
|
|
|
|
|
product = count * (to_len-1);
|
|
|
|
|
if (product / (to_len-1) != count) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
result_len = self_len + product;
|
|
|
|
|
if (result_len < 0) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if ( (result = (PyStringObject *)
|
|
|
|
|
PyString_FromStringAndSize(NULL, result_len)) == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start = self_s;
|
|
|
|
|
end = self_s + self_len;
|
|
|
|
|
while (count-- > 0) {
|
|
|
|
|
next = findchar(start, end-start, from_c);
|
|
|
|
|
if (next == NULL)
|
|
|
|
|
break;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (next == start) {
|
|
|
|
|
/* replace with the 'to' */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
start += 1;
|
|
|
|
|
} else {
|
|
|
|
|
/* copy the unchanged old then the 'to' */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, next-start);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += (next-start);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
start = next+1;
|
2001-05-09 21:05:33 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* Copy the remainder of the remaining string */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, end-start);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
|
|
|
|
}
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* len(self)>=1, len(from)>=2, len(to)>=2, maxcount>=1 */
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace_substring(PyStringObject *self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *from_s, Py_ssize_t from_len,
|
|
|
|
|
const char *to_s, Py_ssize_t to_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount) {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
char *self_s, *result_s;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
char *start, *next, *end;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
Py_ssize_t self_len, result_len;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t count, offset, product;
|
|
|
|
|
PyStringObject *result;
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
self_s = PyString_AS_STRING(self);
|
|
|
|
|
self_len = PyString_GET_SIZE(self);
|
2006-07-30 03:57:04 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
count = countstring(self_s, self_len,
|
|
|
|
|
from_s, from_len,
|
2006-05-26 17:25:22 -03:00
|
|
|
|
0, self_len, FORWARD, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (count == 0) {
|
|
|
|
|
/* no matches, return unchanged */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
2006-07-30 03:57:04 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/* Check for overflow */
|
|
|
|
|
/* result_len = self_len + count * (to_len-from_len) */
|
|
|
|
|
product = count * (to_len-from_len);
|
|
|
|
|
if (product / (to_len-from_len) != count) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
result_len = self_len + product;
|
|
|
|
|
if (result_len < 0) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "replace string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if ( (result = (PyStringObject *)
|
|
|
|
|
PyString_FromStringAndSize(NULL, result_len)) == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
result_s = PyString_AS_STRING(result);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
start = self_s;
|
|
|
|
|
end = self_s + self_len;
|
|
|
|
|
while (count-- > 0) {
|
|
|
|
|
offset = findstring(start, end-start,
|
|
|
|
|
from_s, from_len,
|
|
|
|
|
0, end-start, FORWARD);
|
|
|
|
|
if (offset == -1)
|
|
|
|
|
break;
|
|
|
|
|
next = start+offset;
|
|
|
|
|
if (next == start) {
|
|
|
|
|
/* replace with the 'to' */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
start += from_len;
|
|
|
|
|
} else {
|
|
|
|
|
/* copy the unchanged old then the 'to' */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, next-start);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += (next-start);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, to_s, to_len);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
result_s += to_len;
|
|
|
|
|
start = next+from_len;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
/* Copy the remainder of the remaining string */
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(result_s, start, end-start);
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-05-26 13:32:42 -03:00
|
|
|
|
Py_LOCAL(PyStringObject *)
|
2006-05-25 14:08:14 -03:00
|
|
|
|
replace(PyStringObject *self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *from_s, Py_ssize_t from_len,
|
|
|
|
|
const char *to_s, Py_ssize_t to_len,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
Py_ssize_t maxcount)
|
|
|
|
|
{
|
|
|
|
|
if (maxcount < 0) {
|
|
|
|
|
maxcount = PY_SSIZE_T_MAX;
|
|
|
|
|
} else if (maxcount == 0 || PyString_GET_SIZE(self) == 0) {
|
|
|
|
|
/* nothing to do; return the original string */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
2006-07-30 03:57:04 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (maxcount == 0 ||
|
|
|
|
|
(from_len == 0 && to_len == 0)) {
|
|
|
|
|
/* nothing to do; return the original string */
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Handle zero-length special cases */
|
2006-07-30 03:59:13 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (from_len == 0) {
|
|
|
|
|
/* insert the 'to' string everywhere. */
|
|
|
|
|
/* >>> "Python".replace("", ".") */
|
|
|
|
|
/* '.P.y.t.h.o.n.' */
|
2006-07-30 03:57:04 -03:00
|
|
|
|
return replace_interleave(self, to_s, to_len, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Except for "".replace("", "A") == "A" there is no way beyond this */
|
|
|
|
|
/* point for an empty self string to generate a non-empty string */
|
|
|
|
|
/* Special case so the remaining code always gets a non-empty string */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0) {
|
|
|
|
|
return return_self(self);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (to_len == 0) {
|
|
|
|
|
/* delete all occurances of 'from' string */
|
|
|
|
|
if (from_len == 1) {
|
|
|
|
|
return replace_delete_single_character(
|
2006-07-30 03:57:04 -03:00
|
|
|
|
self, from_s[0], maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
} else {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
return replace_delete_substring(self, from_s, from_len, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Handle special case where both strings have the same length */
|
|
|
|
|
|
|
|
|
|
if (from_len == to_len) {
|
|
|
|
|
if (from_len == 1) {
|
|
|
|
|
return replace_single_character_in_place(
|
|
|
|
|
self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
from_s[0],
|
|
|
|
|
to_s[0],
|
2006-05-25 14:08:14 -03:00
|
|
|
|
maxcount);
|
|
|
|
|
} else {
|
|
|
|
|
return replace_substring_in_place(
|
2006-07-30 03:57:04 -03:00
|
|
|
|
self, from_s, from_len, to_s, to_len, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Otherwise use the more generic algorithms */
|
|
|
|
|
if (from_len == 1) {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
return replace_single_character(self, from_s[0],
|
|
|
|
|
to_s, to_len, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
} else {
|
|
|
|
|
/* len('from')>=2, len('to')>=1 */
|
2006-07-30 03:57:04 -03:00
|
|
|
|
return replace_substring(self, from_s, from_len, to_s, to_len, maxcount);
|
2006-05-25 14:08:14 -03:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(replace__doc__,
|
2003-10-21 23:56:40 -03:00
|
|
|
|
"S.replace (old, new[, count]) -> string\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
|
|
|
|
Return a copy of string S with all occurrences of substring\n\
|
2003-10-21 23:56:40 -03:00
|
|
|
|
old replaced by new. If the optional argument count is\n\
|
|
|
|
|
given, only the first count occurrences are replaced.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_replace(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-04-19 12:38:01 -03:00
|
|
|
|
Py_ssize_t count = -1;
|
2006-05-25 14:08:14 -03:00
|
|
|
|
PyObject *from, *to;
|
2006-07-30 03:57:04 -03:00
|
|
|
|
const char *from_s, *to_s;
|
|
|
|
|
Py_ssize_t from_len, to_len;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "OO|n:replace", &from, &to, &count))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (PyString_Check(from)) {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
from_s = PyString_AS_STRING(from);
|
|
|
|
|
from_len = PyString_GET_SIZE(from);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (PyUnicode_Check(from))
|
2001-04-28 02:38:26 -03:00
|
|
|
|
return PyUnicode_Replace((PyObject *)self,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
from, to, count);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2006-07-30 03:57:04 -03:00
|
|
|
|
else if (PyObject_AsCharBuffer(from, &from_s, &from_len))
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
if (PyString_Check(to)) {
|
2006-07-30 03:57:04 -03:00
|
|
|
|
to_s = PyString_AS_STRING(to);
|
|
|
|
|
to_len = PyString_GET_SIZE(to);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2006-05-25 14:08:14 -03:00
|
|
|
|
else if (PyUnicode_Check(to))
|
2001-04-28 02:38:26 -03:00
|
|
|
|
return PyUnicode_Replace((PyObject *)self,
|
2006-05-25 14:08:14 -03:00
|
|
|
|
from, to, count);
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2006-07-30 03:57:04 -03:00
|
|
|
|
else if (PyObject_AsCharBuffer(to, &to_s, &to_len))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
return (PyObject *)replace((PyStringObject *) self,
|
2006-07-30 03:57:04 -03:00
|
|
|
|
from_s, from_len,
|
|
|
|
|
to_s, to_len, count);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-25 14:08:14 -03:00
|
|
|
|
/** End DALKE **/
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2006-06-11 02:47:14 -03:00
|
|
|
|
/* Matches the end (direction >= 0) or start (direction < 0) of self
|
2006-06-09 15:45:48 -03:00
|
|
|
|
* against substr, using the start and end arguments. Returns
|
|
|
|
|
* -1 on error, 0 if not found and 1 if found.
|
|
|
|
|
*/
|
|
|
|
|
Py_LOCAL(int)
|
|
|
|
|
_string_tailmatch(PyStringObject *self, PyObject *substr, Py_ssize_t start,
|
|
|
|
|
Py_ssize_t end, int direction)
|
|
|
|
|
{
|
|
|
|
|
Py_ssize_t len = PyString_GET_SIZE(self);
|
|
|
|
|
Py_ssize_t slen;
|
|
|
|
|
const char* sub;
|
|
|
|
|
const char* str;
|
|
|
|
|
|
|
|
|
|
if (PyString_Check(substr)) {
|
|
|
|
|
sub = PyString_AS_STRING(substr);
|
|
|
|
|
slen = PyString_GET_SIZE(substr);
|
|
|
|
|
}
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
else if (PyUnicode_Check(substr))
|
|
|
|
|
return PyUnicode_Tailmatch((PyObject *)self,
|
|
|
|
|
substr, start, end, direction);
|
|
|
|
|
#endif
|
|
|
|
|
else if (PyObject_AsCharBuffer(substr, &sub, &slen))
|
|
|
|
|
return -1;
|
|
|
|
|
str = PyString_AS_STRING(self);
|
|
|
|
|
|
|
|
|
|
string_adjust_indices(&start, &end, len);
|
|
|
|
|
|
|
|
|
|
if (direction < 0) {
|
|
|
|
|
/* startswith */
|
|
|
|
|
if (start+slen > len)
|
|
|
|
|
return 0;
|
|
|
|
|
} else {
|
|
|
|
|
/* endswith */
|
|
|
|
|
if (end-start < slen || start > len)
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
|
|
if (end-slen > start)
|
|
|
|
|
start = end - slen;
|
|
|
|
|
}
|
2006-06-11 02:47:14 -03:00
|
|
|
|
if (end-start >= slen)
|
|
|
|
|
return ! memcmp(str+start, sub, slen);
|
|
|
|
|
return 0;
|
2006-06-09 15:45:48 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(startswith__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.startswith(prefix[, start[, end]]) -> bool\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
2003-04-09 16:32:45 -03:00
|
|
|
|
Return True if S starts with the specified prefix, False otherwise.\n\
|
|
|
|
|
With optional start, test S beginning at that position.\n\
|
2006-06-09 15:45:48 -03:00
|
|
|
|
With optional end, stop comparing S at that position.\n\
|
|
|
|
|
prefix can also be a tuple of strings to try.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_startswith(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t start = 0;
|
2006-04-13 04:22:51 -03:00
|
|
|
|
Py_ssize_t end = PY_SSIZE_T_MAX;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *subobj;
|
2006-06-09 15:45:48 -03:00
|
|
|
|
int result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2000-05-08 11:08:05 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:startswith", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
2006-06-09 15:45:48 -03:00
|
|
|
|
if (PyTuple_Check(subobj)) {
|
|
|
|
|
Py_ssize_t i;
|
|
|
|
|
for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
|
|
|
|
|
result = _string_tailmatch(self,
|
|
|
|
|
PyTuple_GET_ITEM(subobj, i),
|
|
|
|
|
start, end, -1);
|
|
|
|
|
if (result == -1)
|
|
|
|
|
return NULL;
|
|
|
|
|
else if (result) {
|
|
|
|
|
Py_RETURN_TRUE;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Py_RETURN_FALSE;
|
2001-01-16 07:54:12 -04:00
|
|
|
|
}
|
2006-06-09 15:45:48 -03:00
|
|
|
|
result = _string_tailmatch(self, subobj, start, end, -1);
|
|
|
|
|
if (result == -1)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2002-06-13 21:50:42 -03:00
|
|
|
|
else
|
2006-06-09 15:45:48 -03:00
|
|
|
|
return PyBool_FromLong(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(endswith__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.endswith(suffix[, start[, end]]) -> bool\n\
|
1999-10-12 16:54:53 -03:00
|
|
|
|
\n\
|
2003-04-09 16:32:45 -03:00
|
|
|
|
Return True if S ends with the specified suffix, False otherwise.\n\
|
|
|
|
|
With optional start, test S beginning at that position.\n\
|
2006-06-09 15:45:48 -03:00
|
|
|
|
With optional end, stop comparing S at that position.\n\
|
|
|
|
|
suffix can also be a tuple of strings to try.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
string_endswith(PyStringObject *self, PyObject *args)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t start = 0;
|
2006-04-13 04:22:51 -03:00
|
|
|
|
Py_ssize_t end = PY_SSIZE_T_MAX;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *subobj;
|
2006-06-09 15:45:48 -03:00
|
|
|
|
int result;
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2000-05-08 11:08:05 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:endswith", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
2006-06-09 15:45:48 -03:00
|
|
|
|
if (PyTuple_Check(subobj)) {
|
|
|
|
|
Py_ssize_t i;
|
|
|
|
|
for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
|
|
|
|
|
result = _string_tailmatch(self,
|
|
|
|
|
PyTuple_GET_ITEM(subobj, i),
|
|
|
|
|
start, end, +1);
|
|
|
|
|
if (result == -1)
|
|
|
|
|
return NULL;
|
|
|
|
|
else if (result) {
|
|
|
|
|
Py_RETURN_TRUE;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Py_RETURN_FALSE;
|
2001-01-16 07:54:12 -04:00
|
|
|
|
}
|
2006-06-09 15:45:48 -03:00
|
|
|
|
result = _string_tailmatch(self, subobj, start, end, +1);
|
|
|
|
|
if (result == -1)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
return NULL;
|
2002-06-13 21:50:42 -03:00
|
|
|
|
else
|
2006-06-09 15:45:48 -03:00
|
|
|
|
return PyBool_FromLong(result);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(encode__doc__,
|
2001-05-15 09:00:02 -03:00
|
|
|
|
"S.encode([encoding[,errors]]) -> object\n\
|
2000-07-06 08:29:01 -03:00
|
|
|
|
\n\
|
2001-05-15 09:00:02 -03:00
|
|
|
|
Encodes S using the codec registered for encoding. encoding defaults\n\
|
|
|
|
|
to the default encoding. errors may be given to set a different error\n\
|
2000-07-06 08:29:01 -03:00
|
|
|
|
handling scheme. Default is 'strict' meaning that encoding errors raise\n\
|
2002-09-02 10:14:32 -03:00
|
|
|
|
a UnicodeEncodeError. Other possible values are 'ignore', 'replace' and\n\
|
|
|
|
|
'xmlcharrefreplace' as well as any other name registered with\n\
|
|
|
|
|
codecs.register_error that is able to handle UnicodeEncodeErrors.");
|
2000-07-06 08:29:01 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_encode(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
char *encoding = NULL;
|
|
|
|
|
char *errors = NULL;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
PyObject *v;
|
2006-03-16 23:29:34 -04:00
|
|
|
|
|
2000-07-06 08:29:01 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|ss:encode", &encoding, &errors))
|
|
|
|
|
return NULL;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
v = PyString_AsEncodedObject((PyObject *)self, encoding, errors);
|
2004-07-08 16:13:55 -03:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
if (!PyString_Check(v) && !PyUnicode_Check(v)) {
|
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"encoder did not return a string/unicode object "
|
|
|
|
|
"(type=%.400s)",
|
|
|
|
|
v->ob_type->tp_name);
|
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
return v;
|
2004-07-08 16:13:55 -03:00
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
2001-05-15 09:00:02 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(decode__doc__,
|
2001-05-15 09:00:02 -03:00
|
|
|
|
"S.decode([encoding[,errors]]) -> object\n\
|
|
|
|
|
\n\
|
|
|
|
|
Decodes S using the codec registered for encoding. encoding defaults\n\
|
|
|
|
|
to the default encoding. errors may be given to set a different error\n\
|
|
|
|
|
handling scheme. Default is 'strict' meaning that encoding errors raise\n\
|
2002-09-02 10:14:32 -03:00
|
|
|
|
a UnicodeDecodeError. Other possible values are 'ignore' and 'replace'\n\
|
|
|
|
|
as well as any other name registerd with codecs.register_error that is\n\
|
|
|
|
|
able to handle UnicodeDecodeErrors.");
|
2001-05-15 09:00:02 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_decode(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
char *encoding = NULL;
|
|
|
|
|
char *errors = NULL;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
PyObject *v;
|
2006-03-16 23:29:34 -04:00
|
|
|
|
|
2001-05-15 09:00:02 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|ss:decode", &encoding, &errors))
|
|
|
|
|
return NULL;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
v = PyString_AsDecodedObject((PyObject *)self, encoding, errors);
|
2004-07-08 16:13:55 -03:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
2004-07-08 14:57:32 -03:00
|
|
|
|
if (!PyString_Check(v) && !PyUnicode_Check(v)) {
|
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"decoder did not return a string/unicode object "
|
|
|
|
|
"(type=%.400s)",
|
|
|
|
|
v->ob_type->tp_name);
|
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
return v;
|
2004-07-08 16:13:55 -03:00
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
2000-07-06 08:29:01 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(expandtabs__doc__,
|
2000-03-10 18:55:18 -04:00
|
|
|
|
"S.expandtabs([tabsize]) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of S where all tab characters are expanded using spaces.\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
If tabsize is not given, a tab size of 8 characters is assumed.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_expandtabs(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
const char *e, *p;
|
|
|
|
|
char *q;
|
2007-06-09 01:06:30 -03:00
|
|
|
|
Py_ssize_t i, j, old_j;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *u;
|
|
|
|
|
int tabsize = 8;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-07-16 09:04:32 -03:00
|
|
|
|
/* First pass: determine size of output string */
|
2007-06-09 01:06:30 -03:00
|
|
|
|
i = j = old_j = 0;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = PyString_AS_STRING(self) + PyString_GET_SIZE(self);
|
|
|
|
|
for (p = PyString_AS_STRING(self); p < e; p++)
|
|
|
|
|
if (*p == '\t') {
|
2007-06-09 01:06:30 -03:00
|
|
|
|
if (tabsize > 0) {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
j += tabsize - (j % tabsize);
|
2007-06-09 01:06:30 -03:00
|
|
|
|
if (old_j > j) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "new string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
old_j = j;
|
|
|
|
|
}
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
j++;
|
|
|
|
|
if (*p == '\n' || *p == '\r') {
|
|
|
|
|
i += j;
|
|
|
|
|
j = 0;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2007-06-09 01:06:30 -03:00
|
|
|
|
if ((i + j) < 0) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError, "new string is too long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
/* Second pass: create output string and fill it */
|
|
|
|
|
u = PyString_FromStringAndSize(NULL, i + j);
|
|
|
|
|
if (!u)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
j = 0;
|
|
|
|
|
q = PyString_AS_STRING(u);
|
|
|
|
|
|
|
|
|
|
for (p = PyString_AS_STRING(self); p < e; p++)
|
|
|
|
|
if (*p == '\t') {
|
|
|
|
|
if (tabsize > 0) {
|
|
|
|
|
i = tabsize - (j % tabsize);
|
|
|
|
|
j += i;
|
|
|
|
|
while (i--)
|
|
|
|
|
*q++ = ' ';
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
j++;
|
|
|
|
|
*q++ = *p;
|
|
|
|
|
if (*p == '\n' || *p == '\r')
|
|
|
|
|
j = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return u;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-02-15 13:27:45 -04:00
|
|
|
|
pad(PyStringObject *self, Py_ssize_t left, Py_ssize_t right, char fill)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
|
|
|
|
PyObject *u;
|
|
|
|
|
|
|
|
|
|
if (left < 0)
|
|
|
|
|
left = 0;
|
|
|
|
|
if (right < 0)
|
|
|
|
|
right = 0;
|
|
|
|
|
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (left == 0 && right == 0 && PyString_CheckExact(self)) {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject *)self;
|
|
|
|
|
}
|
|
|
|
|
|
2001-04-28 02:38:26 -03:00
|
|
|
|
u = PyString_FromStringAndSize(NULL,
|
2000-03-10 18:55:18 -04:00
|
|
|
|
left + PyString_GET_SIZE(self) + right);
|
|
|
|
|
if (u) {
|
|
|
|
|
if (left)
|
|
|
|
|
memset(PyString_AS_STRING(u), fill, left);
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(PyString_AS_STRING(u) + left,
|
2001-04-28 02:38:26 -03:00
|
|
|
|
PyString_AS_STRING(self),
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyString_GET_SIZE(self));
|
|
|
|
|
if (right)
|
|
|
|
|
memset(PyString_AS_STRING(u) + left + PyString_GET_SIZE(self),
|
|
|
|
|
fill, right);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return u;
|
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(ljust__doc__,
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"S.ljust(width[, fillchar]) -> string\n"
|
2001-09-11 23:18:30 -03:00
|
|
|
|
"\n"
|
|
|
|
|
"Return S left justified in a string of length width. Padding is\n"
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"done using the specified fill character (default is a space).");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_ljust(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-04-19 11:50:15 -03:00
|
|
|
|
Py_ssize_t width;
|
2003-11-26 04:21:35 -04:00
|
|
|
|
char fillchar = ' ';
|
|
|
|
|
|
2006-04-19 11:50:15 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "n|c:ljust", &width, &fillchar))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (PyString_GET_SIZE(self) >= width && PyString_CheckExact(self)) {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
2003-11-26 04:21:35 -04:00
|
|
|
|
return pad(self, 0, width - PyString_GET_SIZE(self), fillchar);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(rjust__doc__,
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"S.rjust(width[, fillchar]) -> string\n"
|
2001-09-11 23:18:30 -03:00
|
|
|
|
"\n"
|
|
|
|
|
"Return S right justified in a string of length width. Padding is\n"
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"done using the specified fill character (default is a space)");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_rjust(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-04-19 11:50:15 -03:00
|
|
|
|
Py_ssize_t width;
|
2003-11-26 04:21:35 -04:00
|
|
|
|
char fillchar = ' ';
|
|
|
|
|
|
2006-04-19 11:50:15 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "n|c:rjust", &width, &fillchar))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (PyString_GET_SIZE(self) >= width && PyString_CheckExact(self)) {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
2003-11-26 04:21:35 -04:00
|
|
|
|
return pad(self, width - PyString_GET_SIZE(self), 0, fillchar);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(center__doc__,
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"S.center(width[, fillchar]) -> string\n"
|
2001-09-11 23:18:30 -03:00
|
|
|
|
"\n"
|
2003-11-26 04:21:35 -04:00
|
|
|
|
"Return S centered in a string of length width. Padding is\n"
|
|
|
|
|
"done using the specified fill character (default is a space)");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_center(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t marg, left;
|
2006-04-19 11:50:15 -03:00
|
|
|
|
Py_ssize_t width;
|
2003-11-26 04:21:35 -04:00
|
|
|
|
char fillchar = ' ';
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2006-04-19 11:50:15 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "n|c:center", &width, &fillchar))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2001-09-11 23:18:30 -03:00
|
|
|
|
if (PyString_GET_SIZE(self) >= width && PyString_CheckExact(self)) {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
marg = width - PyString_GET_SIZE(self);
|
|
|
|
|
left = marg / 2 + (marg & width & 1);
|
|
|
|
|
|
2003-11-26 04:21:35 -04:00
|
|
|
|
return pad(self, left, marg - left, fillchar);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(zfill__doc__,
|
2002-04-15 10:36:47 -03:00
|
|
|
|
"S.zfill(width) -> string\n"
|
|
|
|
|
"\n"
|
|
|
|
|
"Pad a numeric string S with zeros on the left, to fill a field\n"
|
2002-06-13 17:33:02 -03:00
|
|
|
|
"of the specified width. The string S is never truncated.");
|
2002-04-15 10:36:47 -03:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_zfill(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-02-16 10:32:27 -04:00
|
|
|
|
Py_ssize_t fill;
|
2002-04-15 10:36:47 -03:00
|
|
|
|
PyObject *s;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
char *p;
|
2006-04-19 11:50:15 -03:00
|
|
|
|
Py_ssize_t width;
|
2002-04-15 10:36:47 -03:00
|
|
|
|
|
2006-04-19 11:50:15 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "n:zfill", &width))
|
2002-04-15 10:36:47 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (PyString_GET_SIZE(self) >= width) {
|
2002-04-15 15:42:15 -03:00
|
|
|
|
if (PyString_CheckExact(self)) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
return PyString_FromStringAndSize(
|
|
|
|
|
PyString_AS_STRING(self),
|
|
|
|
|
PyString_GET_SIZE(self)
|
|
|
|
|
);
|
2002-04-15 10:36:47 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fill = width - PyString_GET_SIZE(self);
|
|
|
|
|
|
|
|
|
|
s = pad(self, fill, 0, '0');
|
|
|
|
|
|
|
|
|
|
if (s == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
p = PyString_AS_STRING(s);
|
|
|
|
|
if (p[fill] == '+' || p[fill] == '-') {
|
|
|
|
|
/* move sign to beginning of string */
|
|
|
|
|
p[0] = p[fill];
|
|
|
|
|
p[fill] = '0';
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return (PyObject*) s;
|
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(isspace__doc__,
|
2003-10-18 06:55:08 -03:00
|
|
|
|
"S.isspace() -> bool\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return True if all characters in S are whitespace\n\
|
|
|
|
|
and there is at least one character in S, False otherwise.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_isspace(PyStringObject *self)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char *e;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isspace(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2000-06-14 06:18:32 -03:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-06-14 06:18:32 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isspace(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(isalpha__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.isalpha() -> bool\n\
|
2000-07-05 06:47:46 -03:00
|
|
|
|
\n\
|
2003-10-18 06:55:08 -03:00
|
|
|
|
Return True if all characters in S are alphabetic\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
and there is at least one character in S, False otherwise.");
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_isalpha(PyStringObject *self)
|
2000-07-05 06:47:46 -03:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
register const unsigned char *e;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isalpha(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isalpha(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(isalnum__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.isalnum() -> bool\n\
|
2000-07-05 06:47:46 -03:00
|
|
|
|
\n\
|
2003-10-18 06:55:08 -03:00
|
|
|
|
Return True if all characters in S are alphanumeric\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
and there is at least one character in S, False otherwise.");
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_isalnum(PyStringObject *self)
|
2000-07-05 06:47:46 -03:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
register const unsigned char *e;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isalnum(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
|
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isalnum(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-07-05 06:47:46 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(isdigit__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.isdigit() -> bool\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
\n\
|
2003-10-18 06:55:08 -03:00
|
|
|
|
Return True if all characters in S are digits\n\
|
|
|
|
|
and there is at least one character in S, False otherwise.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_isdigit(PyStringObject *self)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char *e;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isdigit(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2000-06-14 06:18:32 -03:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-06-14 06:18:32 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isdigit(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(1);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(islower__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.islower() -> bool\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
\n\
|
2002-04-03 18:41:51 -04:00
|
|
|
|
Return True if all cased characters in S are lowercase and there is\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
at least one cased character in S, False otherwise.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_islower(PyStringObject *self)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char *e;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
int cased;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(islower(*p) != 0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2000-06-14 06:18:32 -03:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-06-14 06:18:32 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (isupper(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (!cased && islower(*p))
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(cased);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(isupper__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.isupper() -> bool\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
\n\
|
2003-10-18 06:55:08 -03:00
|
|
|
|
Return True if all cased characters in S are uppercase and there is\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
at least one cased character in S, False otherwise.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_isupper(PyStringObject *self)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char *e;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
int cased;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(isupper(*p) != 0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2000-06-14 06:18:32 -03:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-06-14 06:18:32 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (islower(*p))
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
else if (!cased && isupper(*p))
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(cased);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(istitle__doc__,
|
2002-04-03 18:41:51 -04:00
|
|
|
|
"S.istitle() -> bool\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
\n\
|
2003-10-18 06:55:08 -03:00
|
|
|
|
Return True if S is a titlecased string and there is at least one\n\
|
|
|
|
|
character in S, i.e. uppercase characters may only follow uncased\n\
|
|
|
|
|
characters and lowercase characters only cased ones. Return False\n\
|
|
|
|
|
otherwise.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
2001-08-16 10:15:00 -03:00
|
|
|
|
string_istitle(PyStringObject *self, PyObject *uncased)
|
2000-03-10 18:55:18 -04:00
|
|
|
|
{
|
2000-07-09 04:04:36 -03:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char *e;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
int cased, previous_is_cased;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(isupper(*p) != 0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2000-06-14 06:18:32 -03:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-06-14 06:18:32 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
previous_is_cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
2000-05-05 17:44:24 -03:00
|
|
|
|
register const unsigned char ch = *p;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
if (isupper(ch)) {
|
|
|
|
|
if (previous_is_cased)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
else if (islower(ch)) {
|
|
|
|
|
if (!previous_is_cased)
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(0);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
previous_is_cased = 0;
|
|
|
|
|
}
|
2002-04-03 18:41:51 -04:00
|
|
|
|
return PyBool_FromLong(cased);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(splitlines__doc__,
|
2001-10-13 12:57:55 -03:00
|
|
|
|
"S.splitlines([keepends]) -> list of strings\n\
|
2000-03-10 18:55:18 -04:00
|
|
|
|
\n\
|
|
|
|
|
Return a list of the lines in S, breaking at line boundaries.\n\
|
2000-04-11 12:39:26 -03:00
|
|
|
|
Line breaks are not included in the resulting list unless keepends\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
is given and true.");
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_splitlines(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
|
register Py_ssize_t i;
|
|
|
|
|
register Py_ssize_t j;
|
|
|
|
|
Py_ssize_t len;
|
2000-04-11 12:39:26 -03:00
|
|
|
|
int keepends = 0;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
PyObject *list;
|
|
|
|
|
PyObject *str;
|
|
|
|
|
char *data;
|
|
|
|
|
|
2000-04-11 12:39:26 -03:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|i:splitlines", &keepends))
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
data = PyString_AS_STRING(self);
|
|
|
|
|
len = PyString_GET_SIZE(self);
|
|
|
|
|
|
2006-05-26 19:49:03 -03:00
|
|
|
|
/* This does not use the preallocated list because splitlines is
|
|
|
|
|
usually run with hundreds of newlines. The overhead of
|
|
|
|
|
switching between PyList_SET_ITEM and append causes about a
|
|
|
|
|
2-3% slowdown for that common case. A smarter implementation
|
|
|
|
|
could move the if check out, so the SET_ITEMs are done first
|
|
|
|
|
and the appends only done when the prealloc buffer is full.
|
|
|
|
|
That's too much work for little gain.*/
|
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
list = PyList_New(0);
|
|
|
|
|
if (!list)
|
|
|
|
|
goto onError;
|
|
|
|
|
|
|
|
|
|
for (i = j = 0; i < len; ) {
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t eol;
|
2000-04-11 12:39:26 -03:00
|
|
|
|
|
2000-03-10 18:55:18 -04:00
|
|
|
|
/* Find a line and append it */
|
|
|
|
|
while (i < len && data[i] != '\n' && data[i] != '\r')
|
|
|
|
|
i++;
|
|
|
|
|
|
|
|
|
|
/* Skip the line break reading CRLF as one line break */
|
2000-04-11 12:39:26 -03:00
|
|
|
|
eol = i;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
if (i < len) {
|
|
|
|
|
if (data[i] == '\r' && i + 1 < len &&
|
|
|
|
|
data[i+1] == '\n')
|
|
|
|
|
i += 2;
|
|
|
|
|
else
|
|
|
|
|
i++;
|
2000-04-11 12:39:26 -03:00
|
|
|
|
if (keepends)
|
|
|
|
|
eol = i;
|
2000-03-10 18:55:18 -04:00
|
|
|
|
}
|
2000-04-11 12:39:26 -03:00
|
|
|
|
SPLIT_APPEND(data, j, eol);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
j = i;
|
|
|
|
|
}
|
|
|
|
|
if (j < len) {
|
|
|
|
|
SPLIT_APPEND(data, j, len);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return list;
|
|
|
|
|
|
|
|
|
|
onError:
|
2006-03-07 11:39:21 -04:00
|
|
|
|
Py_XDECREF(list);
|
2000-03-10 18:55:18 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#undef SPLIT_APPEND
|
2006-05-26 11:00:45 -03:00
|
|
|
|
#undef SPLIT_ADD
|
|
|
|
|
#undef MAX_PREALLOC
|
|
|
|
|
#undef PREALLOC_SIZE
|
2000-03-10 18:55:18 -04:00
|
|
|
|
|
2003-01-29 13:58:45 -04:00
|
|
|
|
static PyObject *
|
|
|
|
|
string_getnewargs(PyStringObject *v)
|
|
|
|
|
{
|
|
|
|
|
return Py_BuildValue("(s#)", v->ob_sval, v->ob_size);
|
|
|
|
|
}
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
2001-04-28 02:38:26 -03:00
|
|
|
|
static PyMethodDef
|
1999-10-12 16:54:53 -03:00
|
|
|
|
string_methods[] = {
|
2000-03-10 18:55:18 -04:00
|
|
|
|
/* Counterparts of the obsolete stropmodule functions; except
|
|
|
|
|
string.maketrans(). */
|
2002-04-15 10:48:52 -03:00
|
|
|
|
{"join", (PyCFunction)string_join, METH_O, join__doc__},
|
|
|
|
|
{"split", (PyCFunction)string_split, METH_VARARGS, split__doc__},
|
2003-12-15 14:49:53 -04:00
|
|
|
|
{"rsplit", (PyCFunction)string_rsplit, METH_VARARGS, rsplit__doc__},
|
2002-04-15 10:48:52 -03:00
|
|
|
|
{"lower", (PyCFunction)string_lower, METH_NOARGS, lower__doc__},
|
|
|
|
|
{"upper", (PyCFunction)string_upper, METH_NOARGS, upper__doc__},
|
2001-08-16 10:15:00 -03:00
|
|
|
|
{"islower", (PyCFunction)string_islower, METH_NOARGS, islower__doc__},
|
|
|
|
|
{"isupper", (PyCFunction)string_isupper, METH_NOARGS, isupper__doc__},
|
|
|
|
|
{"isspace", (PyCFunction)string_isspace, METH_NOARGS, isspace__doc__},
|
|
|
|
|
{"isdigit", (PyCFunction)string_isdigit, METH_NOARGS, isdigit__doc__},
|
|
|
|
|
{"istitle", (PyCFunction)string_istitle, METH_NOARGS, istitle__doc__},
|
|
|
|
|
{"isalpha", (PyCFunction)string_isalpha, METH_NOARGS, isalpha__doc__},
|
|
|
|
|
{"isalnum", (PyCFunction)string_isalnum, METH_NOARGS, isalnum__doc__},
|
2002-04-15 10:48:52 -03:00
|
|
|
|
{"capitalize", (PyCFunction)string_capitalize, METH_NOARGS,
|
|
|
|
|
capitalize__doc__},
|
|
|
|
|
{"count", (PyCFunction)string_count, METH_VARARGS, count__doc__},
|
|
|
|
|
{"endswith", (PyCFunction)string_endswith, METH_VARARGS,
|
|
|
|
|
endswith__doc__},
|
2006-05-26 06:46:59 -03:00
|
|
|
|
{"partition", (PyCFunction)string_partition, METH_O, partition__doc__},
|
2002-04-15 10:48:52 -03:00
|
|
|
|
{"find", (PyCFunction)string_find, METH_VARARGS, find__doc__},
|
|
|
|
|
{"index", (PyCFunction)string_index, METH_VARARGS, index__doc__},
|
|
|
|
|
{"lstrip", (PyCFunction)string_lstrip, METH_VARARGS, lstrip__doc__},
|
|
|
|
|
{"replace", (PyCFunction)string_replace, METH_VARARGS, replace__doc__},
|
|
|
|
|
{"rfind", (PyCFunction)string_rfind, METH_VARARGS, rfind__doc__},
|
|
|
|
|
{"rindex", (PyCFunction)string_rindex, METH_VARARGS, rindex__doc__},
|
|
|
|
|
{"rstrip", (PyCFunction)string_rstrip, METH_VARARGS, rstrip__doc__},
|
2006-05-26 15:15:38 -03:00
|
|
|
|
{"rpartition", (PyCFunction)string_rpartition, METH_O,
|
|
|
|
|
rpartition__doc__},
|
2002-04-15 10:48:52 -03:00
|
|
|
|
{"startswith", (PyCFunction)string_startswith, METH_VARARGS,
|
|
|
|
|
startswith__doc__},
|
|
|
|
|
{"strip", (PyCFunction)string_strip, METH_VARARGS, strip__doc__},
|
|
|
|
|
{"swapcase", (PyCFunction)string_swapcase, METH_NOARGS,
|
|
|
|
|
swapcase__doc__},
|
|
|
|
|
{"translate", (PyCFunction)string_translate, METH_VARARGS,
|
|
|
|
|
translate__doc__},
|
|
|
|
|
{"title", (PyCFunction)string_title, METH_NOARGS, title__doc__},
|
|
|
|
|
{"ljust", (PyCFunction)string_ljust, METH_VARARGS, ljust__doc__},
|
|
|
|
|
{"rjust", (PyCFunction)string_rjust, METH_VARARGS, rjust__doc__},
|
|
|
|
|
{"center", (PyCFunction)string_center, METH_VARARGS, center__doc__},
|
|
|
|
|
{"zfill", (PyCFunction)string_zfill, METH_VARARGS, zfill__doc__},
|
|
|
|
|
{"encode", (PyCFunction)string_encode, METH_VARARGS, encode__doc__},
|
|
|
|
|
{"decode", (PyCFunction)string_decode, METH_VARARGS, decode__doc__},
|
|
|
|
|
{"expandtabs", (PyCFunction)string_expandtabs, METH_VARARGS,
|
|
|
|
|
expandtabs__doc__},
|
|
|
|
|
{"splitlines", (PyCFunction)string_splitlines, METH_VARARGS,
|
|
|
|
|
splitlines__doc__},
|
2003-01-29 13:58:45 -04:00
|
|
|
|
{"__getnewargs__", (PyCFunction)string_getnewargs, METH_NOARGS},
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
|
};
|
|
|
|
|
|
2002-07-17 13:30:39 -03:00
|
|
|
|
static PyObject *
|
2001-08-30 00:11:59 -03:00
|
|
|
|
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
|
|
|
|
|
|
1999-10-12 16:54:53 -03:00
|
|
|
|
static PyObject *
|
2001-08-02 01:15:00 -03:00
|
|
|
|
string_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
1999-10-12 16:54:53 -03:00
|
|
|
|
{
|
2001-08-02 01:15:00 -03:00
|
|
|
|
PyObject *x = NULL;
|
2006-02-27 12:46:16 -04:00
|
|
|
|
static char *kwlist[] = {"object", 0};
|
2001-08-02 01:15:00 -03:00
|
|
|
|
|
2001-08-30 00:11:59 -03:00
|
|
|
|
if (type != &PyString_Type)
|
|
|
|
|
return str_subtype_new(type, args, kwds);
|
2001-08-02 01:15:00 -03:00
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:str", kwlist, &x))
|
|
|
|
|
return NULL;
|
|
|
|
|
if (x == NULL)
|
|
|
|
|
return PyString_FromString("");
|
|
|
|
|
return PyObject_Str(x);
|
1999-10-12 16:54:53 -03:00
|
|
|
|
}
|
|
|
|
|
|
2001-08-30 00:11:59 -03:00
|
|
|
|
static PyObject *
|
|
|
|
|
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
|
|
|
{
|
2001-09-12 02:18:58 -03:00
|
|
|
|
PyObject *tmp, *pnew;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t n;
|
2001-08-30 00:11:59 -03:00
|
|
|
|
|
|
|
|
|
assert(PyType_IsSubtype(type, &PyString_Type));
|
|
|
|
|
tmp = string_new(&PyString_Type, args, kwds);
|
|
|
|
|
if (tmp == NULL)
|
|
|
|
|
return NULL;
|
2001-09-10 22:41:59 -03:00
|
|
|
|
assert(PyString_CheckExact(tmp));
|
2001-09-12 02:18:58 -03:00
|
|
|
|
n = PyString_GET_SIZE(tmp);
|
|
|
|
|
pnew = type->tp_alloc(type, n);
|
|
|
|
|
if (pnew != NULL) {
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(PyString_AS_STRING(pnew), PyString_AS_STRING(tmp), n+1);
|
2001-09-12 02:18:58 -03:00
|
|
|
|
((PyStringObject *)pnew)->ob_shash =
|
|
|
|
|
((PyStringObject *)tmp)->ob_shash;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
((PyStringObject *)pnew)->ob_sstate = SSTATE_NOT_INTERNED;
|
2001-09-12 02:18:58 -03:00
|
|
|
|
}
|
2001-08-31 13:11:15 -03:00
|
|
|
|
Py_DECREF(tmp);
|
2001-09-12 02:18:58 -03:00
|
|
|
|
return pnew;
|
2001-08-30 00:11:59 -03:00
|
|
|
|
}
|
|
|
|
|
|
2002-05-24 16:01:59 -03:00
|
|
|
|
static PyObject *
|
|
|
|
|
basestring_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
|
|
|
{
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-05-31 16:58:02 -03:00
|
|
|
|
"The basestring type cannot be instantiated");
|
2002-05-24 16:01:59 -03:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2002-11-18 12:09:38 -04:00
|
|
|
|
static PyObject *
|
|
|
|
|
string_mod(PyObject *v, PyObject *w)
|
|
|
|
|
{
|
|
|
|
|
if (!PyString_Check(v)) {
|
|
|
|
|
Py_INCREF(Py_NotImplemented);
|
|
|
|
|
return Py_NotImplemented;
|
|
|
|
|
}
|
|
|
|
|
return PyString_Format(v, w);
|
|
|
|
|
}
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(basestring_doc,
|
|
|
|
|
"Type basestring cannot be instantiated; it is the base for str and unicode.");
|
2002-05-24 16:01:59 -03:00
|
|
|
|
|
2002-11-18 12:09:38 -04:00
|
|
|
|
static PyNumberMethods string_as_number = {
|
|
|
|
|
0, /*nb_add*/
|
|
|
|
|
0, /*nb_subtract*/
|
|
|
|
|
0, /*nb_multiply*/
|
|
|
|
|
0, /*nb_divide*/
|
|
|
|
|
string_mod, /*nb_remainder*/
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
2002-05-24 16:01:59 -03:00
|
|
|
|
PyTypeObject PyBaseString_Type = {
|
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
|
|
|
0,
|
2002-05-31 16:58:02 -03:00
|
|
|
|
"basestring",
|
2002-05-24 16:01:59 -03:00
|
|
|
|
0,
|
|
|
|
|
0,
|
|
|
|
|
0, /* tp_dealloc */
|
|
|
|
|
0, /* tp_print */
|
|
|
|
|
0, /* tp_getattr */
|
|
|
|
|
0, /* tp_setattr */
|
|
|
|
|
0, /* tp_compare */
|
|
|
|
|
0, /* tp_repr */
|
|
|
|
|
0, /* tp_as_number */
|
|
|
|
|
0, /* tp_as_sequence */
|
|
|
|
|
0, /* tp_as_mapping */
|
|
|
|
|
0, /* tp_hash */
|
|
|
|
|
0, /* tp_call */
|
|
|
|
|
0, /* tp_str */
|
|
|
|
|
0, /* tp_getattro */
|
|
|
|
|
0, /* tp_setattro */
|
|
|
|
|
0, /* tp_as_buffer */
|
|
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
|
|
|
basestring_doc, /* tp_doc */
|
|
|
|
|
0, /* tp_traverse */
|
|
|
|
|
0, /* tp_clear */
|
|
|
|
|
0, /* tp_richcompare */
|
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
|
0, /* tp_iter */
|
|
|
|
|
0, /* tp_iternext */
|
|
|
|
|
0, /* tp_methods */
|
|
|
|
|
0, /* tp_members */
|
|
|
|
|
0, /* tp_getset */
|
|
|
|
|
&PyBaseObject_Type, /* tp_base */
|
|
|
|
|
0, /* tp_dict */
|
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
|
0, /* tp_descr_set */
|
|
|
|
|
0, /* tp_dictoffset */
|
|
|
|
|
0, /* tp_init */
|
|
|
|
|
0, /* tp_alloc */
|
|
|
|
|
basestring_new, /* tp_new */
|
|
|
|
|
0, /* tp_free */
|
|
|
|
|
};
|
|
|
|
|
|
2002-06-13 17:33:02 -03:00
|
|
|
|
PyDoc_STRVAR(string_doc,
|
2001-08-02 01:15:00 -03:00
|
|
|
|
"str(object) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a nice string representation of the object.\n\
|
2002-06-13 17:33:02 -03:00
|
|
|
|
If the argument is a string, the return value is the same object.");
|
1999-10-12 16:54:53 -03:00
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyTypeObject PyString_Type = {
|
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
0,
|
2001-08-02 01:15:00 -03:00
|
|
|
|
"str",
|
1997-05-02 00:12:38 -03:00
|
|
|
|
sizeof(PyStringObject),
|
1990-10-14 09:07:46 -03:00
|
|
|
|
sizeof(char),
|
2006-03-30 07:57:00 -04:00
|
|
|
|
string_dealloc, /* tp_dealloc */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
(printfunc)string_print, /* tp_print */
|
|
|
|
|
0, /* tp_getattr */
|
|
|
|
|
0, /* tp_setattr */
|
|
|
|
|
0, /* tp_compare */
|
2006-03-30 07:57:00 -04:00
|
|
|
|
string_repr, /* tp_repr */
|
2002-11-18 12:09:38 -04:00
|
|
|
|
&string_as_number, /* tp_as_number */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
&string_as_sequence, /* tp_as_sequence */
|
2002-06-11 07:55:12 -03:00
|
|
|
|
&string_as_mapping, /* tp_as_mapping */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
(hashfunc)string_hash, /* tp_hash */
|
|
|
|
|
0, /* tp_call */
|
2006-03-30 07:57:00 -04:00
|
|
|
|
string_str, /* tp_str */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
|
|
|
0, /* tp_setattro */
|
|
|
|
|
&string_as_buffer, /* tp_as_buffer */
|
2006-03-16 23:29:34 -04:00
|
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
|
2002-11-18 12:09:38 -04:00
|
|
|
|
Py_TPFLAGS_BASETYPE, /* tp_flags */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
string_doc, /* tp_doc */
|
|
|
|
|
0, /* tp_traverse */
|
|
|
|
|
0, /* tp_clear */
|
|
|
|
|
(richcmpfunc)string_richcompare, /* tp_richcompare */
|
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
|
0, /* tp_iter */
|
|
|
|
|
0, /* tp_iternext */
|
|
|
|
|
string_methods, /* tp_methods */
|
|
|
|
|
0, /* tp_members */
|
|
|
|
|
0, /* tp_getset */
|
2002-05-24 16:01:59 -03:00
|
|
|
|
&PyBaseString_Type, /* tp_base */
|
2001-08-02 01:15:00 -03:00
|
|
|
|
0, /* tp_dict */
|
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
|
0, /* tp_descr_set */
|
|
|
|
|
0, /* tp_dictoffset */
|
|
|
|
|
0, /* tp_init */
|
|
|
|
|
0, /* tp_alloc */
|
|
|
|
|
string_new, /* tp_new */
|
2002-04-12 00:05:19 -03:00
|
|
|
|
PyObject_Del, /* tp_free */
|
1990-10-14 09:07:46 -03:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
void
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_Concat(register PyObject **pv, register PyObject *w)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
1997-05-02 00:12:38 -03:00
|
|
|
|
register PyObject *v;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
if (*pv == NULL)
|
|
|
|
|
return;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (w == NULL || !PyString_Check(*pv)) {
|
|
|
|
|
Py_DECREF(*pv);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
*pv = NULL;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
return;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
v = string_concat((PyStringObject *) *pv, w);
|
|
|
|
|
Py_DECREF(*pv);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
*pv = v;
|
|
|
|
|
}
|
|
|
|
|
|
1994-08-30 05:19:36 -03:00
|
|
|
|
void
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_ConcatAndDel(register PyObject **pv, register PyObject *w)
|
1994-08-30 05:19:36 -03:00
|
|
|
|
{
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyString_Concat(pv, w);
|
|
|
|
|
Py_XDECREF(w);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
1990-10-14 09:07:46 -03:00
|
|
|
|
/* The following function breaks the notion that strings are immutable:
|
|
|
|
|
it changes the size of a string. We get away with this only if there
|
|
|
|
|
is only one module referencing the object. You can also think of it
|
|
|
|
|
as creating a new string object and destroying the old one, only
|
|
|
|
|
more efficiently. In any case, don't use this if the string may
|
2002-04-27 15:44:32 -03:00
|
|
|
|
already be known to some other part of the code...
|
|
|
|
|
Note that if there's not enough memory to resize the string, the original
|
|
|
|
|
string object at *pv is deallocated, *pv is set to NULL, an "out of
|
|
|
|
|
memory" exception is set, and -1 is returned. Else (on success) 0 is
|
|
|
|
|
returned, and the value in *pv may or may not be the same as on input.
|
|
|
|
|
As always, an extra byte is allocated for a trailing \0 byte (newsize
|
|
|
|
|
does *not* include that), and a trailing \0 byte is stored.
|
|
|
|
|
*/
|
1990-10-14 09:07:46 -03:00
|
|
|
|
|
|
|
|
|
int
|
2006-02-15 13:27:45 -04:00
|
|
|
|
_PyString_Resize(PyObject **pv, Py_ssize_t newsize)
|
1990-10-14 09:07:46 -03:00
|
|
|
|
{
|
1997-05-02 00:12:38 -03:00
|
|
|
|
register PyObject *v;
|
|
|
|
|
register PyStringObject *sv;
|
1990-11-18 13:30:23 -04:00
|
|
|
|
v = *pv;
|
2004-08-07 17:58:32 -03:00
|
|
|
|
if (!PyString_Check(v) || v->ob_refcnt != 1 || newsize < 0 ||
|
|
|
|
|
PyString_CHECK_INTERNED(v)) {
|
1990-10-14 09:07:46 -03:00
|
|
|
|
*pv = 0;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-21 19:15:08 -03:00
|
|
|
|
return -1;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
1990-11-18 13:30:23 -04:00
|
|
|
|
/* XXX UNREF/NEWREF interface should be more symmetrical */
|
object.h special-build macro minefield: renamed all the new lexical
helper macros to something saner, and used them appropriately in other
files too, to reduce #ifdef blocks.
classobject.c, instance_dealloc(): One of my worst Python Memories is
trying to fix this routine a few years ago when COUNT_ALLOCS was defined
but Py_TRACE_REFS wasn't. The special-build code here is way too
complicated. Now it's much simpler. Difference: in a Py_TRACE_REFS
build, the instance is no longer in the doubly-linked list of live
objects while its __del__ method is executing, and that may be visible
via sys.getobjects() called from a __del__ method. Tough -- the object
is presumed dead while its __del__ is executing anyway, and not calling
_Py_NewReference() at the start allows enormous code simplification.
typeobject.c, call_finalizer(): The special-build instance_dealloc()
pain apparently spread to here too via cut-'n-paste, and this is much
simpler now too. In addition, I didn't understand why this routine
was calling _PyObject_GC_TRACK() after a resurrection, since there's no
plausible way _PyObject_GC_UNTRACK() could have been called on the
object by this point. I suspect it was left over from pasting the
instance_delloc() code. Instead asserted that the object is still
tracked. Caution: I suspect we don't have a test that actually
exercises the subtype_dealloc() __del__-resurrected-me code.
2002-07-11 03:23:50 -03:00
|
|
|
|
_Py_DEC_REFTOTAL;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
_Py_ForgetReference(v);
|
|
|
|
|
*pv = (PyObject *)
|
2004-06-27 14:24:49 -03:00
|
|
|
|
PyObject_REALLOC((char *)v, sizeof(PyStringObject) + newsize);
|
1990-10-14 09:07:46 -03:00
|
|
|
|
if (*pv == NULL) {
|
2002-04-12 00:05:19 -03:00
|
|
|
|
PyObject_Del(v);
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_NoMemory();
|
1990-10-21 19:15:08 -03:00
|
|
|
|
return -1;
|
1990-10-14 09:07:46 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
_Py_NewReference(*pv);
|
|
|
|
|
sv = (PyStringObject *) *pv;
|
1990-11-18 13:30:23 -04:00
|
|
|
|
sv->ob_size = newsize;
|
|
|
|
|
sv->ob_sval[newsize] = '\0';
|
2004-10-25 22:52:37 -03:00
|
|
|
|
sv->ob_shash = -1; /* invalidate cached hash value */
|
1990-10-14 09:07:46 -03:00
|
|
|
|
return 0;
|
|
|
|
|
}
|
1993-03-16 08:15:04 -04:00
|
|
|
|
|
|
|
|
|
/* Helpers for formatstring */
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
2006-02-16 11:59:12 -04:00
|
|
|
|
getnextarg(PyObject *args, Py_ssize_t arglen, Py_ssize_t *p_argidx)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
{
|
2006-02-16 11:59:12 -04:00
|
|
|
|
Py_ssize_t argidx = *p_argidx;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (argidx < arglen) {
|
|
|
|
|
(*p_argidx)++;
|
|
|
|
|
if (arglen < 0)
|
|
|
|
|
return args;
|
|
|
|
|
else
|
1997-05-02 00:12:38 -03:00
|
|
|
|
return PyTuple_GetItem(args, argidx);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"not enough arguments for format string");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2000-09-21 02:43:11 -03:00
|
|
|
|
/* Format codes
|
|
|
|
|
* F_LJUST '-'
|
|
|
|
|
* F_SIGN '+'
|
|
|
|
|
* F_BLANK ' '
|
|
|
|
|
* F_ALT '#'
|
|
|
|
|
* F_ZERO '0'
|
|
|
|
|
*/
|
1993-03-16 08:15:04 -04:00
|
|
|
|
#define F_LJUST (1<<0)
|
|
|
|
|
#define F_SIGN (1<<1)
|
|
|
|
|
#define F_BLANK (1<<2)
|
|
|
|
|
#define F_ALT (1<<3)
|
|
|
|
|
#define F_ZERO (1<<4)
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(int)
|
2000-07-09 04:04:36 -03:00
|
|
|
|
formatfloat(char *buf, size_t buflen, int flags,
|
|
|
|
|
int prec, int type, PyObject *v)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
{
|
2000-06-30 07:29:57 -03:00
|
|
|
|
/* fmt = '%#.' + `prec` + `type`
|
|
|
|
|
worst case length = 3 + 10 (len of INT_MAX) + 1 = 14 (use 20)*/
|
1993-03-16 08:15:04 -04:00
|
|
|
|
char fmt[20];
|
|
|
|
|
double x;
|
2002-07-28 13:44:23 -03:00
|
|
|
|
x = PyFloat_AsDouble(v);
|
|
|
|
|
if (x == -1.0 && PyErr_Occurred()) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError, "float argument required");
|
1997-01-21 12:12:09 -04:00
|
|
|
|
return -1;
|
2002-07-28 13:44:23 -03:00
|
|
|
|
}
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 6;
|
|
|
|
|
if (type == 'f' && fabs(x)/1e25 >= 1e25)
|
|
|
|
|
type = 'g';
|
2002-12-29 15:44:06 -04:00
|
|
|
|
/* Worst case length calc to ensure no buffer overrun:
|
|
|
|
|
|
|
|
|
|
'g' formats:
|
2000-06-30 07:29:57 -03:00
|
|
|
|
fmt = %#.<prec>g
|
|
|
|
|
buf = '-' + [0-9]*prec + '.' + 'e+' + (longest exp
|
2001-04-28 02:38:26 -03:00
|
|
|
|
for any double rep.)
|
2000-06-30 07:29:57 -03:00
|
|
|
|
len = 1 + prec + 1 + 2 + 5 = 9 + prec
|
2002-12-29 15:44:06 -04:00
|
|
|
|
|
|
|
|
|
'f' formats:
|
|
|
|
|
buf = '-' + [0-9]*x + '.' + [0-9]*prec (with x < 50)
|
|
|
|
|
len = 1 + 50 + 1 + prec = 52 + prec
|
|
|
|
|
|
2000-06-30 07:29:57 -03:00
|
|
|
|
If prec=0 the effective precision is 1 (the leading digit is
|
2006-03-16 23:29:34 -04:00
|
|
|
|
always given), therefore increase the length by one.
|
2002-12-29 15:44:06 -04:00
|
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
if ((type == 'g' && buflen <= (size_t)10 + (size_t)prec) ||
|
|
|
|
|
(type == 'f' && buflen <= (size_t)53 + (size_t)prec)) {
|
2000-06-30 07:29:57 -03:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
2000-10-24 16:57:45 -03:00
|
|
|
|
"formatted float is too long (precision too large?)");
|
2000-06-30 07:29:57 -03:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
2002-12-29 15:44:06 -04:00
|
|
|
|
PyOS_snprintf(fmt, sizeof(fmt), "%%%s.%d%c",
|
|
|
|
|
(flags&F_ALT) ? "#" : "",
|
|
|
|
|
prec, type);
|
2004-06-08 15:52:54 -03:00
|
|
|
|
PyOS_ascii_formatd(buf, buflen, fmt, x);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return (int)strlen(buf);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
|
2000-09-21 02:43:11 -03:00
|
|
|
|
/* _PyString_FormatLong emulates the format codes d, u, o, x and X, and
|
|
|
|
|
* the F_ALT flag, for Python's long (unbounded) ints. It's not used for
|
|
|
|
|
* Python's regular ints.
|
|
|
|
|
* Return value: a new PyString*, or NULL if error.
|
|
|
|
|
* . *pbuf is set to point into it,
|
|
|
|
|
* *plen set to the # of chars following that.
|
|
|
|
|
* Caller must decref it when done using pbuf.
|
|
|
|
|
* The string starting at *pbuf is of the form
|
|
|
|
|
* "-"? ("0x" | "0X")? digit+
|
|
|
|
|
* "0x"/"0X" are present only for x and X conversions, with F_ALT
|
2001-04-28 02:38:26 -03:00
|
|
|
|
* set in flags. The case of hex digits will be correct,
|
2000-09-21 02:43:11 -03:00
|
|
|
|
* There will be at least prec digits, zero-filled on the left if
|
|
|
|
|
* necessary to get that many.
|
|
|
|
|
* val object to be converted
|
|
|
|
|
* flags bitmask of format flags; only F_ALT is looked at
|
|
|
|
|
* prec minimum number of digits; 0-fill on left if needed
|
|
|
|
|
* type a character in [duoxX]; u acts the same as d
|
|
|
|
|
*
|
|
|
|
|
* CAUTION: o, x and X conversions on regular ints can never
|
|
|
|
|
* produce a '-' sign, but can for Python's unbounded ints.
|
|
|
|
|
*/
|
|
|
|
|
PyObject*
|
|
|
|
|
_PyString_FormatLong(PyObject *val, int flags, int prec, int type,
|
|
|
|
|
char **pbuf, int *plen)
|
|
|
|
|
{
|
|
|
|
|
PyObject *result = NULL;
|
|
|
|
|
char *buf;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
int sign; /* 1 if '-', else 0 */
|
|
|
|
|
int len; /* number of characters */
|
2006-03-07 08:08:51 -04:00
|
|
|
|
Py_ssize_t llen;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
int numdigits; /* len == numnondigits + numdigits */
|
|
|
|
|
int numnondigits = 0;
|
|
|
|
|
|
|
|
|
|
switch (type) {
|
|
|
|
|
case 'd':
|
|
|
|
|
case 'u':
|
|
|
|
|
result = val->ob_type->tp_str(val);
|
|
|
|
|
break;
|
|
|
|
|
case 'o':
|
|
|
|
|
result = val->ob_type->tp_as_number->nb_oct(val);
|
|
|
|
|
break;
|
|
|
|
|
case 'x':
|
|
|
|
|
case 'X':
|
|
|
|
|
numnondigits = 2;
|
|
|
|
|
result = val->ob_type->tp_as_number->nb_hex(val);
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
assert(!"'type' not in [duoxX]");
|
|
|
|
|
}
|
|
|
|
|
if (!result)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2006-08-13 15:11:08 -03:00
|
|
|
|
buf = PyString_AsString(result);
|
2006-08-14 17:25:39 -03:00
|
|
|
|
if (!buf) {
|
|
|
|
|
Py_DECREF(result);
|
2006-08-13 15:11:08 -03:00
|
|
|
|
return NULL;
|
2006-08-14 17:25:39 -03:00
|
|
|
|
}
|
2006-08-13 15:11:08 -03:00
|
|
|
|
|
2000-09-21 02:43:11 -03:00
|
|
|
|
/* To modify the string in-place, there can only be one reference. */
|
|
|
|
|
if (result->ob_refcnt != 1) {
|
|
|
|
|
PyErr_BadInternalCall();
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2006-03-07 08:08:51 -04:00
|
|
|
|
llen = PyString_Size(result);
|
2006-10-04 08:44:06 -03:00
|
|
|
|
if (llen > INT_MAX) {
|
2006-03-07 08:08:51 -04:00
|
|
|
|
PyErr_SetString(PyExc_ValueError, "string too large in _PyString_FormatLong");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
len = (int)llen;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if (buf[len-1] == 'L') {
|
|
|
|
|
--len;
|
|
|
|
|
buf[len] = '\0';
|
|
|
|
|
}
|
|
|
|
|
sign = buf[0] == '-';
|
|
|
|
|
numnondigits += sign;
|
|
|
|
|
numdigits = len - numnondigits;
|
|
|
|
|
assert(numdigits > 0);
|
|
|
|
|
|
2001-04-12 15:38:48 -03:00
|
|
|
|
/* Get rid of base marker unless F_ALT */
|
|
|
|
|
if ((flags & F_ALT) == 0) {
|
2000-09-21 02:43:11 -03:00
|
|
|
|
/* Need to skip 0x, 0X or 0. */
|
|
|
|
|
int skipped = 0;
|
|
|
|
|
switch (type) {
|
|
|
|
|
case 'o':
|
|
|
|
|
assert(buf[sign] == '0');
|
|
|
|
|
/* If 0 is only digit, leave it alone. */
|
|
|
|
|
if (numdigits > 1) {
|
|
|
|
|
skipped = 1;
|
|
|
|
|
--numdigits;
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
case 'x':
|
|
|
|
|
case 'X':
|
|
|
|
|
assert(buf[sign] == '0');
|
|
|
|
|
assert(buf[sign + 1] == 'x');
|
|
|
|
|
skipped = 2;
|
|
|
|
|
numnondigits -= 2;
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
if (skipped) {
|
|
|
|
|
buf += skipped;
|
|
|
|
|
len -= skipped;
|
|
|
|
|
if (sign)
|
|
|
|
|
buf[0] = '-';
|
|
|
|
|
}
|
|
|
|
|
assert(len == numnondigits + numdigits);
|
|
|
|
|
assert(numdigits > 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Fill with leading zeroes to meet minimum width. */
|
|
|
|
|
if (prec > numdigits) {
|
|
|
|
|
PyObject *r1 = PyString_FromStringAndSize(NULL,
|
|
|
|
|
numnondigits + prec);
|
|
|
|
|
char *b1;
|
|
|
|
|
if (!r1) {
|
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
b1 = PyString_AS_STRING(r1);
|
|
|
|
|
for (i = 0; i < numnondigits; ++i)
|
|
|
|
|
*b1++ = *buf++;
|
|
|
|
|
for (i = 0; i < prec - numdigits; i++)
|
|
|
|
|
*b1++ = '0';
|
|
|
|
|
for (i = 0; i < numdigits; i++)
|
|
|
|
|
*b1++ = *buf++;
|
|
|
|
|
*b1 = '\0';
|
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
result = r1;
|
|
|
|
|
buf = PyString_AS_STRING(result);
|
|
|
|
|
len = numnondigits + prec;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Fix up case for hex conversions. */
|
2005-06-29 20:29:56 -03:00
|
|
|
|
if (type == 'X') {
|
|
|
|
|
/* Need to convert all lower case letters to upper case.
|
|
|
|
|
and need to convert 0x to 0X (and -0x to -0X). */
|
2000-09-21 02:43:11 -03:00
|
|
|
|
for (i = 0; i < len; i++)
|
2005-06-29 20:29:56 -03:00
|
|
|
|
if (buf[i] >= 'a' && buf[i] <= 'x')
|
|
|
|
|
buf[i] -= 'a'-'A';
|
2000-09-21 02:43:11 -03:00
|
|
|
|
}
|
|
|
|
|
*pbuf = buf;
|
|
|
|
|
*plen = len;
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(int)
|
2000-07-09 04:04:36 -03:00
|
|
|
|
formatint(char *buf, size_t buflen, int flags,
|
|
|
|
|
int prec, int type, PyObject *v)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
{
|
2000-06-30 07:29:57 -03:00
|
|
|
|
/* fmt = '%#.' + `prec` + 'l' + `type`
|
2000-09-21 02:43:11 -03:00
|
|
|
|
worst case length = 3 + 19 (worst len of INT_MAX on 64-bit machine)
|
|
|
|
|
+ 1 + 1 = 24 */
|
|
|
|
|
char fmt[64]; /* plenty big enough! */
|
2003-11-29 19:52:13 -04:00
|
|
|
|
char *sign;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
long x;
|
2002-02-28 07:38:24 -04:00
|
|
|
|
|
2002-07-28 13:44:23 -03:00
|
|
|
|
x = PyInt_AsLong(v);
|
|
|
|
|
if (x == -1 && PyErr_Occurred()) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError, "int argument required");
|
1997-01-21 12:12:09 -04:00
|
|
|
|
return -1;
|
2002-07-28 13:44:23 -03:00
|
|
|
|
}
|
2003-11-29 19:52:13 -04:00
|
|
|
|
if (x < 0 && type == 'u') {
|
|
|
|
|
type = 'd';
|
2002-08-11 01:24:12 -03:00
|
|
|
|
}
|
2003-11-29 19:52:13 -04:00
|
|
|
|
if (x < 0 && (type == 'x' || type == 'X' || type == 'o'))
|
|
|
|
|
sign = "-";
|
|
|
|
|
else
|
|
|
|
|
sign = "";
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 1;
|
2002-02-28 07:38:24 -04:00
|
|
|
|
|
|
|
|
|
if ((flags & F_ALT) &&
|
|
|
|
|
(type == 'x' || type == 'X')) {
|
2002-04-15 10:48:52 -03:00
|
|
|
|
/* When converting under %#x or %#X, there are a number
|
2002-02-28 07:38:24 -04:00
|
|
|
|
* of issues that cause pain:
|
|
|
|
|
* - when 0 is being converted, the C standard leaves off
|
|
|
|
|
* the '0x' or '0X', which is inconsistent with other
|
|
|
|
|
* %#x/%#X conversions and inconsistent with Python's
|
|
|
|
|
* hex() function
|
|
|
|
|
* - there are platforms that violate the standard and
|
|
|
|
|
* convert 0 with the '0x' or '0X'
|
|
|
|
|
* (Metrowerks, Compaq Tru64)
|
|
|
|
|
* - there are platforms that give '0x' when converting
|
2002-04-15 10:48:52 -03:00
|
|
|
|
* under %#X, but convert 0 in accordance with the
|
2002-02-28 07:38:24 -04:00
|
|
|
|
* standard (OS/2 EMX)
|
2002-04-15 10:48:52 -03:00
|
|
|
|
*
|
2002-02-28 07:38:24 -04:00
|
|
|
|
* We can achieve the desired consistency by inserting our
|
|
|
|
|
* own '0x' or '0X' prefix, and substituting %x/%X in place
|
|
|
|
|
* of %#x/%#X.
|
|
|
|
|
*
|
|
|
|
|
* Note that this is the same approach as used in
|
|
|
|
|
* formatint() in unicodeobject.c
|
|
|
|
|
*/
|
2003-11-29 19:52:13 -04:00
|
|
|
|
PyOS_snprintf(fmt, sizeof(fmt), "%s0%c%%.%dl%c",
|
|
|
|
|
sign, type, prec, type);
|
2002-02-28 07:38:24 -04:00
|
|
|
|
}
|
|
|
|
|
else {
|
2003-11-29 19:52:13 -04:00
|
|
|
|
PyOS_snprintf(fmt, sizeof(fmt), "%s%%%s.%dl%c",
|
|
|
|
|
sign, (flags&F_ALT) ? "#" : "",
|
2002-02-28 07:38:24 -04:00
|
|
|
|
prec, type);
|
|
|
|
|
}
|
|
|
|
|
|
2003-11-29 19:52:13 -04:00
|
|
|
|
/* buf = '+'/'-'/'' + '0'/'0x'/'' + '[0-9]'*max(prec, len(x in octal))
|
|
|
|
|
* worst case buf = '-0x' + [0-9]*prec, where prec >= 11
|
2002-02-28 07:38:24 -04:00
|
|
|
|
*/
|
2003-11-29 19:52:13 -04:00
|
|
|
|
if (buflen <= 14 || buflen <= (size_t)3 + (size_t)prec) {
|
2000-06-30 07:29:57 -03:00
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
2002-04-15 10:48:52 -03:00
|
|
|
|
"formatted integer is too long (precision too large?)");
|
2000-06-30 07:29:57 -03:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
2003-11-29 19:52:13 -04:00
|
|
|
|
if (sign[0])
|
|
|
|
|
PyOS_snprintf(buf, buflen, fmt, -x);
|
|
|
|
|
else
|
|
|
|
|
PyOS_snprintf(buf, buflen, fmt, x);
|
2006-02-15 13:27:45 -04:00
|
|
|
|
return (int)strlen(buf);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
|
2006-05-27 11:58:20 -03:00
|
|
|
|
Py_LOCAL_INLINE(int)
|
2000-07-09 04:04:36 -03:00
|
|
|
|
formatchar(char *buf, size_t buflen, PyObject *v)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
{
|
2000-06-30 07:29:57 -03:00
|
|
|
|
/* presume that the buffer is at least 2 characters long */
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (PyString_Check(v)) {
|
|
|
|
|
if (!PyArg_Parse(v, "c;%c requires int or char", &buf[0]))
|
1997-01-21 12:12:09 -04:00
|
|
|
|
return -1;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
else {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (!PyArg_Parse(v, "b;%c requires int or char", &buf[0]))
|
1997-01-21 12:12:09 -04:00
|
|
|
|
return -1;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
buf[1] = '\0';
|
1997-01-21 12:12:09 -04:00
|
|
|
|
return 1;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
|
2000-06-30 07:29:57 -03:00
|
|
|
|
/* fmt%(v1,v2,...) is roughly equivalent to sprintf(fmt, v1, v2, ...)
|
|
|
|
|
|
|
|
|
|
FORMATBUFLEN is the length of the buffer in which the floats, ints, &
|
|
|
|
|
chars are formatted. XXX This is a magic number. Each formatting
|
|
|
|
|
routine does bounds checking to ensure no overflow, but a better
|
|
|
|
|
solution may be to malloc a buffer of appropriate size for each
|
|
|
|
|
format. For now, the current solution is sufficient.
|
|
|
|
|
*/
|
|
|
|
|
#define FORMATBUFLEN (size_t)120
|
1993-03-16 08:15:04 -04:00
|
|
|
|
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_Format(PyObject *format, PyObject *args)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
{
|
|
|
|
|
char *fmt, *res;
|
2006-02-16 10:32:27 -04:00
|
|
|
|
Py_ssize_t arglen, argidx;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t reslen, rescnt, fmtcnt;
|
1996-05-21 19:44:20 -03:00
|
|
|
|
int args_owned = 0;
|
2001-08-17 15:39:25 -03:00
|
|
|
|
PyObject *result, *orig_args;
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
PyObject *v, *w;
|
|
|
|
|
#endif
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *dict = NULL;
|
|
|
|
|
if (format == NULL || !PyString_Check(format) || args == NULL) {
|
|
|
|
|
PyErr_BadInternalCall();
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2000-04-10 10:47:21 -03:00
|
|
|
|
orig_args = args;
|
2001-12-06 11:18:48 -04:00
|
|
|
|
fmt = PyString_AS_STRING(format);
|
|
|
|
|
fmtcnt = PyString_GET_SIZE(format);
|
1993-05-12 05:24:20 -03:00
|
|
|
|
reslen = rescnt = fmtcnt + 100;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
result = PyString_FromStringAndSize((char *)NULL, reslen);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
res = PyString_AsString(result);
|
|
|
|
|
if (PyTuple_Check(args)) {
|
2001-12-06 11:18:48 -04:00
|
|
|
|
arglen = PyTuple_GET_SIZE(args);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
argidx = 0;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
arglen = -1;
|
|
|
|
|
argidx = -2;
|
|
|
|
|
}
|
2002-11-12 19:01:12 -04:00
|
|
|
|
if (args->ob_type->tp_as_mapping && !PyTuple_Check(args) &&
|
|
|
|
|
!PyObject_TypeCheck(args, &PyBaseString_Type))
|
1994-08-30 05:19:36 -03:00
|
|
|
|
dict = args;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
while (--fmtcnt >= 0) {
|
|
|
|
|
if (*fmt != '%') {
|
|
|
|
|
if (--rescnt < 0) {
|
1993-05-12 05:24:20 -03:00
|
|
|
|
rescnt = fmtcnt + 100;
|
|
|
|
|
reslen += rescnt;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (_PyString_Resize(&result, reslen) < 0)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return NULL;
|
2001-12-06 11:18:48 -04:00
|
|
|
|
res = PyString_AS_STRING(result)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
+ reslen - rescnt;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
--rescnt;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
*res++ = *fmt++;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
/* Got a format specifier */
|
|
|
|
|
int flags = 0;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t width = -1;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
int prec = -1;
|
1993-11-11 10:51:57 -04:00
|
|
|
|
int c = '\0';
|
1993-03-16 08:15:04 -04:00
|
|
|
|
int fill;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *v = NULL;
|
|
|
|
|
PyObject *temp = NULL;
|
2000-06-30 07:29:57 -03:00
|
|
|
|
char *pbuf;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
int sign;
|
2006-03-07 08:08:51 -04:00
|
|
|
|
Py_ssize_t len;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
char formatbuf[FORMATBUFLEN];
|
|
|
|
|
/* For format{float,int,char}() */
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-04-10 10:47:21 -03:00
|
|
|
|
char *fmt_start = fmt;
|
2006-03-07 08:08:51 -04:00
|
|
|
|
Py_ssize_t argidx_start = argidx;
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
1996-12-05 17:58:58 -04:00
|
|
|
|
fmt++;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
if (*fmt == '(') {
|
|
|
|
|
char *keystart;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t keylen;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyObject *key;
|
1997-09-08 15:30:11 -03:00
|
|
|
|
int pcount = 1;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
|
|
|
|
|
if (dict == NULL) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2001-04-28 02:38:26 -03:00
|
|
|
|
"format requires a mapping");
|
1994-08-30 05:19:36 -03:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
++fmt;
|
|
|
|
|
--fmtcnt;
|
|
|
|
|
keystart = fmt;
|
1997-09-08 15:30:11 -03:00
|
|
|
|
/* Skip over balanced parentheses */
|
|
|
|
|
while (pcount > 0 && --fmtcnt >= 0) {
|
|
|
|
|
if (*fmt == ')')
|
|
|
|
|
--pcount;
|
|
|
|
|
else if (*fmt == '(')
|
|
|
|
|
++pcount;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
fmt++;
|
1997-09-08 15:30:11 -03:00
|
|
|
|
}
|
|
|
|
|
keylen = fmt - keystart - 1;
|
|
|
|
|
if (fmtcnt < 0 || pcount > 0) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
1994-08-30 05:19:36 -03:00
|
|
|
|
"incomplete format key");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
key = PyString_FromStringAndSize(keystart,
|
|
|
|
|
keylen);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
if (key == NULL)
|
|
|
|
|
goto error;
|
1996-05-21 19:44:20 -03:00
|
|
|
|
if (args_owned) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(args);
|
1996-05-21 19:44:20 -03:00
|
|
|
|
args_owned = 0;
|
|
|
|
|
}
|
|
|
|
|
args = PyObject_GetItem(dict, key);
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(key);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
if (args == NULL) {
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
1996-05-21 19:44:20 -03:00
|
|
|
|
args_owned = 1;
|
1994-08-30 05:19:36 -03:00
|
|
|
|
arglen = -1;
|
|
|
|
|
argidx = -2;
|
|
|
|
|
}
|
1993-03-16 08:15:04 -04:00
|
|
|
|
while (--fmtcnt >= 0) {
|
|
|
|
|
switch (c = *fmt++) {
|
|
|
|
|
case '-': flags |= F_LJUST; continue;
|
|
|
|
|
case '+': flags |= F_SIGN; continue;
|
|
|
|
|
case ' ': flags |= F_BLANK; continue;
|
|
|
|
|
case '#': flags |= F_ALT; continue;
|
|
|
|
|
case '0': flags |= F_ZERO; continue;
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
if (c == '*') {
|
|
|
|
|
v = getnextarg(args, arglen, &argidx);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (!PyInt_Check(v)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"* wants int");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
width = PyInt_AsLong(v);
|
1999-06-07 12:12:32 -03:00
|
|
|
|
if (width < 0) {
|
|
|
|
|
flags |= F_LJUST;
|
|
|
|
|
width = -width;
|
|
|
|
|
}
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
1995-02-10 13:00:37 -04:00
|
|
|
|
else if (c >= 0 && isdigit(c)) {
|
1993-03-16 08:15:04 -04:00
|
|
|
|
width = c - '0';
|
|
|
|
|
while (--fmtcnt >= 0) {
|
1995-02-10 13:00:37 -04:00
|
|
|
|
c = Py_CHARMASK(*fmt++);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (!isdigit(c))
|
|
|
|
|
break;
|
|
|
|
|
if ((width*10) / 10 != width) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_ValueError,
|
|
|
|
|
"width too big");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
width = width*10 + (c - '0');
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (c == '.') {
|
|
|
|
|
prec = 0;
|
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
if (c == '*') {
|
|
|
|
|
v = getnextarg(args, arglen, &argidx);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
1997-05-02 00:12:38 -03:00
|
|
|
|
if (!PyInt_Check(v)) {
|
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_TypeError,
|
|
|
|
|
"* wants int");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
prec = PyInt_AsLong(v);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 0;
|
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
1995-02-10 13:00:37 -04:00
|
|
|
|
else if (c >= 0 && isdigit(c)) {
|
1993-03-16 08:15:04 -04:00
|
|
|
|
prec = c - '0';
|
|
|
|
|
while (--fmtcnt >= 0) {
|
1995-02-10 13:00:37 -04:00
|
|
|
|
c = Py_CHARMASK(*fmt++);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (!isdigit(c))
|
|
|
|
|
break;
|
|
|
|
|
if ((prec*10) / 10 != prec) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_ValueError,
|
1993-03-16 08:15:04 -04:00
|
|
|
|
"prec too big");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
prec = prec*10 + (c - '0');
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} /* prec */
|
|
|
|
|
if (fmtcnt >= 0) {
|
|
|
|
|
if (c == 'h' || c == 'l' || c == 'L') {
|
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (fmtcnt < 0) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"incomplete format");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
if (c != '%') {
|
|
|
|
|
v = getnextarg(args, arglen, &argidx);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
sign = 0;
|
|
|
|
|
fill = ' ';
|
|
|
|
|
switch (c) {
|
|
|
|
|
case '%':
|
2000-06-30 07:29:57 -03:00
|
|
|
|
pbuf = "%";
|
1993-03-16 08:15:04 -04:00
|
|
|
|
len = 1;
|
|
|
|
|
break;
|
|
|
|
|
case 's':
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2005-08-31 20:02:05 -03:00
|
|
|
|
if (PyUnicode_Check(v)) {
|
|
|
|
|
fmt = fmt_start;
|
|
|
|
|
argidx = argidx_start;
|
|
|
|
|
goto unicode;
|
|
|
|
|
}
|
2005-10-01 14:06:00 -03:00
|
|
|
|
#endif
|
2005-08-12 14:34:58 -03:00
|
|
|
|
temp = _PyObject_Str(v);
|
2005-10-01 14:06:00 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2005-08-12 14:34:58 -03:00
|
|
|
|
if (temp != NULL && PyUnicode_Check(temp)) {
|
|
|
|
|
Py_DECREF(temp);
|
2000-04-10 10:47:21 -03:00
|
|
|
|
fmt = fmt_start;
|
2001-05-02 11:21:53 -03:00
|
|
|
|
argidx = argidx_start;
|
2000-04-10 10:47:21 -03:00
|
|
|
|
goto unicode;
|
|
|
|
|
}
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif
|
2002-10-09 16:07:53 -03:00
|
|
|
|
/* Fall through */
|
2003-06-18 11:17:01 -03:00
|
|
|
|
case 'r':
|
2005-08-12 14:34:58 -03:00
|
|
|
|
if (c == 'r')
|
2000-04-11 12:39:26 -03:00
|
|
|
|
temp = PyObject_Repr(v);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
if (temp == NULL)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
1998-06-09 12:08:41 -03:00
|
|
|
|
if (!PyString_Check(temp)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-10-09 16:14:30 -03:00
|
|
|
|
"%s argument has non-string str()");
|
2001-12-06 11:18:48 -04:00
|
|
|
|
Py_DECREF(temp);
|
1998-06-09 12:08:41 -03:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
2001-12-06 11:18:48 -04:00
|
|
|
|
pbuf = PyString_AS_STRING(temp);
|
|
|
|
|
len = PyString_GET_SIZE(temp);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
if (prec >= 0 && len > prec)
|
|
|
|
|
len = prec;
|
|
|
|
|
break;
|
|
|
|
|
case 'i':
|
|
|
|
|
case 'd':
|
|
|
|
|
case 'u':
|
|
|
|
|
case 'o':
|
|
|
|
|
case 'x':
|
|
|
|
|
case 'X':
|
|
|
|
|
if (c == 'i')
|
|
|
|
|
c = 'd';
|
2000-11-30 01:22:44 -04:00
|
|
|
|
if (PyLong_Check(v)) {
|
2006-03-07 08:08:51 -04:00
|
|
|
|
int ilen;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
temp = _PyString_FormatLong(v, flags,
|
2006-03-07 08:08:51 -04:00
|
|
|
|
prec, c, &pbuf, &ilen);
|
|
|
|
|
len = ilen;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if (!temp)
|
|
|
|
|
goto error;
|
|
|
|
|
sign = 1;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
pbuf = formatbuf;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
len = formatint(pbuf,
|
|
|
|
|
sizeof(formatbuf),
|
2000-09-21 02:43:11 -03:00
|
|
|
|
flags, prec, c, v);
|
|
|
|
|
if (len < 0)
|
|
|
|
|
goto error;
|
2003-11-29 19:52:13 -04:00
|
|
|
|
sign = 1;
|
1997-01-29 02:00:24 -04:00
|
|
|
|
}
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if (flags & F_ZERO)
|
|
|
|
|
fill = '0';
|
1993-03-16 08:15:04 -04:00
|
|
|
|
break;
|
|
|
|
|
case 'e':
|
|
|
|
|
case 'E':
|
|
|
|
|
case 'f':
|
2003-08-27 01:55:52 -03:00
|
|
|
|
case 'F':
|
1993-03-16 08:15:04 -04:00
|
|
|
|
case 'g':
|
|
|
|
|
case 'G':
|
2003-08-27 01:55:52 -03:00
|
|
|
|
if (c == 'F')
|
|
|
|
|
c = 'f';
|
2000-06-30 07:29:57 -03:00
|
|
|
|
pbuf = formatbuf;
|
2002-04-15 10:48:52 -03:00
|
|
|
|
len = formatfloat(pbuf, sizeof(formatbuf),
|
|
|
|
|
flags, prec, c, v);
|
1997-01-21 12:12:09 -04:00
|
|
|
|
if (len < 0)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
sign = 1;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if (flags & F_ZERO)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
fill = '0';
|
|
|
|
|
break;
|
|
|
|
|
case 'c':
|
2003-03-31 14:07:50 -04:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
if (PyUnicode_Check(v)) {
|
|
|
|
|
fmt = fmt_start;
|
|
|
|
|
argidx = argidx_start;
|
|
|
|
|
goto unicode;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
2000-06-30 07:29:57 -03:00
|
|
|
|
pbuf = formatbuf;
|
|
|
|
|
len = formatchar(pbuf, sizeof(formatbuf), v);
|
1997-01-21 12:12:09 -04:00
|
|
|
|
if (len < 0)
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
1997-09-08 15:30:11 -03:00
|
|
|
|
PyErr_Format(PyExc_ValueError,
|
2000-12-15 09:07:46 -04:00
|
|
|
|
"unsupported format character '%c' (0x%x) "
|
2006-10-04 08:44:06 -03:00
|
|
|
|
"at index %zd",
|
2002-09-12 11:43:41 -03:00
|
|
|
|
c, c,
|
2006-10-04 08:44:06 -03:00
|
|
|
|
(Py_ssize_t)(fmt - 1 -
|
|
|
|
|
PyString_AsString(format)));
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
if (sign) {
|
2000-06-30 07:29:57 -03:00
|
|
|
|
if (*pbuf == '-' || *pbuf == '+') {
|
|
|
|
|
sign = *pbuf++;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
len--;
|
|
|
|
|
}
|
|
|
|
|
else if (flags & F_SIGN)
|
|
|
|
|
sign = '+';
|
|
|
|
|
else if (flags & F_BLANK)
|
|
|
|
|
sign = ' ';
|
|
|
|
|
else
|
2000-09-21 02:43:11 -03:00
|
|
|
|
sign = 0;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
if (width < len)
|
|
|
|
|
width = len;
|
2002-10-10 21:43:48 -03:00
|
|
|
|
if (rescnt - (sign != 0) < width) {
|
1993-05-12 05:24:20 -03:00
|
|
|
|
reslen -= rescnt;
|
|
|
|
|
rescnt = width + fmtcnt + 100;
|
|
|
|
|
reslen += rescnt;
|
2002-10-10 21:43:48 -03:00
|
|
|
|
if (reslen < 0) {
|
|
|
|
|
Py_DECREF(result);
|
2007-02-26 09:51:34 -04:00
|
|
|
|
Py_XDECREF(temp);
|
2002-10-10 21:43:48 -03:00
|
|
|
|
return PyErr_NoMemory();
|
|
|
|
|
}
|
2007-02-26 09:51:34 -04:00
|
|
|
|
if (_PyString_Resize(&result, reslen) < 0) {
|
|
|
|
|
Py_XDECREF(temp);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return NULL;
|
2007-02-26 09:51:34 -04:00
|
|
|
|
}
|
2001-12-06 11:18:48 -04:00
|
|
|
|
res = PyString_AS_STRING(result)
|
1997-05-02 00:12:38 -03:00
|
|
|
|
+ reslen - rescnt;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
}
|
|
|
|
|
if (sign) {
|
1993-11-11 11:03:51 -04:00
|
|
|
|
if (fill != ' ')
|
|
|
|
|
*res++ = sign;
|
1993-03-16 08:15:04 -04:00
|
|
|
|
rescnt--;
|
|
|
|
|
if (width > len)
|
|
|
|
|
width--;
|
|
|
|
|
}
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if ((flags & F_ALT) && (c == 'x' || c == 'X')) {
|
|
|
|
|
assert(pbuf[0] == '0');
|
2001-04-12 15:38:48 -03:00
|
|
|
|
assert(pbuf[1] == c);
|
|
|
|
|
if (fill != ' ') {
|
|
|
|
|
*res++ = *pbuf++;
|
|
|
|
|
*res++ = *pbuf++;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
}
|
2001-04-12 15:38:48 -03:00
|
|
|
|
rescnt -= 2;
|
|
|
|
|
width -= 2;
|
|
|
|
|
if (width < 0)
|
|
|
|
|
width = 0;
|
|
|
|
|
len -= 2;
|
2000-09-21 02:43:11 -03:00
|
|
|
|
}
|
|
|
|
|
if (width > len && !(flags & F_LJUST)) {
|
1993-03-16 08:15:04 -04:00
|
|
|
|
do {
|
|
|
|
|
--rescnt;
|
|
|
|
|
*res++ = fill;
|
|
|
|
|
} while (--width > len);
|
|
|
|
|
}
|
2000-09-21 02:43:11 -03:00
|
|
|
|
if (fill == ' ') {
|
|
|
|
|
if (sign)
|
|
|
|
|
*res++ = sign;
|
|
|
|
|
if ((flags & F_ALT) &&
|
2001-04-12 15:38:48 -03:00
|
|
|
|
(c == 'x' || c == 'X')) {
|
|
|
|
|
assert(pbuf[0] == '0');
|
|
|
|
|
assert(pbuf[1] == c);
|
2000-09-21 02:43:11 -03:00
|
|
|
|
*res++ = *pbuf++;
|
|
|
|
|
*res++ = *pbuf++;
|
|
|
|
|
}
|
|
|
|
|
}
|
2006-05-28 09:06:46 -03:00
|
|
|
|
Py_MEMCPY(res, pbuf, len);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
res += len;
|
|
|
|
|
rescnt -= len;
|
|
|
|
|
while (--width >= len) {
|
|
|
|
|
--rescnt;
|
|
|
|
|
*res++ = ' ';
|
|
|
|
|
}
|
1995-02-10 13:00:37 -04:00
|
|
|
|
if (dict && (argidx < arglen) && c != '%') {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-05-21 12:14:57 -03:00
|
|
|
|
"not all arguments converted during string formatting");
|
2007-02-26 09:51:34 -04:00
|
|
|
|
Py_XDECREF(temp);
|
1994-08-30 05:19:36 -03:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_XDECREF(temp);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
} /* '%' */
|
|
|
|
|
} /* until end */
|
1995-02-27 06:13:23 -04:00
|
|
|
|
if (argidx < arglen && !dict) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-05-21 12:14:57 -03:00
|
|
|
|
"not all arguments converted during string formatting");
|
1993-03-16 08:15:04 -04:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1998-04-10 19:16:39 -03:00
|
|
|
|
if (args_owned) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(args);
|
1998-04-10 19:16:39 -03:00
|
|
|
|
}
|
1997-05-02 00:12:38 -03:00
|
|
|
|
_PyString_Resize(&result, reslen - rescnt);
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return result;
|
2000-04-10 10:47:21 -03:00
|
|
|
|
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#ifdef Py_USING_UNICODE
|
2000-04-10 10:47:21 -03:00
|
|
|
|
unicode:
|
|
|
|
|
if (args_owned) {
|
|
|
|
|
Py_DECREF(args);
|
|
|
|
|
args_owned = 0;
|
|
|
|
|
}
|
2001-05-02 11:21:53 -03:00
|
|
|
|
/* Fiddle args right (remove the first argidx arguments) */
|
2000-04-10 10:47:21 -03:00
|
|
|
|
if (PyTuple_Check(orig_args) && argidx > 0) {
|
|
|
|
|
PyObject *v;
|
2006-02-16 10:32:27 -04:00
|
|
|
|
Py_ssize_t n = PyTuple_GET_SIZE(orig_args) - argidx;
|
2000-04-10 10:47:21 -03:00
|
|
|
|
v = PyTuple_New(n);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
|
|
|
|
while (--n >= 0) {
|
|
|
|
|
PyObject *w = PyTuple_GET_ITEM(orig_args, n + argidx);
|
|
|
|
|
Py_INCREF(w);
|
|
|
|
|
PyTuple_SET_ITEM(v, n, w);
|
|
|
|
|
}
|
|
|
|
|
args = v;
|
|
|
|
|
} else {
|
|
|
|
|
Py_INCREF(orig_args);
|
|
|
|
|
args = orig_args;
|
|
|
|
|
}
|
2000-10-07 05:54:09 -03:00
|
|
|
|
args_owned = 1;
|
|
|
|
|
/* Take what we have of the result and let the Unicode formatting
|
|
|
|
|
function format the rest of the input. */
|
2000-04-10 10:47:21 -03:00
|
|
|
|
rescnt = res - PyString_AS_STRING(result);
|
2000-10-07 05:54:09 -03:00
|
|
|
|
if (_PyString_Resize(&result, rescnt))
|
|
|
|
|
goto error;
|
2000-04-10 10:47:21 -03:00
|
|
|
|
fmtcnt = PyString_GET_SIZE(format) - \
|
|
|
|
|
(fmt - PyString_AS_STRING(format));
|
2000-10-07 05:54:09 -03:00
|
|
|
|
format = PyUnicode_Decode(fmt, fmtcnt, NULL, NULL);
|
|
|
|
|
if (format == NULL)
|
2000-04-10 10:47:21 -03:00
|
|
|
|
goto error;
|
2000-10-07 05:54:09 -03:00
|
|
|
|
v = PyUnicode_Format(format, args);
|
2000-04-10 10:47:21 -03:00
|
|
|
|
Py_DECREF(format);
|
2000-10-07 05:54:09 -03:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
|
|
|
|
/* Paste what we have (result) to what the Unicode formatting
|
|
|
|
|
function returned (v) and return the result (or error) */
|
|
|
|
|
w = PyUnicode_Concat(result, v);
|
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
Py_DECREF(v);
|
2000-04-10 10:47:21 -03:00
|
|
|
|
Py_DECREF(args);
|
2000-10-07 05:54:09 -03:00
|
|
|
|
return w;
|
2001-08-17 15:39:25 -03:00
|
|
|
|
#endif /* Py_USING_UNICODE */
|
2001-04-28 02:38:26 -03:00
|
|
|
|
|
1993-03-16 08:15:04 -04:00
|
|
|
|
error:
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(result);
|
1998-04-10 19:16:39 -03:00
|
|
|
|
if (args_owned) {
|
1997-05-02 00:12:38 -03:00
|
|
|
|
Py_DECREF(args);
|
1998-04-10 19:16:39 -03:00
|
|
|
|
}
|
1993-03-16 08:15:04 -04:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-01-18 03:55:05 -04:00
|
|
|
|
|
|
|
|
|
void
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_InternInPlace(PyObject **p)
|
1997-01-18 03:55:05 -04:00
|
|
|
|
{
|
|
|
|
|
register PyStringObject *s = (PyStringObject *)(*p);
|
|
|
|
|
PyObject *t;
|
|
|
|
|
if (s == NULL || !PyString_Check(s))
|
|
|
|
|
Py_FatalError("PyString_InternInPlace: strings only please!");
|
2004-08-07 16:20:05 -03:00
|
|
|
|
/* If it's a string subclass, we don't really know what putting
|
|
|
|
|
it in the interned dict might do. */
|
|
|
|
|
if (!PyString_CheckExact(s))
|
|
|
|
|
return;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
if (PyString_CHECK_INTERNED(s))
|
1997-01-18 03:55:05 -04:00
|
|
|
|
return;
|
|
|
|
|
if (interned == NULL) {
|
|
|
|
|
interned = PyDict_New();
|
2002-08-19 18:43:18 -03:00
|
|
|
|
if (interned == NULL) {
|
|
|
|
|
PyErr_Clear(); /* Don't leave an exception */
|
1997-01-18 03:55:05 -04:00
|
|
|
|
return;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
}
|
1997-01-18 03:55:05 -04:00
|
|
|
|
}
|
2004-08-07 16:20:05 -03:00
|
|
|
|
t = PyDict_GetItem(interned, (PyObject *)s);
|
|
|
|
|
if (t) {
|
1997-01-18 03:55:05 -04:00
|
|
|
|
Py_INCREF(t);
|
2002-08-19 18:43:18 -03:00
|
|
|
|
Py_DECREF(*p);
|
|
|
|
|
*p = t;
|
1997-01-18 03:55:05 -04:00
|
|
|
|
return;
|
|
|
|
|
}
|
2002-08-19 18:43:18 -03:00
|
|
|
|
|
2004-08-07 16:27:39 -03:00
|
|
|
|
if (PyDict_SetItem(interned, (PyObject *)s, (PyObject *)s) < 0) {
|
2004-08-07 16:20:05 -03:00
|
|
|
|
PyErr_Clear();
|
2002-08-19 18:43:18 -03:00
|
|
|
|
return;
|
1997-01-18 03:55:05 -04:00
|
|
|
|
}
|
2004-08-07 16:20:05 -03:00
|
|
|
|
/* The two references in interned are not counted by refcnt.
|
|
|
|
|
The string deallocator will take care of this */
|
2004-08-07 16:27:39 -03:00
|
|
|
|
s->ob_refcnt -= 2;
|
2004-08-07 16:20:05 -03:00
|
|
|
|
PyString_CHECK_INTERNED(s) = SSTATE_INTERNED_MORTAL;
|
1997-01-18 03:55:05 -04:00
|
|
|
|
}
|
|
|
|
|
|
2002-08-19 18:43:18 -03:00
|
|
|
|
void
|
|
|
|
|
PyString_InternImmortal(PyObject **p)
|
|
|
|
|
{
|
|
|
|
|
PyString_InternInPlace(p);
|
|
|
|
|
if (PyString_CHECK_INTERNED(*p) != SSTATE_INTERNED_IMMORTAL) {
|
|
|
|
|
PyString_CHECK_INTERNED(*p) = SSTATE_INTERNED_IMMORTAL;
|
|
|
|
|
Py_INCREF(*p);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
1997-01-18 03:55:05 -04:00
|
|
|
|
|
|
|
|
|
PyObject *
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_InternFromString(const char *cp)
|
1997-01-18 03:55:05 -04:00
|
|
|
|
{
|
|
|
|
|
PyObject *s = PyString_FromString(cp);
|
|
|
|
|
if (s == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
PyString_InternInPlace(&s);
|
|
|
|
|
return s;
|
|
|
|
|
}
|
|
|
|
|
|
1997-08-01 23:57:45 -03:00
|
|
|
|
void
|
2000-07-09 04:04:36 -03:00
|
|
|
|
PyString_Fini(void)
|
1997-08-01 23:57:45 -03:00
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
for (i = 0; i < UCHAR_MAX + 1; i++) {
|
|
|
|
|
Py_XDECREF(characters[i]);
|
|
|
|
|
characters[i] = NULL;
|
|
|
|
|
}
|
|
|
|
|
Py_XDECREF(nullstring);
|
|
|
|
|
nullstring = NULL;
|
|
|
|
|
}
|
2001-02-23 12:40:48 -04:00
|
|
|
|
|
|
|
|
|
void _Py_ReleaseInternedStrings(void)
|
|
|
|
|
{
|
2002-08-19 18:43:18 -03:00
|
|
|
|
PyObject *keys;
|
|
|
|
|
PyStringObject *s;
|
2006-02-15 13:27:45 -04:00
|
|
|
|
Py_ssize_t i, n;
|
2002-08-19 18:43:18 -03:00
|
|
|
|
|
|
|
|
|
if (interned == NULL || !PyDict_Check(interned))
|
|
|
|
|
return;
|
|
|
|
|
keys = PyDict_Keys(interned);
|
|
|
|
|
if (keys == NULL || !PyList_Check(keys)) {
|
|
|
|
|
PyErr_Clear();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Since _Py_ReleaseInternedStrings() is intended to help a leak
|
|
|
|
|
detector, interned strings are not forcibly deallocated; rather, we
|
|
|
|
|
give them their stolen references back, and then clear and DECREF
|
|
|
|
|
the interned dict. */
|
2006-03-16 23:29:34 -04:00
|
|
|
|
|
2002-08-19 18:43:18 -03:00
|
|
|
|
fprintf(stderr, "releasing interned strings\n");
|
|
|
|
|
n = PyList_GET_SIZE(keys);
|
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
|
s = (PyStringObject *) PyList_GET_ITEM(keys, i);
|
|
|
|
|
switch (s->ob_sstate) {
|
|
|
|
|
case SSTATE_NOT_INTERNED:
|
|
|
|
|
/* XXX Shouldn't happen */
|
|
|
|
|
break;
|
|
|
|
|
case SSTATE_INTERNED_IMMORTAL:
|
|
|
|
|
s->ob_refcnt += 1;
|
|
|
|
|
break;
|
|
|
|
|
case SSTATE_INTERNED_MORTAL:
|
|
|
|
|
s->ob_refcnt += 2;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
Py_FatalError("Inconsistent interned string state.");
|
|
|
|
|
}
|
|
|
|
|
s->ob_sstate = SSTATE_NOT_INTERNED;
|
2001-02-23 12:40:48 -04:00
|
|
|
|
}
|
2002-08-19 18:43:18 -03:00
|
|
|
|
Py_DECREF(keys);
|
|
|
|
|
PyDict_Clear(interned);
|
|
|
|
|
Py_DECREF(interned);
|
|
|
|
|
interned = NULL;
|
2001-02-23 12:40:48 -04:00
|
|
|
|
}
|