Fixed issue #57. Updated file writing to use tempfile.mkstemp

gooey/python_bindings/modules.py

@@ -1,36 +1,27 @@
-__author__ = 'Chris'
 
 import os
 import time
 import hashlib
 from itertools import dropwhile
 import sys
+import tempfile
 
 sys.path.append(os.path.dirname(__file__))
 
-def generate_pyfilename():
-  '''
-  generates a random filename by hashing the current time stamp
-  Leading numbers are dropped from the filename for import compatibility
-  '''
-  hash = hashlib.md5(str(time.time())).hexdigest()
-  return ''.join(dropwhile(lambda c: c.isdigit(), hash))
-
 def load(module_source):
-  tmp_filename = generate_pyfilename()
-  tmp_filedir = os.path.dirname(__file__)
-  tmp_filepath = os.path.join(tmp_filedir, tmp_filename)
-
-  tmp_py_file = tmp_filepath + '.py'
-  tmp_pyc_file = tmp_filepath + '.pyc'
-
+  descriptor, tmpfilepath = tempfile.mkstemp(suffix='.py')
+  print descriptor, tmpfilepath
+  tmpfiledir = os.path.dirname(tmpfilepath)
+  tmpfilename = os.path.splitext(os.path.split(tmpfilepath)[-1])[0]
+  print tmpfilename
+  sys.path.append(tmpfiledir)
   try:
-    with open(tmp_py_file, 'w') as f:
+    with open(tmpfilepath, 'w') as f:
       f.write(module_source)
-    return __import__(tmp_filename)
+    return __import__(tmpfilename)
   finally:
-    os.remove(tmp_py_file)
-    os.remove(tmp_pyc_file)
+    os.close(descriptor)
+    os.remove(tmpfilepath)
 
 if __name__ == '__main__':
   pass