在Shell命令替换中分离Python脚本的输入提示与返回值

针对你遇到的命令替换捕获所有输出导致输入提示不实时显示的问题，这里提供几种实用的解决方法：

方法1：将输入提示输出到标准错误流（stderr） #

Python的input()函数默认把提示信息输出到标准输出流（stdout），而命令替换$(...)会捕获全部stdout内容。我们可以手动将提示信息输出到stderr，这样stdout只保留需要返回的序列号，提示会实时显示在终端。

修改你的Python脚本：

import sys

# 把输入提示输出到stderr
print("Enter ID (0-indexed):", file=sys.stderr)
id = input()
# ... 此处为计算serial_number的逻辑 ...
print(serial_number)

Bash脚本无需改动，执行时提示会直接显示，serial_number变量只会拿到序列号。

方法2：由Shell脚本处理输入，将ID作为参数传递给Python #

这是你已经想到的方案，把输入逻辑转移到Shell脚本中，再将用户输入的ID作为命令行参数传给Python脚本，Python只负责计算并返回结果。

修改后的Bash脚本：

read -p "Enter ID (0-indexed): " user_id
serial_number=$(./get_serial_no.py "$user_id")
echo "$serial_number"

对应的Python脚本修改：

import sys

# 从命令行参数获取ID（注意sys.argv[0]是脚本名，sys.argv[1]才是传入的ID）
id = sys.argv[1]
# ... 此处为计算serial_number的逻辑 ...
print(serial_number)

方法3：将序列号写入临时文件 #

如果不想调整输入逻辑，可以让Python脚本把计算出的序列号写入临时文件，输入提示正常输出到stdout（这样执行脚本时用户能看到提示并输入），之后Shell脚本从临时文件读取序列号。

修改后的Python脚本：

id = input("Enter ID (0-indexed):\n")
# ... 此处为计算serial_number的逻辑 ...
# 将序列号写入临时文件
with open("/tmp/serial_temp.txt", "w") as temp_file:
    temp_file.write(serial_number)

修改后的Bash脚本：

# 执行Python脚本，此时会显示提示并等待用户输入
./get_serial_no.py
# 从临时文件读取序列号
serial_number=$(cat /tmp/serial_temp.txt)
echo "$serial_number"
# 可选：用完临时文件后删除
rm /tmp/serial_temp.txt

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