Scrap-Dji / storage /manager.py
joel
Initial deployment: Scrap-Dji with API
dfdddb1
raw
history blame contribute delete
669 Bytes
import os
from utils.config import STORAGE_PATH
class StorageManager:
def __init__(self, base_path=STORAGE_PATH):
self.base_path = base_path
os.makedirs(self.base_path, exist_ok=True)
def save_file(self, file_content: bytes, filename: str, subfolder: str = "") -> str:
folder = os.path.join(self.base_path, subfolder)
os.makedirs(folder, exist_ok=True)
path = os.path.join(folder, filename)
with open(path, "wb") as f:
f.write(file_content)
return path
def get_file_path(self, filename: str, subfolder: str = "") -> str:
return os.path.join(self.base_path, subfolder, filename)