New feature: python scriptlets.

[pakfire.git] / python / pakfire / packages / packager.py

diff --git a/python/pakfire/packages/packager.py b/python/pakfire/packages/packager.py
index 3ff3c384d81a69edd58861f248fc1083042460db..c756b0c5aed60363b96341573e8d8cc5815ffb80 100644 (file)
--- a/python/pakfire/packages/packager.py
+++ b/python/pakfire/packages/packager.py
@@ -455,7 +455,9 @@ class BinaryPackager(Packager):
                        # Write script to a file.
                        scriptlet_file = self.mktemp()
 
-                       if scriptlet["lang"] == "bin":
+                       lang = scriptlet["lang"]
+
+                       if lang == "bin":
                                path = lang["path"]
                                try:
                                        f = open(path, "b")
@@ -474,14 +476,13 @@ class BinaryPackager(Packager):
                                f.close()
                                s.close()
 
-                       elif scriptlet["lang"] == "shell":
+                       elif lang == "shell":
                                s = open(scriptlet_file, "w")
 
                                # Write shell script to file.
                                s.write("#!/bin/sh -e\n\n")
                                s.write(scriptlet["scriptlet"])
                                s.write("\n\nexit 0\n")
-
                                s.close()
 
                                if scriptlet_name in SCRIPTS_PREREQUIRES:
@@ -490,6 +491,12 @@ class BinaryPackager(Packager):
 
                                        prerequires += self.builder.find_prerequires(scriptlet_file)
 
+                       elif lang == "python":
+                               # Write the code to the scriptlet file.
+                               s = open(scriptlet_file, "w")
+                               s.write(scriptlet["scriptlet"])
+                               s.close()
+
                        else:
                                raise Exception, "Unknown scriptlet language: %s" % scriptlet["lang"]