simplify string handling

zarr-developers · Aug 3, 2024 · da6083e · da6083e
1 parent ebbfbe0
commit da6083e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/src/zarr/store/local.py b/src/zarr/store/local.py
@@ -196,7 +196,7 @@ async def list_prefix(self, prefix: str) -> AsyncGenerator[str, None]:
         to_strip = str(self.root) + "/"
         for p in (self.root / prefix).rglob("*"):
             if p.is_file():
-                yield str(p).replace(to_strip, "")
+                yield str(p).removeprefix(to_strip)
 
     async def list_dir(self, prefix: str) -> AsyncGenerator[str, None]:
         """

diff --git a/src/zarr/store/remote.py b/src/zarr/store/remote.py
@@ -205,10 +205,6 @@ async def list_dir(self, prefix: str) -> AsyncGenerator[str, None]:
             yield onefile
 
     async def list_prefix(self, prefix: str) -> AsyncGenerator[str, None]:
-        if prefix == "":
-            find_str = "/".join([self.path, prefix])
-        else:
-            find_str = "/".join([self.path, prefix])
-
+        find_str = "/".join([self.path, prefix])
         for onefile in await self._fs._find(find_str):
             yield onefile.removeprefix(find_str)