在将日志模块的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()
函数确认是否要覆盖它。在得到“yes”响应后,我会记录一个警告“Overwriting existing directory”。
在连续两次调用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
太慢了,跟不上其余的代码?
1条答案
按热度按时间qaxu7uf21#
input
将其提示符写入stdout,而日志记录转到stderr。这两个流是独立的,可以不可预测地交叉。或者:
1.将日志记录调用替换为
print
,1.将日志记录配置为转到stdout而不是stderr,或者
1.在可重用函数中,将
input(prompt)
替换为print(prompt, file=sys.stderr, end=''); input()
。