Add support for mqtt version parameter

aws · Dec 22, 2023 · a366de8 · a366de8
1 parent cc023a9
commit a366de8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/samples/utils/command_line_utils.py b/samples/utils/command_line_utils.py
@@ -436,6 +436,7 @@ def parse_sample_input_fleet_provisioning():
         cmdUtils.register_command(CommandLineUtils.m_cmd_csr, "<path>", "Path to CSR in Pem format (optional).")
         cmdUtils.register_command(CommandLineUtils.m_cmd_template_name, "<str>", "The name of your provisioning template.")
         cmdUtils.register_command(CommandLineUtils.m_cmd_template_parameters, "<json>", "Template parameters json.")
+        cmdUtils.register_command(CommandLineUtils.m_cmd_mqtt_version, "<int>", "MQTT Version")
         cmdUtils.get_args()
 
         cmdData = CommandLineUtils.CmdData()
@@ -451,6 +452,7 @@ def parse_sample_input_fleet_provisioning():
         cmdData.input_template_name = cmdUtils.get_command_required(CommandLineUtils.m_cmd_template_name)
         cmdData.input_template_parameters = cmdUtils.get_command_required(CommandLineUtils.m_cmd_template_parameters)
         cmdData.input_is_ci = cmdUtils.get_command(CommandLineUtils.m_cmd_is_ci, None) != None
+        cmdData.input_mqtt_version = int(cmdUtils.get_command(CommandLineUtils.m_cmd_mqtt_version, 5))
         return cmdData
 
     def parse_sample_input_jobs():