Remove the temporary directory containing cacert.pem when done

AustinSchuh · brandjon · commit df29268c2951 · 2019-08-01T18:41:39.000-04:00
Our verification machines were filling up /tmp/ with this temporary
folder.  This patch removes it when piptool exits.
diff --git a/rules_python/piptool.py b/rules_python/piptool.py
@@ -73,7 +73,9 @@ def extract_packages(package_names):
 def pip_main(argv):
     # Extract the certificates from the PAR following the example of get-pip.py
     # https://github.com/pypa/get-pip/blob/430ba37776ae2ad89/template.py#L164-L168
-    cert_path = os.path.join(tempfile.mkdtemp(), "cacert.pem")
+    cert_tmpdir = tempfile.mkdtemp()
+    cert_path = os.path.join(cert_tmpdir, "cacert.pem")
+    atexit.register(lambda: shutil.rmtree(cert_tmpdir, ignore_errors=True))
     with open(cert_path, "wb") as cert:
       cert.write(pkgutil.get_data("pip._vendor.requests", "cacert.pem"))
     argv = ["--disable-pip-version-check", "--cert", cert_path] + argv
