import os

class Config:
    def __init__(self):
        self.set_ict()
        self.use_imagen = True
        
    def set_fermat(self):
        """
        내부 서버용
        """
        self.config = 'dev'
        self.remote_folder = "/home/fermat/STORAGE/01.Projects/A2TEC/K_EYEWEAR/02.ML_DATA/Image_generator_result"
        self.sftp_host = "192.168.200.230"
        self.sftp_port = 22
        self.sftp_id = "fermat"
        self.sftp_pw = "1234"
        
        self.api_path = "./gemini.env"
                
    def set_dev(self):
        """
        개발용
        """
        self.config = 'dev'
        self.remote_folder = "/home/fermat/project/FM_TEST_REST_SERVER/result"
        self.sftp_host = "192.168.200.231"
        self.sftp_port = 22
        self.sftp_id = "fermat"
        self.sftp_pw = "fermat3514"
        self.db_pw = "1234"
        
        self.api_path = "./gemini.env"
        
        if not os.path.exists(self.remote_folder):
            os.makedirs(self.remote_folder)
            
    def set_ict(self):
        self.config = 'release'
        self.local_folder = "./result" # 결과물 저장 폴더
        
        self.db_pw = "Fermat3514!"
        
        self.api_path = "../gemini.env"
        
        if not os.path.exists(self.local_folder):
            os.makedirs(self.local_folder)

rest_config = Config()
# rest_config.set_dev()