Merge PR #11302: Add --fuzz, --real, --user to timing scripts

Reviewed-by: SkySkimmer
Ack-by: Zimmi48
Reviewed-by: ejgallego
Ack-by: jfehrle

1 files changed, 8 insertions, 16 deletions

diff --git a/tools/make-one-time-file.py b/tools/make-one-time-file.py
index ad0a04ab07..3df7d7e584 100755
--- a/tools/make-one-time-file.py
+++ b/tools/make-one-time-file.py
@@ -3,19 +3,11 @@ import sys
 from TimeFileMaker import *
 
 if __name__ == '__main__':
-    USAGE = 'Usage: %s FILE_NAME [OUTPUT_FILE_NAME ..]' % sys.argv[0]
-    HELP_STRING = r'''Formats timing information from the output of `make TIMED=1` into a sorted table.
-
-The input is expected to contain lines in the format:
-FILE_NAME (...user: NUMBER_IN_SECONDS...)
-'''
-    if len(sys.argv) < 2 or '--help' in sys.argv[1:] or '-h' in sys.argv[1:]:
-        print(USAGE)
-        if '--help' in sys.argv[1:] or '-h' in sys.argv[1:]:
-            print(HELP_STRING)
-            if len(sys.argv) == 2: sys.exit(0)
-        sys.exit(1)
-    else:
-        times_dict = get_times(sys.argv[1])
-        table = make_table_string(times_dict)
-        print_or_write_table(table, sys.argv[2:])
+    parser = argparse.ArgumentParser(description=r'''Formats timing information from the output of `make TIMED=1` into a sorted table.''')
+    add_real(parser)
+    add_file_name(parser)
+    add_output_file_name(parser)
+    args = parser.parse_args()
+    times_dict = get_times(args.FILE_NAME, use_real=args.real)
+    table = make_table_string(times_dict)
+    print_or_write_table(table, args.OUTPUT_FILE_NAME)