support both %runscript and %post-umount to perform scripts before packaging

[tools/mic.git] / mic / kickstart / __init__.py

diff --git a/mic/kickstart/__init__.py b/mic/kickstart/__init__.py
index 76cc525..bc97e2e 100755 (executable)
--- a/mic/kickstart/__init__.py
+++ b/mic/kickstart/__init__.py
@@ -823,10 +823,11 @@ def get_post_scripts(ks):
 def get_sign_scripts(ks):
     scripts = []
     for s in ks.handler.scripts:
-        if s.type != ksparser.KS_SCRIPT_RUN:
-            continue
-        scripts.append(s)
+        if (s.type == ksparser.KS_SCRIPT_RUN or \
+            s.type == ksparser.KS_SCRIPT_UMOUNT):
+            scripts.append(s)
     return scripts
+
 def add_repo(ks, repostr):
     args = repostr.split()
     repoobj = ks.handler.repo.parse(args[1:])