allow save_py_cahces to select what it does

ajslater · Jan 26, 2022 · 2249bfc · 2249bfc
1 parent 3aa0d5a
commit 2249bfc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -39,7 +39,7 @@ services:
     container_name: codex-save-cache
     volumes:
       - ./cache/packages:/app/cache/packages
-    command: ./python_cacher/save_py_caches.py prune
+    command: ./python_cacher/save_py_caches.py pip poetry prune
   codex-frontend-lint:
     <<: *codex-dist-builder
     container_name: codex-frontent-lint
@@ -92,7 +92,7 @@ services:
     container_name: codex-save-final-cache
     volumes:
       - ./cache/packages:/app/cache/packages
-    command: ./python_cacher/save_py_caches.py
+    command: ./python_cacher/save_py_caches.py pip
   codex:
     env_file:
       - .env

diff --git a/python_cacher/README.md b/python_cacher/README.md
@@ -0,0 +1,3 @@
+# python\_cacher
+
+Some scripts for managing the circleci python package & wheels caches and linking wheels to them for use by `pip find-links`
diff --git a/python_cacher/save_py_caches.py b/python_cacher/save_py_caches.py
@@ -90,10 +90,12 @@ def main(args):
     arch = run(("uname", "-m"))[0]
     arch_cache_path = CACHE_PATH / arch
     arch_cache_path.mkdir(parents=True, exist_ok=True)
-    if "prune" in args:
-        prune_poetry_artifacts()
-    save_cache(POETRY_CACHE_PATH, arch_cache_path)
-    save_cache(PIP_CACHE_PATH, arch_cache_path)
+    if "poetry" in args:
+        if "prune" in args:
+            prune_poetry_artifacts()
+        save_cache(POETRY_CACHE_PATH, arch_cache_path)
+    if "pip" in args:
+        save_cache(PIP_CACHE_PATH, arch_cache_path)
 
 
 if __name__ == "__main__":