gen_partition.py: Enhance gen_partition.py to handle command line

This backported commit adds support of option "-m" missing in the initially
imported utility.

Original commit: https://git.codelinaro.org/clo/le/meta-qti-bsp/-/commit/b920abedc2ec26f2e32e965b31d3338ec606ce6e

Now gen_partition.py will take input file, output xml and
partition-imagefile mapping that a user passes through command line.
output xml will have image filenames set by user mapped to partitions
rather than the entries specified in the input conf file if -m is specified.

Fixes: 7586c81 ("Import gen_partition.py utility")
Signed-off-by: Oleksandr Suvorov <[email protected]>

Author: Divya Theja

gen_partition.py

@@ -64,6 +64,8 @@ def usage():
 # store entries read from input file
 disk_entry = None
 partition_entries_dict = {}
+# store partition image map passed from command line
+partition_image_map = {}
 input_file = None
 output_xml = None
 
@@ -119,6 +121,8 @@ def partition_options(argv):
          partition_entry["filename"] = arg
       elif opt in ['--sparse']:
          partition_entry["sparse"] = arg
+      if partition_entry["label"] in partition_image_map.keys():
+         partition_entry["filename"] = partition_image_map[partition_entry["label"]]
    return partition_entry
 
 def parse_partition_entry(partition_entry):
@@ -227,6 +231,14 @@ def generate_partition_xml (disk_entry, partition_entries_dict, output_xml):
           input_file=arg
         elif opt in ["-o"]:
           output_xml=arg
+        elif opt in ["-m"]:
+          for mapping in arg.split(','):
+            tags=mapping.split("=")
+            if len(tags) > 1:
+              partition_image_map[tags[0]]=tags[1]
+        else:
+          usage()
+
    except Exception as argerr:
       print (str(argerr))
       usage()