Add possibility to parse list of partition files

Kickstarter was not able to correctly parse config files that contained
a list of partition files in "Part" section. Now it is possible to add
several partition files in that section.

Change-Id: I723769ef8326576e52205850eca7f69bc6aa9472
Signed-off-by: Katarzyna Gorska <k.gorska@samsung.com>

diff --git a/kswriter/KSWriter.py b/kswriter/KSWriter.py
index b5e97a3..bae1e36 100644 (file)
--- a/kswriter/KSWriter.py
+++ b/kswriter/KSWriter.py
@@ -138,9 +138,19 @@ class KSWriter():
         ptab = ""
         for g in [ plat, img ]:
             if g.has_key("Part"):
-                f = open("%s/partitions/%s" %(meta_root, g['Part']) )
-                ptab = f.read()
-                f.close()
+                if type(g['Part']) == str:
+                    t = [g['Part']]
+                elif type(g['Part']) == list:
+                    t = g['Part']
+                ptab = ""
+                for p in t:
+                    if os.path.exists('%s/partitions/%s' %(meta_root,p)):
+                        f = open("%s/partitions/%s" %(meta_root, p) )
+                        ptab += f.read()
+                        ptab += "\n"
+                        f.close()
+                    else:
+                        raise KSMetaError('%s/partitions/%s.run not found, aborting.' %(meta_root,p))
 
         conf['Part'] = ptab
         conf['Post'] = postscript