Spaces:
Sleeping
Sleeping
| 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) |