如何要求用户在terminal输入python脚本参数？

2023-11-13

比如：

python run.py hello

需要用到argparse库，run.py的代码如下：

import argparse

def main():
    parser = argparse.ArgumentParser(description="请输入hello或者yes")
    parser.add_argument("command", choices=['hello', 'yes'], 
                        help="Command to execute: 'hello' or 'yes'")
    
    args = parser.parse_args()

    if args.command == 'hello':
        print("hello world")
    elif args.command == 'yes':
        print("yes man")

if __name__ == "__main__":
    main()

以上代码指用户可以输入hello或者yes获取不同的输出。

以上功能可以用sys模块来完成，但是sys实现起来比较功能比较少，一般很少用，仅供参考：

import sys

def main():
    if len(sys.argv) ! = 2:
        print("Usage: python fun.py [hello/yes]")
        sys.exit(1)

    command = sys.argv[1]

    if command == 'hello':
        print("hello world")
    elif command == 'yes':
        print("yes man")
    else:
        print("Invalid command. Use 'hello' or 'yes'.")

if __name__ == "__main__":
    main()