reivilibre/per4m-synapse
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: giltracer can import before tracing

Browse Source
This commit is contained in:
Maarten A. Breddels 2020-12-15 14:57:53 +01:00
parent 06d2ea5d30
commit b4488bff82
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
per4m/giltracer.py
View File

@ -93,6 +93,7 @@ def main(argv=sys.argv):
formatter_class=argparse.RawDescriptionHelpFormatter, formatter_class=argparse.RawDescriptionHelpFormatter,
usage=usage) usage=usage)
parser.add_argument('--module', '-m') parser.add_argument('--module', '-m')
parser.add_argument('--import', dest="import_", help="Comma seperated list of modules to import before tracing (cleans up tracing output)")
parser.add_argument('--verbose', '-v', action='count', default=1) parser.add_argument('--verbose', '-v', action='count', default=1)
parser.add_argument('--quiet', '-q', action='count', default=0) parser.add_argument('--quiet', '-q', action='count', default=0)
parser.add_argument('--output', '-o', dest="output", default='result.html', help="Output filename (default %(default)s)") parser.add_argument('--output', '-o', dest="output", default='result.html', help="Output filename (default %(default)s)")
@ -100,6 +101,12 @@ def main(argv=sys.argv):
args = parser.parse_args(argv[1:]) args = parser.parse_args(argv[1:])
verbose = args.verbose - args.quiet verbose = args.verbose - args.quiet
if args.import_:
for module in args.import_.split(','):
if verbose >= 2:
print(f'importing {module}')
__import__(module)
with GilTracer(verbose=verbose) as gt: with GilTracer(verbose=verbose) as gt:
with viztracer.VizTracer(output_file="viztracer.json", verbose=verbose): with viztracer.VizTracer(output_file="viztracer.json", verbose=verbose):
if args.module: if args.module: