__init__.py 500 B

1234567891011121314
  1. from .files import FileStore
  2. from .local import LocalFileStore
  3. from .memory import InMemoryFileStore
  4. from .s3 import S3FileStore
  5. def get_file_store(file_store: str, file_store_path: str | None = None) -> FileStore:
  6. if file_store == 'local':
  7. if file_store_path is None:
  8. raise ValueError('file_store_path is required for local file store')
  9. return LocalFileStore(file_store_path)
  10. elif file_store == 's3':
  11. return S3FileStore()
  12. return InMemoryFileStore()