Commit f564761e authored by Guido van Rossum's avatar Guido van Rossum

Make str and tuple types subclassable.

parent 8e9a68d3
...@@ -2683,13 +2683,17 @@ string_methods[] = { ...@@ -2683,13 +2683,17 @@ string_methods[] = {
{NULL, NULL} /* sentinel */ {NULL, NULL} /* sentinel */
}; };
staticforward PyObject *
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
static PyObject * static PyObject *
string_new(PyTypeObject *type, PyObject *args, PyObject *kwds) string_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{ {
PyObject *x = NULL; PyObject *x = NULL;
static char *kwlist[] = {"object", 0}; static char *kwlist[] = {"object", 0};
assert(type == &PyString_Type); if (type != &PyString_Type)
return str_subtype_new(type, args, kwds);
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:str", kwlist, &x)) if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:str", kwlist, &x))
return NULL; return NULL;
if (x == NULL) if (x == NULL)
...@@ -2697,6 +2701,24 @@ string_new(PyTypeObject *type, PyObject *args, PyObject *kwds) ...@@ -2697,6 +2701,24 @@ string_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
return PyObject_Str(x); return PyObject_Str(x);
} }
static PyObject *
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyObject *tmp, *new;
int n;
assert(PyType_IsSubtype(type, &PyString_Type));
tmp = string_new(&PyString_Type, args, kwds);
if (tmp == NULL)
return NULL;
assert(PyString_Check(tmp));
new = type->tp_alloc(type, n = PyString_GET_SIZE(tmp));
if (new == NULL)
return NULL;
memcpy(PyString_AS_STRING(new), PyString_AS_STRING(tmp), n+1);
return new;
}
static char string_doc[] = static char string_doc[] =
"str(object) -> string\n\ "str(object) -> string\n\
\n\ \n\
...@@ -2724,7 +2746,7 @@ PyTypeObject PyString_Type = { ...@@ -2724,7 +2746,7 @@ PyTypeObject PyString_Type = {
PyObject_GenericGetAttr, /* tp_getattro */ PyObject_GenericGetAttr, /* tp_getattro */
0, /* tp_setattro */ 0, /* tp_setattro */
&string_as_buffer, /* tp_as_buffer */ &string_as_buffer, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT, /* tp_flags */ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
string_doc, /* tp_doc */ string_doc, /* tp_doc */
0, /* tp_traverse */ 0, /* tp_traverse */
0, /* tp_clear */ 0, /* tp_clear */
......
...@@ -475,13 +475,17 @@ tuplerichcompare(PyObject *v, PyObject *w, int op) ...@@ -475,13 +475,17 @@ tuplerichcompare(PyObject *v, PyObject *w, int op)
return PyObject_RichCompare(vt->ob_item[i], wt->ob_item[i], op); return PyObject_RichCompare(vt->ob_item[i], wt->ob_item[i], op);
} }
staticforward PyObject *
tuple_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
static PyObject * static PyObject *
tuple_new(PyTypeObject *type, PyObject *args, PyObject *kwds) tuple_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{ {
PyObject *arg = NULL; PyObject *arg = NULL;
static char *kwlist[] = {"sequence", 0}; static char *kwlist[] = {"sequence", 0};
assert(type == &PyTuple_Type); if (type != &PyTuple_Type)
return tuple_subtype_new(type, args, kwds);
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:tuple", kwlist, &arg)) if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:tuple", kwlist, &arg))
return NULL; return NULL;
...@@ -491,6 +495,29 @@ tuple_new(PyTypeObject *type, PyObject *args, PyObject *kwds) ...@@ -491,6 +495,29 @@ tuple_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
return PySequence_Tuple(arg); return PySequence_Tuple(arg);
} }
static PyObject *
tuple_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyObject *tmp, *new, *item;
int i, n;
assert(PyType_IsSubtype(type, &PyTuple_Type));
tmp = tuple_new(&PyTuple_Type, args, kwds);
if (tmp == NULL)
return NULL;
assert(PyTuple_Check(tmp));
new = type->tp_alloc(type, n = PyTuple_GET_SIZE(tmp));
if (new == NULL)
return NULL;
for (i = 0; i < n; i++) {
item = PyTuple_GET_ITEM(tmp, i);
Py_INCREF(item);
PyTuple_SET_ITEM(new, i, item);
}
Py_DECREF(tmp);
return new;
}
static char tuple_doc[] = static char tuple_doc[] =
"tuple(sequence) -> list\n\ "tuple(sequence) -> list\n\
\n\ \n\
...@@ -529,7 +556,8 @@ PyTypeObject PyTuple_Type = { ...@@ -529,7 +556,8 @@ PyTypeObject PyTuple_Type = {
PyObject_GenericGetAttr, /* tp_getattro */ PyObject_GenericGetAttr, /* tp_getattro */
0, /* tp_setattro */ 0, /* tp_setattro */
0, /* tp_as_buffer */ 0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
Py_TPFLAGS_BASETYPE, /* tp_flags */
tuple_doc, /* tp_doc */ tuple_doc, /* tp_doc */
(traverseproc)tupletraverse, /* tp_traverse */ (traverseproc)tupletraverse, /* tp_traverse */
0, /* tp_clear */ 0, /* tp_clear */
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment