future imports fix

py-src/ltmain.py

@@ -14,13 +14,17 @@
 import asyncore
 import signal
 import lttools
+import __future__
+import operator
+import functools
 
 oldout = sys.stdout
 olderr = sys.stderr
 stop = False
 local = True
 threads = []
 currentClient = 0
+compilerFlags = 0
 
 class Printer():
   cur = ""
@@ -147,6 +151,18 @@ def stopped():
   global stop
   return stop
 
+def featureFlags(code):
+  futureItems = __future__.__dict__.iteritems()
+  #ensure we only try to get flags for features
+  featureNames = [k for  k,v in futureItems if hasattr(v, 'compiler_flag')]
+  names = [n for n in code.co_names[1:] if n in featureNames]
+
+  return [getattr(__future__, name).compiler_flag for name in names]
+
+def addFlags(current, new):
+  new.append(current)
+  return functools.reduce(operator.or_, new)
+
 def handleEval(data):
   result = None
   code = cleanCode(data[2]["code"])
@@ -190,11 +206,14 @@ def handleEval(data):
       loc = form[0]
       isEval = False
       try:
-        code= compile(ensureUtf(code), ensureUtf(data[2]["name"]), 'eval')
+        code= compile(ensureUtf(code), ensureUtf(data[2]["name"]), 'eval', compilerFlags)
         isEval = True
       except:
         try:
-          code= compile(ensureUtf(code), ensureUtf(data[2]["name"]), 'exec')
+          global compilerFlags
+          code= compile(ensureUtf(code), ensureUtf(data[2]["name"]), 'exec', compilerFlags)
+          if code.co_names[0] == '__future__':
+            compilerFlags = addFlags(compilerFlags, featureFlags(code))
         except:
           e = traceback.format_exc()
           send(data[0], "editor.eval.python.exception", {"ex": cleanTrace(e), "meta": loc})