1991-02-19 20:39:46 +08:00
|
|
|
|
/***********************************************************
|
2000-07-01 07:50:40 +08:00
|
|
|
|
Copyright (c) 2000, BeOpen.com.
|
|
|
|
|
Copyright (c) 1995-2000, Corporation for National Research Initiatives.
|
|
|
|
|
Copyright (c) 1990-1995, Stichting Mathematisch Centrum.
|
|
|
|
|
All rights reserved.
|
1996-10-25 22:44:06 +08:00
|
|
|
|
|
2000-07-01 07:50:40 +08:00
|
|
|
|
See the file "Misc/COPYRIGHT" for information on usage and
|
|
|
|
|
redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
|
1991-02-19 20:39:46 +08:00
|
|
|
|
******************************************************************/
|
|
|
|
|
|
1990-10-14 20:07:46 +08:00
|
|
|
|
/* String object implementation */
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
#include "Python.h"
|
1990-10-14 20:07:46 +08:00
|
|
|
|
|
1997-06-04 02:03:18 +08:00
|
|
|
|
#include "mymath.h"
|
1994-08-30 16:19:36 +08:00
|
|
|
|
#include <ctype.h>
|
|
|
|
|
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
int null_strings, one_strings;
|
|
|
|
|
#endif
|
|
|
|
|
|
1994-09-28 23:51:32 +08:00
|
|
|
|
#ifdef HAVE_LIMITS_H
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#include <limits.h>
|
|
|
|
|
#else
|
|
|
|
|
#ifndef UCHAR_MAX
|
|
|
|
|
#define UCHAR_MAX 255
|
|
|
|
|
#endif
|
|
|
|
|
#endif
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyStringObject *characters[UCHAR_MAX + 1];
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyStringObject *nullstring;
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#endif
|
1993-10-22 20:04:32 +08:00
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
Newsizedstringobject() and newstringobject() try in certain cases
|
|
|
|
|
to share string objects. When the size of the string is zero,
|
|
|
|
|
these routines always return a pointer to the same string object;
|
|
|
|
|
when the size is one, they return a pointer to an already existing
|
|
|
|
|
object if the contents of the string is known. For
|
|
|
|
|
newstringobject() this is always the case, for
|
|
|
|
|
newsizedstringobject() this is the case when the first argument in
|
|
|
|
|
not NULL.
|
|
|
|
|
A common practice to allocate a string and then fill it in or
|
|
|
|
|
change it must be done carefully. It is only allowed to change the
|
|
|
|
|
contents of the string if the obect was gotten from
|
|
|
|
|
newsizedstringobject() with a NULL first argument, because in the
|
|
|
|
|
future these routines may try to do even more sharing of objects.
|
|
|
|
|
*/
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_FromStringAndSize(const char *str, int size)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyStringObject *op;
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
1993-10-22 20:04:32 +08:00
|
|
|
|
if (size == 0 && (op = nullstring) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
null_strings++;
|
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (size == 1 && str != NULL &&
|
|
|
|
|
(op = characters[*str & UCHAR_MAX]) != NULL)
|
|
|
|
|
{
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#endif /* DONT_SHARE_SHORT_STRINGS */
|
2000-05-04 07:44:39 +08:00
|
|
|
|
|
|
|
|
|
/* PyObject_NewVar is inlined */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
op = (PyStringObject *)
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_MALLOC(sizeof(PyStringObject) + size * sizeof(char));
|
1990-10-22 06:15:08 +08:00
|
|
|
|
if (op == NULL)
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
op->ob_shash = -1;
|
1997-01-18 15:55:05 +08:00
|
|
|
|
#endif
|
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
op->ob_sinterned = NULL;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#endif
|
1990-10-22 06:15:08 +08:00
|
|
|
|
if (str != NULL)
|
|
|
|
|
memcpy(op->ob_sval, str, size);
|
|
|
|
|
op->ob_sval[size] = '\0';
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
1993-10-22 20:04:32 +08:00
|
|
|
|
if (size == 0) {
|
|
|
|
|
nullstring = op;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
} else if (size == 1 && str != NULL) {
|
|
|
|
|
characters[*str & UCHAR_MAX] = op;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_FromString(const char *str)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
2000-06-30 18:29:57 +08:00
|
|
|
|
register size_t size = strlen(str);
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyStringObject *op;
|
2000-06-30 18:29:57 +08:00
|
|
|
|
if (size > INT_MAX) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"string is too long for a Python string");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
1993-10-22 20:04:32 +08:00
|
|
|
|
if (size == 0 && (op = nullstring) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
null_strings++;
|
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
|
|
|
|
if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) {
|
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
|
|
|
|
return (PyObject *)op;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#endif /* DONT_SHARE_SHORT_STRINGS */
|
2000-05-04 07:44:39 +08:00
|
|
|
|
|
|
|
|
|
/* PyObject_NewVar is inlined */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
op = (PyStringObject *)
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_MALLOC(sizeof(PyStringObject) + size * sizeof(char));
|
1990-10-22 06:15:08 +08:00
|
|
|
|
if (op == NULL)
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
op->ob_shash = -1;
|
1997-01-18 15:55:05 +08:00
|
|
|
|
#endif
|
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
op->ob_sinterned = NULL;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#endif
|
1990-10-22 06:15:08 +08:00
|
|
|
|
strcpy(op->ob_sval, str);
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
1993-10-22 20:04:32 +08:00
|
|
|
|
if (size == 0) {
|
|
|
|
|
nullstring = op;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
} else if (size == 1) {
|
|
|
|
|
characters[*str & UCHAR_MAX] = op;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(op);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
}
|
1993-11-01 21:46:50 +08:00
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
2000-07-06 19:29:01 +08:00
|
|
|
|
PyObject *PyString_Decode(const char *s,
|
|
|
|
|
int size,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *buffer = NULL, *str;
|
|
|
|
|
|
|
|
|
|
if (encoding == NULL)
|
|
|
|
|
encoding = PyUnicode_GetDefaultEncoding();
|
|
|
|
|
|
|
|
|
|
/* Decode via the codec registry */
|
|
|
|
|
buffer = PyBuffer_FromMemory((void *)s, size);
|
|
|
|
|
if (buffer == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
str = PyCodec_Decode(buffer, encoding, errors);
|
|
|
|
|
if (str == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
/* Convert Unicode to a string using the default encoding */
|
|
|
|
|
if (PyUnicode_Check(str)) {
|
|
|
|
|
PyObject *temp = str;
|
|
|
|
|
str = PyUnicode_AsEncodedString(str, NULL, NULL);
|
|
|
|
|
Py_DECREF(temp);
|
|
|
|
|
if (str == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
|
|
|
|
if (!PyString_Check(str)) {
|
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2000-07-12 10:58:28 +08:00
|
|
|
|
"decoder did not return a string object (type=%.400s)",
|
2000-07-06 19:29:01 +08:00
|
|
|
|
str->ob_type->tp_name);
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
|
|
|
|
Py_DECREF(buffer);
|
|
|
|
|
return str;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
Py_XDECREF(buffer);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_Encode(const char *s,
|
|
|
|
|
int size,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v, *str;
|
|
|
|
|
|
|
|
|
|
str = PyString_FromStringAndSize(s, size);
|
|
|
|
|
if (str == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
v = PyString_AsEncodedString(str, encoding, errors);
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
return v;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PyObject *PyString_AsEncodedString(PyObject *str,
|
|
|
|
|
const char *encoding,
|
|
|
|
|
const char *errors)
|
|
|
|
|
{
|
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
|
|
if (!PyString_Check(str)) {
|
|
|
|
|
PyErr_BadArgument();
|
|
|
|
|
goto onError;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (encoding == NULL)
|
|
|
|
|
encoding = PyUnicode_GetDefaultEncoding();
|
|
|
|
|
|
|
|
|
|
/* Encode via the codec registry */
|
|
|
|
|
v = PyCodec_Encode(str, encoding, errors);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto onError;
|
|
|
|
|
/* 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;
|
|
|
|
|
}
|
|
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
return v;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
1993-06-17 20:35:49 +08:00
|
|
|
|
static void
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_dealloc(PyObject *op)
|
1992-03-28 01:31:02 +08:00
|
|
|
|
{
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_DEL(op);
|
1992-03-28 01:31:02 +08:00
|
|
|
|
}
|
|
|
|
|
|
1995-01-03 03:07:15 +08:00
|
|
|
|
int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_Size(register PyObject *op)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyString_Check(op)) {
|
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return ((PyStringObject *)op) -> ob_size;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*const*/ char *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_AsString(register PyObject *op)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyString_Check(op)) {
|
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return ((PyStringObject *)op) -> ob_sval;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Methods */
|
|
|
|
|
|
1991-06-08 07:58:57 +09:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_print(PyStringObject *op, FILE *fp, int flags)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
char c;
|
1993-10-26 23:25:16 +08:00
|
|
|
|
int quote;
|
1991-06-08 07:58:57 +09:00
|
|
|
|
/* XXX Ought to check for interrupts when writing long strings */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (flags & Py_PRINT_RAW) {
|
1990-10-14 20:07:46 +08:00
|
|
|
|
fwrite(op->ob_sval, 1, (int) op->ob_size, fp);
|
1991-06-08 07:58:57 +09:00
|
|
|
|
return 0;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
1993-10-26 23:25:16 +08:00
|
|
|
|
|
2000-07-16 20:04:32 +08:00
|
|
|
|
/* figure out which quote to use; single is preferred */
|
1993-10-26 23:25:16 +08:00
|
|
|
|
quote = '\'';
|
|
|
|
|
if (strchr(op->ob_sval, '\'') && !strchr(op->ob_sval, '"'))
|
|
|
|
|
quote = '"';
|
|
|
|
|
|
|
|
|
|
fputc(quote, fp);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
for (i = 0; i < op->ob_size; i++) {
|
|
|
|
|
c = op->ob_sval[i];
|
1993-10-26 23:25:16 +08:00
|
|
|
|
if (c == quote || c == '\\')
|
1990-10-14 20:07:46 +08:00
|
|
|
|
fprintf(fp, "\\%c", c);
|
|
|
|
|
else if (c < ' ' || c >= 0177)
|
1993-10-26 23:25:16 +08:00
|
|
|
|
fprintf(fp, "\\%03o", c & 0377);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
else
|
1993-10-26 23:25:16 +08:00
|
|
|
|
fputc(c, fp);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
1993-10-26 23:25:16 +08:00
|
|
|
|
fputc(quote, fp);
|
1991-06-08 07:58:57 +09:00
|
|
|
|
return 0;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_repr(register PyStringObject *op)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
2000-06-30 18:29:57 +08:00
|
|
|
|
size_t newsize = 2 + 4 * op->ob_size * sizeof(char);
|
|
|
|
|
PyObject *v;
|
|
|
|
|
if (newsize > INT_MAX) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"string is too large to make repr");
|
|
|
|
|
}
|
|
|
|
|
v = PyString_FromStringAndSize((char *)NULL, newsize);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
if (v == NULL) {
|
1991-06-08 07:58:57 +09:00
|
|
|
|
return NULL;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
register int i;
|
|
|
|
|
register char c;
|
|
|
|
|
register char *p;
|
1993-10-26 23:25:16 +08:00
|
|
|
|
int quote;
|
|
|
|
|
|
2000-07-16 20:04:32 +08:00
|
|
|
|
/* figure out which quote to use; single is preferred */
|
1993-10-26 23:25:16 +08:00
|
|
|
|
quote = '\'';
|
|
|
|
|
if (strchr(op->ob_sval, '\'') && !strchr(op->ob_sval, '"'))
|
|
|
|
|
quote = '"';
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
p = ((PyStringObject *)v)->ob_sval;
|
1993-10-26 23:25:16 +08:00
|
|
|
|
*p++ = quote;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
for (i = 0; i < op->ob_size; i++) {
|
|
|
|
|
c = op->ob_sval[i];
|
1993-10-26 23:25:16 +08:00
|
|
|
|
if (c == quote || c == '\\')
|
1990-10-14 20:07:46 +08:00
|
|
|
|
*p++ = '\\', *p++ = c;
|
|
|
|
|
else if (c < ' ' || c >= 0177) {
|
1993-10-26 23:25:16 +08:00
|
|
|
|
sprintf(p, "\\%03o", c & 0377);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
while (*p != '\0')
|
|
|
|
|
p++;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
*p++ = c;
|
|
|
|
|
}
|
1993-10-26 23:25:16 +08:00
|
|
|
|
*p++ = quote;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
*p = '\0';
|
1997-05-02 11:12:38 +08:00
|
|
|
|
_PyString_Resize(
|
|
|
|
|
&v, (int) (p - ((PyStringObject *)v)->ob_sval));
|
1990-10-22 06:15:08 +08:00
|
|
|
|
return v;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_length(PyStringObject *a)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
|
|
|
|
return a->ob_size;
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_concat(register PyStringObject *a, register PyObject *bb)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
|
|
|
|
register unsigned int size;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyStringObject *op;
|
|
|
|
|
if (!PyString_Check(bb)) {
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (PyUnicode_Check(bb))
|
|
|
|
|
return PyUnicode_Concat((PyObject *)a, bb);
|
2000-06-01 11:12:13 +08:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"cannot add type \"%.200s\" to string",
|
|
|
|
|
bb->ob_type->tp_name);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
#define b ((PyStringObject *)bb)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
/* Optimize cases with empty left or right operand */
|
|
|
|
|
if (a->ob_size == 0) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(bb);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return bb;
|
|
|
|
|
}
|
|
|
|
|
if (b->ob_size == 0) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
size = a->ob_size + b->ob_size;
|
2000-05-04 07:44:39 +08:00
|
|
|
|
/* PyObject_NewVar is inlined */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
op = (PyStringObject *)
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_MALLOC(sizeof(PyStringObject) + size * sizeof(char));
|
1990-10-22 06:15:08 +08:00
|
|
|
|
if (op == NULL)
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
op->ob_shash = -1;
|
1997-01-18 15:55:05 +08:00
|
|
|
|
#endif
|
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
op->ob_sinterned = NULL;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#endif
|
1990-10-22 06:15:08 +08:00
|
|
|
|
memcpy(op->ob_sval, a->ob_sval, (int) a->ob_size);
|
|
|
|
|
memcpy(op->ob_sval + a->ob_size, b->ob_sval, (int) b->ob_size);
|
|
|
|
|
op->ob_sval[size] = '\0';
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
#undef b
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_repeat(register PyStringObject *a, register int n)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
|
|
|
|
register int i;
|
1997-04-10 03:41:24 +08:00
|
|
|
|
register int size;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyStringObject *op;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
if (n < 0)
|
|
|
|
|
n = 0;
|
|
|
|
|
size = a->ob_size * n;
|
|
|
|
|
if (size == a->ob_size) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
2000-05-04 07:44:39 +08:00
|
|
|
|
/* PyObject_NewVar is inlined */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
op = (PyStringObject *)
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_MALLOC(sizeof(PyStringObject) + size * sizeof(char));
|
1990-10-22 06:15:08 +08:00
|
|
|
|
if (op == NULL)
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyErr_NoMemory();
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_INIT_VAR(op, &PyString_Type, size);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
op->ob_shash = -1;
|
1997-01-18 15:55:05 +08:00
|
|
|
|
#endif
|
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
op->ob_sinterned = NULL;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#endif
|
1990-10-22 06:15:08 +08:00
|
|
|
|
for (i = 0; i < size; i += a->ob_size)
|
|
|
|
|
memcpy(op->ob_sval+i, a->ob_sval, (int) a->ob_size);
|
|
|
|
|
op->ob_sval[size] = '\0';
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return (PyObject *) op;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* String slice a[i:j] consists of characters a[i] ... a[j-1] */
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_slice(register PyStringObject *a, register int i, register int j)
|
|
|
|
|
/* j -- may be negative! */
|
1990-10-14 20:07:46 +08: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;
|
|
|
|
|
if (i == 0 && j == a->ob_size) { /* It's the same as a */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(a);
|
|
|
|
|
return (PyObject *)a;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
if (j < i)
|
|
|
|
|
j = i;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyString_FromStringAndSize(a->ob_sval + i, (int) (j-i));
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
2000-03-07 23:53:43 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_contains(PyObject *a, PyObject *el)
|
2000-03-07 23:53:43 +08:00
|
|
|
|
{
|
|
|
|
|
register char *s, *end;
|
|
|
|
|
register char c;
|
2000-04-10 21:47:21 +08:00
|
|
|
|
if (PyUnicode_Check(el))
|
2000-03-13 23:56:08 +08:00
|
|
|
|
return PyUnicode_Contains(a, el);
|
2000-04-10 21:47:21 +08:00
|
|
|
|
if (!PyString_Check(el) || PyString_Size(el) != 1) {
|
2000-03-07 23:53:43 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2000-06-09 22:04:53 +08:00
|
|
|
|
"'in <string>' requires character as left operand");
|
2000-03-07 23:53:43 +08:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
c = PyString_AsString(el)[0];
|
|
|
|
|
s = PyString_AsString(a);
|
|
|
|
|
end = s + PyString_Size(a);
|
|
|
|
|
while (s < end) {
|
|
|
|
|
if (c == *s++)
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_item(PyStringObject *a, register int i)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1991-04-04 18:48:33 +08:00
|
|
|
|
int c;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *v;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_IndexError, "string index out of range");
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1991-04-04 18:48:33 +08:00
|
|
|
|
c = a->ob_sval[i] & UCHAR_MAX;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
v = (PyObject *) characters[c];
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef COUNT_ALLOCS
|
|
|
|
|
if (v != NULL)
|
|
|
|
|
one_strings++;
|
|
|
|
|
#endif
|
1991-04-04 18:48:33 +08:00
|
|
|
|
if (v == NULL) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
v = PyString_FromStringAndSize((char *)NULL, 1);
|
1991-04-04 18:48:33 +08:00
|
|
|
|
if (v == NULL)
|
|
|
|
|
return NULL;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
characters[c] = (PyStringObject *) v;
|
|
|
|
|
((PyStringObject *)v)->ob_sval[0] = c;
|
1991-04-04 18:48:33 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_INCREF(v);
|
1991-04-04 18:48:33 +08:00
|
|
|
|
return v;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_compare(PyStringObject *a, PyStringObject *b)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1991-02-14 07:18:39 +08:00
|
|
|
|
int len_a = a->ob_size, len_b = b->ob_size;
|
|
|
|
|
int min_len = (len_a < len_b) ? len_a : len_b;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
int cmp;
|
|
|
|
|
if (min_len > 0) {
|
1996-10-23 22:19:40 +08:00
|
|
|
|
cmp = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval);
|
1993-10-22 20:04:32 +08:00
|
|
|
|
if (cmp == 0)
|
|
|
|
|
cmp = memcmp(a->ob_sval, b->ob_sval, min_len);
|
|
|
|
|
if (cmp != 0)
|
|
|
|
|
return cmp;
|
|
|
|
|
}
|
1991-02-14 07:18:39 +08:00
|
|
|
|
return (len_a < len_b) ? -1 : (len_a > len_b) ? 1 : 0;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
1993-03-29 18:43:31 +08:00
|
|
|
|
static long
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_hash(PyStringObject *a)
|
1993-03-29 18:43:31 +08:00
|
|
|
|
{
|
1993-10-22 20:04:32 +08:00
|
|
|
|
register int len;
|
|
|
|
|
register unsigned char *p;
|
|
|
|
|
register long x;
|
|
|
|
|
|
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
if (a->ob_shash != -1)
|
|
|
|
|
return a->ob_shash;
|
1997-02-15 00:29:22 +08:00
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
if (a->ob_sinterned != NULL)
|
|
|
|
|
return (a->ob_shash =
|
1997-05-02 11:12:38 +08:00
|
|
|
|
((PyStringObject *)(a->ob_sinterned))->ob_shash);
|
1997-02-15 00:29:22 +08:00
|
|
|
|
#endif
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#endif
|
|
|
|
|
len = a->ob_size;
|
|
|
|
|
p = (unsigned char *) a->ob_sval;
|
|
|
|
|
x = *p << 7;
|
1993-03-29 18:43:31 +08:00
|
|
|
|
while (--len >= 0)
|
1996-09-12 04:22:48 +08:00
|
|
|
|
x = (1000003*x) ^ *p++;
|
1993-03-29 18:43:31 +08:00
|
|
|
|
x ^= a->ob_size;
|
|
|
|
|
if (x == -1)
|
|
|
|
|
x = -2;
|
1993-10-22 20:04:32 +08:00
|
|
|
|
#ifdef CACHE_HASH
|
|
|
|
|
a->ob_shash = x;
|
|
|
|
|
#endif
|
1993-03-29 18:43:31 +08:00
|
|
|
|
return x;
|
|
|
|
|
}
|
|
|
|
|
|
1997-05-06 06:15:02 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_buffer_getreadbuf(PyStringObject *self, int index, const void **ptr)
|
1997-05-06 06:15:02 +08:00
|
|
|
|
{
|
|
|
|
|
if ( index != 0 ) {
|
1997-09-09 02:30:11 +08:00
|
|
|
|
PyErr_SetString(PyExc_SystemError,
|
1998-10-08 10:18:52 +08:00
|
|
|
|
"accessing non-existent string segment");
|
1997-05-06 06:15:02 +08:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
*ptr = (void *)self->ob_sval;
|
|
|
|
|
return self->ob_size;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_buffer_getwritebuf(PyStringObject *self, int index, const void **ptr)
|
1997-05-06 06:15:02 +08:00
|
|
|
|
{
|
1997-09-09 02:30:11 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
1998-10-01 23:59:48 +08:00
|
|
|
|
"Cannot use string as modifiable buffer");
|
1997-05-06 06:15:02 +08:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_buffer_getsegcount(PyStringObject *self, int *lenp)
|
1997-05-06 06:15:02 +08:00
|
|
|
|
{
|
|
|
|
|
if ( lenp )
|
|
|
|
|
*lenp = self->ob_size;
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
1998-10-08 10:18:52 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_buffer_getcharbuf(PyStringObject *self, int index, const char **ptr)
|
1998-10-08 10:18:52 +08: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 11:12:38 +08:00
|
|
|
|
static PySequenceMethods string_as_sequence = {
|
1994-08-30 16:19:36 +08:00
|
|
|
|
(inquiry)string_length, /*sq_length*/
|
|
|
|
|
(binaryfunc)string_concat, /*sq_concat*/
|
|
|
|
|
(intargfunc)string_repeat, /*sq_repeat*/
|
|
|
|
|
(intargfunc)string_item, /*sq_item*/
|
|
|
|
|
(intintargfunc)string_slice, /*sq_slice*/
|
1991-06-05 04:36:32 +09:00
|
|
|
|
0, /*sq_ass_item*/
|
|
|
|
|
0, /*sq_ass_slice*/
|
2000-03-07 23:53:43 +08:00
|
|
|
|
(objobjproc)string_contains /*sq_contains*/
|
1990-10-14 20:07:46 +08:00
|
|
|
|
};
|
|
|
|
|
|
1997-05-06 06:15:02 +08:00
|
|
|
|
static PyBufferProcs string_as_buffer = {
|
|
|
|
|
(getreadbufferproc)string_buffer_getreadbuf,
|
|
|
|
|
(getwritebufferproc)string_buffer_getwritebuf,
|
|
|
|
|
(getsegcountproc)string_buffer_getsegcount,
|
1998-10-08 10:18:52 +08:00
|
|
|
|
(getcharbufferproc)string_buffer_getcharbuf,
|
1997-05-06 06:15:02 +08:00
|
|
|
|
};
|
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define LEFTSTRIP 0
|
|
|
|
|
#define RIGHTSTRIP 1
|
|
|
|
|
#define BOTHSTRIP 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 16:02:21 +08:00
|
|
|
|
split_whitespace(const char *s, int len, int maxsplit)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
int i, j, err;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
PyObject* item;
|
|
|
|
|
PyObject *list = PyList_New(0);
|
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
for (i = j = 0; i < len; ) {
|
|
|
|
|
while (i < len && isspace(Py_CHARMASK(s[i])))
|
|
|
|
|
i++;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
j = i;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
while (i < len && !isspace(Py_CHARMASK(s[i])))
|
|
|
|
|
i++;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (j < i) {
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (maxsplit-- <= 0)
|
|
|
|
|
break;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
item = PyString_FromStringAndSize(s+j, (int)(i-j));
|
|
|
|
|
if (item == NULL)
|
|
|
|
|
goto finally;
|
|
|
|
|
err = PyList_Append(list, item);
|
|
|
|
|
Py_DECREF(item);
|
|
|
|
|
if (err < 0)
|
|
|
|
|
goto finally;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
while (i < len && isspace(Py_CHARMASK(s[i])))
|
|
|
|
|
i++;
|
|
|
|
|
j = i;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (j < len) {
|
|
|
|
|
item = PyString_FromStringAndSize(s+j, (int)(len - j));
|
|
|
|
|
if (item == NULL)
|
|
|
|
|
goto finally;
|
|
|
|
|
err = PyList_Append(list, item);
|
|
|
|
|
Py_DECREF(item);
|
|
|
|
|
if (err < 0)
|
|
|
|
|
goto finally;
|
|
|
|
|
}
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return list;
|
|
|
|
|
finally:
|
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char split__doc__[] =
|
|
|
|
|
"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-11 06:55:18 +08:00
|
|
|
|
delimiter string. If maxsplit is given, at most maxsplit\n\
|
|
|
|
|
splits are done. If sep is not specified, any whitespace string\n\
|
|
|
|
|
is a separator.";
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_split(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
int len = PyString_GET_SIZE(self), n, i, j, err;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
int maxsplit = -1;
|
|
|
|
|
const char *s = PyString_AS_STRING(self), *sub;
|
|
|
|
|
PyObject *list, *item, *subobj = Py_None;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "|Oi:split", &subobj, &maxsplit))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (maxsplit < 0)
|
|
|
|
|
maxsplit = INT_MAX;
|
|
|
|
|
if (subobj == Py_None)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return split_whitespace(s, len, maxsplit);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
n = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyUnicode_Split((PyObject *)self, subobj, maxsplit);
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &n))
|
|
|
|
|
return NULL;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (n == 0) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError, "empty separator");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
list = PyList_New(0);
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
i = j = 0;
|
|
|
|
|
while (i+n <= len) {
|
2000-06-20 23:47:54 +08:00
|
|
|
|
if (s[i] == sub[0] && memcmp(s+i, sub, n) == 0) {
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (maxsplit-- <= 0)
|
|
|
|
|
break;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
item = PyString_FromStringAndSize(s+j, (int)(i-j));
|
|
|
|
|
if (item == NULL)
|
|
|
|
|
goto fail;
|
|
|
|
|
err = PyList_Append(list, item);
|
|
|
|
|
Py_DECREF(item);
|
|
|
|
|
if (err < 0)
|
|
|
|
|
goto fail;
|
|
|
|
|
i = j = i + n;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
i++;
|
|
|
|
|
}
|
|
|
|
|
item = PyString_FromStringAndSize(s+j, (int)(len-j));
|
|
|
|
|
if (item == NULL)
|
|
|
|
|
goto fail;
|
|
|
|
|
err = PyList_Append(list, item);
|
|
|
|
|
Py_DECREF(item);
|
|
|
|
|
if (err < 0)
|
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
|
|
return list;
|
|
|
|
|
|
|
|
|
|
fail:
|
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char join__doc__[] =
|
|
|
|
|
"S.join(sequence) -> string\n\
|
|
|
|
|
\n\
|
2000-03-11 06:55:18 +08:00
|
|
|
|
Return a string which is the concatenation of the strings in the\n\
|
|
|
|
|
sequence. The separator between elements is S.";
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_join(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *sep = PyString_AS_STRING(self);
|
|
|
|
|
int seplen = PyString_GET_SIZE(self);
|
|
|
|
|
PyObject *res = NULL;
|
|
|
|
|
int reslen = 0;
|
|
|
|
|
char *p;
|
|
|
|
|
int seqlen = 0;
|
|
|
|
|
int sz = 100;
|
2000-07-12 04:55:38 +08:00
|
|
|
|
int i, slen, sz_incr;
|
2000-07-11 05:30:28 +08:00
|
|
|
|
PyObject *orig, *seq, *item;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-07-11 05:30:28 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O:join", &orig))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-07-11 12:58:12 +08:00
|
|
|
|
if (!(seq = PySequence_Fast(orig, ""))) {
|
2000-07-11 05:30:28 +08:00
|
|
|
|
if (PyErr_ExceptionMatches(PyExc_TypeError))
|
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
|
"sequence expected, %.80s found",
|
|
|
|
|
orig->ob_type->tp_name);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-07-11 05:30:28 +08:00
|
|
|
|
}
|
2000-07-11 12:58:12 +08:00
|
|
|
|
/* From here on out, errors go through finally: for proper
|
|
|
|
|
* reference count manipulations.
|
|
|
|
|
*/
|
2000-07-12 21:05:33 +08:00
|
|
|
|
seqlen = PySequence_Size(seq);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (seqlen == 1) {
|
2000-07-11 05:30:28 +08:00
|
|
|
|
item = PySequence_Fast_GET_ITEM(seq, 0);
|
|
|
|
|
Py_INCREF(item);
|
2000-07-11 12:58:12 +08:00
|
|
|
|
Py_DECREF(seq);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return item;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
2000-07-11 05:30:28 +08:00
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (!(res = PyString_FromStringAndSize((char*)NULL, sz)))
|
2000-07-11 12:58:12 +08:00
|
|
|
|
goto finally;
|
|
|
|
|
|
2000-07-12 04:55:38 +08:00
|
|
|
|
p = PyString_AS_STRING(res);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-07-11 05:30:28 +08:00
|
|
|
|
for (i = 0; i < seqlen; i++) {
|
|
|
|
|
item = PySequence_Fast_GET_ITEM(seq, i);
|
|
|
|
|
if (!PyString_Check(item)){
|
|
|
|
|
if (PyUnicode_Check(item)) {
|
|
|
|
|
Py_DECREF(res);
|
2000-07-11 12:58:12 +08:00
|
|
|
|
Py_DECREF(seq);
|
|
|
|
|
return PyUnicode_Join((PyObject *)self, seq);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
2000-07-11 05:30:28 +08:00
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2000-07-12 04:55:38 +08:00
|
|
|
|
"sequence item %i: expected string,"
|
|
|
|
|
" %.80s found",
|
2000-07-11 05:30:28 +08:00
|
|
|
|
i, item->ob_type->tp_name);
|
|
|
|
|
goto finally;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
2000-07-11 05:30:28 +08:00
|
|
|
|
slen = PyString_GET_SIZE(item);
|
|
|
|
|
while (reslen + slen + seplen >= sz) {
|
2000-07-12 04:55:38 +08:00
|
|
|
|
/* at least double the size of the string */
|
|
|
|
|
sz_incr = slen + seplen > sz ? slen + seplen : sz;
|
|
|
|
|
if (_PyString_Resize(&res, sz + sz_incr)) {
|
2000-03-06 22:52:18 +08:00
|
|
|
|
goto finally;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
}
|
2000-07-12 04:55:38 +08:00
|
|
|
|
sz += sz_incr;
|
|
|
|
|
p = PyString_AS_STRING(res) + reslen;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
2000-07-11 05:30:28 +08:00
|
|
|
|
if (i > 0) {
|
|
|
|
|
memcpy(p, sep, seplen);
|
|
|
|
|
p += seplen;
|
|
|
|
|
reslen += seplen;
|
|
|
|
|
}
|
|
|
|
|
memcpy(p, PyString_AS_STRING(item), slen);
|
|
|
|
|
p += slen;
|
|
|
|
|
reslen += slen;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
|
|
|
|
if (_PyString_Resize(&res, reslen))
|
|
|
|
|
goto finally;
|
2000-07-11 11:28:17 +08:00
|
|
|
|
Py_DECREF(seq);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return res;
|
|
|
|
|
|
|
|
|
|
finally:
|
2000-07-11 11:28:17 +08:00
|
|
|
|
Py_DECREF(seq);
|
2000-07-11 12:58:12 +08:00
|
|
|
|
Py_XDECREF(res);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static long
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_find_internal(PyStringObject *self, PyObject *args, int dir)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char *s = PyString_AS_STRING(self), *sub;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int len = PyString_GET_SIZE(self);
|
|
|
|
|
int n, i = 0, last = INT_MAX;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *subobj;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-05-08 22:08:05 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:find/rfind/index/rindex",
|
|
|
|
|
&subobj, _PyEval_SliceIndex, &i, _PyEval_SliceIndex, &last))
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return -2;
|
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
n = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyUnicode_Find((PyObject *)self, subobj, i, last, 1);
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &n))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return -2;
|
|
|
|
|
|
|
|
|
|
if (last > len)
|
|
|
|
|
last = len;
|
|
|
|
|
if (last < 0)
|
|
|
|
|
last += len;
|
|
|
|
|
if (last < 0)
|
|
|
|
|
last = 0;
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i += len;
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i = 0;
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (dir > 0) {
|
|
|
|
|
if (n == 0 && i <= last)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return (long)i;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
last -= n;
|
|
|
|
|
for (; i <= last; ++i)
|
2000-06-20 23:47:54 +08:00
|
|
|
|
if (s[i] == sub[0] && memcmp(&s[i], sub, n) == 0)
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return (long)i;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
int j;
|
|
|
|
|
|
|
|
|
|
if (n == 0 && i <= last)
|
|
|
|
|
return (long)last;
|
|
|
|
|
for (j = last-n; j >= i; --j)
|
2000-06-20 23:47:54 +08:00
|
|
|
|
if (s[j] == sub[0] && memcmp(&s[j], sub, n) == 0)
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return (long)j;
|
|
|
|
|
}
|
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char find__doc__[] =
|
|
|
|
|
"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\
|
|
|
|
|
Return -1 on failure.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_find(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
long result = string_find_internal(self, args, +1);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
return PyInt_FromLong(result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char index__doc__[] =
|
|
|
|
|
"S.index(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Like S.find() but raise ValueError when the substring is not found.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_index(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
long result = string_find_internal(self, args, +1);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
if (result == -1) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"substring not found in string.index");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char rfind__doc__[] =
|
|
|
|
|
"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\
|
|
|
|
|
Return -1 on failure.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_rfind(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
long result = string_find_internal(self, args, -1);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
return PyInt_FromLong(result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char rindex__doc__[] =
|
|
|
|
|
"S.rindex(sub [,start [,end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Like S.rfind() but raise ValueError when the substring is not found.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_rindex(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
long result = string_find_internal(self, args, -1);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (result == -2)
|
|
|
|
|
return NULL;
|
|
|
|
|
if (result == -1) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"substring not found in string.rindex");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
do_strip(PyStringObject *self, PyObject *args, int striptype)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self);
|
|
|
|
|
int len = PyString_GET_SIZE(self), i, j;
|
|
|
|
|
|
2000-02-29 21:59:29 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, ":strip"))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
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++;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (i == 0 && j == len) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*)self;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
return PyString_FromStringAndSize(s+i, j-i);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char strip__doc__[] =
|
|
|
|
|
"S.strip() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with leading and trailing\n\
|
|
|
|
|
whitespace removed.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_strip(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
return do_strip(self, args, BOTHSTRIP);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char lstrip__doc__[] =
|
|
|
|
|
"S.lstrip() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with leading whitespace removed.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_lstrip(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
return do_strip(self, args, LEFTSTRIP);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char rstrip__doc__[] =
|
|
|
|
|
"S.rstrip() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with trailing whitespace removed.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_rstrip(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
return do_strip(self, args, RIGHTSTRIP);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char lower__doc__[] =
|
|
|
|
|
"S.lower() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S converted to lowercase.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_lower(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
|
|
|
|
int i, n = PyString_GET_SIZE(self);
|
|
|
|
|
PyObject *new;
|
|
|
|
|
|
2000-02-29 21:59:29 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, ":lower"))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
new = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (new == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
s_new = PyString_AsString(new);
|
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (isupper(c)) {
|
|
|
|
|
*s_new = tolower(c);
|
|
|
|
|
} else
|
|
|
|
|
*s_new = c;
|
|
|
|
|
s_new++;
|
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char upper__doc__[] =
|
|
|
|
|
"S.upper() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S converted to uppercase.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_upper(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
|
|
|
|
int i, n = PyString_GET_SIZE(self);
|
|
|
|
|
PyObject *new;
|
|
|
|
|
|
2000-02-29 21:59:29 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, ":upper"))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
new = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (new == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
s_new = PyString_AsString(new);
|
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
|
int c = Py_CHARMASK(*s++);
|
|
|
|
|
if (islower(c)) {
|
|
|
|
|
*s_new = toupper(c);
|
|
|
|
|
} else
|
|
|
|
|
*s_new = c;
|
|
|
|
|
s_new++;
|
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
static char title__doc__[] =
|
|
|
|
|
"S.title() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a titlecased version of S, i.e. words start with uppercase\n\
|
|
|
|
|
characters, all remaining cased characters have lowercase.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_title(PyUnicodeObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
|
|
|
|
int i, n = PyString_GET_SIZE(self);
|
|
|
|
|
int previous_is_cased = 0;
|
|
|
|
|
PyObject *new;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, ":title"))
|
|
|
|
|
return NULL;
|
|
|
|
|
new = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (new == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
s_new = PyString_AsString(new);
|
|
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
static char capitalize__doc__[] =
|
|
|
|
|
"S.capitalize() -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of the string S with only its first character\n\
|
|
|
|
|
capitalized.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_capitalize(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
|
|
|
|
int i, n = PyString_GET_SIZE(self);
|
|
|
|
|
PyObject *new;
|
|
|
|
|
|
2000-02-29 21:59:29 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, ":capitalize"))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
new = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (new == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
s_new = PyString_AsString(new);
|
|
|
|
|
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++;
|
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char count__doc__[] =
|
|
|
|
|
"S.count(sub[, start[, end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return the number of occurrences of substring sub in string\n\
|
|
|
|
|
S[start:end]. Optional arguments start and end are\n\
|
|
|
|
|
interpreted as in slice notation.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_count(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char *s = PyString_AS_STRING(self), *sub;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int len = PyString_GET_SIZE(self), n;
|
|
|
|
|
int i = 0, last = INT_MAX;
|
|
|
|
|
int m, r;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *subobj;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-05-08 22:08:05 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:count", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &i, _PyEval_SliceIndex, &last))
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return NULL;
|
2000-05-08 22:08:05 +08:00
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
n = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyInt_FromLong(
|
|
|
|
|
PyUnicode_Count((PyObject *)self, subobj, i, last));
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &n))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (last > len)
|
|
|
|
|
last = len;
|
|
|
|
|
if (last < 0)
|
|
|
|
|
last += len;
|
|
|
|
|
if (last < 0)
|
|
|
|
|
last = 0;
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i += len;
|
|
|
|
|
if (i < 0)
|
|
|
|
|
i = 0;
|
|
|
|
|
m = last + 1 - n;
|
|
|
|
|
if (n == 0)
|
|
|
|
|
return PyInt_FromLong((long) (m-i));
|
|
|
|
|
|
|
|
|
|
r = 0;
|
|
|
|
|
while (i < m) {
|
|
|
|
|
if (!memcmp(s+i, sub, n)) {
|
|
|
|
|
r++;
|
|
|
|
|
i += n;
|
|
|
|
|
} else {
|
|
|
|
|
i++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong((long) r);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char swapcase__doc__[] =
|
|
|
|
|
"S.swapcase() -> string\n\
|
|
|
|
|
\n\
|
2000-03-11 06:55:18 +08:00
|
|
|
|
Return a copy of the string S with uppercase characters\n\
|
1999-10-13 03:54:53 +08:00
|
|
|
|
converted to lowercase and vice versa.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_swapcase(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *s = PyString_AS_STRING(self), *s_new;
|
|
|
|
|
int i, n = PyString_GET_SIZE(self);
|
|
|
|
|
PyObject *new;
|
|
|
|
|
|
2000-02-29 21:59:29 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, ":swapcase"))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
new = PyString_FromStringAndSize(NULL, n);
|
|
|
|
|
if (new == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
s_new = PyString_AsString(new);
|
|
|
|
|
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++;
|
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char translate__doc__[] =
|
|
|
|
|
"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\
|
|
|
|
|
translation table, which must be a string of length 256.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_translate(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
register char *input, *output;
|
|
|
|
|
register const char *table;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
register int i, c, changed = 0;
|
|
|
|
|
PyObject *input_obj = (PyObject*)self;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char *table1, *output_start, *del_table=NULL;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int inlen, tablen, dellen = 0;
|
|
|
|
|
PyObject *result;
|
|
|
|
|
int trans_table[256];
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *tableobj, *delobj = NULL;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O:translate",
|
|
|
|
|
&tableobj, &delobj))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (PyString_Check(tableobj)) {
|
|
|
|
|
table1 = PyString_AS_STRING(tableobj);
|
|
|
|
|
tablen = PyString_GET_SIZE(tableobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(tableobj)) {
|
|
|
|
|
/* Unicode .translate() does not support the deletechars
|
|
|
|
|
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);
|
|
|
|
|
}
|
|
|
|
|
else if (PyObject_AsCharBuffer(tableobj, &table1, &tablen))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (delobj != NULL) {
|
|
|
|
|
if (PyString_Check(delobj)) {
|
|
|
|
|
del_table = PyString_AS_STRING(delobj);
|
|
|
|
|
dellen = PyString_GET_SIZE(delobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(delobj)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"deletions are implemented differently for unicode");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
else if (PyObject_AsCharBuffer(delobj, &del_table, &dellen))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (tablen != 256) {
|
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"translation table must be 256 characters long");
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
del_table = NULL;
|
|
|
|
|
dellen = 0;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
table = table1;
|
|
|
|
|
inlen = PyString_Size(input_obj);
|
|
|
|
|
result = PyString_FromStringAndSize((char *)NULL, inlen);
|
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
output_start = output = PyString_AsString(result);
|
|
|
|
|
input = PyString_AsString(input_obj);
|
|
|
|
|
|
|
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
if (changed)
|
|
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
if (!changed) {
|
|
|
|
|
Py_DECREF(result);
|
|
|
|
|
Py_INCREF(input_obj);
|
|
|
|
|
return input_obj;
|
|
|
|
|
}
|
|
|
|
|
/* Fix the size of the resulting string */
|
|
|
|
|
if (inlen > 0 &&_PyString_Resize(&result, output-output_start))
|
|
|
|
|
return NULL;
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* What follows is used for implementing replace(). Perry Stoll. */
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
mymemfind
|
|
|
|
|
|
|
|
|
|
strstr replacement for arbitrary blocks of memory.
|
|
|
|
|
|
2000-03-21 00:36:48 +08:00
|
|
|
|
Locates the first occurrence in the memory pointed to by MEM of the
|
1999-10-13 03:54:53 +08:00
|
|
|
|
contents of memory pointed to by PAT. Returns the index into MEM if
|
|
|
|
|
found, or -1 if not found. If len of PAT is greater than length of
|
|
|
|
|
MEM, the function returns -1.
|
|
|
|
|
*/
|
|
|
|
|
static int
|
2000-07-09 16:02:21 +08:00
|
|
|
|
mymemfind(const char *mem, int len, const char *pat, int pat_len)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
register int ii;
|
|
|
|
|
|
|
|
|
|
/* pattern can not occur in the last pat_len-1 chars */
|
|
|
|
|
len -= pat_len;
|
|
|
|
|
|
|
|
|
|
for (ii = 0; ii <= len; ii++) {
|
2000-06-20 23:47:54 +08:00
|
|
|
|
if (mem[ii] == pat[0] && memcmp(&mem[ii], pat, pat_len) == 0) {
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return ii;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
mymemcnt
|
|
|
|
|
|
|
|
|
|
Return the number of distinct times PAT is found in MEM.
|
|
|
|
|
meaning mem=1111 and pat==11 returns 2.
|
|
|
|
|
mem=11111 and pat==11 also return 2.
|
|
|
|
|
*/
|
|
|
|
|
static int
|
2000-07-09 16:02:21 +08:00
|
|
|
|
mymemcnt(const char *mem, int len, const char *pat, int pat_len)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
register int offset = 0;
|
|
|
|
|
int nfound = 0;
|
|
|
|
|
|
|
|
|
|
while (len >= 0) {
|
|
|
|
|
offset = mymemfind(mem, len, pat, pat_len);
|
|
|
|
|
if (offset == -1)
|
|
|
|
|
break;
|
|
|
|
|
mem += offset + pat_len;
|
|
|
|
|
len -= offset + pat_len;
|
|
|
|
|
nfound++;
|
|
|
|
|
}
|
|
|
|
|
return nfound;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
mymemreplace
|
|
|
|
|
|
2000-07-16 20:04:32 +08:00
|
|
|
|
Return a string in which all occurrences of PAT in memory STR are
|
1999-10-13 03:54:53 +08:00
|
|
|
|
replaced with SUB.
|
|
|
|
|
|
2000-07-16 20:04:32 +08:00
|
|
|
|
If length of PAT is less than length of STR or there are no occurrences
|
1999-10-13 03:54:53 +08:00
|
|
|
|
of PAT in STR, then the original string is returned. Otherwise, a new
|
|
|
|
|
string is allocated here and returned.
|
|
|
|
|
|
|
|
|
|
on return, out_len is:
|
|
|
|
|
the length of output string, or
|
|
|
|
|
-1 if the input string is returned, or
|
|
|
|
|
unchanged if an error occurs (no memory).
|
|
|
|
|
|
|
|
|
|
return value is:
|
|
|
|
|
the new string allocated locally, or
|
|
|
|
|
NULL if an error occurred.
|
|
|
|
|
*/
|
|
|
|
|
static char *
|
2000-07-09 16:02:21 +08:00
|
|
|
|
mymemreplace(const char *str, int len, /* input string */
|
|
|
|
|
const char *pat, int pat_len, /* pattern string to find */
|
|
|
|
|
const char *sub, int sub_len, /* substitution string */
|
|
|
|
|
int count, /* number of replacements */
|
2000-07-09 15:04:36 +08:00
|
|
|
|
int *out_len)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
char *out_s;
|
|
|
|
|
char *new_s;
|
|
|
|
|
int nfound, offset, new_len;
|
|
|
|
|
|
|
|
|
|
if (len == 0 || pat_len > len)
|
|
|
|
|
goto return_same;
|
|
|
|
|
|
|
|
|
|
/* find length of output string */
|
|
|
|
|
nfound = mymemcnt(str, len, pat, pat_len);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (count < 0)
|
|
|
|
|
count = INT_MAX;
|
|
|
|
|
else if (nfound > count)
|
|
|
|
|
nfound = count;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (nfound == 0)
|
|
|
|
|
goto return_same;
|
|
|
|
|
new_len = len + nfound*(sub_len - pat_len);
|
|
|
|
|
|
2000-05-04 07:44:39 +08:00
|
|
|
|
new_s = (char *)PyMem_MALLOC(new_len);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (new_s == NULL) return NULL;
|
|
|
|
|
|
|
|
|
|
*out_len = new_len;
|
|
|
|
|
out_s = new_s;
|
|
|
|
|
|
|
|
|
|
while (len > 0) {
|
|
|
|
|
/* find index of next instance of pattern */
|
|
|
|
|
offset = mymemfind(str, len, pat, pat_len);
|
|
|
|
|
/* if not found, break out of loop */
|
|
|
|
|
if (offset == -1) break;
|
|
|
|
|
|
|
|
|
|
/* copy non matching part of input string */
|
|
|
|
|
memcpy(new_s, str, offset); /* copy part of str before pat */
|
|
|
|
|
str += offset + pat_len; /* move str past pattern */
|
|
|
|
|
len -= offset + pat_len; /* reduce length of str remaining */
|
|
|
|
|
|
|
|
|
|
/* copy substitute into the output string */
|
|
|
|
|
new_s += offset; /* move new_s to dest for sub string */
|
|
|
|
|
memcpy(new_s, sub, sub_len); /* copy substring into new_s */
|
|
|
|
|
new_s += sub_len; /* offset new_s past sub string */
|
|
|
|
|
|
|
|
|
|
/* break when we've done count replacements */
|
|
|
|
|
if (--count == 0) break;
|
|
|
|
|
}
|
|
|
|
|
/* copy any remaining values into output string */
|
|
|
|
|
if (len > 0)
|
|
|
|
|
memcpy(new_s, str, len);
|
|
|
|
|
return out_s;
|
|
|
|
|
|
|
|
|
|
return_same:
|
|
|
|
|
*out_len = -1;
|
2000-07-09 16:02:21 +08:00
|
|
|
|
return (char*)str; /* have to cast away constness here */
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char replace__doc__[] =
|
|
|
|
|
"S.replace (old, new[, maxsplit]) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of string S with all occurrences of substring\n\
|
|
|
|
|
old replaced by new. If the optional argument maxsplit is\n\
|
|
|
|
|
given, only the first maxsplit occurrences are replaced.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_replace(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char *str = PyString_AS_STRING(self), *sub, *repl;
|
|
|
|
|
char *new_s;
|
|
|
|
|
int len = PyString_GET_SIZE(self), sub_len, repl_len, out_len;
|
|
|
|
|
int count = -1;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
PyObject *new;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *subobj, *replobj;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "OO|i:replace",
|
|
|
|
|
&subobj, &replobj, &count))
|
|
|
|
|
return NULL;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
sub = PyString_AS_STRING(subobj);
|
|
|
|
|
sub_len = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyUnicode_Replace((PyObject *)self,
|
|
|
|
|
subobj, replobj, count);
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &sub, &sub_len))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (PyString_Check(replobj)) {
|
|
|
|
|
repl = PyString_AS_STRING(replobj);
|
|
|
|
|
repl_len = PyString_GET_SIZE(replobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(replobj))
|
|
|
|
|
return PyUnicode_Replace((PyObject *)self,
|
|
|
|
|
subobj, replobj, count);
|
|
|
|
|
else if (PyObject_AsCharBuffer(replobj, &repl, &repl_len))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-03-13 23:56:08 +08:00
|
|
|
|
if (sub_len <= 0) {
|
2000-03-21 00:36:48 +08:00
|
|
|
|
PyErr_SetString(PyExc_ValueError, "empty pattern string");
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2000-03-11 06:55:18 +08:00
|
|
|
|
new_s = mymemreplace(str,len,sub,sub_len,repl,repl_len,count,&out_len);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
if (new_s == NULL) {
|
|
|
|
|
PyErr_NoMemory();
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
if (out_len == -1) {
|
|
|
|
|
/* we're returning another reference to self */
|
|
|
|
|
new = (PyObject*)self;
|
|
|
|
|
Py_INCREF(new);
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
new = PyString_FromStringAndSize(new_s, out_len);
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyMem_FREE(new_s);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
}
|
|
|
|
|
return new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char startswith__doc__[] =
|
|
|
|
|
"S.startswith(prefix[, start[, end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if S starts with the specified prefix, otherwise return 0. With\n\
|
|
|
|
|
optional start, test S beginning at that position. With optional end, stop\n\
|
|
|
|
|
comparing S at that position.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_startswith(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char* str = PyString_AS_STRING(self);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int len = PyString_GET_SIZE(self);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char* prefix;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int plen;
|
|
|
|
|
int start = 0;
|
|
|
|
|
int end = -1;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *subobj;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-05-08 22:08:05 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:startswith", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
prefix = PyString_AS_STRING(subobj);
|
|
|
|
|
plen = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyInt_FromLong(
|
|
|
|
|
PyUnicode_Tailmatch((PyObject *)self,
|
|
|
|
|
subobj, start, end, -1));
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &prefix, &plen))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* adopt Java semantics for index out of range. it is legal for
|
|
|
|
|
* offset to be == plen, but this only returns true if prefix is
|
|
|
|
|
* the empty string.
|
|
|
|
|
*/
|
|
|
|
|
if (start < 0 || start+plen > len)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
|
|
|
|
if (!memcmp(str+start, prefix, plen)) {
|
|
|
|
|
/* did the match end after the specified end? */
|
|
|
|
|
if (end < 0)
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
else if (end - start < plen)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
else
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
}
|
|
|
|
|
else return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char endswith__doc__[] =
|
|
|
|
|
"S.endswith(suffix[, start[, end]]) -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if S ends with the specified suffix, otherwise return 0. With\n\
|
|
|
|
|
optional start, test S beginning at that position. With optional end, stop\n\
|
|
|
|
|
comparing S at that position.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_endswith(PyStringObject *self, PyObject *args)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char* str = PyString_AS_STRING(self);
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int len = PyString_GET_SIZE(self);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
const char* suffix;
|
|
|
|
|
int slen;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
int start = 0;
|
|
|
|
|
int end = -1;
|
|
|
|
|
int lower, upper;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *subobj;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-05-08 22:08:05 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "O|O&O&:endswith", &subobj,
|
|
|
|
|
_PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
if (PyString_Check(subobj)) {
|
|
|
|
|
suffix = PyString_AS_STRING(subobj);
|
|
|
|
|
slen = PyString_GET_SIZE(subobj);
|
|
|
|
|
}
|
|
|
|
|
else if (PyUnicode_Check(subobj))
|
|
|
|
|
return PyInt_FromLong(
|
|
|
|
|
PyUnicode_Tailmatch((PyObject *)self,
|
|
|
|
|
subobj, start, end, +1));
|
|
|
|
|
else if (PyObject_AsCharBuffer(subobj, &suffix, &slen))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (start < 0 || start > len || slen > len)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
|
|
|
|
upper = (end >= 0 && end <= len) ? end : len;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
lower = (upper - slen) > start ? (upper - slen) : start;
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (upper-lower >= slen && !memcmp(str+lower, suffix, slen))
|
1999-10-13 03:54:53 +08:00
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
else return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-07-06 19:29:01 +08:00
|
|
|
|
static char encode__doc__[] =
|
|
|
|
|
"S.encode([encoding[,errors]]) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return an encoded string version of S. Default encoding is the current\n\
|
|
|
|
|
default string encoding. errors may be given to set a different error\n\
|
|
|
|
|
handling scheme. Default is 'strict' meaning that encoding errors raise\n\
|
|
|
|
|
a ValueError. Other possible values are 'ignore' and 'replace'.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_encode(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
char *encoding = NULL;
|
|
|
|
|
char *errors = NULL;
|
|
|
|
|
if (!PyArg_ParseTuple(args, "|ss:encode", &encoding, &errors))
|
|
|
|
|
return NULL;
|
|
|
|
|
return PyString_AsEncodedString((PyObject *)self, encoding, errors);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
static char expandtabs__doc__[] =
|
|
|
|
|
"S.expandtabs([tabsize]) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return a copy of S where all tab characters are expanded using spaces.\n\
|
|
|
|
|
If tabsize is not given, a tab size of 8 characters is assumed.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_expandtabs(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
const char *e, *p;
|
|
|
|
|
char *q;
|
|
|
|
|
int i, j;
|
|
|
|
|
PyObject *u;
|
|
|
|
|
int tabsize = 8;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
2000-07-16 20:04:32 +08:00
|
|
|
|
/* First pass: determine size of output string */
|
2000-03-11 06:55:18 +08:00
|
|
|
|
i = j = 0;
|
|
|
|
|
e = PyString_AS_STRING(self) + PyString_GET_SIZE(self);
|
|
|
|
|
for (p = PyString_AS_STRING(self); p < e; p++)
|
|
|
|
|
if (*p == '\t') {
|
|
|
|
|
if (tabsize > 0)
|
|
|
|
|
j += tabsize - (j % tabsize);
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
j++;
|
|
|
|
|
if (*p == '\n' || *p == '\r') {
|
|
|
|
|
i += j;
|
|
|
|
|
j = 0;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* 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;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static
|
|
|
|
|
PyObject *pad(PyStringObject *self,
|
|
|
|
|
int left,
|
|
|
|
|
int right,
|
|
|
|
|
char fill)
|
|
|
|
|
{
|
|
|
|
|
PyObject *u;
|
|
|
|
|
|
|
|
|
|
if (left < 0)
|
|
|
|
|
left = 0;
|
|
|
|
|
if (right < 0)
|
|
|
|
|
right = 0;
|
|
|
|
|
|
|
|
|
|
if (left == 0 && right == 0) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject *)self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
u = PyString_FromStringAndSize(NULL,
|
|
|
|
|
left + PyString_GET_SIZE(self) + right);
|
|
|
|
|
if (u) {
|
|
|
|
|
if (left)
|
|
|
|
|
memset(PyString_AS_STRING(u), fill, left);
|
|
|
|
|
memcpy(PyString_AS_STRING(u) + left,
|
|
|
|
|
PyString_AS_STRING(self),
|
|
|
|
|
PyString_GET_SIZE(self));
|
|
|
|
|
if (right)
|
|
|
|
|
memset(PyString_AS_STRING(u) + left + PyString_GET_SIZE(self),
|
|
|
|
|
fill, right);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return u;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static char ljust__doc__[] =
|
|
|
|
|
"S.ljust(width) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return S left justified in a string of length width. Padding is\n\
|
|
|
|
|
done using spaces.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_ljust(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
int width;
|
|
|
|
|
if (!PyArg_ParseTuple(args, "i:ljust", &width))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (PyString_GET_SIZE(self) >= width) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return pad(self, 0, width - PyString_GET_SIZE(self), ' ');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char rjust__doc__[] =
|
|
|
|
|
"S.rjust(width) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return S right justified in a string of length width. Padding is\n\
|
|
|
|
|
done using spaces.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_rjust(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
int width;
|
|
|
|
|
if (!PyArg_ParseTuple(args, "i:rjust", &width))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (PyString_GET_SIZE(self) >= width) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return pad(self, width - PyString_GET_SIZE(self), 0, ' ');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char center__doc__[] =
|
|
|
|
|
"S.center(width) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return S centered in a string of length width. Padding is done\n\
|
|
|
|
|
using spaces.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_center(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
int marg, left;
|
|
|
|
|
int width;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "i:center", &width))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (PyString_GET_SIZE(self) >= width) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
marg = width - PyString_GET_SIZE(self);
|
|
|
|
|
left = marg / 2 + (marg & width & 1);
|
|
|
|
|
|
|
|
|
|
return pad(self, left, marg - left, ' ');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
|
static char zfill__doc__[] =
|
|
|
|
|
"S.zfill(width) -> string\n\
|
|
|
|
|
\n\
|
|
|
|
|
Pad a numeric string x with zeros on the left, to fill a field\n\
|
|
|
|
|
of the specified width. The string x is never truncated.";
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
|
string_zfill(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
int fill;
|
|
|
|
|
PyObject *u;
|
|
|
|
|
char *str;
|
|
|
|
|
|
|
|
|
|
int width;
|
|
|
|
|
if (!PyArg_ParseTuple(args, "i:zfill", &width))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (PyString_GET_SIZE(self) >= width) {
|
|
|
|
|
Py_INCREF(self);
|
|
|
|
|
return (PyObject*) self;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fill = width - PyString_GET_SIZE(self);
|
|
|
|
|
|
|
|
|
|
u = pad(self, fill, 0, '0');
|
|
|
|
|
if (u == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
str = PyString_AS_STRING(u);
|
|
|
|
|
if (str[fill] == '+' || str[fill] == '-') {
|
|
|
|
|
/* move sign to beginning of string */
|
|
|
|
|
str[0] = str[fill];
|
|
|
|
|
str[fill] = '0';
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return u;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
static char isspace__doc__[] =
|
|
|
|
|
"S.isspace() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if there are only whitespace characters in S,\n\
|
|
|
|
|
0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_isspace(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char *e;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isspace(*p))
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
|
2000-06-14 17:18:32 +08:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isspace(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-07-05 17:47:46 +08:00
|
|
|
|
static char isalpha__doc__[] =
|
|
|
|
|
"S.isalpha() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if all characters in S are alphabetic\n\
|
|
|
|
|
and there is at least one character in S, 0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_isalpha(PyUnicodeObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-07-05 17:47:46 +08:00
|
|
|
|
register const unsigned char *e;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isalpha(*p))
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
|
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isalpha(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char isalnum__doc__[] =
|
|
|
|
|
"S.isalnum() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if all characters in S are alphanumeric\n\
|
|
|
|
|
and there is at least one character in S, 0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_isalnum(PyUnicodeObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-07-05 17:47:46 +08:00
|
|
|
|
register const unsigned char *e;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isalnum(*p))
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
|
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isalnum(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
static char isdigit__doc__[] =
|
|
|
|
|
"S.isdigit() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if there are only digit characters in S,\n\
|
|
|
|
|
0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_isdigit(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char *e;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1 &&
|
|
|
|
|
isdigit(*p))
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
|
2000-06-14 17:18:32 +08:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (!isdigit(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char islower__doc__[] =
|
|
|
|
|
"S.islower() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if all cased characters in S are lowercase and there is\n\
|
|
|
|
|
at least one cased character in S, 0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_islower(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char *e;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
int cased;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
|
|
|
|
return PyInt_FromLong(islower(*p) != 0);
|
|
|
|
|
|
2000-06-14 17:18:32 +08:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (isupper(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
else if (!cased && islower(*p))
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(cased);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char isupper__doc__[] =
|
|
|
|
|
"S.isupper() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if all cased characters in S are uppercase and there is\n\
|
|
|
|
|
at least one cased character in S, 0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_isupper(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char *e;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
int cased;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
|
|
|
|
return PyInt_FromLong(isupper(*p) != 0);
|
|
|
|
|
|
2000-06-14 17:18:32 +08:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
|
|
|
|
if (islower(*p))
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
else if (!cased && isupper(*p))
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(cased);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char istitle__doc__[] =
|
|
|
|
|
"S.istitle() -> int\n\
|
|
|
|
|
\n\
|
|
|
|
|
Return 1 if S is a titlecased string, i.e. uppercase characters\n\
|
|
|
|
|
may only follow uncased characters and lowercase characters only cased\n\
|
|
|
|
|
ones. Return 0 otherwise.";
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_istitle(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
2000-07-09 15:04:36 +08:00
|
|
|
|
register const unsigned char *p
|
|
|
|
|
= (unsigned char *) PyString_AS_STRING(self);
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char *e;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
int cased, previous_is_cased;
|
|
|
|
|
|
|
|
|
|
if (!PyArg_NoArgs(args))
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
/* Shortcut for single character strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 1)
|
|
|
|
|
return PyInt_FromLong(isupper(*p) != 0);
|
|
|
|
|
|
2000-06-14 17:18:32 +08:00
|
|
|
|
/* Special case for empty strings */
|
|
|
|
|
if (PyString_GET_SIZE(self) == 0)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
|
2000-03-11 06:55:18 +08:00
|
|
|
|
e = p + PyString_GET_SIZE(self);
|
|
|
|
|
cased = 0;
|
|
|
|
|
previous_is_cased = 0;
|
|
|
|
|
for (; p < e; p++) {
|
2000-05-06 04:44:24 +08:00
|
|
|
|
register const unsigned char ch = *p;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
if (isupper(ch)) {
|
|
|
|
|
if (previous_is_cased)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
else if (islower(ch)) {
|
|
|
|
|
if (!previous_is_cased)
|
|
|
|
|
return PyInt_FromLong(0);
|
|
|
|
|
previous_is_cased = 1;
|
|
|
|
|
cased = 1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
previous_is_cased = 0;
|
|
|
|
|
}
|
|
|
|
|
return PyInt_FromLong(cased);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static char splitlines__doc__[] =
|
2000-04-11 23:39:26 +08:00
|
|
|
|
"S.splitlines([keepends]]) -> list of strings\n\
|
2000-03-11 06:55:18 +08:00
|
|
|
|
\n\
|
|
|
|
|
Return a list of the lines in S, breaking at line boundaries.\n\
|
2000-04-11 23:39:26 +08:00
|
|
|
|
Line breaks are not included in the resulting list unless keepends\n\
|
|
|
|
|
is given and true.";
|
2000-03-11 06:55:18 +08:00
|
|
|
|
|
|
|
|
|
#define SPLIT_APPEND(data, left, right) \
|
|
|
|
|
str = PyString_FromStringAndSize(data + left, right - left); \
|
|
|
|
|
if (!str) \
|
|
|
|
|
goto onError; \
|
|
|
|
|
if (PyList_Append(list, str)) { \
|
|
|
|
|
Py_DECREF(str); \
|
|
|
|
|
goto onError; \
|
|
|
|
|
} \
|
|
|
|
|
else \
|
|
|
|
|
Py_DECREF(str);
|
|
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
|
string_splitlines(PyStringObject *self, PyObject *args)
|
|
|
|
|
{
|
|
|
|
|
register int i;
|
|
|
|
|
register int j;
|
|
|
|
|
int len;
|
2000-04-11 23:39:26 +08:00
|
|
|
|
int keepends = 0;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
PyObject *list;
|
|
|
|
|
PyObject *str;
|
|
|
|
|
char *data;
|
|
|
|
|
|
2000-04-11 23:39:26 +08:00
|
|
|
|
if (!PyArg_ParseTuple(args, "|i:splitlines", &keepends))
|
2000-03-11 06:55:18 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
data = PyString_AS_STRING(self);
|
|
|
|
|
len = PyString_GET_SIZE(self);
|
|
|
|
|
|
|
|
|
|
list = PyList_New(0);
|
|
|
|
|
if (!list)
|
|
|
|
|
goto onError;
|
|
|
|
|
|
|
|
|
|
for (i = j = 0; i < len; ) {
|
2000-04-11 23:39:26 +08:00
|
|
|
|
int eol;
|
|
|
|
|
|
2000-03-11 06:55:18 +08: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 23:39:26 +08:00
|
|
|
|
eol = i;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
if (i < len) {
|
|
|
|
|
if (data[i] == '\r' && i + 1 < len &&
|
|
|
|
|
data[i+1] == '\n')
|
|
|
|
|
i += 2;
|
|
|
|
|
else
|
|
|
|
|
i++;
|
2000-04-11 23:39:26 +08:00
|
|
|
|
if (keepends)
|
|
|
|
|
eol = i;
|
2000-03-11 06:55:18 +08:00
|
|
|
|
}
|
2000-04-11 23:39:26 +08:00
|
|
|
|
SPLIT_APPEND(data, j, eol);
|
2000-03-11 06:55:18 +08:00
|
|
|
|
j = i;
|
|
|
|
|
}
|
|
|
|
|
if (j < len) {
|
|
|
|
|
SPLIT_APPEND(data, j, len);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return list;
|
|
|
|
|
|
|
|
|
|
onError:
|
|
|
|
|
Py_DECREF(list);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#undef SPLIT_APPEND
|
|
|
|
|
|
1999-10-13 03:54:53 +08:00
|
|
|
|
|
|
|
|
|
static PyMethodDef
|
|
|
|
|
string_methods[] = {
|
2000-03-11 06:55:18 +08:00
|
|
|
|
/* Counterparts of the obsolete stropmodule functions; except
|
|
|
|
|
string.maketrans(). */
|
|
|
|
|
{"join", (PyCFunction)string_join, 1, join__doc__},
|
|
|
|
|
{"split", (PyCFunction)string_split, 1, split__doc__},
|
|
|
|
|
{"lower", (PyCFunction)string_lower, 1, lower__doc__},
|
|
|
|
|
{"upper", (PyCFunction)string_upper, 1, upper__doc__},
|
|
|
|
|
{"islower", (PyCFunction)string_islower, 0, islower__doc__},
|
|
|
|
|
{"isupper", (PyCFunction)string_isupper, 0, isupper__doc__},
|
|
|
|
|
{"isspace", (PyCFunction)string_isspace, 0, isspace__doc__},
|
|
|
|
|
{"isdigit", (PyCFunction)string_isdigit, 0, isdigit__doc__},
|
|
|
|
|
{"istitle", (PyCFunction)string_istitle, 0, istitle__doc__},
|
2000-07-05 17:47:46 +08:00
|
|
|
|
{"isalpha", (PyCFunction)string_isalpha, 0, isalpha__doc__},
|
|
|
|
|
{"isalnum", (PyCFunction)string_isalnum, 0, isalnum__doc__},
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{"capitalize", (PyCFunction)string_capitalize, 1, capitalize__doc__},
|
|
|
|
|
{"count", (PyCFunction)string_count, 1, count__doc__},
|
|
|
|
|
{"endswith", (PyCFunction)string_endswith, 1, endswith__doc__},
|
|
|
|
|
{"find", (PyCFunction)string_find, 1, find__doc__},
|
|
|
|
|
{"index", (PyCFunction)string_index, 1, index__doc__},
|
|
|
|
|
{"lstrip", (PyCFunction)string_lstrip, 1, lstrip__doc__},
|
|
|
|
|
{"replace", (PyCFunction)string_replace, 1, replace__doc__},
|
|
|
|
|
{"rfind", (PyCFunction)string_rfind, 1, rfind__doc__},
|
|
|
|
|
{"rindex", (PyCFunction)string_rindex, 1, rindex__doc__},
|
|
|
|
|
{"rstrip", (PyCFunction)string_rstrip, 1, rstrip__doc__},
|
|
|
|
|
{"startswith", (PyCFunction)string_startswith, 1, startswith__doc__},
|
|
|
|
|
{"strip", (PyCFunction)string_strip, 1, strip__doc__},
|
|
|
|
|
{"swapcase", (PyCFunction)string_swapcase, 1, swapcase__doc__},
|
2000-03-11 06:55:18 +08:00
|
|
|
|
{"translate", (PyCFunction)string_translate, 1, translate__doc__},
|
|
|
|
|
{"title", (PyCFunction)string_title, 1, title__doc__},
|
|
|
|
|
{"ljust", (PyCFunction)string_ljust, 1, ljust__doc__},
|
|
|
|
|
{"rjust", (PyCFunction)string_rjust, 1, rjust__doc__},
|
|
|
|
|
{"center", (PyCFunction)string_center, 1, center__doc__},
|
2000-07-06 19:29:01 +08:00
|
|
|
|
{"encode", (PyCFunction)string_encode, 1, encode__doc__},
|
2000-03-11 06:55:18 +08:00
|
|
|
|
{"expandtabs", (PyCFunction)string_expandtabs, 1, expandtabs__doc__},
|
|
|
|
|
{"splitlines", (PyCFunction)string_splitlines, 1, splitlines__doc__},
|
|
|
|
|
#if 0
|
|
|
|
|
{"zfill", (PyCFunction)string_zfill, 1, zfill__doc__},
|
|
|
|
|
#endif
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
string_getattr(PyStringObject *s, char *name)
|
1999-10-13 03:54:53 +08:00
|
|
|
|
{
|
|
|
|
|
return Py_FindMethod(string_methods, (PyObject*)s, name);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyTypeObject PyString_Type = {
|
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
0,
|
|
|
|
|
"string",
|
1997-05-02 11:12:38 +08:00
|
|
|
|
sizeof(PyStringObject),
|
1990-10-14 20:07:46 +08:00
|
|
|
|
sizeof(char),
|
1994-08-30 16:19:36 +08:00
|
|
|
|
(destructor)string_dealloc, /*tp_dealloc*/
|
|
|
|
|
(printfunc)string_print, /*tp_print*/
|
1999-10-13 03:54:53 +08:00
|
|
|
|
(getattrfunc)string_getattr, /*tp_getattr*/
|
1990-10-14 20:07:46 +08:00
|
|
|
|
0, /*tp_setattr*/
|
1994-08-30 16:19:36 +08:00
|
|
|
|
(cmpfunc)string_compare, /*tp_compare*/
|
|
|
|
|
(reprfunc)string_repr, /*tp_repr*/
|
1990-10-14 20:07:46 +08:00
|
|
|
|
0, /*tp_as_number*/
|
|
|
|
|
&string_as_sequence, /*tp_as_sequence*/
|
|
|
|
|
0, /*tp_as_mapping*/
|
1994-08-30 16:19:36 +08:00
|
|
|
|
(hashfunc)string_hash, /*tp_hash*/
|
1997-01-18 15:55:05 +08:00
|
|
|
|
0, /*tp_call*/
|
|
|
|
|
0, /*tp_str*/
|
|
|
|
|
0, /*tp_getattro*/
|
|
|
|
|
0, /*tp_setattro*/
|
1997-05-06 06:15:02 +08:00
|
|
|
|
&string_as_buffer, /*tp_as_buffer*/
|
1998-10-08 10:18:52 +08:00
|
|
|
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/
|
1997-01-18 15:55:05 +08:00
|
|
|
|
0, /*tp_doc*/
|
1990-10-14 20:07:46 +08:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
void
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_Concat(register PyObject **pv, register PyObject *w)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyObject *v;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (*pv == NULL)
|
|
|
|
|
return;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (w == NULL || !PyString_Check(*pv)) {
|
|
|
|
|
Py_DECREF(*pv);
|
1994-08-30 16:19:36 +08:00
|
|
|
|
*pv = NULL;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
v = string_concat((PyStringObject *) *pv, w);
|
|
|
|
|
Py_DECREF(*pv);
|
1990-10-14 20:07:46 +08:00
|
|
|
|
*pv = v;
|
|
|
|
|
}
|
|
|
|
|
|
1994-08-30 16:19:36 +08:00
|
|
|
|
void
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_ConcatAndDel(register PyObject **pv, register PyObject *w)
|
1994-08-30 16:19:36 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyString_Concat(pv, w);
|
|
|
|
|
Py_XDECREF(w);
|
1994-08-30 16:19:36 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
1990-10-14 20:07:46 +08: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
|
|
|
|
|
already be known to some other part of the code... */
|
|
|
|
|
|
|
|
|
|
int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
_PyString_Resize(PyObject **pv, int newsize)
|
1990-10-14 20:07:46 +08:00
|
|
|
|
{
|
1997-05-02 11:12:38 +08:00
|
|
|
|
register PyObject *v;
|
|
|
|
|
register PyStringObject *sv;
|
1990-11-19 01:30:23 +08:00
|
|
|
|
v = *pv;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyString_Check(v) || v->ob_refcnt != 1) {
|
1990-10-14 20:07:46 +08:00
|
|
|
|
*pv = 0;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(v);
|
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-22 06:15:08 +08:00
|
|
|
|
return -1;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
1990-11-19 01:30:23 +08:00
|
|
|
|
/* XXX UNREF/NEWREF interface should be more symmetrical */
|
1996-05-24 06:46:51 +08:00
|
|
|
|
#ifdef Py_REF_DEBUG
|
1995-03-30 00:57:48 +08:00
|
|
|
|
--_Py_RefTotal;
|
1990-11-19 01:30:23 +08:00
|
|
|
|
#endif
|
1997-05-02 11:12:38 +08:00
|
|
|
|
_Py_ForgetReference(v);
|
|
|
|
|
*pv = (PyObject *)
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_REALLOC((char *)v,
|
1997-05-02 11:12:38 +08:00
|
|
|
|
sizeof(PyStringObject) + newsize * sizeof(char));
|
1990-10-14 20:07:46 +08:00
|
|
|
|
if (*pv == NULL) {
|
2000-05-04 07:44:39 +08:00
|
|
|
|
PyObject_DEL(v);
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_NoMemory();
|
1990-10-22 06:15:08 +08:00
|
|
|
|
return -1;
|
1990-10-14 20:07:46 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
_Py_NewReference(*pv);
|
|
|
|
|
sv = (PyStringObject *) *pv;
|
1990-11-19 01:30:23 +08:00
|
|
|
|
sv->ob_size = newsize;
|
|
|
|
|
sv->ob_sval[newsize] = '\0';
|
1990-10-14 20:07:46 +08:00
|
|
|
|
return 0;
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
|
|
|
|
|
/* Helpers for formatstring */
|
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
static PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
getnextarg(PyObject *args, int arglen, int *p_argidx)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
{
|
|
|
|
|
int argidx = *p_argidx;
|
|
|
|
|
if (argidx < arglen) {
|
|
|
|
|
(*p_argidx)++;
|
|
|
|
|
if (arglen < 0)
|
|
|
|
|
return args;
|
|
|
|
|
else
|
1997-05-02 11:12:38 +08:00
|
|
|
|
return PyTuple_GetItem(args, argidx);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"not enough arguments for format string");
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#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)
|
|
|
|
|
|
1997-01-22 00:12:09 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
formatfloat(char *buf, size_t buflen, int flags,
|
|
|
|
|
int prec, int type, PyObject *v)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
{
|
2000-06-30 18:29:57 +08:00
|
|
|
|
/* fmt = '%#.' + `prec` + `type`
|
|
|
|
|
worst case length = 3 + 10 (len of INT_MAX) + 1 = 14 (use 20)*/
|
1993-03-16 20:15:04 +08:00
|
|
|
|
char fmt[20];
|
|
|
|
|
double x;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyArg_Parse(v, "d;float argument required", &x))
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return -1;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 6;
|
|
|
|
|
if (type == 'f' && fabs(x)/1e25 >= 1e25)
|
|
|
|
|
type = 'g';
|
|
|
|
|
sprintf(fmt, "%%%s.%d%c", (flags&F_ALT) ? "#" : "", prec, type);
|
2000-06-30 18:29:57 +08:00
|
|
|
|
/* worst case length calc to ensure no buffer overrun:
|
|
|
|
|
fmt = %#.<prec>g
|
|
|
|
|
buf = '-' + [0-9]*prec + '.' + 'e+' + (longest exp
|
|
|
|
|
for any double rep.)
|
|
|
|
|
len = 1 + prec + 1 + 2 + 5 = 9 + prec
|
|
|
|
|
If prec=0 the effective precision is 1 (the leading digit is
|
|
|
|
|
always given), therefore increase by one to 10+prec. */
|
|
|
|
|
if (buflen <= (size_t)10 + (size_t)prec) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"formatted float is too long (precision too long?)");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
sprintf(buf, fmt, x);
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return strlen(buf);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
|
1997-01-22 00:12:09 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
formatint(char *buf, size_t buflen, int flags,
|
|
|
|
|
int prec, int type, PyObject *v)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
{
|
2000-06-30 18:29:57 +08:00
|
|
|
|
/* fmt = '%#.' + `prec` + 'l' + `type`
|
|
|
|
|
worst case length = 3 + 10 (len of INT_MAX) + 1 + 1 = 15 (use 20)*/
|
1993-03-16 20:15:04 +08:00
|
|
|
|
char fmt[20];
|
|
|
|
|
long x;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyArg_Parse(v, "l;int argument required", &x))
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return -1;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 1;
|
|
|
|
|
sprintf(fmt, "%%%s.%dl%c", (flags&F_ALT) ? "#" : "", prec, type);
|
2000-06-30 18:29:57 +08:00
|
|
|
|
/* buf = '+'/'-'/'0'/'0x' + '[0-9]'*max(prec,len(x in octal))
|
|
|
|
|
worst case buf = '0x' + [0-9]*prec, where prec >= 11 */
|
|
|
|
|
if (buflen <= 13 || buflen <= (size_t)2+(size_t)prec) {
|
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
|
"formatted integer is too long (precision too long?)");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
sprintf(buf, fmt, x);
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return strlen(buf);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
|
1997-01-22 00:12:09 +08:00
|
|
|
|
static int
|
2000-07-09 15:04:36 +08:00
|
|
|
|
formatchar(char *buf, size_t buflen, PyObject *v)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
{
|
2000-06-30 18:29:57 +08:00
|
|
|
|
/* presume that the buffer is at least 2 characters long */
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (PyString_Check(v)) {
|
|
|
|
|
if (!PyArg_Parse(v, "c;%c requires int or char", &buf[0]))
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return -1;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
else {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (!PyArg_Parse(v, "b;%c requires int or char", &buf[0]))
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return -1;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
buf[1] = '\0';
|
1997-01-22 00:12:09 +08:00
|
|
|
|
return 1;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
|
1994-08-30 16:19:36 +08:00
|
|
|
|
|
2000-06-30 18:29:57 +08: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 20:15:04 +08:00
|
|
|
|
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_Format(PyObject *format, PyObject *args)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
{
|
|
|
|
|
char *fmt, *res;
|
|
|
|
|
int fmtcnt, rescnt, reslen, arglen, argidx;
|
1996-05-22 06:44:20 +08:00
|
|
|
|
int args_owned = 0;
|
2000-04-10 21:47:21 +08:00
|
|
|
|
PyObject *result, *orig_args;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *dict = NULL;
|
|
|
|
|
if (format == NULL || !PyString_Check(format) || args == NULL) {
|
|
|
|
|
PyErr_BadInternalCall();
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2000-04-10 21:47:21 +08:00
|
|
|
|
orig_args = args;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
fmt = PyString_AsString(format);
|
|
|
|
|
fmtcnt = PyString_Size(format);
|
1993-05-12 16:24:20 +08:00
|
|
|
|
reslen = rescnt = fmtcnt + 100;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
result = PyString_FromStringAndSize((char *)NULL, reslen);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (result == NULL)
|
|
|
|
|
return NULL;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
res = PyString_AsString(result);
|
|
|
|
|
if (PyTuple_Check(args)) {
|
|
|
|
|
arglen = PyTuple_Size(args);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
argidx = 0;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
arglen = -1;
|
|
|
|
|
argidx = -2;
|
|
|
|
|
}
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (args->ob_type->tp_as_mapping)
|
|
|
|
|
dict = args;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
while (--fmtcnt >= 0) {
|
|
|
|
|
if (*fmt != '%') {
|
|
|
|
|
if (--rescnt < 0) {
|
1993-05-12 16:24:20 +08:00
|
|
|
|
rescnt = fmtcnt + 100;
|
|
|
|
|
reslen += rescnt;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (_PyString_Resize(&result, reslen) < 0)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return NULL;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
res = PyString_AsString(result)
|
|
|
|
|
+ reslen - rescnt;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
--rescnt;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
*res++ = *fmt++;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
/* Got a format specifier */
|
|
|
|
|
int flags = 0;
|
|
|
|
|
int width = -1;
|
|
|
|
|
int prec = -1;
|
|
|
|
|
int size = 0;
|
1993-11-11 22:51:57 +08:00
|
|
|
|
int c = '\0';
|
1993-03-16 20:15:04 +08:00
|
|
|
|
int fill;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *v = NULL;
|
|
|
|
|
PyObject *temp = NULL;
|
2000-06-30 18:29:57 +08:00
|
|
|
|
char *pbuf;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
int sign;
|
|
|
|
|
int len;
|
2000-06-30 18:29:57 +08:00
|
|
|
|
char formatbuf[FORMATBUFLEN]; /* For format{float,int,char}() */
|
2000-04-10 21:47:21 +08:00
|
|
|
|
char *fmt_start = fmt;
|
|
|
|
|
|
1996-12-06 05:58:58 +08:00
|
|
|
|
fmt++;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (*fmt == '(') {
|
|
|
|
|
char *keystart;
|
|
|
|
|
int keylen;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyObject *key;
|
1997-09-09 02:30:11 +08:00
|
|
|
|
int pcount = 1;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
|
|
|
|
|
if (dict == NULL) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
1994-08-30 16:19:36 +08:00
|
|
|
|
"format requires a mapping");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
++fmt;
|
|
|
|
|
--fmtcnt;
|
|
|
|
|
keystart = fmt;
|
1997-09-09 02:30:11 +08:00
|
|
|
|
/* Skip over balanced parentheses */
|
|
|
|
|
while (pcount > 0 && --fmtcnt >= 0) {
|
|
|
|
|
if (*fmt == ')')
|
|
|
|
|
--pcount;
|
|
|
|
|
else if (*fmt == '(')
|
|
|
|
|
++pcount;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
fmt++;
|
1997-09-09 02:30:11 +08:00
|
|
|
|
}
|
|
|
|
|
keylen = fmt - keystart - 1;
|
|
|
|
|
if (fmtcnt < 0 || pcount > 0) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
1994-08-30 16:19:36 +08:00
|
|
|
|
"incomplete format key");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
key = PyString_FromStringAndSize(keystart,
|
|
|
|
|
keylen);
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (key == NULL)
|
|
|
|
|
goto error;
|
1996-05-22 06:44:20 +08:00
|
|
|
|
if (args_owned) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(args);
|
1996-05-22 06:44:20 +08:00
|
|
|
|
args_owned = 0;
|
|
|
|
|
}
|
|
|
|
|
args = PyObject_GetItem(dict, key);
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(key);
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (args == NULL) {
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
1996-05-22 06:44:20 +08:00
|
|
|
|
args_owned = 1;
|
1994-08-30 16:19:36 +08:00
|
|
|
|
arglen = -1;
|
|
|
|
|
argidx = -2;
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08: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 11:12:38 +08:00
|
|
|
|
if (!PyInt_Check(v)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"* wants int");
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
width = PyInt_AsLong(v);
|
1999-06-07 23:12:32 +08:00
|
|
|
|
if (width < 0) {
|
|
|
|
|
flags |= F_LJUST;
|
|
|
|
|
width = -width;
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
1995-02-11 01:00:37 +08:00
|
|
|
|
else if (c >= 0 && isdigit(c)) {
|
1993-03-16 20:15:04 +08:00
|
|
|
|
width = c - '0';
|
|
|
|
|
while (--fmtcnt >= 0) {
|
1995-02-11 01:00:37 +08:00
|
|
|
|
c = Py_CHARMASK(*fmt++);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (!isdigit(c))
|
|
|
|
|
break;
|
|
|
|
|
if ((width*10) / 10 != width) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_ValueError,
|
|
|
|
|
"width too big");
|
1993-03-16 20:15:04 +08: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 11:12:38 +08:00
|
|
|
|
if (!PyInt_Check(v)) {
|
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_TypeError,
|
|
|
|
|
"* wants int");
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
prec = PyInt_AsLong(v);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (prec < 0)
|
|
|
|
|
prec = 0;
|
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
1995-02-11 01:00:37 +08:00
|
|
|
|
else if (c >= 0 && isdigit(c)) {
|
1993-03-16 20:15:04 +08:00
|
|
|
|
prec = c - '0';
|
|
|
|
|
while (--fmtcnt >= 0) {
|
1995-02-11 01:00:37 +08:00
|
|
|
|
c = Py_CHARMASK(*fmt++);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
if (!isdigit(c))
|
|
|
|
|
break;
|
|
|
|
|
if ((prec*10) / 10 != prec) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(
|
|
|
|
|
PyExc_ValueError,
|
1993-03-16 20:15:04 +08:00
|
|
|
|
"prec too big");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
prec = prec*10 + (c - '0');
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} /* prec */
|
|
|
|
|
if (fmtcnt >= 0) {
|
|
|
|
|
if (c == 'h' || c == 'l' || c == 'L') {
|
|
|
|
|
size = c;
|
|
|
|
|
if (--fmtcnt >= 0)
|
|
|
|
|
c = *fmt++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (fmtcnt < 0) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
|
"incomplete format");
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
if (c != '%') {
|
|
|
|
|
v = getnextarg(args, arglen, &argidx);
|
|
|
|
|
if (v == NULL)
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
sign = 0;
|
|
|
|
|
fill = ' ';
|
|
|
|
|
switch (c) {
|
|
|
|
|
case '%':
|
2000-06-30 18:29:57 +08:00
|
|
|
|
pbuf = "%";
|
1993-03-16 20:15:04 +08:00
|
|
|
|
len = 1;
|
|
|
|
|
break;
|
|
|
|
|
case 's':
|
2000-04-10 21:47:21 +08:00
|
|
|
|
case 'r':
|
|
|
|
|
if (PyUnicode_Check(v)) {
|
|
|
|
|
fmt = fmt_start;
|
|
|
|
|
goto unicode;
|
|
|
|
|
}
|
2000-04-11 23:39:26 +08:00
|
|
|
|
if (c == 's')
|
1997-05-02 11:12:38 +08:00
|
|
|
|
temp = PyObject_Str(v);
|
2000-04-11 23:39:26 +08:00
|
|
|
|
else
|
|
|
|
|
temp = PyObject_Repr(v);
|
1994-08-30 16:19:36 +08:00
|
|
|
|
if (temp == NULL)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
1998-06-09 23:08:41 +08:00
|
|
|
|
if (!PyString_Check(temp)) {
|
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"%s argument has non-string str()");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2000-06-30 18:29:57 +08:00
|
|
|
|
pbuf = PyString_AsString(temp);
|
1997-05-02 11:12:38 +08:00
|
|
|
|
len = PyString_Size(temp);
|
1993-03-16 20:15:04 +08: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-06-30 18:29:57 +08:00
|
|
|
|
pbuf = formatbuf;
|
|
|
|
|
len = formatint(pbuf, sizeof(formatbuf), flags, prec, c, v);
|
1997-01-22 00:12:09 +08:00
|
|
|
|
if (len < 0)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
sign = (c == 'd');
|
1997-01-29 14:00:24 +08:00
|
|
|
|
if (flags&F_ZERO) {
|
1993-03-16 20:15:04 +08:00
|
|
|
|
fill = '0';
|
1997-01-29 14:00:24 +08:00
|
|
|
|
if ((flags&F_ALT) &&
|
|
|
|
|
(c == 'x' || c == 'X') &&
|
2000-06-30 18:29:57 +08:00
|
|
|
|
pbuf[0] == '0' && pbuf[1] == c) {
|
|
|
|
|
*res++ = *pbuf++;
|
|
|
|
|
*res++ = *pbuf++;
|
1997-01-29 14:00:24 +08:00
|
|
|
|
rescnt -= 2;
|
|
|
|
|
len -= 2;
|
|
|
|
|
width -= 2;
|
|
|
|
|
if (width < 0)
|
|
|
|
|
width = 0;
|
|
|
|
|
}
|
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
break;
|
|
|
|
|
case 'e':
|
|
|
|
|
case 'E':
|
|
|
|
|
case 'f':
|
|
|
|
|
case 'g':
|
|
|
|
|
case 'G':
|
2000-06-30 18:29:57 +08:00
|
|
|
|
pbuf = formatbuf;
|
|
|
|
|
len = formatfloat(pbuf, sizeof(formatbuf), flags, prec, c, v);
|
1997-01-22 00:12:09 +08:00
|
|
|
|
if (len < 0)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
sign = 1;
|
|
|
|
|
if (flags&F_ZERO)
|
|
|
|
|
fill = '0';
|
|
|
|
|
break;
|
|
|
|
|
case 'c':
|
2000-06-30 18:29:57 +08:00
|
|
|
|
pbuf = formatbuf;
|
|
|
|
|
len = formatchar(pbuf, sizeof(formatbuf), v);
|
1997-01-22 00:12:09 +08:00
|
|
|
|
if (len < 0)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
1997-09-09 02:30:11 +08:00
|
|
|
|
PyErr_Format(PyExc_ValueError,
|
|
|
|
|
"unsupported format character '%c' (0x%x)",
|
|
|
|
|
c, c);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
if (sign) {
|
2000-06-30 18:29:57 +08:00
|
|
|
|
if (*pbuf == '-' || *pbuf == '+') {
|
|
|
|
|
sign = *pbuf++;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
len--;
|
|
|
|
|
}
|
|
|
|
|
else if (flags & F_SIGN)
|
|
|
|
|
sign = '+';
|
|
|
|
|
else if (flags & F_BLANK)
|
|
|
|
|
sign = ' ';
|
|
|
|
|
else
|
|
|
|
|
sign = '\0';
|
|
|
|
|
}
|
|
|
|
|
if (width < len)
|
|
|
|
|
width = len;
|
|
|
|
|
if (rescnt < width + (sign != '\0')) {
|
1993-05-12 16:24:20 +08:00
|
|
|
|
reslen -= rescnt;
|
|
|
|
|
rescnt = width + fmtcnt + 100;
|
|
|
|
|
reslen += rescnt;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
if (_PyString_Resize(&result, reslen) < 0)
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return NULL;
|
1997-05-02 11:12:38 +08:00
|
|
|
|
res = PyString_AsString(result)
|
|
|
|
|
+ reslen - rescnt;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
}
|
|
|
|
|
if (sign) {
|
1993-11-11 23:03:51 +08:00
|
|
|
|
if (fill != ' ')
|
|
|
|
|
*res++ = sign;
|
1993-03-16 20:15:04 +08:00
|
|
|
|
rescnt--;
|
|
|
|
|
if (width > len)
|
|
|
|
|
width--;
|
|
|
|
|
}
|
|
|
|
|
if (width > len && !(flags&F_LJUST)) {
|
|
|
|
|
do {
|
|
|
|
|
--rescnt;
|
|
|
|
|
*res++ = fill;
|
|
|
|
|
} while (--width > len);
|
|
|
|
|
}
|
1993-11-11 23:03:51 +08:00
|
|
|
|
if (sign && fill == ' ')
|
1993-11-11 22:51:57 +08:00
|
|
|
|
*res++ = sign;
|
2000-06-30 18:29:57 +08:00
|
|
|
|
memcpy(res, pbuf, len);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
res += len;
|
|
|
|
|
rescnt -= len;
|
|
|
|
|
while (--width >= len) {
|
|
|
|
|
--rescnt;
|
|
|
|
|
*res++ = ' ';
|
|
|
|
|
}
|
1995-02-11 01:00:37 +08:00
|
|
|
|
if (dict && (argidx < arglen) && c != '%') {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
1994-08-30 16:19:36 +08:00
|
|
|
|
"not all arguments converted");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_XDECREF(temp);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
} /* '%' */
|
|
|
|
|
} /* until end */
|
1995-02-27 18:13:23 +08:00
|
|
|
|
if (argidx < arglen && !dict) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
|
"not all arguments converted");
|
1993-03-16 20:15:04 +08:00
|
|
|
|
goto error;
|
|
|
|
|
}
|
1998-04-11 06:16:39 +08:00
|
|
|
|
if (args_owned) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(args);
|
1998-04-11 06:16:39 +08:00
|
|
|
|
}
|
1997-05-02 11:12:38 +08:00
|
|
|
|
_PyString_Resize(&result, reslen - rescnt);
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return result;
|
2000-04-10 21:47:21 +08:00
|
|
|
|
|
|
|
|
|
unicode:
|
|
|
|
|
if (args_owned) {
|
|
|
|
|
Py_DECREF(args);
|
|
|
|
|
args_owned = 0;
|
|
|
|
|
}
|
|
|
|
|
/* Fiddle args right (remove the first argidx-1 arguments) */
|
|
|
|
|
--argidx;
|
|
|
|
|
if (PyTuple_Check(orig_args) && argidx > 0) {
|
|
|
|
|
PyObject *v;
|
|
|
|
|
int n = PyTuple_GET_SIZE(orig_args) - argidx;
|
|
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
/* Paste rest of format string to what we have of the result
|
|
|
|
|
string; we reuse result for this */
|
|
|
|
|
rescnt = res - PyString_AS_STRING(result);
|
|
|
|
|
fmtcnt = PyString_GET_SIZE(format) - \
|
|
|
|
|
(fmt - PyString_AS_STRING(format));
|
|
|
|
|
if (_PyString_Resize(&result, rescnt + fmtcnt)) {
|
|
|
|
|
Py_DECREF(args);
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
memcpy(PyString_AS_STRING(result) + rescnt, fmt, fmtcnt);
|
|
|
|
|
format = result;
|
|
|
|
|
/* Let Unicode do its magic */
|
|
|
|
|
result = PyUnicode_Format(format, args);
|
|
|
|
|
Py_DECREF(format);
|
|
|
|
|
Py_DECREF(args);
|
|
|
|
|
return result;
|
|
|
|
|
|
1993-03-16 20:15:04 +08:00
|
|
|
|
error:
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(result);
|
1998-04-11 06:16:39 +08:00
|
|
|
|
if (args_owned) {
|
1997-05-02 11:12:38 +08:00
|
|
|
|
Py_DECREF(args);
|
1998-04-11 06:16:39 +08:00
|
|
|
|
}
|
1993-03-16 20:15:04 +08:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
1997-01-18 15:55:05 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
|
|
|
|
|
static PyObject *interned;
|
|
|
|
|
|
|
|
|
|
void
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_InternInPlace(PyObject **p)
|
1997-01-18 15:55:05 +08:00
|
|
|
|
{
|
|
|
|
|
register PyStringObject *s = (PyStringObject *)(*p);
|
|
|
|
|
PyObject *t;
|
|
|
|
|
if (s == NULL || !PyString_Check(s))
|
|
|
|
|
Py_FatalError("PyString_InternInPlace: strings only please!");
|
|
|
|
|
if ((t = s->ob_sinterned) != NULL) {
|
|
|
|
|
if (t == (PyObject *)s)
|
|
|
|
|
return;
|
|
|
|
|
Py_INCREF(t);
|
|
|
|
|
*p = t;
|
|
|
|
|
Py_DECREF(s);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (interned == NULL) {
|
|
|
|
|
interned = PyDict_New();
|
|
|
|
|
if (interned == NULL)
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if ((t = PyDict_GetItem(interned, (PyObject *)s)) != NULL) {
|
|
|
|
|
Py_INCREF(t);
|
|
|
|
|
*p = s->ob_sinterned = t;
|
|
|
|
|
Py_DECREF(s);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
t = (PyObject *)s;
|
|
|
|
|
if (PyDict_SetItem(interned, t, t) == 0) {
|
|
|
|
|
s->ob_sinterned = t;
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
PyErr_Clear();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
PyObject *
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_InternFromString(const char *cp)
|
1997-01-18 15:55:05 +08:00
|
|
|
|
{
|
|
|
|
|
PyObject *s = PyString_FromString(cp);
|
|
|
|
|
if (s == NULL)
|
|
|
|
|
return NULL;
|
|
|
|
|
PyString_InternInPlace(&s);
|
|
|
|
|
return s;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif
|
1997-08-02 10:57:45 +08:00
|
|
|
|
|
|
|
|
|
void
|
2000-07-09 15:04:36 +08:00
|
|
|
|
PyString_Fini(void)
|
1997-08-02 10:57:45 +08:00
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
for (i = 0; i < UCHAR_MAX + 1; i++) {
|
|
|
|
|
Py_XDECREF(characters[i]);
|
|
|
|
|
characters[i] = NULL;
|
|
|
|
|
}
|
|
|
|
|
#ifndef DONT_SHARE_SHORT_STRINGS
|
|
|
|
|
Py_XDECREF(nullstring);
|
|
|
|
|
nullstring = NULL;
|
|
|
|
|
#endif
|
1997-08-05 10:15:12 +08:00
|
|
|
|
#ifdef INTERN_STRINGS
|
|
|
|
|
if (interned) {
|
|
|
|
|
int pos, changed;
|
|
|
|
|
PyObject *key, *value;
|
|
|
|
|
do {
|
|
|
|
|
changed = 0;
|
|
|
|
|
pos = 0;
|
|
|
|
|
while (PyDict_Next(interned, &pos, &key, &value)) {
|
|
|
|
|
if (key->ob_refcnt == 2 && key == value) {
|
|
|
|
|
PyDict_DelItem(interned, key);
|
|
|
|
|
changed = 1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} while (changed);
|
|
|
|
|
}
|
|
|
|
|
#endif
|
1997-08-02 10:57:45 +08:00
|
|
|
|
}
|