File size: 1,125 Bytes
dfb775d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | """Always-available local filesystem `StorageProvider` implementation.
Canonical mindxtrain2.md §Part 4 `storage.local_fs`. Acts as the dev/CI fallback
when no remote provider is configured.
"""
from __future__ import annotations
import shutil
from pathlib import Path
from mindxtrain.storage.provider import StorageProvider, StorageRef
class LocalFsProvider(StorageProvider):
name = "local_fs"
def __init__(self, root: Path) -> None:
self.root = root.resolve()
self.root.mkdir(parents=True, exist_ok=True)
def put_dir(self, src: Path, key: str) -> StorageRef:
dest = self.root / key
if dest.exists():
shutil.rmtree(dest)
shutil.copytree(src, dest)
return StorageRef(provider=self.name, uri=str(dest))
def get_dir(self, ref: StorageRef, dest: Path) -> Path:
if ref.provider != self.name:
msg = f"ref provider {ref.provider!r} != {self.name!r}"
raise ValueError(msg)
src = Path(ref.uri)
if dest.exists():
shutil.rmtree(dest)
shutil.copytree(src, dest)
return dest
|