py/stream: Add mp_stream_close() helper function.

1 files changed, 8 insertions, 0 deletions

diff --git a/py/stream.c b/py/stream.c
index 9b1d5fd2d..ebdbe26b4 100644
--- a/py/stream.c
+++ b/py/stream.c
@@ -31,6 +31,7 @@
 #include "py/nlr.h"
 #include "py/objstr.h"
 #include "py/stream.h"
+#include "py/runtime.h"
 
 #if MICROPY_STREAMS_NON_BLOCK
 #include <errno.h>
@@ -104,6 +105,13 @@ const mp_stream_p_t *mp_get_stream_raise(mp_obj_t self_in, int flags) {
     return stream_p;
 }
 
+mp_obj_t mp_stream_close(mp_obj_t stream) {
+    // TODO: Still consider using ioctl for close
+    mp_obj_t dest[2];
+    mp_load_method(stream, MP_QSTR_close, dest);
+    return mp_call_method_n_kw(0, 0, dest);
+}
+
 STATIC mp_obj_t stream_read_generic(size_t n_args, const mp_obj_t *args, byte flags) {
     const mp_stream_p_t *stream_p = mp_get_stream_raise(args[0], MP_STREAM_OP_READ);