From b4488bff8228b718ac7022748aaa045de9a04727 Mon Sep 17 00:00:00 2001 From: "Maarten A. Breddels" Date: Tue, 15 Dec 2020 14:57:53 +0100 Subject: [PATCH] feat: giltracer can import before tracing --- per4m/giltracer.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/per4m/giltracer.py b/per4m/giltracer.py index b8675e1..2589a40 100644 --- a/per4m/giltracer.py +++ b/per4m/giltracer.py @@ -93,6 +93,7 @@ def main(argv=sys.argv): formatter_class=argparse.RawDescriptionHelpFormatter, usage=usage) 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('--quiet', '-q', action='count', default=0) 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:]) 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 viztracer.VizTracer(output_file="viztracer.json", verbose=verbose): if args.module: