lzybetter/myAssistant
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

测试插件

Browse Source
This commit is contained in:
lzybetter
2023-09-09 14:54:25 +08:00
parent 1bd2a9dbc2
commit c0f2ac9847
3 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
util/config.py
View File

@@ -3,6 +3,7 @@ import os
class Config:
__BASE_PATH = os.getcwd()
__CONFIG_PATH = os.path.join(__BASE_PATH, 'config')
__CONFIG_NAME = 'test.yaml'
@@ -61,3 +62,14 @@ class Config:
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):
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))