Temporarily creating a build directory

LouisStAmour · LouisStAmour · commit df6ea4dd98e3 · 2021-03-13T00:59:27.000-05:00
diff --git a/python/pip_install/extract_wheels/__init__.py b/python/pip_install/extract_wheels/__init__.py
@@ -12,6 +12,7 @@
 import subprocess
 import sys
 import json
+import tempfile
 
 from python.pip_install.extract_wheels.lib import bazel, requirements
 
@@ -83,12 +84,24 @@ def main() -> None:
     # Pip is run with the working directory changed to the folder containing the requirements.txt file, to allow for
     # relative requirements to be correctly resolved. The --wheel-dir is therefore required to be repointed back to the
     # current calling working directory (the repo root in .../external/name), where the wheel files should be written to
-    pip_args = [sys.executable, "-m", "pip", "--isolated", "wheel", "-r", args.requirements, "--wheel-dir", os.getcwd()]
-    if args.extra_pip_args:
-        pip_args += json.loads(args.extra_pip_args)["args"]
-
-    # Assumes any errors are logged by pip so do nothing. This command will fail if pip fails
-    subprocess.run(pip_args, check=True, cwd=str(pathlib.Path(args.requirements).parent.resolve()))
+    with tempfile.TemporaryDirectory(dir = os.environ['TEST_TMPDIR']) as build_dir:
+        pip_args = [
+            sys.executable, 
+            "-m", "pip", 
+            "--isolated", 
+            "wheel", 
+            "-r", args.requirements, 
+            "--wheel-dir", os.getcwd(), 
+            "--build", build_dir,
+        ]
+        if args.extra_pip_args:
+            pip_args += json.loads(args.extra_pip_args)["args"]
+        
+        # Assumes any errors are logged by pip so do nothing. This command will fail if pip fails
+        subprocess.run(pip_args, 
+                       check=True, 
+                       cwd=str(pathlib.Path(args.requirements).parent.resolve()),
+                       env={"TMPDIR": build_dir})
 
     extras = requirements.parse_extras(args.requirements)
 
