rpc_wrappers: data_logger_erase: longer timeout

Make the `data_logger_erase` command have a longer, configurable, RPC
timeout.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/rpc_wrappers/data_logger_erase.py

@@ -16,9 +16,12 @@ def add_parser(cls, parser):
             action="store_true",
             help="Erase entire address space, not just written blocks",
         )
+        # Erasing a complete flash chip can take a long time
+        parser.add_argument("--timeout", type=int, default=60000, help="Duration to wait for erase to complete")
 
     def __init__(self, args):
         self.infuse_id = args.id
+        self.timeout = args.timeout
         if args.onboard:
             self.logger = defs.rpc_enum_data_logger.FLASH_ONBOARD
         elif args.removable:
@@ -27,6 +30,9 @@ def __init__(self, args):
             raise NotImplementedError
         self.erase_all = 1 if args.erase_all else 0
 
+    def command_timeout_ms(self) -> int:
+        return self.timeout
+
     def request_struct(self):
         return self.request(self.logger, self.erase_all)