Python脚本高效更新命令行输出

在Python脚本中，特别是使用进度条或交互式命令行工具（如Inquirer）时，动态更新输出至关重要。本文介绍几种方法，有效清除之前的输出并更新至光标位置，尤其针对Inquirer库的特殊情况。

挑战：动态更新输出

许多Python脚本需要在运行过程中更新输出内容，例如显示进度或实时反馈。直接打印新内容会导致输出累积，影响用户体验。 尤其在使用Inquirer库时，之前的输出可能会干扰用户交互界面。

解决方案：多种方法实现动态更新

1. 利用\r回车符:

   \r回车符将光标移动到当前行首，但不换行。 通过end="\r"和flush=True，可以覆盖之前的输出。

   立即学习“Python免费学习笔记（深入）”；

   import time
   
   for i in range(10):
       print(f"进度: {i*10}%", end="\r", flush=True)
       time.sleep(1)
   print() # 最后添加换行，避免光标停留在行首

2. ANSI转义码:

   

   macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

   让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

   下载

   ANSI转义码提供更精细的终端控制。 \033[2J\033[H可以清除整个屏幕并把光标移动到左上角。 \033[K清除当前行从光标到行尾的内容。

   import sys
   
   def clear_line():
       sys.stdout.write("\033[K")
       sys.stdout.flush()
   
   def clear_screen():
       sys.stdout.write("\033[2J\033[H")
       sys.stdout.flush()
   
   # 使用示例:
   clear_screen()
   print("新内容")

3. 逐行清除:

   如果只需要清除特定行数，可以结合ANSI转义码中的光标移动指令\033[A (向上移动光标) 和 \033[K (清除当前行) 实现。

   import sys
   
   def clear_lines(num_lines):
       for _ in range(num_lines):
           sys.stdout.write("\033[1A\033[K")
       sys.stdout.flush()
   
   # 使用示例: 清除前两行
   clear_lines(2)

Inquirer库的兼容性

使用Inquirer库时，需要在调用inquirer.prompt() 之前 清除之前的输出，确保用户界面不受干扰。

import inquirer
import sys

def clear_screen():
    sys.stdout.write("\033[2J\033[H")
    sys.stdout.flush()

clear_screen() # 清空屏幕

questions = [
    inquirer.List(
        '选择',
        message="请选择一个选项",
        choices=['选项A', '选项B', '选项C'],
    ),
]

answers = inquirer.prompt(questions)
print(f"你选择了: {answers['选择']}")

总结

选择哪种方法取决于你的需求。\r适合简单地更新单行输出；ANSI转义码提供更强大的控制能力；逐行清除则适用于更精细的输出管理。 在使用Inquirer等交互式库时，记得在调用库函数之前清除之前的输出，以保证良好的用户体验。 注意，ANSI转义码的兼容性取决于终端。