Hide passwords in upgrade scripts and fix compact_ce testcase

This commit is contained in:
obdev
2024-04-24 08:23:04 +00:00
committed by ob-robot
parent e36e4917bf
commit 5de554fc81
6 changed files with 147 additions and 36 deletions

View File

@ -14,6 +14,7 @@ import actions
import upgrade_health_checker
import tenant_upgrade_action
import upgrade_post_checker
import re
# 由于用了/*+read_consistency(WEAK) */来查询,因此升级期间不能允许创建或删除租户
@ -22,6 +23,11 @@ class UpgradeParams:
sql_dump_filename = config.post_upgrade_sql_filename
rollback_sql_filename = config.post_upgrade_rollback_sql_filename
class PasswordMaskingFormatter(logging.Formatter):
def format(self, record):
s = super(PasswordMaskingFormatter, self).format(record)
return re.sub(r'password=\".*?\"', 'password=\"******\"', s)
def config_logging_module(log_filenamme):
logging.basicConfig(level=logging.INFO,\
format='[%(asctime)s] %(levelname)s %(filename)s:%(lineno)d %(message)s',\
@ -29,15 +35,18 @@ def config_logging_module(log_filenamme):
filename=log_filenamme,\
filemode='w')
# 定义日志打印格式
formatter = logging.Formatter('[%(asctime)s] %(levelname)s %(filename)s:%(lineno)d %(message)s', '%Y-%m-%d %H:%M:%S')
formatter = PasswordMaskingFormatter('[%(asctime)s] %(levelname)s %(filename)s:%(lineno)d %(message)s', '%Y-%m-%d %H:%M:%S')
#######################################
# 定义一个Handler打印INFO及以上级别的日志到sys.stdout
stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setLevel(logging.INFO)
# 设置日志打印格式
stdout_handler.setFormatter(formatter)
# 定义好的stdout_handler日志handler添加到root logger
# 定义一个Handler处理文件输出
file_handler = logging.FileHandler(log_filenamme, mode='w')
file_handler.setLevel(logging.INFO)
file_handler.setFormatter(formatter)
logging.getLogger('').addHandler(stdout_handler)
logging.getLogger('').addHandler(file_handler)
def print_stats():
logging.info('==================================================================================')