|
|
@@ -4,6 +4,7 @@ import json |
|
|
|
import logging |
|
|
|
import os |
|
|
|
import pickle |
|
|
|
import copy |
|
|
|
|
|
|
|
from common.log import logger |
|
|
|
|
|
|
@@ -233,20 +234,26 @@ config = Config() |
|
|
|
|
|
|
|
|
|
|
|
def drag_sensitive(config): |
|
|
|
if isinstance(config, str): |
|
|
|
conf_dict: dict = json.loads(config) |
|
|
|
for key in conf_dict: |
|
|
|
if "key" in key or "secret" in key: |
|
|
|
if isinstance(key, str): |
|
|
|
conf_dict[key] = conf_dict[key][0:3] + "*" * 5 + conf_dict[key][-3:] |
|
|
|
return json.dumps(conf_dict, indent=4) |
|
|
|
elif isinstance(config, dict): |
|
|
|
for key in config: |
|
|
|
if "key" in key or "secret" in key: |
|
|
|
if isinstance(key, str): |
|
|
|
config[key] = config[key][0:3] + "*" * 5 + config[key][-3:] |
|
|
|
try: |
|
|
|
if isinstance(config, str): |
|
|
|
conf_dict: dict = json.loads(config) |
|
|
|
conf_dict_copy = copy.deepcopy(conf_dict) |
|
|
|
for key in conf_dict_copy: |
|
|
|
if "key" in key or "secret" in key: |
|
|
|
if isinstance(key, str): |
|
|
|
conf_dict_copy[key] = conf_dict_copy[key][0:3] + "*" * 5 + conf_dict_copy[key][-3:] |
|
|
|
return json.dumps(conf_dict_copy, indent=4) |
|
|
|
|
|
|
|
elif isinstance(config, dict): |
|
|
|
config_copy = copy.deepcopy(config) |
|
|
|
for key in config: |
|
|
|
if "key" in key or "secret" in key: |
|
|
|
if isinstance(key, str): |
|
|
|
config_copy[key] = config_copy[key][0:3] + "*" * 5 + config_copy[key][-3:] |
|
|
|
return config_copy |
|
|
|
except Exception as e: |
|
|
|
logger.exception(e) |
|
|
|
return config |
|
|
|
|
|
|
|
return config |
|
|
|
|
|
|
|
|
|
|
@@ -258,7 +265,7 @@ def load_config(): |
|
|
|
config_path = "./config-template.json" |
|
|
|
|
|
|
|
config_str = read_file(config_path) |
|
|
|
logger.debug("[INIT] config str: {}".format(config_str)) |
|
|
|
logger.debug("[INIT] config str: {}".format(drag_sensitive(config_str))) |
|
|
|
|
|
|
|
# 将json字符串反序列化为dict类型 |
|
|
|
config = Config(json.loads(config_str)) |
|
|
@@ -283,13 +290,11 @@ def load_config(): |
|
|
|
logger.setLevel(logging.DEBUG) |
|
|
|
logger.debug("[INIT] set log level to DEBUG") |
|
|
|
|
|
|
|
logger.info("[INIT] load config: {}".format(config)) |
|
|
|
logger.info("[INIT] load config: {}".format(drag_sensitive(config))) |
|
|
|
|
|
|
|
config.load_user_datas() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_root(): |
|
|
|
return os.path.dirname(os.path.abspath(__file__)) |
|
|
|
|
|
|
|