feat: giltracer can import before tracing
This commit is contained in:
parent
06d2ea5d30
commit
b4488bff82
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user