[tor-commits] [thandy/master] Add a multiplatform ScriptWrapper

[nickm at torproject.org]

commit c01e9c0bcef743442c8e44641b8cb0f51585eb6e
Author: Tomás Touceda <chiiph at torproject.org>
Date:   Sun Nov 6 14:05:17 2011 -0300

    Add a multiplatform ScriptWrapper
---
 lib/thandy/packagesys/ThpPackages.py |   36 +++++++++++++++++++++++++++++++--
 1 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/lib/thandy/packagesys/ThpPackages.py b/lib/thandy/packagesys/ThpPackages.py
index af66b80..d90af0c 100644
--- a/lib/thandy/packagesys/ThpPackages.py
+++ b/lib/thandy/packagesys/ThpPackages.py
@@ -9,6 +9,7 @@ import shutil
 import subprocess
 import sys
 import errno
+import traceback
 
 from lockfile import LockFile, AlreadyLocked, LockFailed
 
@@ -334,9 +335,9 @@ class ThpPackage(object):
                     env['THP_VERBOSE'] = "1"
                     env['THP_PURGE'] = "0"
                     env['THP_TEMP_DIR'] = self._tmp_path
-
-                    sw = ScriptWrapper(os.path.join(self._tmp_path, "meta", 
-                                       "scripts", script[0]), env)
+                    
+                    sw = ScriptBundleWrapper(os.path.join(self._tmp_path, "meta", 
+                                                          "scripts", script[0]), env)
 
                     for type in script[1]:
                         self._scripts[type] = sw
@@ -397,3 +398,32 @@ class ScriptWrapper(object):
                                          env=self._env)
         self._process.wait()
         return self._process.returncode
+
+class ScriptBundleWrapper(object):
+    """ Wrapper for the scripts that runs the code without spawning
+        another python process. This is mostly for handling
+        multiplatform code. """
+    def __init__(self, path = None, env = None):
+        super(ScriptBundleWrapper, self).__init__()
+        self._path = path
+        self._env = env
+
+    def run(self):
+        """ Runs the script using the code class. """
+        loc = {}
+        glob = {}
+        old_environ = os.environ.copy()
+        os.environ.clear()
+        os.environ.update(self._env)
+        try:
+            execfile(self._path, loc, glob)
+        except SystemExit, e:
+            return 1
+        except:
+            traceback.print_exc()
+            return 1
+        finally:
+            os.environ.clear()
+            os.environ.update(old_environ)
+
+        return 0