在 python 中测试命令行输入脚本有效性的最佳方法是什么?

What is the best way to test the validity of command line input to script in python?

当您在 python 中编写命令行程序时,您需要在对其进行任何计算之前测试输入的有效性。执行此操作的最佳方法是什么?

一个好的做法是使用调用 sys.exit(1) 的验证函数。例如,您可以执行以下操作:

import sys

def usage():
    print("You need 3 arguments")
    print("Usage: script.py arg1 arg2 arg3")
    sys.exit(1)

argc = len(sys.argv)
if argc < 3:
    usage()
# And then with the rest of your script here.

您可以使用 Python 的 argparse 模块。它允许您向脚本添加选项和标志,并允许您的脚本在无法识别参数时轻松抛出错误。