triage: consistent {url: container} mapping

Author: olupton

.github/triage/jax_toolbox_triage/pyxis.py

@@ -1,9 +1,15 @@
+import hashlib
 import logging
 import pathlib
 import secrets
 import subprocess
 import typing
 
+# Used to make sure the {url: name} mapping is consistent within a process, but that
+# names are not re-used by different invocations of the triage tool. Using a consistent
+# mapping is a simple way of avoiding re-creating the container multiple times, e.g.
+# for .exists()
+_process_token = secrets.token_bytes()
 
 class PyxisContainer:
     def __init__(
@@ -16,7 +22,7 @@ def __init__(
         self._logger = logger
         mount_str = ",".join(map(lambda t: f"{t[0]}:{t[1]}", mounts))
         self._mount_args = [f"--container-mounts={mount_str}"] if mount_str else []
-        self._name = secrets.token_urlsafe()
+        self._name = hashlib.sha256(url.encode() + _process_token).hexdigest()
         self._url = url
 
     def __enter__(self):