Skip to content

Commit 22d85ec

Browse files
committed
py: Use new code pattern for parsing kw args with mp_arg_parse_all.
Makes code easier to read and more maintainable.
1 parent 8bb4931 commit 22d85ec

File tree

3 files changed

+25
-17
lines changed

3 files changed

+25
-17
lines changed

py/objenumerate.c

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,23 +39,25 @@ typedef struct _mp_obj_enumerate_t {
3939

4040
STATIC mp_obj_t enumerate_iternext(mp_obj_t self_in);
4141

42-
STATIC const mp_arg_t enumerate_make_new_args[] = {
43-
{ MP_QSTR_iterable, MP_ARG_REQUIRED | MP_ARG_OBJ, {.u_obj = MP_OBJ_NULL} },
44-
{ MP_QSTR_start, MP_ARG_INT, {.u_int = 0} },
45-
};
46-
#define ENUMERATE_MAKE_NEW_NUM_ARGS MP_ARRAY_SIZE(enumerate_make_new_args)
47-
4842
STATIC mp_obj_t enumerate_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
4943
#if MICROPY_CPYTHON_COMPAT
44+
static const mp_arg_t allowed_args[] = {
45+
{ MP_QSTR_iterable, MP_ARG_REQUIRED | MP_ARG_OBJ, {.u_obj = MP_OBJ_NULL} },
46+
{ MP_QSTR_start, MP_ARG_INT, {.u_int = 0} },
47+
};
48+
5049
// parse args
51-
mp_arg_val_t vals[ENUMERATE_MAKE_NEW_NUM_ARGS];
52-
mp_arg_parse_all_kw_array(n_args, n_kw, args, ENUMERATE_MAKE_NEW_NUM_ARGS, enumerate_make_new_args, vals);
50+
struct {
51+
mp_arg_val_t iterable, start;
52+
} arg_vals;
53+
mp_arg_parse_all_kw_array(n_args, n_kw, args,
54+
MP_ARRAY_SIZE(allowed_args), allowed_args, (mp_arg_val_t*)&arg_vals);
5355

5456
// create enumerate object
5557
mp_obj_enumerate_t *o = m_new_obj(mp_obj_enumerate_t);
5658
o->base.type = type;
57-
o->iter = mp_getiter(vals[0].u_obj);
58-
o->cur = vals[1].u_int;
59+
o->iter = mp_getiter(arg_vals.iterable.u_obj);
60+
o->cur = arg_vals.start.u_int;
5961
#else
6062
(void)n_kw;
6163
mp_obj_enumerate_t *o = m_new_obj(mp_obj_enumerate_t);

py/objlist.c

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -318,16 +318,19 @@ mp_obj_t mp_obj_list_sort(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_
318318
};
319319

320320
// parse args
321-
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
322-
mp_arg_parse_all(n_args - 1, pos_args + 1, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
321+
struct {
322+
mp_arg_val_t key, reverse;
323+
} args;
324+
mp_arg_parse_all(n_args - 1, pos_args + 1, kw_args,
325+
MP_ARRAY_SIZE(allowed_args), allowed_args, (mp_arg_val_t*)&args);
323326

324327
assert(MP_OBJ_IS_TYPE(pos_args[0], &mp_type_list));
325328
mp_obj_list_t *self = MP_OBJ_TO_PTR(pos_args[0]);
326329

327330
if (self->len > 1) {
328331
mp_quicksort(self->items, self->items + self->len - 1,
329-
args[0].u_obj == mp_const_none ? MP_OBJ_NULL : args[0].u_obj,
330-
args[1].u_bool ? mp_const_false : mp_const_true);
332+
args.key.u_obj == mp_const_none ? MP_OBJ_NULL : args.key.u_obj,
333+
args.reverse.u_bool ? mp_const_false : mp_const_true);
331334
}
332335

333336
return mp_const_none;

py/objstr.c

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -557,11 +557,14 @@ STATIC mp_obj_t str_splitlines(size_t n_args, const mp_obj_t *pos_args, mp_map_t
557557
};
558558

559559
// parse args
560-
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
561-
mp_arg_parse_all(n_args - 1, pos_args + 1, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
560+
struct {
561+
mp_arg_val_t keepends;
562+
} args;
563+
mp_arg_parse_all(n_args - 1, pos_args + 1, kw_args,
564+
MP_ARRAY_SIZE(allowed_args), allowed_args, (mp_arg_val_t*)&args);
562565

563566
mp_obj_t new_args[2] = {pos_args[0], MP_OBJ_NEW_QSTR(MP_QSTR__backslash_n)};
564-
return str_split_internal(2, new_args, SPLITLINES | (args[0].u_bool ? KEEP : 0));
567+
return str_split_internal(2, new_args, SPLITLINES | (args.keepends.u_bool ? KEEP : 0));
565568
}
566569
#endif
567570

0 commit comments

Comments
 (0)