1#!/usr/bin/env python3 2 3# portions copyright 2001, Autonomous Zones Industries, Inc., all rights... 4# err... reserved and offered to the public under the terms of the 5# Python 2.2 license. 6# Author: Zooko O'Whielacronx 7# http://zooko.com/ 8# mailto:zooko@zooko.com 9# 10# Copyright 2000, Mojam Media, Inc., all rights reserved. 11# Author: Skip Montanaro 12# 13# Copyright 1999, Bioreason, Inc., all rights reserved. 14# Author: Andrew Dalke 15# 16# Copyright 1995-1997, Automatrix, Inc., all rights reserved. 17# Author: Skip Montanaro 18# 19# Copyright 1991-1995, Stichting Mathematisch Centrum, all rights reserved. 20# 21# 22# Permission to use, copy, modify, and distribute this Python software and 23# its associated documentation for any purpose without fee is hereby 24# granted, provided that the above copyright notice appears in all copies, 25# and that both that copyright notice and this permission notice appear in 26# supporting documentation, and that the name of neither Automatrix, 27# Bioreason or Mojam Media be used in advertising or publicity pertaining to 28# distribution of the software without specific, written prior permission. 29# 30"""program/module to trace Python program or function execution 31 32Sample use, command line: 33 trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs 34 trace.py -t --ignore-dir '$prefix' spam.py eggs 35 trace.py --trackcalls spam.py eggs 36 37Sample use, programmatically 38 import sys 39 40 # create a Trace object, telling it what to ignore, and whether to 41 # do tracing or line-counting or both. 42 tracer = trace.Trace(ignoredirs=[sys.base_prefix, sys.base_exec_prefix,], 43 trace=0, count=1) 44 # run the new command using the given tracer 45 tracer.run('main()') 46 # make a report, placing output in /tmp 47 r = tracer.results() 48 r.write_results(show_missing=True, coverdir="/tmp") 49""" 50__all__ = ['Trace', 'CoverageResults'] 51 52import linecache 53import os 54import sys 55import sysconfig 56import token 57import tokenize 58import inspect 59import gc 60import dis 61import pickle 62from time import monotonic as _time 63 64import threading 65 66PRAGMA_NOCOVER = "#pragma NO COVER" 67 68class _Ignore: 69 def __init__(self, modules=None, dirs=None): 70 self._mods = set() if not modules else set(modules) 71 self._dirs = [] if not dirs else [os.path.normpath(d) 72 for d in dirs] 73 self._ignore = { '<string>': 1 } 74 75 def names(self, filename, modulename): 76 if modulename in self._ignore: 77 return self._ignore[modulename] 78 79 # haven't seen this one before, so see if the module name is 80 # on the ignore list. 81 if modulename in self._mods: # Identical names, so ignore 82 self._ignore[modulename] = 1 83 return 1 84 85 # check if the module is a proper submodule of something on 86 # the ignore list 87 for mod in self._mods: 88 # Need to take some care since ignoring 89 # "cmp" mustn't mean ignoring "cmpcache" but ignoring 90 # "Spam" must also mean ignoring "Spam.Eggs". 91 if modulename.startswith(mod + '.'): 92 self._ignore[modulename] = 1 93 return 1 94 95 # Now check that filename isn't in one of the directories 96 if filename is None: 97 # must be a built-in, so we must ignore 98 self._ignore[modulename] = 1 99 return 1 100 101 # Ignore a file when it contains one of the ignorable paths 102 for d in self._dirs: 103 # The '+ os.sep' is to ensure that d is a parent directory, 104 # as compared to cases like: 105 # d = "/usr/local" 106 # filename = "/usr/local.py" 107 # or 108 # d = "/usr/local.py" 109 # filename = "/usr/local.py" 110 if filename.startswith(d + os.sep): 111 self._ignore[modulename] = 1 112 return 1 113 114 # Tried the different ways, so we don't ignore this module 115 self._ignore[modulename] = 0 116 return 0 117 118def _modname(path): 119 """Return a plausible module name for the patch.""" 120 121 base = os.path.basename(path) 122 filename, ext = os.path.splitext(base) 123 return filename 124 125def _fullmodname(path): 126 """Return a plausible module name for the path.""" 127 128 # If the file 'path' is part of a package, then the filename isn't 129 # enough to uniquely identify it. Try to do the right thing by 130 # looking in sys.path for the longest matching prefix. We'll 131 # assume that the rest is the package name. 132 133 comparepath = os.path.normcase(path) 134 longest = "" 135 for dir in sys.path: 136 dir = os.path.normcase(dir) 137 if comparepath.startswith(dir) and comparepath[len(dir)] == os.sep: 138 if len(dir) > len(longest): 139 longest = dir 140 141 if longest: 142 base = path[len(longest) + 1:] 143 else: 144 base = path 145 # the drive letter is never part of the module name 146 drive, base = os.path.splitdrive(base) 147 base = base.replace(os.sep, ".") 148 if os.altsep: 149 base = base.replace(os.altsep, ".") 150 filename, ext = os.path.splitext(base) 151 return filename.lstrip(".") 152 153class CoverageResults: 154 def __init__(self, counts=None, calledfuncs=None, infile=None, 155 callers=None, outfile=None): 156 self.counts = counts 157 if self.counts is None: 158 self.counts = {} 159 self.counter = self.counts.copy() # map (filename, lineno) to count 160 self.calledfuncs = calledfuncs 161 if self.calledfuncs is None: 162 self.calledfuncs = {} 163 self.calledfuncs = self.calledfuncs.copy() 164 self.callers = callers 165 if self.callers is None: 166 self.callers = {} 167 self.callers = self.callers.copy() 168 self.infile = infile 169 self.outfile = outfile 170 if self.infile: 171 # Try to merge existing counts file. 172 try: 173 with open(self.infile, 'rb') as f: 174 counts, calledfuncs, callers = pickle.load(f) 175 self.update(self.__class__(counts, calledfuncs, callers)) 176 except (OSError, EOFError, ValueError) as err: 177 print(("Skipping counts file %r: %s" 178 % (self.infile, err)), file=sys.stderr) 179 180 def is_ignored_filename(self, filename): 181 """Return True if the filename does not refer to a file 182 we want to have reported. 183 """ 184 return filename.startswith('<') and filename.endswith('>') 185 186 def update(self, other): 187 """Merge in the data from another CoverageResults""" 188 counts = self.counts 189 calledfuncs = self.calledfuncs 190 callers = self.callers 191 other_counts = other.counts 192 other_calledfuncs = other.calledfuncs 193 other_callers = other.callers 194 195 for key in other_counts: 196 counts[key] = counts.get(key, 0) + other_counts[key] 197 198 for key in other_calledfuncs: 199 calledfuncs[key] = 1 200 201 for key in other_callers: 202 callers[key] = 1 203 204 def write_results(self, show_missing=True, summary=False, coverdir=None): 205 """ 206 Write the coverage results. 207 208 :param show_missing: Show lines that had no hits. 209 :param summary: Include coverage summary per module. 210 :param coverdir: If None, the results of each module are placed in its 211 directory, otherwise it is included in the directory 212 specified. 213 """ 214 if self.calledfuncs: 215 print() 216 print("functions called:") 217 calls = self.calledfuncs 218 for filename, modulename, funcname in sorted(calls): 219 print(("filename: %s, modulename: %s, funcname: %s" 220 % (filename, modulename, funcname))) 221 222 if self.callers: 223 print() 224 print("calling relationships:") 225 lastfile = lastcfile = "" 226 for ((pfile, pmod, pfunc), (cfile, cmod, cfunc)) \ 227 in sorted(self.callers): 228 if pfile != lastfile: 229 print() 230 print("***", pfile, "***") 231 lastfile = pfile 232 lastcfile = "" 233 if cfile != pfile and lastcfile != cfile: 234 print(" -->", cfile) 235 lastcfile = cfile 236 print(" %s.%s -> %s.%s" % (pmod, pfunc, cmod, cfunc)) 237 238 # turn the counts data ("(filename, lineno) = count") into something 239 # accessible on a per-file basis 240 per_file = {} 241 for filename, lineno in self.counts: 242 lines_hit = per_file[filename] = per_file.get(filename, {}) 243 lines_hit[lineno] = self.counts[(filename, lineno)] 244 245 # accumulate summary info, if needed 246 sums = {} 247 248 for filename, count in per_file.items(): 249 if self.is_ignored_filename(filename): 250 continue 251 252 if filename.endswith(".pyc"): 253 filename = filename[:-1] 254 255 if coverdir is None: 256 dir = os.path.dirname(os.path.abspath(filename)) 257 modulename = _modname(filename) 258 else: 259 dir = coverdir 260 if not os.path.exists(dir): 261 os.makedirs(dir) 262 modulename = _fullmodname(filename) 263 264 # If desired, get a list of the line numbers which represent 265 # executable content (returned as a dict for better lookup speed) 266 if show_missing: 267 lnotab = _find_executable_linenos(filename) 268 else: 269 lnotab = {} 270 source = linecache.getlines(filename) 271 coverpath = os.path.join(dir, modulename + ".cover") 272 with open(filename, 'rb') as fp: 273 encoding, _ = tokenize.detect_encoding(fp.readline) 274 n_hits, n_lines = self.write_results_file(coverpath, source, 275 lnotab, count, encoding) 276 if summary and n_lines: 277 percent = int(100 * n_hits / n_lines) 278 sums[modulename] = n_lines, percent, modulename, filename 279 280 281 if summary and sums: 282 print("lines cov% module (path)") 283 for m in sorted(sums): 284 n_lines, percent, modulename, filename = sums[m] 285 print("%5d %3d%% %s (%s)" % sums[m]) 286 287 if self.outfile: 288 # try and store counts and module info into self.outfile 289 try: 290 with open(self.outfile, 'wb') as f: 291 pickle.dump((self.counts, self.calledfuncs, self.callers), 292 f, 1) 293 except OSError as err: 294 print("Can't save counts files because %s" % err, file=sys.stderr) 295 296 def write_results_file(self, path, lines, lnotab, lines_hit, encoding=None): 297 """Return a coverage results file in path.""" 298 # ``lnotab`` is a dict of executable lines, or a line number "table" 299 300 try: 301 outfile = open(path, "w", encoding=encoding) 302 except OSError as err: 303 print(("trace: Could not open %r for writing: %s " 304 "- skipping" % (path, err)), file=sys.stderr) 305 return 0, 0 306 307 n_lines = 0 308 n_hits = 0 309 with outfile: 310 for lineno, line in enumerate(lines, 1): 311 # do the blank/comment match to try to mark more lines 312 # (help the reader find stuff that hasn't been covered) 313 if lineno in lines_hit: 314 outfile.write("%5d: " % lines_hit[lineno]) 315 n_hits += 1 316 n_lines += 1 317 elif lineno in lnotab and not PRAGMA_NOCOVER in line: 318 # Highlight never-executed lines, unless the line contains 319 # #pragma: NO COVER 320 outfile.write(">>>>>> ") 321 n_lines += 1 322 else: 323 outfile.write(" ") 324 outfile.write(line.expandtabs(8)) 325 326 return n_hits, n_lines 327 328def _find_lines_from_code(code, strs): 329 """Return dict where keys are lines in the line number table.""" 330 linenos = {} 331 332 for _, lineno in dis.findlinestarts(code): 333 if lineno not in strs: 334 linenos[lineno] = 1 335 336 return linenos 337 338def _find_lines(code, strs): 339 """Return lineno dict for all code objects reachable from code.""" 340 # get all of the lineno information from the code of this scope level 341 linenos = _find_lines_from_code(code, strs) 342 343 # and check the constants for references to other code objects 344 for c in code.co_consts: 345 if inspect.iscode(c): 346 # find another code object, so recurse into it 347 linenos.update(_find_lines(c, strs)) 348 return linenos 349 350def _find_strings(filename, encoding=None): 351 """Return a dict of possible docstring positions. 352 353 The dict maps line numbers to strings. There is an entry for 354 line that contains only a string or a part of a triple-quoted 355 string. 356 """ 357 d = {} 358 # If the first token is a string, then it's the module docstring. 359 # Add this special case so that the test in the loop passes. 360 prev_ttype = token.INDENT 361 with open(filename, encoding=encoding) as f: 362 tok = tokenize.generate_tokens(f.readline) 363 for ttype, tstr, start, end, line in tok: 364 if ttype == token.STRING: 365 if prev_ttype == token.INDENT: 366 sline, scol = start 367 eline, ecol = end 368 for i in range(sline, eline + 1): 369 d[i] = 1 370 prev_ttype = ttype 371 return d 372 373def _find_executable_linenos(filename): 374 """Return dict where keys are line numbers in the line number table.""" 375 try: 376 with tokenize.open(filename) as f: 377 prog = f.read() 378 encoding = f.encoding 379 except OSError as err: 380 print(("Not printing coverage data for %r: %s" 381 % (filename, err)), file=sys.stderr) 382 return {} 383 code = compile(prog, filename, "exec") 384 strs = _find_strings(filename, encoding) 385 return _find_lines(code, strs) 386 387class Trace: 388 def __init__(self, count=1, trace=1, countfuncs=0, countcallers=0, 389 ignoremods=(), ignoredirs=(), infile=None, outfile=None, 390 timing=False): 391 """ 392 @param count true iff it should count number of times each 393 line is executed 394 @param trace true iff it should print out each line that is 395 being counted 396 @param countfuncs true iff it should just output a list of 397 (filename, modulename, funcname,) for functions 398 that were called at least once; This overrides 399 `count' and `trace' 400 @param ignoremods a list of the names of modules to ignore 401 @param ignoredirs a list of the names of directories to ignore 402 all of the (recursive) contents of 403 @param infile file from which to read stored counts to be 404 added into the results 405 @param outfile file in which to write the results 406 @param timing true iff timing information be displayed 407 """ 408 self.infile = infile 409 self.outfile = outfile 410 self.ignore = _Ignore(ignoremods, ignoredirs) 411 self.counts = {} # keys are (filename, linenumber) 412 self.pathtobasename = {} # for memoizing os.path.basename 413 self.donothing = 0 414 self.trace = trace 415 self._calledfuncs = {} 416 self._callers = {} 417 self._caller_cache = {} 418 self.start_time = None 419 if timing: 420 self.start_time = _time() 421 if countcallers: 422 self.globaltrace = self.globaltrace_trackcallers 423 elif countfuncs: 424 self.globaltrace = self.globaltrace_countfuncs 425 elif trace and count: 426 self.globaltrace = self.globaltrace_lt 427 self.localtrace = self.localtrace_trace_and_count 428 elif trace: 429 self.globaltrace = self.globaltrace_lt 430 self.localtrace = self.localtrace_trace 431 elif count: 432 self.globaltrace = self.globaltrace_lt 433 self.localtrace = self.localtrace_count 434 else: 435 # Ahem -- do nothing? Okay. 436 self.donothing = 1 437 438 def run(self, cmd): 439 import __main__ 440 dict = __main__.__dict__ 441 self.runctx(cmd, dict, dict) 442 443 def runctx(self, cmd, globals=None, locals=None): 444 if globals is None: globals = {} 445 if locals is None: locals = {} 446 if not self.donothing: 447 threading.settrace(self.globaltrace) 448 sys.settrace(self.globaltrace) 449 try: 450 exec(cmd, globals, locals) 451 finally: 452 if not self.donothing: 453 sys.settrace(None) 454 threading.settrace(None) 455 456 def runfunc(*args, **kw): 457 if len(args) >= 2: 458 self, func, *args = args 459 elif not args: 460 raise TypeError("descriptor 'runfunc' of 'Trace' object " 461 "needs an argument") 462 elif 'func' in kw: 463 func = kw.pop('func') 464 self, *args = args 465 import warnings 466 warnings.warn("Passing 'func' as keyword argument is deprecated", 467 DeprecationWarning, stacklevel=2) 468 else: 469 raise TypeError('runfunc expected at least 1 positional argument, ' 470 'got %d' % (len(args)-1)) 471 472 result = None 473 if not self.donothing: 474 sys.settrace(self.globaltrace) 475 try: 476 result = func(*args, **kw) 477 finally: 478 if not self.donothing: 479 sys.settrace(None) 480 return result 481 runfunc.__text_signature__ = '($self, func, /, *args, **kw)' 482 483 def file_module_function_of(self, frame): 484 code = frame.f_code 485 filename = code.co_filename 486 if filename: 487 modulename = _modname(filename) 488 else: 489 modulename = None 490 491 funcname = code.co_name 492 clsname = None 493 if code in self._caller_cache: 494 if self._caller_cache[code] is not None: 495 clsname = self._caller_cache[code] 496 else: 497 self._caller_cache[code] = None 498 ## use of gc.get_referrers() was suggested by Michael Hudson 499 # all functions which refer to this code object 500 funcs = [f for f in gc.get_referrers(code) 501 if inspect.isfunction(f)] 502 # require len(func) == 1 to avoid ambiguity caused by calls to 503 # new.function(): "In the face of ambiguity, refuse the 504 # temptation to guess." 505 if len(funcs) == 1: 506 dicts = [d for d in gc.get_referrers(funcs[0]) 507 if isinstance(d, dict)] 508 if len(dicts) == 1: 509 classes = [c for c in gc.get_referrers(dicts[0]) 510 if hasattr(c, "__bases__")] 511 if len(classes) == 1: 512 # ditto for new.classobj() 513 clsname = classes[0].__name__ 514 # cache the result - assumption is that new.* is 515 # not called later to disturb this relationship 516 # _caller_cache could be flushed if functions in 517 # the new module get called. 518 self._caller_cache[code] = clsname 519 if clsname is not None: 520 funcname = "%s.%s" % (clsname, funcname) 521 522 return filename, modulename, funcname 523 524 def globaltrace_trackcallers(self, frame, why, arg): 525 """Handler for call events. 526 527 Adds information about who called who to the self._callers dict. 528 """ 529 if why == 'call': 530 # XXX Should do a better job of identifying methods 531 this_func = self.file_module_function_of(frame) 532 parent_func = self.file_module_function_of(frame.f_back) 533 self._callers[(parent_func, this_func)] = 1 534 535 def globaltrace_countfuncs(self, frame, why, arg): 536 """Handler for call events. 537 538 Adds (filename, modulename, funcname) to the self._calledfuncs dict. 539 """ 540 if why == 'call': 541 this_func = self.file_module_function_of(frame) 542 self._calledfuncs[this_func] = 1 543 544 def globaltrace_lt(self, frame, why, arg): 545 """Handler for call events. 546 547 If the code block being entered is to be ignored, returns `None', 548 else returns self.localtrace. 549 """ 550 if why == 'call': 551 code = frame.f_code 552 filename = frame.f_globals.get('__file__', None) 553 if filename: 554 # XXX _modname() doesn't work right for packages, so 555 # the ignore support won't work right for packages 556 modulename = _modname(filename) 557 if modulename is not None: 558 ignore_it = self.ignore.names(filename, modulename) 559 if not ignore_it: 560 if self.trace: 561 print((" --- modulename: %s, funcname: %s" 562 % (modulename, code.co_name))) 563 return self.localtrace 564 else: 565 return None 566 567 def localtrace_trace_and_count(self, frame, why, arg): 568 if why == "line": 569 # record the file name and line number of every trace 570 filename = frame.f_code.co_filename 571 lineno = frame.f_lineno 572 key = filename, lineno 573 self.counts[key] = self.counts.get(key, 0) + 1 574 575 if self.start_time: 576 print('%.2f' % (_time() - self.start_time), end=' ') 577 bname = os.path.basename(filename) 578 print("%s(%d): %s" % (bname, lineno, 579 linecache.getline(filename, lineno)), end='') 580 return self.localtrace 581 582 def localtrace_trace(self, frame, why, arg): 583 if why == "line": 584 # record the file name and line number of every trace 585 filename = frame.f_code.co_filename 586 lineno = frame.f_lineno 587 588 if self.start_time: 589 print('%.2f' % (_time() - self.start_time), end=' ') 590 bname = os.path.basename(filename) 591 print("%s(%d): %s" % (bname, lineno, 592 linecache.getline(filename, lineno)), end='') 593 return self.localtrace 594 595 def localtrace_count(self, frame, why, arg): 596 if why == "line": 597 filename = frame.f_code.co_filename 598 lineno = frame.f_lineno 599 key = filename, lineno 600 self.counts[key] = self.counts.get(key, 0) + 1 601 return self.localtrace 602 603 def results(self): 604 return CoverageResults(self.counts, infile=self.infile, 605 outfile=self.outfile, 606 calledfuncs=self._calledfuncs, 607 callers=self._callers) 608 609def main(): 610 import argparse 611 612 parser = argparse.ArgumentParser() 613 parser.add_argument('--version', action='version', version='trace 2.0') 614 615 grp = parser.add_argument_group('Main options', 616 'One of these (or --report) must be given') 617 618 grp.add_argument('-c', '--count', action='store_true', 619 help='Count the number of times each line is executed and write ' 620 'the counts to <module>.cover for each module executed, in ' 621 'the module\'s directory. See also --coverdir, --file, ' 622 '--no-report below.') 623 grp.add_argument('-t', '--trace', action='store_true', 624 help='Print each line to sys.stdout before it is executed') 625 grp.add_argument('-l', '--listfuncs', action='store_true', 626 help='Keep track of which functions are executed at least once ' 627 'and write the results to sys.stdout after the program exits. ' 628 'Cannot be specified alongside --trace or --count.') 629 grp.add_argument('-T', '--trackcalls', action='store_true', 630 help='Keep track of caller/called pairs and write the results to ' 631 'sys.stdout after the program exits.') 632 633 grp = parser.add_argument_group('Modifiers') 634 635 _grp = grp.add_mutually_exclusive_group() 636 _grp.add_argument('-r', '--report', action='store_true', 637 help='Generate a report from a counts file; does not execute any ' 638 'code. --file must specify the results file to read, which ' 639 'must have been created in a previous run with --count ' 640 '--file=FILE') 641 _grp.add_argument('-R', '--no-report', action='store_true', 642 help='Do not generate the coverage report files. ' 643 'Useful if you want to accumulate over several runs.') 644 645 grp.add_argument('-f', '--file', 646 help='File to accumulate counts over several runs') 647 grp.add_argument('-C', '--coverdir', 648 help='Directory where the report files go. The coverage report ' 649 'for <package>.<module> will be written to file ' 650 '<dir>/<package>/<module>.cover') 651 grp.add_argument('-m', '--missing', action='store_true', 652 help='Annotate executable lines that were not executed with ' 653 '">>>>>> "') 654 grp.add_argument('-s', '--summary', action='store_true', 655 help='Write a brief summary for each file to sys.stdout. ' 656 'Can only be used with --count or --report') 657 grp.add_argument('-g', '--timing', action='store_true', 658 help='Prefix each line with the time since the program started. ' 659 'Only used while tracing') 660 661 grp = parser.add_argument_group('Filters', 662 'Can be specified multiple times') 663 grp.add_argument('--ignore-module', action='append', default=[], 664 help='Ignore the given module(s) and its submodules ' 665 '(if it is a package). Accepts comma separated list of ' 666 'module names.') 667 grp.add_argument('--ignore-dir', action='append', default=[], 668 help='Ignore files in the given directory ' 669 '(multiple directories can be joined by os.pathsep).') 670 671 parser.add_argument('--module', action='store_true', default=False, 672 help='Trace a module. ') 673 parser.add_argument('progname', nargs='?', 674 help='file to run as main program') 675 parser.add_argument('arguments', nargs=argparse.REMAINDER, 676 help='arguments to the program') 677 678 opts = parser.parse_args() 679 680 if opts.ignore_dir: 681 _prefix = sysconfig.get_path("stdlib") 682 _exec_prefix = sysconfig.get_path("platstdlib") 683 684 def parse_ignore_dir(s): 685 s = os.path.expanduser(os.path.expandvars(s)) 686 s = s.replace('$prefix', _prefix).replace('$exec_prefix', _exec_prefix) 687 return os.path.normpath(s) 688 689 opts.ignore_module = [mod.strip() 690 for i in opts.ignore_module for mod in i.split(',')] 691 opts.ignore_dir = [parse_ignore_dir(s) 692 for i in opts.ignore_dir for s in i.split(os.pathsep)] 693 694 if opts.report: 695 if not opts.file: 696 parser.error('-r/--report requires -f/--file') 697 results = CoverageResults(infile=opts.file, outfile=opts.file) 698 return results.write_results(opts.missing, opts.summary, opts.coverdir) 699 700 if not any([opts.trace, opts.count, opts.listfuncs, opts.trackcalls]): 701 parser.error('must specify one of --trace, --count, --report, ' 702 '--listfuncs, or --trackcalls') 703 704 if opts.listfuncs and (opts.count or opts.trace): 705 parser.error('cannot specify both --listfuncs and (--trace or --count)') 706 707 if opts.summary and not opts.count: 708 parser.error('--summary can only be used with --count or --report') 709 710 if opts.progname is None: 711 parser.error('progname is missing: required with the main options') 712 713 t = Trace(opts.count, opts.trace, countfuncs=opts.listfuncs, 714 countcallers=opts.trackcalls, ignoremods=opts.ignore_module, 715 ignoredirs=opts.ignore_dir, infile=opts.file, 716 outfile=opts.file, timing=opts.timing) 717 try: 718 if opts.module: 719 import runpy 720 module_name = opts.progname 721 mod_name, mod_spec, code = runpy._get_module_details(module_name) 722 sys.argv = [code.co_filename, *opts.arguments] 723 globs = { 724 '__name__': '__main__', 725 '__file__': code.co_filename, 726 '__package__': mod_spec.parent, 727 '__loader__': mod_spec.loader, 728 '__spec__': mod_spec, 729 '__cached__': None, 730 } 731 else: 732 sys.argv = [opts.progname, *opts.arguments] 733 sys.path[0] = os.path.dirname(opts.progname) 734 735 with open(opts.progname, 'rb') as fp: 736 code = compile(fp.read(), opts.progname, 'exec') 737 # try to emulate __main__ namespace as much as possible 738 globs = { 739 '__file__': opts.progname, 740 '__name__': '__main__', 741 '__package__': None, 742 '__cached__': None, 743 } 744 t.runctx(code, globs, globs) 745 except OSError as err: 746 sys.exit("Cannot run file %r because: %s" % (sys.argv[0], err)) 747 except SystemExit: 748 pass 749 750 results = t.results() 751 752 if not opts.no_report: 753 results.write_results(opts.missing, opts.summary, opts.coverdir) 754 755if __name__=='__main__': 756 main() 757