stmhal/wdt: Implement keyword args to WDT constructor.

1 files changed, 12 insertions, 6 deletions

diff --git a/stmhal/wdt.c b/stmhal/wdt.c
index 6e1172caf6..d9a089d011 100644
--- a/stmhal/wdt.c
+++ b/stmhal/wdt.c
@@ -37,17 +37,23 @@ typedef struct _pyb_wdt_obj_t {
 
 STATIC pyb_wdt_obj_t pyb_wdt = {{&pyb_wdt_type}};
 
-STATIC mp_obj_t pyb_wdt_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
-    // check arguments
-    mp_arg_check_num(n_args, n_kw, 2, 2, false);
-
-    mp_int_t id = mp_obj_get_int(args[0]);
+STATIC mp_obj_t pyb_wdt_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *all_args) {
+    // parse arguments
+    enum { ARG_id, ARG_timeout };
+    static const mp_arg_t allowed_args[] = {
+        { MP_QSTR_id, MP_ARG_INT, {.u_int = 0} },
+        { MP_QSTR_timeout, MP_ARG_INT, {.u_int = 5000} },
+    };
+    mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
+    mp_arg_parse_all_kw_array(n_args, n_kw, all_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
+
+    mp_int_t id = args[ARG_id].u_int;
     if (id != 0) {
         nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "WDT(%d) does not exist", id));
     }
 
     // timeout is in milliseconds
-    mp_int_t timeout = mp_obj_get_int(args[1]);
+    mp_int_t timeout = args[ARG_timeout].u_int;
 
     // compute prescaler
     uint32_t prescaler;