add storage factory (#3922)

This commit is contained in:
Jyong
2024-04-29 18:22:03 +08:00
committed by GitHub
parent c5e2659771
commit 338e4669e5
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