python - 将命令行参数转换为 Python 中的标准输入

Question

如何将接受命令行参数的脚本转换为能够通过标准用户输入接收参数？

目前我有一个文件，要处理它，我需要运行python abc.py query 而我希望该文件能够通过 python abc.py 运行，然后在下一行中它应该要求用户通过 raw_input 输入，响应应该是传递给参数(即“查询”)。

Answer 1

如果不更改 abc.py 本身，就无法实现这一点。

如果您可以更改abc.py，那么一个好方法是让它检查命令行参数，如果不存在，则要求它......如下所示:

if __name__ == '__main__':
    if len(sys.argv) > 1:
        file_to_process = sys.argv[1]
    else:
        file_to_process = raw_input("Enter file to process: ").strip()