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

Question

在python中编写命令行程序时，需要在对其进行任何计算之前测试输入的有效性。什么是最好的方法呢？

Answer 1

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

Answer 2

一个好的做法是使用调用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.