cli argument

tmtcc.py

@@ -166,16 +166,18 @@ class PusHandler(SpecificApidHandlerBase):
         wrapper: VerificationWrapper,
         printer: FsfwTmTcPrinter,
         raw_logger: RawTmtcTimedLogWrapper,
+        hk_level: int
     ):
         super().__init__(PUS_APID, None)
         self.printer = printer
         self.verif_wrapper = wrapper
         self.raw_logger = raw_logger
+        self.hk_level = hk_level
 
     def handle_tm(self, packet: bytes, _user_args: any):
         # with open("tc.bin", "wb") as of:
         #    of.write(packet)
-        pus_factory_hook(packet, self.verif_wrapper, self.printer, self.raw_logger)
+        pus_factory_hook(packet, self.verif_wrapper, self.printer, self.raw_logger, self.hk_level)
 
 
 class UnknownApidHandler(GenericApidHandlerBase):
@@ -335,13 +337,18 @@ class TcHandler(TcHandlerBase):
                 _LOGGER.info("Finished CFDP queue")
 
 
-def setup_params() -> SetupWrapper:
+def setup_params() -> (SetupWrapper, int):
     hook_obj = EiveHookObject(default_json_path())
     params = SetupParams()
     parser_wrapper = PreArgsParsingWrapper()
     parser_wrapper.create_default_parent_parser()
     parser_wrapper.create_default_parser()
-    parser_wrapper.add_def_proc_and_cfdp_as_subparsers()
+    tmtc_parser, cfdp_parser = parser_wrapper.add_def_proc_and_cfdp_as_subparsers()
+    tmtc_parser.add_argument(
+        "--hk",
+        help="HK output level",
+        default=2,
+    )
     post_arg_parsing_wrapper = parser_wrapper.parse(
         setup_params=params, hook_obj=hook_obj
     )
@@ -352,13 +359,14 @@ def setup_params() -> SetupWrapper:
         post_arg_parsing_wrapper.set_params_with_prompts(proc_param_wrapper)
     else:
         post_arg_parsing_wrapper.set_params_without_prompts(proc_param_wrapper)
+    hk_level = post_arg_parsing_wrapper.args_raw.hk
     params.apid = PUS_APID
     if params.com_if is None:
         raise ValueError("could not determine a COM interface.")
     setup_wrapper = SetupWrapper(
         hook_obj=hook_obj, setup_params=params, proc_param_wrapper=proc_param_wrapper
     )
-    return setup_wrapper
+    return setup_wrapper, hk_level
 
 
 def setup_cfdp_handler() -> CfdpInCcsdsWrapper:
@@ -400,12 +408,13 @@ def setup_tmtc_handlers(
     printer: FsfwTmTcPrinter,
     raw_logger: RawTmtcTimedLogWrapper,
     gui: bool,
+    hk_level: int
 ) -> (CcsdsTmHandler, TcHandler):
     cfdp_in_ccsds_wrapper = setup_cfdp_handler()
     verification_wrapper = VerificationWrapper(
         verificator, _LOGGER, printer.file_logger
     )
-    pus_handler = PusHandler(verification_wrapper, printer, raw_logger)
+    pus_handler = PusHandler(verification_wrapper, printer, raw_logger, hk_level)
     ccsds_handler = CustomCcsdsTmHandler(generic_handler=UnknownApidHandler(None))
     ccsds_handler.add_apid_handler(pus_handler)
     ccsds_handler.add_apid_handler(cfdp_in_ccsds_wrapper)
@@ -444,7 +453,7 @@ def main():  # noqa C901: Complexity okay here.
     # TODO: -V CLI argument to enable this?
     _LOGGER.setLevel(_LOG_LEVEL)
     try:
-        setup_wrapper = setup_params()
+        setup_wrapper, hk_level = setup_params()
     except KeyboardInterrupt as e:
         _LOGGER.info(f"{e}. Exiting")
         sys.exit(0)
@@ -457,7 +466,7 @@ def main():  # noqa C901: Complexity okay here.
     )
     pus_verificator = PusVerificator()
     ccsds_handler, tc_handler = setup_tmtc_handlers(
-        pus_verificator, printer, raw_logger, setup_wrapper.params.use_gui
+        pus_verificator, printer, raw_logger, setup_wrapper.params.use_gui, hk_level
     )
 
     tmtccmd.setup(setup_wrapper)