mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-31 05:58:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

#8046: add context manager protocol support to mmap objects. Also add closed property.

Browse source
This commit is contained in:
Georg Brandl 2010-08-01 14:50:00 +00:00
parent 120d633871
commit 0bccc185b4
4 changed files with 73 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

35
Modules/mmapmodule.c
View file

@ -651,6 +651,31 @@ mmap_move_method(mmap_object *self, PyObject *args)
}
}
static PyObject *
mmap_closed_get(mmap_object *self)
{
#ifdef MS_WINDOWS
return PyBool_FromLong(self->map_handle == NULL ? 1 : 0);
#elif defined(UNIX)
return PyBool_FromLong(self->data == NULL ? 1 : 0);
#endif
}
static PyObject *
mmap__enter__method(mmap_object *self, PyObject *args)
{
CHECK_VALID(NULL);
Py_INCREF(self);
return (PyObject *)self;
}
static PyObject *
mmap__exit__method(PyObject *self, PyObject *args)
{
return PyObject_CallMethod(self, "close", NULL);
}
static struct PyMethodDef mmap_object_methods[] = {
{"close", (PyCFunction) mmap_close_method, METH_NOARGS},
{"find", (PyCFunction) mmap_find_method, METH_VARARGS},
@ -666,9 +691,17 @@ static struct PyMethodDef mmap_object_methods[] = {
{"tell", (PyCFunction) mmap_tell_method, METH_NOARGS},
{"write", (PyCFunction) mmap_write_method, METH_VARARGS},
{"write_byte", (PyCFunction) mmap_write_byte_method, METH_VARARGS},
{"__enter__", (PyCFunction) mmap__enter__method, METH_NOARGS},
{"__exit__", (PyCFunction) mmap__exit__method, METH_VARARGS},
{NULL, NULL} /* sentinel */
};
static PyGetSetDef mmap_object_getset[] = {
{"closed", (getter) mmap_closed_get, NULL, NULL},
{NULL}
};
/* Functions for treating an mmap'ed file as a buffer */
static int
@ -975,7 +1008,7 @@ static PyTypeObject mmap_object_type = {
0, /* tp_iternext */
mmap_object_methods, /* tp_methods */
0, /* tp_members */
0, /* tp_getset */
mmap_object_getset, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
0, /* tp_descr_get */