日志流Handler()和input()之间的Python控制台干扰



当使用日志模块的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。这两个流是独立的,并且可以不可预测地交织。

任一:

  1. 用CCD_ 7替换日志记录调用
  2. 将日志记录配置为转到stdout而不是stderr,或者
  3. 在可重用函数中用print(prompt, file=sys.stderr, end=''); input()替换input(prompt)

最新更新