在 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 模块。它允许您向脚本添加选项和标志,并允许您的脚本在无法识别参数时轻松抛出错误。
当您在 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 模块。它允许您向脚本添加选项和标志,并允许您的脚本在无法识别参数时轻松抛出错误。