当使用日志模块的StreamHandler()
输出和input()
行时,我在控制台输出中遇到了一个小麻烦。
我用以下功能配置我的记录器:
def configure_logger(filename='log'):
"""Configures a logger which outputs at level INFO and above to both
the console and a file called 'filename'.
"""
file_handler = logging.FileHandler(filename, mode='w')
stream_handler = logging.StreamHandler()
file_formatter = logging.Formatter(datefmt="%d/%m/%Y %H:%M:%S",
fmt='%(levelname)-8s %(asctime)s '
'%(name)-50s - %(message)s')
stream_formatter = logging.Formatter(fmt='%(levelname)-8s %(name)-50s - '
'%(message)s')
file_handler.setFormatter(file_formatter)
stream_handler.setFormatter(stream_formatter)
logging.basicConfig(handlers=(file_handler, stream_handler),
level=logging.INFO)
稍后,我连续两次调用以下函数来创建两个目录:
def create_directory(directory):
"""Creates a directory, prompting the user to confirm overwrite if
it already exists
"""
logger = logging.getLogger(f'{__name__}.create_directory')
logger.info(f'Creating directory {directory}')
try:
os.mkdir(directory)
except FileExistsError:
overwrite = input(f'Directory {directory} already '
f'exists. Overwrite? (y/n): ')
overwrite.lower()
while overwrite not in ['y', 'yes', 'n', 'no']:
overwrite = input(f'Enter yes/y or no/n?: ')
overwrite.lower()
if overwrite in ['y', 'yes']:
logger.warning('Overwriting existing directory')
shutil.rmtree(directory)
os.mkdir(directory)
else:
logger.info("Exiting")
sys.exit()
如果我试图创建的目录已经存在,我会提示用户并要求他们确认要使用input()
函数覆盖它。响应"是"后,我会记录一条警告"正在覆盖现有目录"。
在连续两次调用create_directory()
之后,我在控制台中得到以下内容:
INFO sowfatools.create_directory - Creating directory time_histories
Directory time_histories already exists. Overwrite? (y/n): >? y
Directory time_history_plots already exists. Overwrite? (y/n): WARNING sowfatools.create_directory - Overwriting existing directory
INFO sowfatools.create_directory - Creating directory time_history_plots
>?
第二条输入线(来自第二个函数调用(在第一个函数调用发出警告之前打印到控制台。事实上,它也出现在第二次调用中的第一条日志消息之前。
为什么会这样?我该怎么修?它不会使我的代码无法使用,但它是一个麻烦。日志StreamHandler
是不是太慢了,跟不上其余的代码?
input
将其提示写入stdout,而日志记录则进入stderr。这两个流是独立的,并且可以不可预测地交织。
任一:
- 用CCD_ 7替换日志记录调用
- 将日志记录配置为转到stdout而不是stderr,或者
- 在可重用函数中用
print(prompt, file=sys.stderr, end=''); input()
替换input(prompt)