如何在Python的input输入框后实时显示文件扩展名？

要实现输入时实时在内容后方显示.pdf扩展名，标准的input()函数做不到——它会阻塞直到用户按下回车，无法实时监听输入内容变化。我们可以通过直接读取标准输入的字符，结合ANSI终端转义序列控制屏幕输出和光标位置，实现实时渲染效果。

完整实现代码 #

import sys
import os

def input_with_extension(prompt, extension):
    # 初始化输入内容
    user_input = []
    # 拼接基础提示文本（包含当前脚本所在目录的images路径）
    base_prompt = f"{prompt} {os.path.dirname(os.path.realpath(__file__))}\\images\\"
    
    while True:
        # 读取单个字符（不自动回显）
        char = sys.stdin.read(1)
        # 处理回车，结束输入流程
        if char in ('\n', '\r'):
            print()  # 换行避免和后续输出重叠
            break
        # 处理退格键，删除最后输入的字符
        elif char == '\x08' and user_input:
            user_input.pop()
        # 处理可打印字符，加入输入列表
        elif char.isprintable():
            user_input.append(char)
        
        # 构建当前要显示的完整行内容
        current_content = ''.join(user_input)
        full_line = f"{base_prompt}{current_content}{extension}"
        
        # 使用ANSI转义序列刷新当前行：移到行首→清除行内容→打印新内容
        sys.stdout.write(f"\r\033[K{full_line}")
        # 将光标移回输入内容的末尾，保证正常输入
        cursor_pos = len(base_prompt) + len(current_content)
        sys.stdout.write(f"\033[{cursor_pos}G")
        sys.stdout.flush()
    
    return ''.join(user_input)

# 调用示例
docname = input_with_extension("Where do you wanna save your document?", ".pdf")
# 后续处理：拼接扩展名（仅为逻辑处理，实时显示只是视觉效果）
final_filename = docname + ".pdf"
print(f"最终文件名：{final_filename}")

核心逻辑说明 #

• 逐字符监听：用sys.stdin.read(1)替代阻塞式的input()，实时捕获每个输入字符。
• ANSI转义控制：
  • \r：将光标移至行首
  • \033[K：清除光标到行尾的所有内容
  • \033[{n}G：将光标定位到第n列（从1开始计数）
• 特殊键兼容：支持退格键删除输入内容，回车触发输入结束，和原生输入体验一致。
• 光标同步：每次更新显示后，精准计算光标位置并移动，避免出现光标跳转异常。

对应解决你提到的问题 #

1. 替换了原代码中无法运行的嵌套循环逻辑，改用逐字符监听的可行方案。
2. 实时获取输入内容的长度，用于计算光标位置和渲染完整行。
3. 循环在输入过程中同步执行，每输入一个字符就刷新一次显示。
4. 通过精准的光标控制，避免了输入异常，体验和原生input()一致。

内容的提问来源于stack exchange，提问作者tygo