mirror of https://github.com/python/cpython
325 lines
9.2 KiB
C
325 lines
9.2 KiB
C
/*[clinic input]
|
|
preserve
|
|
[clinic start generated code]*/
|
|
|
|
#if defined(Py_BUILD_CORE) && !defined(Py_BUILD_CORE_MODULE)
|
|
# include "pycore_gc.h" // PyGC_Head
|
|
# include "pycore_runtime.h" // _Py_ID()
|
|
#endif
|
|
#include "pycore_abstract.h" // _PyNumber_Index()
|
|
#include "pycore_modsupport.h" // _PyArg_UnpackKeywords()
|
|
|
|
PyDoc_STRVAR(stringlib_expandtabs__doc__,
|
|
"expandtabs($self, /, tabsize=8)\n"
|
|
"--\n"
|
|
"\n"
|
|
"Return a copy where all tab characters are expanded using spaces.\n"
|
|
"\n"
|
|
"If tabsize is not given, a tab size of 8 characters is assumed.");
|
|
|
|
#define STRINGLIB_EXPANDTABS_METHODDEF \
|
|
{"expandtabs", _PyCFunction_CAST(stringlib_expandtabs), METH_FASTCALL|METH_KEYWORDS, stringlib_expandtabs__doc__},
|
|
|
|
static PyObject *
|
|
stringlib_expandtabs_impl(PyObject *self, int tabsize);
|
|
|
|
static PyObject *
|
|
stringlib_expandtabs(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
|
|
{
|
|
PyObject *return_value = NULL;
|
|
#if defined(Py_BUILD_CORE) && !defined(Py_BUILD_CORE_MODULE)
|
|
|
|
#define NUM_KEYWORDS 1
|
|
static struct {
|
|
PyGC_Head _this_is_not_used;
|
|
PyObject_VAR_HEAD
|
|
PyObject *ob_item[NUM_KEYWORDS];
|
|
} _kwtuple = {
|
|
.ob_base = PyVarObject_HEAD_INIT(&PyTuple_Type, NUM_KEYWORDS)
|
|
.ob_item = { &_Py_ID(tabsize), },
|
|
};
|
|
#undef NUM_KEYWORDS
|
|
#define KWTUPLE (&_kwtuple.ob_base.ob_base)
|
|
|
|
#else // !Py_BUILD_CORE
|
|
# define KWTUPLE NULL
|
|
#endif // !Py_BUILD_CORE
|
|
|
|
static const char * const _keywords[] = {"tabsize", NULL};
|
|
static _PyArg_Parser _parser = {
|
|
.keywords = _keywords,
|
|
.fname = "expandtabs",
|
|
.kwtuple = KWTUPLE,
|
|
};
|
|
#undef KWTUPLE
|
|
PyObject *argsbuf[1];
|
|
Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 0;
|
|
int tabsize = 8;
|
|
|
|
args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 0, 1, 0, argsbuf);
|
|
if (!args) {
|
|
goto exit;
|
|
}
|
|
if (!noptargs) {
|
|
goto skip_optional_pos;
|
|
}
|
|
tabsize = PyLong_AsInt(args[0]);
|
|
if (tabsize == -1 && PyErr_Occurred()) {
|
|
goto exit;
|
|
}
|
|
skip_optional_pos:
|
|
return_value = stringlib_expandtabs_impl(self, tabsize);
|
|
|
|
exit:
|
|
return return_value;
|
|
}
|
|
|
|
PyDoc_STRVAR(stringlib_ljust__doc__,
|
|
"ljust($self, width, fillchar=b\' \', /)\n"
|
|
"--\n"
|
|
"\n"
|
|
"Return a left-justified string of length width.\n"
|
|
"\n"
|
|
"Padding is done using the specified fill character.");
|
|
|
|
#define STRINGLIB_LJUST_METHODDEF \
|
|
{"ljust", _PyCFunction_CAST(stringlib_ljust), METH_FASTCALL, stringlib_ljust__doc__},
|
|
|
|
static PyObject *
|
|
stringlib_ljust_impl(PyObject *self, Py_ssize_t width, char fillchar);
|
|
|
|
static PyObject *
|
|
stringlib_ljust(PyObject *self, PyObject *const *args, Py_ssize_t nargs)
|
|
{
|
|
PyObject *return_value = NULL;
|
|
Py_ssize_t width;
|
|
char fillchar = ' ';
|
|
|
|
if (!_PyArg_CheckPositional("ljust", nargs, 1, 2)) {
|
|
goto exit;
|
|
}
|
|
{
|
|
Py_ssize_t ival = -1;
|
|
PyObject *iobj = _PyNumber_Index(args[0]);
|
|
if (iobj != NULL) {
|
|
ival = PyLong_AsSsize_t(iobj);
|
|
Py_DECREF(iobj);
|
|
}
|
|
if (ival == -1 && PyErr_Occurred()) {
|
|
goto exit;
|
|
}
|
|
width = ival;
|
|
}
|
|
if (nargs < 2) {
|
|
goto skip_optional;
|
|
}
|
|
if (PyBytes_Check(args[1])) {
|
|
if (PyBytes_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"ljust(): argument 2 must be a byte string of length 1, "
|
|
"not a bytes object of length %zd",
|
|
PyBytes_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyBytes_AS_STRING(args[1])[0];
|
|
}
|
|
else if (PyByteArray_Check(args[1])) {
|
|
if (PyByteArray_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"ljust(): argument 2 must be a byte string of length 1, "
|
|
"not a bytearray object of length %zd",
|
|
PyByteArray_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyByteArray_AS_STRING(args[1])[0];
|
|
}
|
|
else {
|
|
_PyArg_BadArgument("ljust", "argument 2", "a byte string of length 1", args[1]);
|
|
goto exit;
|
|
}
|
|
skip_optional:
|
|
return_value = stringlib_ljust_impl(self, width, fillchar);
|
|
|
|
exit:
|
|
return return_value;
|
|
}
|
|
|
|
PyDoc_STRVAR(stringlib_rjust__doc__,
|
|
"rjust($self, width, fillchar=b\' \', /)\n"
|
|
"--\n"
|
|
"\n"
|
|
"Return a right-justified string of length width.\n"
|
|
"\n"
|
|
"Padding is done using the specified fill character.");
|
|
|
|
#define STRINGLIB_RJUST_METHODDEF \
|
|
{"rjust", _PyCFunction_CAST(stringlib_rjust), METH_FASTCALL, stringlib_rjust__doc__},
|
|
|
|
static PyObject *
|
|
stringlib_rjust_impl(PyObject *self, Py_ssize_t width, char fillchar);
|
|
|
|
static PyObject *
|
|
stringlib_rjust(PyObject *self, PyObject *const *args, Py_ssize_t nargs)
|
|
{
|
|
PyObject *return_value = NULL;
|
|
Py_ssize_t width;
|
|
char fillchar = ' ';
|
|
|
|
if (!_PyArg_CheckPositional("rjust", nargs, 1, 2)) {
|
|
goto exit;
|
|
}
|
|
{
|
|
Py_ssize_t ival = -1;
|
|
PyObject *iobj = _PyNumber_Index(args[0]);
|
|
if (iobj != NULL) {
|
|
ival = PyLong_AsSsize_t(iobj);
|
|
Py_DECREF(iobj);
|
|
}
|
|
if (ival == -1 && PyErr_Occurred()) {
|
|
goto exit;
|
|
}
|
|
width = ival;
|
|
}
|
|
if (nargs < 2) {
|
|
goto skip_optional;
|
|
}
|
|
if (PyBytes_Check(args[1])) {
|
|
if (PyBytes_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"rjust(): argument 2 must be a byte string of length 1, "
|
|
"not a bytes object of length %zd",
|
|
PyBytes_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyBytes_AS_STRING(args[1])[0];
|
|
}
|
|
else if (PyByteArray_Check(args[1])) {
|
|
if (PyByteArray_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"rjust(): argument 2 must be a byte string of length 1, "
|
|
"not a bytearray object of length %zd",
|
|
PyByteArray_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyByteArray_AS_STRING(args[1])[0];
|
|
}
|
|
else {
|
|
_PyArg_BadArgument("rjust", "argument 2", "a byte string of length 1", args[1]);
|
|
goto exit;
|
|
}
|
|
skip_optional:
|
|
return_value = stringlib_rjust_impl(self, width, fillchar);
|
|
|
|
exit:
|
|
return return_value;
|
|
}
|
|
|
|
PyDoc_STRVAR(stringlib_center__doc__,
|
|
"center($self, width, fillchar=b\' \', /)\n"
|
|
"--\n"
|
|
"\n"
|
|
"Return a centered string of length width.\n"
|
|
"\n"
|
|
"Padding is done using the specified fill character.");
|
|
|
|
#define STRINGLIB_CENTER_METHODDEF \
|
|
{"center", _PyCFunction_CAST(stringlib_center), METH_FASTCALL, stringlib_center__doc__},
|
|
|
|
static PyObject *
|
|
stringlib_center_impl(PyObject *self, Py_ssize_t width, char fillchar);
|
|
|
|
static PyObject *
|
|
stringlib_center(PyObject *self, PyObject *const *args, Py_ssize_t nargs)
|
|
{
|
|
PyObject *return_value = NULL;
|
|
Py_ssize_t width;
|
|
char fillchar = ' ';
|
|
|
|
if (!_PyArg_CheckPositional("center", nargs, 1, 2)) {
|
|
goto exit;
|
|
}
|
|
{
|
|
Py_ssize_t ival = -1;
|
|
PyObject *iobj = _PyNumber_Index(args[0]);
|
|
if (iobj != NULL) {
|
|
ival = PyLong_AsSsize_t(iobj);
|
|
Py_DECREF(iobj);
|
|
}
|
|
if (ival == -1 && PyErr_Occurred()) {
|
|
goto exit;
|
|
}
|
|
width = ival;
|
|
}
|
|
if (nargs < 2) {
|
|
goto skip_optional;
|
|
}
|
|
if (PyBytes_Check(args[1])) {
|
|
if (PyBytes_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"center(): argument 2 must be a byte string of length 1, "
|
|
"not a bytes object of length %zd",
|
|
PyBytes_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyBytes_AS_STRING(args[1])[0];
|
|
}
|
|
else if (PyByteArray_Check(args[1])) {
|
|
if (PyByteArray_GET_SIZE(args[1]) != 1) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"center(): argument 2 must be a byte string of length 1, "
|
|
"not a bytearray object of length %zd",
|
|
PyByteArray_GET_SIZE(args[1]));
|
|
goto exit;
|
|
}
|
|
fillchar = PyByteArray_AS_STRING(args[1])[0];
|
|
}
|
|
else {
|
|
_PyArg_BadArgument("center", "argument 2", "a byte string of length 1", args[1]);
|
|
goto exit;
|
|
}
|
|
skip_optional:
|
|
return_value = stringlib_center_impl(self, width, fillchar);
|
|
|
|
exit:
|
|
return return_value;
|
|
}
|
|
|
|
PyDoc_STRVAR(stringlib_zfill__doc__,
|
|
"zfill($self, width, /)\n"
|
|
"--\n"
|
|
"\n"
|
|
"Pad a numeric string with zeros on the left, to fill a field of the given width.\n"
|
|
"\n"
|
|
"The original string is never truncated.");
|
|
|
|
#define STRINGLIB_ZFILL_METHODDEF \
|
|
{"zfill", (PyCFunction)stringlib_zfill, METH_O, stringlib_zfill__doc__},
|
|
|
|
static PyObject *
|
|
stringlib_zfill_impl(PyObject *self, Py_ssize_t width);
|
|
|
|
static PyObject *
|
|
stringlib_zfill(PyObject *self, PyObject *arg)
|
|
{
|
|
PyObject *return_value = NULL;
|
|
Py_ssize_t width;
|
|
|
|
{
|
|
Py_ssize_t ival = -1;
|
|
PyObject *iobj = _PyNumber_Index(arg);
|
|
if (iobj != NULL) {
|
|
ival = PyLong_AsSsize_t(iobj);
|
|
Py_DECREF(iobj);
|
|
}
|
|
if (ival == -1 && PyErr_Occurred()) {
|
|
goto exit;
|
|
}
|
|
width = ival;
|
|
}
|
|
return_value = stringlib_zfill_impl(self, width);
|
|
|
|
exit:
|
|
return return_value;
|
|
}
|
|
/*[clinic end generated code: output=06dd79019356b6bb input=a9049054013a1b77]*/
|