add storage factory (#3922)

This commit is contained in:
Jyong 2024-04-29 18:22:03 +08:00 • committed by GitHub
commit 338e4669e5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 384 additions and 211 deletions

View file

@ -0,0 +1,38 @@
"""Abstract interface for file storage implementations."""
from abc import ABC, abstractmethod
from collections.abc import Generator
from flask import Flask
class BaseStorage(ABC):
"""Interface for file storage.
"""
app = None
def __init__(self, app: Flask):
self.app = app
@abstractmethod
def save(self, filename, data):
raise NotImplementedError
@abstractmethod
def load_once(self, filename: str) -> bytes:
raise NotImplementedError
@abstractmethod
def load_stream(self, filename: str) -> Generator:
raise NotImplementedError
@abstractmethod
def download(self, filename, target_filepath):
raise NotImplementedError
@abstractmethod
def exists(self, filename):
raise NotImplementedError
@abstractmethod
def delete(self, filename):
raise NotImplementedError