Add extra_pip_args to pip_repository (bazel-contrib#41)

gergelyfabian · web-flow · commit a83aa9b45abc · 2020-07-09T08:42:19.000+10:00
diff --git a/defs.bzl b/defs.bzl
@@ -34,6 +34,12 @@ def _pip_repository_impl(rctx):
         "@%s" % rctx.attr.name,
     ]
 
+    if rctx.attr.extra_pip_args:
+        args += [
+            "--extra_pip_args",
+            "\"" + " ".join(rctx.attr.extra_pip_args) + "\"",
+        ]
+
     result = rctx.execute(
         args,
         environment = {
@@ -62,6 +68,9 @@ python_interpreter.
         # 600 is documented as default here: https://docs.bazel.build/versions/master/skylark/lib/repository_ctx.html#execute
         "timeout": attr.int(default = 600),
         "quiet": attr.bool(default = True),
+        "extra_pip_args": attr.string_list(
+            doc = "Extra arguments to pass on to pip. Must not contain spaces.",
+        ),
     },
     implementation = _pip_repository_impl,
 )
diff --git a/extract_wheels/__init__.py b/extract_wheels/__init__.py
@@ -63,9 +63,13 @@ def main() -> None:
         required=True,
         help="The external repo name to install dependencies. In the format '@{REPO_NAME}'",
     )
+    parser.add_argument('--extra_pip_args', action='store',
+                        help=('Extra arguments to pass down to pip.'))
     args = parser.parse_args()
 
     pip_args = [sys.executable, "-m", "pip", "wheel", "-r", args.requirements]
+    if args.extra_pip_args:
+        pip_args += args.extra_pip_args.strip("\"").split()
     # Assumes any errors are logged by pip so do nothing. This command will fail if pip fails
     subprocess.run(pip_args, check=True)
 
