From 3bab2010ee15ef3a4e115403ae6544fb07859b7a Mon Sep 17 00:00:00 2001 From: "sanika.inamdar" Date: Fri, 22 Nov 2024 12:27:57 +0530 Subject: [PATCH] feat(mfg_gen): provision for specifying prefix counter for encrypted key files --- tools/mass_mfg/mfg_gen.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/mass_mfg/mfg_gen.py b/tools/mass_mfg/mfg_gen.py index 00260b46c3..3f303f8a8e 100644 --- a/tools/mass_mfg/mfg_gen.py +++ b/tools/mass_mfg/mfg_gen.py @@ -283,10 +283,14 @@ def create_intermediate_csv(args, keys_in_values_file, keys_repeat, is_encr=Fals prefix_number = f'{prefix_num_start:0{prefix_num_digits}}' csv_filename = args.prefix + '-' + prefix_number + '-' + file_identifier_value + '.' + 'csv' bin_filename = args.prefix + '-' + prefix_number + '-' + file_identifier_value + '.' + 'bin' + if args.keygen: + args.keyfile = 'keys-' + args.prefix + '-' + prefix_number + '-' + file_identifier_value prefix_num_start += 1 else: csv_filename = args.prefix + '-' + file_identifier_value + '.' + 'csv' bin_filename = args.prefix + '-' + file_identifier_value + '.' + 'bin' + if args.keygen: + args.keyfile = 'keys-' + args.prefix + '-' + file_identifier_value output_csv_file = output_csv_target_dir + csv_filename # Verify if output csv file does not exist @@ -304,8 +308,6 @@ def create_intermediate_csv(args, keys_in_values_file, keys_repeat, is_encr=Fals args.input = output_csv_file args.output = os.path.join('bin', bin_filename) - if args.keygen: - args.keyfile = 'keys-' + args.prefix + '-' + file_identifier_value if is_encr: nvs_partition_gen.encrypt(args)