ilusha/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython synced 2026-09-29 12:10:30 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

bpo-15913: Implement PyBuffer_SizeFromFormat() (GH-13873)

Browse source 
Implement PyBuffer_SizeFromFormat() function (previously
documented but not implemented): call struct.calcsize().
This commit is contained in:
Joannah Nanjekye 2019-08-20 11:46:36 -03:00 • committed by Victor Stinner
parent 18f8dcfa10
commit 9e66aba999
6 changed files with 83 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

42
Objects/abstract.c Normal file → Executable file
View file

@ -495,6 +495,48 @@ _Py_add_one_to_index_C(int nd, Py_ssize_t *index, const Py_ssize_t *shape)
}
}
Py_ssize_t
PyBuffer_SizeFromFormat(const char *format)
{
PyObject *structmodule = NULL;
PyObject *calcsize = NULL;
PyObject *res = NULL;
PyObject *fmt = NULL;
Py_ssize_t itemsize = -1;
structmodule = PyImport_ImportModule("struct");
if (structmodule == NULL) {
return itemsize;
}
calcsize = PyObject_GetAttrString(structmodule, "calcsize");
if (calcsize == NULL) {
goto done;
}
fmt = PyUnicode_FromString(format);
if (fmt == NULL) {
goto done;
}
res = PyObject_CallFunctionObjArgs(calcsize, fmt, NULL);
if (res == NULL) {
goto done;
}
itemsize = PyLong_AsSsize_t(res);
if (itemsize < 0) {
goto done;
}
done:
Py_DECREF(structmodule);
Py_XDECREF(calcsize);
Py_XDECREF(fmt);
Py_XDECREF(res);
return itemsize;
}
int
PyBuffer_FromContiguous(Py_buffer *view, void *buf, Py_ssize_t len, char fort)
{