From b11b85adaac9ac2e460647e59f951f5086cd2d4e Mon Sep 17 00:00:00 2001 From: Damien George Date: Fri, 18 Apr 2014 22:59:24 +0100 Subject: py: Allow to pass buffer protocol flags to get_buffer helper funcs. --- py/obj.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'py/obj.c') diff --git a/py/obj.c b/py/obj.c index e0a712cb0..4b7251447 100644 --- a/py/obj.c +++ b/py/obj.c @@ -357,20 +357,20 @@ mp_obj_t mp_identity(mp_obj_t self) { } MP_DEFINE_CONST_FUN_OBJ_1(mp_identity_obj, mp_identity); -bool mp_get_buffer(mp_obj_t obj, mp_buffer_info_t *bufinfo) { +bool mp_get_buffer(mp_obj_t obj, mp_buffer_info_t *bufinfo, int flags) { mp_obj_type_t *type = mp_obj_get_type(obj); if (type->buffer_p.get_buffer == NULL) { return false; } - int ret = type->buffer_p.get_buffer(obj, bufinfo, MP_BUFFER_READ); + int ret = type->buffer_p.get_buffer(obj, bufinfo, flags); if (ret != 0 || bufinfo->buf == NULL) { return false; } return true; } -void mp_get_buffer_raise(mp_obj_t obj, mp_buffer_info_t *bufinfo) { - if (!mp_get_buffer(obj, bufinfo)) { +void mp_get_buffer_raise(mp_obj_t obj, mp_buffer_info_t *bufinfo, int flags) { + if (!mp_get_buffer(obj, bufinfo, flags)) { nlr_raise(mp_obj_new_exception_msg(&mp_type_TypeError, "object with buffer protocol required")); } } -- cgit v1.2.3