Add __file__ in run context

klen · klen · commit b79b2e991e48 · 2014-03-27T01:41:15.000+07:00
diff --git a/pymode/run.py b/pymode/run.py
@@ -28,7 +28,10 @@ def run_code():
             lines.pop(ix)
 
     context = dict(
-        __name__='__main__', input=env.user_input, raw_input=env.user_input)
+        __name__='__main__',
+        __file__=env.var('expand("%:p")'),
+        input=env.user_input,
+        raw_input=env.user_input)
 
     sys.stdout, stdout_ = StringIO(), sys.stdout
     sys.stderr, stderr_ = StringIO(), sys.stderr
