Mock S3, GCS, and Azure storage with a Python abstract interface

from abc import ABC, abstractmethod
from pathlib import Path


class Storage(ABC):
    @abstractmethod
    def put(self, name: str, data: bytes) -> None:
        pass

    @abstractmethod
    def get(self, name: str) -> bytes:
        pass


class LocalStorage(Storage):
    def __init__(self, base_dir: str = "mock_store"):
        self.base_dir = Path(base_dir)
        self.base_dir.mkdir(exist_ok=True)

    def put(self, name: str, data: bytes) -> None:
        (self.base_dir / name).write_bytes(data)

    def get(self, name: str) -> bytes:
        return (self.base_dir / name).read_bytes()


def main() -> None:
    storage = LocalStorage()
    key = "example.txt"
    content = b"Hello, storage!"

    storage.put(key, content)
    retrieved = storage.get(key)

    print(f"Stored: {content!r}")
    print(f"Retrieved: {retrieved!r}")
    print(f"Match: {content == retrieved}")


if __name__ == "__main__":
    main()