Allow scripts to pass arguments to script_main()

The default behaviour is to use sys.argv.
apache · Mar 14, 2024 · 9d41c48 · 9d41c48
1 parent 0e9f4a4
commit 9d41c48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hunter/main.py b/hunter/main.py
@@ -471,7 +471,7 @@ def main():
     script_main(conf)
 
 
-def script_main(conf: Config):
+def script_main(conf: Config, args: List[str] = None):
     logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.INFO)
 
     parser = argparse.ArgumentParser(description="Hunts performance regressions in Fallout results")
@@ -540,7 +540,7 @@ def script_main(conf: Config):
     )
 
     try:
-        args = parser.parse_args()
+        args = parser.parse_args(args=args)
         hunter = Hunter(conf)
 
         if args.command == "list-groups":