如何为Python子进程输出添加实时时间戳并进行日志记录

#! /usr/bin/env python
import logging
import pexpect
import sys

# 配置日志
# filename='ts.log'：将日志输出到名为 'ts.log' 的文件
# encoding='utf-8'：指定日志文件的编码
# format='%(asctime)s %(levelname)-8s %(message)s'：定义日志消息的格式
#   %(asctime)s：记录时间戳
#   %(levelname)-8s：记录日志级别（左对齐，宽度8个字符）
#   %(message)s：记录日志内容
# level=logging.DEBUG：设置日志记录的最低级别为DEBUG，即所有级别的消息都会被记录
logging.basicConfig(
    filename='ts.log',
    encoding='utf-8',
    format='%(asctime)s %(levelname)-8s %(message)s',
    level=logging.DEBUG
)

# 也可以同时输出到控制台（如果需要）
# console_handler = logging.StreamHandler(sys.stdout)
# console_handler.setFormatter(logging.Formatter('%(asctime)s %(message)s'))
# logging.getLogger().addHandler(console_handler)


def run_command_with_timestamp_logging(cmd: str):
    """
    运行指定的命令，并将其输出实时地记录到日志中，每行带时间戳。

    Args:
        cmd (str): 要执行的命令字符串。
    """
    logging.info(f"Executing command: '{cmd}'") # 记录命令本身
    try:
        # 使用 pexpect.spawn 启动命令
        # encoding="utf-8"：确保正确处理子进程的输出编码
        p = pexpect.spawn(cmd, encoding="utf-8")

        # 循环读取子进程的每一行输出
        # 'l := p.readline()' 是 Python 3.8+ 的赋值表达式（walrus operator）
        # 它在读取行的同时将其赋值给变量 l
        while l := p.readline():
            # 记录子进程的输出行，去除末尾的换行符
            logging.info(l.strip())

        # 等待子进程结束并获取其退出状态
        p.wait()
        logging.info(f"Command '{cmd}' finished with exit status: {p.exitstatus}")

    except pexpect.exceptions.EOF:
        logging.error(f"Command '{cmd}' terminated unexpectedly or reached EOF.")
        logging.error(f"Before EOF: {p.before}") # 记录EOF前捕获到的内容
    except pexpect.exceptions.TIMEOUT:
        logging.error(f"Command '{cmd}' timed out.")
    except Exception as e:
        logging.critical(f"An unexpected error occurred: {e}")

# 示例用法
if __name__ == "__main__":
    print("Running 'ls -l' and logging output to ts.log...")
    run_command_with_timestamp_logging("ls -l")

    print("\nRunning 'docker build .' (example, replace with actual command) and logging output to ts.log...")
    # 注意：如果 docker build . 需要交互或特定环境，可能需要进一步配置 pexpect
    # 例如：run_command_with_timestamp_logging("docker build .")
    # 为了演示，这里使用一个会持续输出的简单命令
    run_command_with_timestamp_logging("bash -c 'for i in {1..5}; do echo \"Line $i from bash\"; sleep 1; done'")

    print("\nCheck ts.log for output.")