diff options
| author | Antoine Pitrou <solipsis@pitrou.net> | 2012-07-29 22:01:44 (GMT) |
|---|---|---|
| committer | Antoine Pitrou <solipsis@pitrou.net> | 2012-07-29 22:01:44 (GMT) |
| commit | 57911f6c1a84dd0eb050908f81d6e60d38054271 (patch) | |
| tree | 539f04bf5b1110f16170097d74467cfd188aa3bd /Modules/_io | |
| parent | a264384fe6de357680ca0cf02cd6024bbba0ba45 (diff) | |
| parent | 8f328d0c1d86096b3d4e5f1c1ac497663e197a0d (diff) | |
| download | cpython-57911f6c1a84dd0eb050908f81d6e60d38054271.zip cpython-57911f6c1a84dd0eb050908f81d6e60d38054271.tar.gz cpython-57911f6c1a84dd0eb050908f81d6e60d38054271.tar.bz2 | |
Issue #15489: Add a __sizeof__ implementation for BytesIO objects.
Patch by Serhiy Storchaka.
Diffstat (limited to 'Modules/_io')
| -rw-r--r-- | Modules/_io/bytesio.c | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/Modules/_io/bytesio.c b/Modules/_io/bytesio.c index 65ec931..3d027e2 100644 --- a/Modules/_io/bytesio.c +++ b/Modules/_io/bytesio.c @@ -834,6 +834,17 @@ bytesio_init(bytesio *self, PyObject *args, PyObject *kwds) return 0; } +static PyObject * +bytesio_sizeof(bytesio *self, void *unused) +{ + Py_ssize_t res; + + res = sizeof(bytesio); + if (self->buf) + res += self->buf_size; + return PyLong_FromSsize_t(res); +} + static int bytesio_traverse(bytesio *self, visitproc visit, void *arg) { @@ -876,6 +887,7 @@ static struct PyMethodDef bytesio_methods[] = { {"truncate", (PyCFunction)bytesio_truncate, METH_VARARGS, truncate_doc}, {"__getstate__", (PyCFunction)bytesio_getstate, METH_NOARGS, NULL}, {"__setstate__", (PyCFunction)bytesio_setstate, METH_O, NULL}, + {"__sizeof__", (PyCFunction)bytesio_sizeof, METH_NOARGS, NULL}, {NULL, NULL} /* sentinel */ }; |