modified pgo_rust so that it can be invoked as a module

This will be used by an upcoming change to rust_uprev to
automatically generate and upload profile data for new
Rust versions.

BUG=b:271017864
TEST=use it to upgrade Rust

Change-Id: I9e382e0bc3317bb1de9d4501391029f457afae82
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/third_party/toolchain-utils/+/4775779
Tested-by: Bob Haarman <inglorion@chromium.org>
Reviewed-by: George Burgess <gbiv@chromium.org>
Commit-Queue: Bob Haarman <inglorion@chromium.org>

1 files changed, 4 insertions, 4 deletions

diff --git a/pgo_tools_rust/pgo_rust.py b/pgo_tools_rust/pgo_rust.py
index 07cd8ac3..fc693169 100755
--- a/pgo_tools_rust/pgo_rust.py
+++ b/pgo_tools_rust/pgo_rust.py
@@ -544,13 +544,13 @@ def upload_profdata(args):
     )
 
 
-def main():
+def main(argv: List[str]) -> int:
     logging.basicConfig(
         stream=sys.stdout, level=logging.NOTSET, format="%(message)s"
     )
 
     parser = argparse.ArgumentParser(
-        prog=sys.argv[0],
+        prog=argv[0],
         description=__doc__,
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
@@ -652,7 +652,7 @@ def main():
         "rustc versions",
     )
 
-    args = parser.parse_args()
+    args = parser.parse_args(argv[1:])
 
     (LOCAL_BASE / "crates").mkdir(parents=True, exist_ok=True)
     (LOCAL_BASE / "llvm-profraw").mkdir(parents=True, exist_ok=True)
@@ -665,4 +665,4 @@ def main():
 
 
 if __name__ == "__main__":
-    sys.exit(main())
+    sys.exit(main(sys.argv))