Feat: improves delete_dir for s3fs-backed FsspecStore

src/zarr/storage/_fsspec.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import warnings
+import inspect
 from typing import TYPE_CHECKING, Any
 
 from zarr.abc.store import ByteRangeRequest, Store
@@ -261,6 +262,24 @@ async def delete(self, key: str) -> None:
         except self.allowed_exceptions:
             pass
 
+    async def delete_dir(self, prefix: str) -> None:
+        # docstring inherited
+        if not self.supports_deletes:
+            raise NotImplementedError('This method is only available for stores that support deletes.')
+        if not self.supports_listing:
+            raise NotImplementedError('This method is only available for stores that support directory listing.')
+        self._check_writable()
+
+        path_to_delete = _dereference_path(self.path, prefix)
+
+        if hasattr(self.fs, "_rm") and inspect.iscoroutinefunction(self.fs._rm):
+            try:
+                await self.fs._rm(path_to_delete, recursive=True)
+            except self.allowed_exceptions:
+                pass
+        else:
+            raise NotImplementedError("The store does not support async deletes")
+
     async def exists(self, key: str) -> bool:
         # docstring inherited
         path = _dereference_path(self.path, key)