Merge pull request #201 from mjpieters/patch-1

Make memory_profiler work with Python 2.7 tracemalloc

Author: fabianp

memory_profiler.py

@@ -48,6 +48,7 @@
 
 if PY2:
     import __builtin__ as builtins
+    from future_builtins import filter
 else:
     import builtins
 
@@ -1122,17 +1123,14 @@ def exec_with_profiler(filename, profiler, backend, passed_args=[]):
     ns = dict(_CLEAN_GLOBALS, profile=profiler)
     _backend = choose_backend(backend)
     sys.argv = [filename] + passed_args
-    if PY2:
-        execfile(filename, ns, ns)
-    else:
+    try:
         if _backend == 'tracemalloc' and has_tracemalloc:
             tracemalloc.start()
-        try:
-            with open(filename) as f:
-                exec(compile(f.read(), filename, 'exec'), ns, ns)
-        finally:
-            if has_tracemalloc and tracemalloc.is_tracing():
-                tracemalloc.stop()
+        with open(filename) as f:
+            exec(compile(f.read(), filename, 'exec'), ns, ns)
+    finally:
+        if has_tracemalloc and tracemalloc.is_tracing():
+            tracemalloc.stop()
 
 
 def run_module_with_profiler(module, profiler, backend, passed_args=[]):