|
|
|
@ -1,3 +1,4 @@
|
|
|
|
|
import os
|
|
|
|
|
import re
|
|
|
|
|
from pathlib import Path
|
|
|
|
|
from typing import Iterator, List, Optional, Sequence, Tuple, Union
|
|
|
|
@ -42,7 +43,7 @@ class LocalFileStore(ByteStore):
|
|
|
|
|
root_path (Union[str, Path]): The root path of the file store. All keys are
|
|
|
|
|
interpreted as paths relative to this root.
|
|
|
|
|
"""
|
|
|
|
|
self.root_path = Path(root_path)
|
|
|
|
|
self.root_path = Path(root_path).absolute()
|
|
|
|
|
|
|
|
|
|
def _get_full_path(self, key: str) -> Path:
|
|
|
|
|
"""Get the full path for a given key relative to the root path.
|
|
|
|
@ -55,7 +56,15 @@ class LocalFileStore(ByteStore):
|
|
|
|
|
"""
|
|
|
|
|
if not re.match(r"^[a-zA-Z0-9_.\-/]+$", key):
|
|
|
|
|
raise InvalidKeyException(f"Invalid characters in key: {key}")
|
|
|
|
|
return self.root_path / key
|
|
|
|
|
full_path = os.path.abspath(self.root_path / key)
|
|
|
|
|
common_path = os.path.commonpath([str(self.root_path), full_path])
|
|
|
|
|
if common_path != str(self.root_path):
|
|
|
|
|
raise InvalidKeyException(
|
|
|
|
|
f"Invalid key: {key}. Key should be relative to the full path."
|
|
|
|
|
f"{self.root_path} vs. {common_path} and full path of {full_path}"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
return Path(full_path)
|
|
|
|
|
|
|
|
|
|
def mget(self, keys: Sequence[str]) -> List[Optional[bytes]]:
|
|
|
|
|
"""Get the values associated with the given keys.
|
|
|
|
|