Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(31)

Side by Side Diff: third_party/coverage-3.7.1/coverage/execfile.py

Issue 225633007: Upgrade to coverage 3.7.1 and have it auto-build itself on first use. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/tools/build
Patch Set: sigh our imports are a mess Created 6 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 """Execute files of Python code.""" 1 """Execute files of Python code."""
2 2
3 import imp, os, sys 3 import imp, marshal, os, sys
4 4
5 from coverage.backward import exec_code_object, open_source 5 from coverage.backward import exec_code_object, open_source
6 from coverage.misc import NoSource, ExceptionDuringRun 6 from coverage.misc import ExceptionDuringRun, NoCode, NoSource
7 7
8 8
9 try: 9 try:
10 # In Py 2.x, the builtins were in __builtin__ 10 # In Py 2.x, the builtins were in __builtin__
11 BUILTINS = sys.modules['__builtin__'] 11 BUILTINS = sys.modules['__builtin__']
12 except KeyError: 12 except KeyError:
13 # In Py 3.x, they're in builtins 13 # In Py 3.x, they're in builtins
14 BUILTINS = sys.modules['builtins'] 14 BUILTINS = sys.modules['builtins']
15 15
16 16
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after
58 searchpath = package.__path__ 58 searchpath = package.__path__
59 openfile, pathname, _ = imp.find_module(name, searchpath) 59 openfile, pathname, _ = imp.find_module(name, searchpath)
60 except ImportError: 60 except ImportError:
61 _, err, _ = sys.exc_info() 61 _, err, _ = sys.exc_info()
62 raise NoSource(str(err)) 62 raise NoSource(str(err))
63 finally: 63 finally:
64 if openfile: 64 if openfile:
65 openfile.close() 65 openfile.close()
66 66
67 # Finally, hand the file off to run_python_file for execution. 67 # Finally, hand the file off to run_python_file for execution.
68 pathname = os.path.abspath(pathname)
68 args[0] = pathname 69 args[0] = pathname
69 run_python_file(pathname, args, package=packagename) 70 run_python_file(pathname, args, package=packagename)
70 71
71 72
72 def run_python_file(filename, args, package=None): 73 def run_python_file(filename, args, package=None):
73 """Run a python file as if it were the main program on the command line. 74 """Run a python file as if it were the main program on the command line.
74 75
75 `filename` is the path to the file to execute, it need not be a .py file. 76 `filename` is the path to the file to execute, it need not be a .py file.
76 `args` is the argument array to present as sys.argv, including the first 77 `args` is the argument array to present as sys.argv, including the first
77 element naming the file being executed. `package` is the name of the 78 element naming the file being executed. `package` is the name of the
78 enclosing package, if any. 79 enclosing package, if any.
79 80
80 """ 81 """
81 # Create a module to serve as __main__ 82 # Create a module to serve as __main__
82 old_main_mod = sys.modules['__main__'] 83 old_main_mod = sys.modules['__main__']
83 main_mod = imp.new_module('__main__') 84 main_mod = imp.new_module('__main__')
84 sys.modules['__main__'] = main_mod 85 sys.modules['__main__'] = main_mod
85 main_mod.__file__ = filename 86 main_mod.__file__ = filename
86 if package: 87 if package:
87 main_mod.__package__ = package 88 main_mod.__package__ = package
88 main_mod.__builtins__ = BUILTINS 89 main_mod.__builtins__ = BUILTINS
89 90
90 # Set sys.argv and the first path element properly. 91 # Set sys.argv properly.
91 old_argv = sys.argv 92 old_argv = sys.argv
92 old_path0 = sys.path[0]
93 sys.argv = args 93 sys.argv = args
94 if package:
95 sys.path[0] = ''
96 else:
97 sys.path[0] = os.path.abspath(os.path.dirname(filename))
98 94
99 try: 95 try:
100 # Open the source file. 96 # Make a code object somehow.
101 try: 97 if filename.endswith(".pyc") or filename.endswith(".pyo"):
102 source_file = open_source(filename) 98 code = make_code_from_pyc(filename)
103 except IOError: 99 else:
104 raise NoSource("No file to run: %r" % filename) 100 code = make_code_from_py(filename)
105 101
106 try: 102 # Execute the code object.
107 source = source_file.read()
108 finally:
109 source_file.close()
110
111 # We have the source. `compile` still needs the last line to be clean,
112 # so make sure it is, then compile a code object from it.
113 if not source or source[-1] != '\n':
114 source += '\n'
115 code = compile(source, filename, "exec")
116
117 # Execute the source file.
118 try: 103 try:
119 exec_code_object(code, main_mod.__dict__) 104 exec_code_object(code, main_mod.__dict__)
120 except SystemExit: 105 except SystemExit:
121 # The user called sys.exit(). Just pass it along to the upper 106 # The user called sys.exit(). Just pass it along to the upper
122 # layers, where it will be handled. 107 # layers, where it will be handled.
123 raise 108 raise
124 except: 109 except:
125 # Something went wrong while executing the user code. 110 # Something went wrong while executing the user code.
126 # Get the exc_info, and pack them into an exception that we can 111 # Get the exc_info, and pack them into an exception that we can
127 # throw up to the outer loop. We peel two layers off the traceback 112 # throw up to the outer loop. We peel two layers off the traceback
128 # so that the coverage.py code doesn't appear in the final printed 113 # so that the coverage.py code doesn't appear in the final printed
129 # traceback. 114 # traceback.
130 typ, err, tb = sys.exc_info() 115 typ, err, tb = sys.exc_info()
131 raise ExceptionDuringRun(typ, err, tb.tb_next.tb_next) 116 raise ExceptionDuringRun(typ, err, tb.tb_next.tb_next)
132 finally: 117 finally:
133 # Restore the old __main__ 118 # Restore the old __main__
134 sys.modules['__main__'] = old_main_mod 119 sys.modules['__main__'] = old_main_mod
135 120
136 # Restore the old argv and path 121 # Restore the old argv and path
137 sys.argv = old_argv 122 sys.argv = old_argv
138 sys.path[0] = old_path0 123
124 def make_code_from_py(filename):
125 """Get source from `filename` and make a code object of it."""
126 # Open the source file.
127 try:
128 source_file = open_source(filename)
129 except IOError:
130 raise NoSource("No file to run: %r" % filename)
131
132 try:
133 source = source_file.read()
134 finally:
135 source_file.close()
136
137 # We have the source. `compile` still needs the last line to be clean,
138 # so make sure it is, then compile a code object from it.
139 if not source or source[-1] != '\n':
140 source += '\n'
141 code = compile(source, filename, "exec")
142
143 return code
144
145
146 def make_code_from_pyc(filename):
147 """Get a code object from a .pyc file."""
148 try:
149 fpyc = open(filename, "rb")
150 except IOError:
151 raise NoCode("No file to run: %r" % filename)
152
153 try:
154 # First four bytes are a version-specific magic number. It has to
155 # match or we won't run the file.
156 magic = fpyc.read(4)
157 if magic != imp.get_magic():
158 raise NoCode("Bad magic number in .pyc file")
159
160 # Skip the junk in the header that we don't need.
161 fpyc.read(4) # Skip the moddate.
162 if sys.version_info >= (3, 3):
163 # 3.3 added another long to the header (size), skip it.
164 fpyc.read(4)
165
166 # The rest of the file is the code object we want.
167 code = marshal.load(fpyc)
168 finally:
169 fpyc.close()
170
171 return code
OLDNEW
« no previous file with comments | « third_party/coverage-3.7.1/coverage/debug.py ('k') | third_party/coverage-3.7.1/coverage/files.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698