summaryrefslogtreecommitdiff
path: root/extmod
diff options
context:
space:
mode:
Diffstat (limited to 'extmod')
-rw-r--r--extmod/modbtree.c10
-rw-r--r--extmod/moduasyncio.c10
-rw-r--r--extmod/moduselect.c5
-rw-r--r--extmod/modussl_axtls.c2
-rw-r--r--extmod/modussl_mbedtls.c2
-rw-r--r--extmod/vfs_fat_file.c8
-rw-r--r--extmod/vfs_lfsx_file.c8
-rw-r--r--extmod/vfs_posix_file.c8
8 files changed, 22 insertions, 31 deletions
diff --git a/extmod/modbtree.c b/extmod/modbtree.c
index 60c6885e6..f21fe3ff9 100644
--- a/extmod/modbtree.c
+++ b/extmod/modbtree.c
@@ -319,15 +319,19 @@ STATIC const mp_rom_map_elem_t btree_locals_dict_table[] = {
STATIC MP_DEFINE_CONST_DICT(btree_locals_dict, btree_locals_dict_table);
+STATIC const mp_getiter_iternext_custom_t btree_getiter_iternext = {
+ .getiter = btree_getiter,
+ .iternext = btree_iternext,
+};
+
STATIC MP_DEFINE_CONST_OBJ_TYPE(
btree_type,
MP_QSTR_btree,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_CUSTOM,
MP_TYPE_NULL_MAKE_NEW,
// Save on qstr's, reuse same as for module
print, btree_print,
- getiter, btree_getiter,
- iternext, btree_iternext,
+ iter, &btree_getiter_iternext,
binary_op, btree_binary_op,
subscr, btree_subscr,
locals_dict, &btree_locals_dict
diff --git a/extmod/moduasyncio.c b/extmod/moduasyncio.c
index 546764209..b0eb8b650 100644
--- a/extmod/moduasyncio.c
+++ b/extmod/moduasyncio.c
@@ -287,14 +287,18 @@ STATIC mp_obj_t task_iternext(mp_obj_t self_in) {
return mp_const_none;
}
+STATIC const mp_getiter_iternext_custom_t task_getiter_iternext = {
+ .getiter = task_getiter,
+ .iternext = task_iternext,
+};
+
STATIC MP_DEFINE_CONST_OBJ_TYPE(
task_type,
MP_QSTR_Task,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_CUSTOM,
task_make_new,
attr, task_attr,
- getiter, task_getiter,
- iternext, task_iternext
+ iter, &task_getiter_iternext
);
/******************************************************************************/
diff --git a/extmod/moduselect.c b/extmod/moduselect.c
index 352b15d98..58bd1169a 100644
--- a/extmod/moduselect.c
+++ b/extmod/moduselect.c
@@ -339,10 +339,9 @@ STATIC MP_DEFINE_CONST_DICT(poll_locals_dict, poll_locals_dict_table);
STATIC MP_DEFINE_CONST_OBJ_TYPE(
mp_type_poll,
MP_QSTR_poll,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_ITERNEXT,
MP_TYPE_NULL_MAKE_NEW,
- getiter, mp_identity_getiter,
- iternext, poll_iternext,
+ iter, poll_iternext,
locals_dict, &poll_locals_dict
);
diff --git a/extmod/modussl_axtls.c b/extmod/modussl_axtls.c
index 78470ea6d..a6d606d56 100644
--- a/extmod/modussl_axtls.c
+++ b/extmod/modussl_axtls.c
@@ -321,8 +321,6 @@ STATIC MP_DEFINE_CONST_OBJ_TYPE(
MP_TYPE_NULL_MAKE_NEW,
// Save on qstr's, reuse same as for module
print, ussl_socket_print,
- getiter, NULL,
- iternext, NULL,
protocol, &ussl_socket_stream_p,
locals_dict, &ussl_socket_locals_dict
);
diff --git a/extmod/modussl_mbedtls.c b/extmod/modussl_mbedtls.c
index 76ca3ac71..50712980b 100644
--- a/extmod/modussl_mbedtls.c
+++ b/extmod/modussl_mbedtls.c
@@ -399,8 +399,6 @@ STATIC MP_DEFINE_CONST_OBJ_TYPE(
MP_TYPE_NULL_MAKE_NEW,
// Save on qstr's, reuse same as for module
print, socket_print,
- getiter, NULL,
- iternext, NULL,
protocol, &ussl_socket_stream_p,
locals_dict, &ussl_socket_locals_dict
);
diff --git a/extmod/vfs_fat_file.c b/extmod/vfs_fat_file.c
index 00980459d..ca2e2b446 100644
--- a/extmod/vfs_fat_file.c
+++ b/extmod/vfs_fat_file.c
@@ -179,11 +179,9 @@ STATIC const mp_stream_p_t vfs_fat_fileio_stream_p = {
MP_DEFINE_CONST_OBJ_TYPE(
mp_type_vfs_fat_fileio,
MP_QSTR_FileIO,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, file_obj_print,
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &vfs_fat_fileio_stream_p,
locals_dict, &vfs_fat_rawfile_locals_dict
);
@@ -198,11 +196,9 @@ STATIC const mp_stream_p_t vfs_fat_textio_stream_p = {
MP_DEFINE_CONST_OBJ_TYPE(
mp_type_vfs_fat_textio,
MP_QSTR_TextIOWrapper,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, file_obj_print,
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &vfs_fat_textio_stream_p,
locals_dict, &vfs_fat_rawfile_locals_dict
);
diff --git a/extmod/vfs_lfsx_file.c b/extmod/vfs_lfsx_file.c
index fda1b97b2..f97641b7b 100644
--- a/extmod/vfs_lfsx_file.c
+++ b/extmod/vfs_lfsx_file.c
@@ -223,11 +223,9 @@ STATIC const mp_stream_p_t MP_VFS_LFSx(fileio_stream_p) = {
MP_DEFINE_CONST_OBJ_TYPE(
MP_TYPE_VFS_LFSx_(_fileio),
MP_QSTR_FileIO,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, MP_VFS_LFSx(file_print),
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &MP_VFS_LFSx(fileio_stream_p),
locals_dict, &MP_VFS_LFSx(file_locals_dict)
);
@@ -242,11 +240,9 @@ STATIC const mp_stream_p_t MP_VFS_LFSx(textio_stream_p) = {
MP_DEFINE_CONST_OBJ_TYPE(
MP_TYPE_VFS_LFSx_(_textio),
MP_QSTR_TextIOWrapper,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, MP_VFS_LFSx(file_print),
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &MP_VFS_LFSx(textio_stream_p),
locals_dict, &MP_VFS_LFSx(file_locals_dict)
);
diff --git a/extmod/vfs_posix_file.c b/extmod/vfs_posix_file.c
index f0b5436fe..729d914d3 100644
--- a/extmod/vfs_posix_file.c
+++ b/extmod/vfs_posix_file.c
@@ -252,11 +252,9 @@ STATIC const mp_stream_p_t vfs_posix_fileio_stream_p = {
MP_DEFINE_CONST_OBJ_TYPE(
mp_type_vfs_posix_fileio,
MP_QSTR_FileIO,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, vfs_posix_file_print,
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &vfs_posix_fileio_stream_p,
locals_dict, &vfs_posix_rawfile_locals_dict
);
@@ -271,11 +269,9 @@ STATIC const mp_stream_p_t vfs_posix_textio_stream_p = {
MP_DEFINE_CONST_OBJ_TYPE(
mp_type_vfs_posix_textio,
MP_QSTR_TextIOWrapper,
- MP_TYPE_FLAG_NONE,
+ MP_TYPE_FLAG_ITER_IS_STREAM,
MP_TYPE_NULL_MAKE_NEW,
print, vfs_posix_file_print,
- getiter, mp_identity_getiter,
- iternext, mp_stream_unbuffered_iter,
protocol, &vfs_posix_textio_stream_p,
locals_dict, &vfs_posix_rawfile_locals_dict
);