Spaces:
Sleeping
Sleeping
File size: 669 Bytes
dfdddb1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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) |