Support specifying hooks in pyperformance

doc/usage.rst

@@ -104,6 +104,7 @@ Usage::
                        [--append FILENAME] [--manifest MANIFEST]
                        [--timeout TIMEOUT] [-b BM_LIST]
                        [--inherit-environ VAR_LIST] [-p PYTHON]
+                       [--hook HOOK]
 
 options::
 
@@ -146,6 +147,9 @@ options::
                         Use the same number of loops as a previous run
                         (i.e., don't recalibrate). Should be a path to a
                         .json file from a previous run.
+  --hook HOOK
+                        Apply the given pyperf hook when running the
+                        benchmarks.
 
 show
 ----

pyperformance/cli.py

@@ -19,6 +19,8 @@
     cmd_compare,
 )
 
+from pyperf import _hooks
+
 
 def comma_separated(values):
     values = [value.strip() for value in values.split(',')]
@@ -93,6 +95,12 @@ def parse_args():
                      help="Specify a timeout in seconds for a single "
                      "benchmark run (default: disabled)",
                      type=check_positive)
+    hook_names = list(_hooks.get_hook_names())
+    cmd.add_argument("--hook",
+                     action="append",
+                     choices=hook_names,
+                     metavar=f"{', '.join(x for x in hook_names if not x.startswith('_'))}",
+                     help="Apply the given pyperf hook(s) when running each benchmark")
     filter_opts(cmd)
 
     # show

pyperformance/run.py

@@ -242,5 +242,8 @@ def get_pyperf_opts(options):
         opts.append('--min-time=%s' % options.min_time)
     if options.timeout:
         opts.append('--timeout=%s' % options.timeout)
+    if options.hook:
+        for hook in options.hook:
+            opts.append('--hook=%s' % hook)
 
     return opts