From f5a01de3a45fa06cdc35c863d8829c8c7083b1ce Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Sat, 4 Apr 2009 08:34:23 -0400 Subject: Even better execution of main files. --- coverage/execfile.py | 30 +++++++++++++++++++++++------- 1 file changed, 23 insertions(+), 7 deletions(-) (limited to 'coverage/execfile.py') diff --git a/coverage/execfile.py b/coverage/execfile.py index bcf5bf4..4820c8e 100644 --- a/coverage/execfile.py +++ b/coverage/execfile.py @@ -1,11 +1,27 @@ """Execute files of Python code.""" -import os, sys +import imp, os, sys -def run_python_file(filename): - mod_globals = { - '__name__': '__main__', - '__file__': filename, - } +def run_python_file(filename, args): + """Run a python source file as if it were the main program on the python + command line. + + `filename` is the path to the file to execute, must be a .py file. + `args` is the argument array to present as sys.argv. + + """ + # Most code that does this does it in a way that leaves __main__ or __file__ + # with the wrong values. Importing the code as __main__ gets all of this + # right automatically. + # + # One difference from python.exe: if I run foo.py from the command line, it + # always uses foo.py. With this code, it might find foo.pyc instead. + + sys.argv = args sys.path[0] = os.path.dirname(filename) - execfile(filename, mod_globals) + + try: + src = open(filename) + imp.load_module('__main__', src, filename, (".py", "r", imp.PY_SOURCE)) + finally: + src.close() -- cgit v1.2.1