82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
import yaml
|
|
import os
|
|
|
|
|
|
class Config:
|
|
|
|
__BASE_PATH = os.getcwd()
|
|
__CONFIG_PATH = os.path.join(__BASE_PATH, 'config')
|
|
__CONFIG_NAME = 'test.yaml'
|
|
__SCHEDULER_DB_FILE_NAME = 'schedule_db.db'
|
|
__LOG_FILE_NAME = 'myAssistant.log'
|
|
__CONFIG_DICT = {}
|
|
|
|
def __init__(self):
|
|
if self.__CONFIG_DICT == {}:
|
|
with open(os.path.join(self.__CONFIG_PATH, self.__CONFIG_NAME)) as f:
|
|
self.__CONFIG_DICT = yaml.safe_load(f)
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
if not hasattr(Config, "_instance"):
|
|
Config._instance = object.__new__(cls)
|
|
return Config._instance
|
|
|
|
@property
|
|
def CONFIG_PATH(self):
|
|
return self.__CONFIG_PATH
|
|
|
|
@property
|
|
def BASE_PATH(self):
|
|
return self.__BASE_PATH
|
|
|
|
def get_scheduler_db_file_path(self):
|
|
try:
|
|
return os.path.join(self.__BASE_PATH, self.__CONFIG_DICT['SCHEDULER_DB_PATH'],
|
|
self.__SCHEDULER_DB_FILE_NAME)
|
|
except:
|
|
return os.path.join(self.__CONFIG_PATH, 'schedule_db', self.__SCHEDULER_DB_FILE_NAME)
|
|
|
|
def set_scheduler_db_path(self, new_path):
|
|
self.__CONFIG_DICT['SCHEDULER_DB_PATH'] = new_path
|
|
with open(os.path.join(self.__CONFIG_PATH, self.__CONFIG_NAME), 'w') as f:
|
|
f.write(yaml.safe_dump(self.__CONFIG_DICT, sort_keys=False))
|
|
|
|
def get_log_file_path(self):
|
|
try:
|
|
return os.path.join(self.__CONFIG_PATH, self.__CONFIG_DICT['LOG_PATH'], self.__LOG_FILE_NAME)
|
|
except:
|
|
return os.path.join(self.__CONFIG_PATH, 'log', self.__LOG_FILE_NAME)
|
|
|
|
def set_log_path(self, new_path):
|
|
self.__CONFIG_DICT['LOG_PATH'] = new_path
|
|
with open(os.path.join(self.__CONFIG_PATH, self.__CONFIG_NAME), 'w') as f:
|
|
f.write(yaml.safe_dump(self.__CONFIG_DICT, sort_keys=False))
|
|
|
|
def get_proxy(self):
|
|
try:
|
|
return self.__CONFIG_DICT['PROXY']
|
|
except:
|
|
return None
|
|
|
|
def set_proxy(self, new_proxy):
|
|
self.__CONFIG_DICT['PROXY'] = new_proxy
|
|
with open(os.path.join(self.__CONFIG_PATH, self.__CONFIG_NAME), 'w') as f:
|
|
f.write(yaml.safe_dump(self.__CONFIG_DICT, sort_keys=False))
|
|
|
|
def get_config(self, config_name, default=None):
|
|
try:
|
|
return self.__CONFIG_DICT[config_name]
|
|
except:
|
|
return default
|
|
|
|
def set_config(self, config_name, value, config_path = ''):
|
|
self.__CONFIG_DICT[config_name] = value
|
|
with open(os.path.join(self.__CONFIG_PATH, self.__CONFIG_NAME), 'w') as f:
|
|
f.write(yaml.safe_dump(self.__CONFIG_DICT, sort_keys=False))
|
|
|
|
def get_plugin_config(self, config_name, config_path):
|
|
print(config_path)
|
|
if config_path != '':
|
|
with open(os.path.join(config_path, self.__CONFIG_NAME)) as f:
|
|
config_d = yaml.safe_load(f)
|
|
return config_d[config_name] |