Add a timeout attribute to pip_import rules.

ChrisCummins · brandjon · commit 6915c78b8baf · 2020-03-05T14:36:50.000-05:00
The combination of a large requirements.txt file and a slow network can
cause the default timeout for repository_ctx.execute() to be reached.
This patch exposes that timeout to the pip_import() caller so they can
manually specify a longer timeout. E.g.

    pip_import(
        name = "requirements",
        requirements = "//long_list_of_requirements.txt",
        timeout = 3600,
    )
diff --git a/python/pip.bzl b/python/pip.bzl
@@ -43,7 +43,7 @@ def _pip_import_impl(repository_ctx):
         ]
 
     # To see the output, pass: quiet=False
-    result = repository_ctx.execute(args)
+    result = repository_ctx.execute(args, timeout=repository_ctx.attr.timeout)
 
     if result.return_code:
         fail("pip_import failed: %s (%s)" % (result.stdout, result.stderr))
@@ -62,6 +62,10 @@ wheels.
             allow_single_file = True,
             doc = "The label of the requirements.txt file.",
         ),
+        "timeout": attr.int(
+            default = 600,
+            doc = "Timeout (in seconds) for repository fetch."
+        ),
         "_script": attr.label(
             executable = True,
             default = Label("//tools:piptool.par"),
