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)