日志记录streamHandler()和input()之间的Python控制台干扰

jjhzyzn0  于 2023-02-02  发布在  Python
关注(0)|答案(1)|浏览(126)

在将日志模块的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太慢了,跟不上其余的代码?

qaxu7uf2

qaxu7uf21#

input将其提示符写入stdout,而日志记录转到stderr。这两个流是独立的,可以不可预测地交叉。
或者:
1.将日志记录调用替换为print
1.将日志记录配置为转到stdout而不是stderr,或者
1.在可重用函数中,将input(prompt)替换为print(prompt, file=sys.stderr, end=''); input()

相关问题