From 913dd0be52f28fbc5036d95722b80222880706fd Mon Sep 17 00:00:00 2001 From: Guido van Rossum Date: Fri, 13 Apr 2007 03:33:53 +0000 Subject: struct.unpack() allows a bytes string too (if it has the right size). --- Modules/_struct.c | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/Modules/_struct.c b/Modules/_struct.c index 3418c30..5fc9991 100644 --- a/Modules/_struct.c +++ b/Modules/_struct.c @@ -1523,10 +1523,10 @@ fail: PyDoc_STRVAR(s_unpack__doc__, -"S.unpack(str) -> (v1, v2, ...)\n\ +"S.unpack(buffer) -> (v1, v2, ...)\n\ \n\ Return tuple containing values unpacked according to this Struct's format.\n\ -Requires len(str) == self.size. See struct.__doc__ for more on format\n\ +Requires len(buffer) == self.size. See struct.__doc__ for more on format\n\ strings."); static PyObject * @@ -1535,6 +1535,10 @@ s_unpack(PyObject *self, PyObject *inputstr) PyStructObject *soself = (PyStructObject *)self; assert(PyStruct_Check(self)); assert(soself->s_codes != NULL); + if (inputstr != NULL && PyBytes_Check(inputstr) && + PyBytes_GET_SIZE(inputstr) == soself->s_size) { + return s_unpack_internal(soself, PyBytes_AS_STRING(inputstr)); + } if (inputstr == NULL || !PyString_Check(inputstr) || PyString_GET_SIZE(inputstr) != soself->s_size) { PyErr_Format(StructError, -- cgit v0.12