Enable automatic cleanup for local store (#175)

Author: kylebarron

obstore/python/obstore/store/__init__.pyi

@@ -26,7 +26,17 @@ class LocalStore:
     store = LocalStore(prefix=Path("."))
     ```
     """
-    def __init__(self, prefix: str | Path | None = None) -> None: ...
+    def __init__(
+        self, prefix: str | Path | None = None, *, automatic_cleanup: bool = False
+    ) -> None:
+        """Create a new LocalStore.
+
+        Args:
+            prefix: Use the specified prefix applied to all paths. Defaults to `None`.
+
+        Keyword Args:
+            automatic_cleanup: if `True`, enables automatic cleanup of empty directories when deleting files. Defaults to False.
+        """
     def __repr__(self) -> str: ...
     @classmethod
     def from_url(cls, url: str) -> LocalStore:

pyo3-object_store/src/local.rs

@@ -29,13 +29,17 @@ impl PyLocalStore {
 #[pymethods]
 impl PyLocalStore {
     #[new]
-    #[pyo3(signature = (prefix = None))]
-    fn py_new(prefix: Option<std::path::PathBuf>) -> PyObjectStoreResult<Self> {
+    #[pyo3(signature = (prefix = None, *, automatic_cleanup=false))]
+    fn py_new(
+        prefix: Option<std::path::PathBuf>,
+        automatic_cleanup: bool,
+    ) -> PyObjectStoreResult<Self> {
         let fs = if let Some(prefix) = prefix {
             LocalFileSystem::new_with_prefix(prefix)?
         } else {
             LocalFileSystem::new()
         };
+        let fs = fs.with_automatic_cleanup(automatic_cleanup);
         Ok(Self(Arc::new(fs)))
     }