diff options
-rw-r--r-- | kswriter/KSWriter.py | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/kswriter/KSWriter.py b/kswriter/KSWriter.py index b5e97a3..bae1e36 100644 --- 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 |