checkpointer 2.14.0__py3-none-any.whl → 2.14.1__py3-none-any.whl

checkpointer/storages/memory_storage.py

@@ -35,3 +35,9 @@ class MemoryStorage(Storage):
           for call_hash, (date, _) in list(calldict.items()):
             if self.expired_dt(date):
               del calldict[call_hash]
+
+  def clear(self):
+    fn_path = self.fn_dir().parent
+    for key in list(item_map.keys()):
+      if key.parent == fn_path:
+        del item_map[key]

checkpointer/storages/pickle_storage.py

@@ -7,6 +7,17 @@ from .storage import Storage
 def filedate(path: Path) -> datetime:
   return datetime.fromtimestamp(path.stat().st_mtime)
 
+def empty_dirs(path: Path):
+  nonempty_count = 0
+  for child in path.iterdir():
+    nonempty_count += 1
+    if child.is_dir():
+      for grand_child in empty_dirs(child):
+        yield grand_child
+        nonempty_count -= child == grand_child
+  if nonempty_count == 0:
+    yield path
+
 class PickleStorage(Storage):
   def get_path(self, call_hash: str):
     return self.fn_dir() / f"{call_hash[:2]}/{call_hash[2:]}.pkl"
@@ -35,7 +46,7 @@ class PickleStorage(Storage):
   def cleanup(self, invalidated=True, expired=True):
     version_path = self.fn_dir()
     fn_path = version_path.parent
-    if invalidated:
+    if invalidated and fn_path.exists():
       old_dirs = [path for path in fn_path.iterdir() if path.is_dir() and path != version_path]
       for path in old_dirs:
         shutil.rmtree(path)
@@ -47,3 +58,13 @@ class PickleStorage(Storage):
           count += 1
           pkl_path.unlink(missing_ok=True)
       print(f"Removed {count} expired checkpoints for {self.cached_fn.__qualname__}")
+    if fn_path.exists():
+      for file in fn_path.glob("**/.DS_Store"):
+        file.unlink()
+      for directory in empty_dirs(fn_path):
+        directory.rmdir()
+
+  def clear(self):
+    fn_path = self.fn_dir().parent
+    if fn_path.exists():
+      shutil.rmtree(fn_path)

checkpointer/storages/storage.py

@@ -44,4 +44,6 @@ class Storage:
 
   def delete(self, call_hash: str) -> None: ...
 
-  def cleanup(self, invalidated=True, expired=True): ...
+  def cleanup(self, invalidated=True, expired=True) -> None: ...
+
+  def clear(self) -> None: ...

{checkpointer-2.14.0.dist-info → checkpointer-2.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: checkpointer
-Version: 2.14.0
+Version: 2.14.1
 Summary: checkpointer adds code-aware caching to Python functions, maintaining correctness and speeding up execution as your code changes.
 Project-URL: Repository, https://github.com/Reddan/checkpointer.git
 Author: Hampus Hallman

{checkpointer-2.14.0.dist-info → checkpointer-2.14.1.dist-info}/RECORD

@@ -8,11 +8,11 @@ checkpointer/print_checkpoint.py,sha256=uUQ493fJCaB4nhp4Ox60govSCiBTIPbBX15zt2Qi
 checkpointer/types.py,sha256=GFqbGACdDxzQX3bb2LmF9UxQVWOEisGvdtobnqCBAOA,1129
 checkpointer/utils.py,sha256=CF28pGytCWV4URhRDFaIDawX8qmCAAByQp0ZkEwrA6c,3014
 checkpointer/storages/__init__.py,sha256=p-r4YrPXn505_S3qLrSXHSlsEtb13w_DFnCt9IiUomk,296
-checkpointer/storages/memory_storage.py,sha256=f242kLEsB4TcqEkEBmJGIewJE9mvv9aCAZ0V9pPB2EQ,1110
-checkpointer/storages/pickle_storage.py,sha256=burbgX-ICPLNMAF9iU3HRZroxJibKfmgJLqTCm6YbbI,1649
-checkpointer/storages/storage.py,sha256=PgpwW_VLkiwD361Kjsz8aoq5FHZLBTqF43yD2mw9dTI,1326
-checkpointer-2.14.0.dist-info/METADATA,sha256=MSJ--2X9N_Ug8sd9VhffONSzhIPA_njmzJhCe3fa_l8,11215
-checkpointer-2.14.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-checkpointer-2.14.0.dist-info/licenses/ATTRIBUTION.md,sha256=WF6L7-sD4s9t9ytVJOhjhpDoZ6TrWpqE3_bMdDIeJxI,1078
-checkpointer-2.14.0.dist-info/licenses/LICENSE,sha256=drXs6vIb7uW49r70UuMz2A1VtOCl626kiTbcmrar1Xo,1072
-checkpointer-2.14.0.dist-info/RECORD,,
+checkpointer/storages/memory_storage.py,sha256=1YHZU-WSqvLA_0og53UZY141ziDgnaBub2mDjDaeAj8,1261
+checkpointer/storages/pickle_storage.py,sha256=PwFbCqYHjR3jiWuTnwxCU9cui7CINvq0wXIOKtf8dOk,2230
+checkpointer/storages/storage.py,sha256=JVxdq1DMhbq83bvflvIuW7okE1CQCa7poQPsj3x0ACg,1366
+checkpointer-2.14.1.dist-info/METADATA,sha256=KupseJCcWRNym_Ckit_F-oGLPIgqUsyKJvjl6ekEDZU,11215
+checkpointer-2.14.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+checkpointer-2.14.1.dist-info/licenses/ATTRIBUTION.md,sha256=WF6L7-sD4s9t9ytVJOhjhpDoZ6TrWpqE3_bMdDIeJxI,1078
+checkpointer-2.14.1.dist-info/licenses/LICENSE,sha256=drXs6vIb7uW49r70UuMz2A1VtOCl626kiTbcmrar1Xo,1072
+checkpointer-2.14.1.dist-info/RECORD,,