1 files changed, 9 insertions, 10 deletions

diff --git a/main.py b/main.py
index d993ee4e..7a09c6ba 100755
--- a/main.py
+++ b/main.py
@@ -23,10 +23,10 @@ if __name__ == '__main__':
 del magic
 
 import getpass
+import imp
 import netrc
 import optparse
 import os
-import re
 import sys
 import time
 import urllib2
@@ -167,16 +167,15 @@ def _MyRepoPath():
 def _MyWrapperPath():
   return os.path.join(os.path.dirname(__file__), 'repo')
 
+_wrapper_module = None
+def WrapperModule():
+  global _wrapper_module
+  if not _wrapper_module:
+    _wrapper_module = imp.load_source('wrapper', _MyWrapperPath())
+  return _wrapper_module
+
 def _CurrentWrapperVersion():
-  VERSION = None
-  pat = re.compile(r'^VERSION *=')
-  fd = open(_MyWrapperPath())
-  for line in fd:
-    if pat.match(line):
-      fd.close()
-      exec line
-      return VERSION
-  raise NameError, 'No VERSION in repo script'
+  return WrapperModule().VERSION
 
 def _CheckWrapperVersion(ver, repo_path):
   if not repo_path: