尝试并在 python 中排除代码块以引发缺少参数的错误

Question

所以我试图在我的代码中实现基本的 try 和 except 逻辑，以便在读取命令行参数时读取和处理异常：

我已经能够做到这一点。但我想在抛出错误时明确说明缺少第一个或第二个参数。

这是我的代码。

import sys
import logging

try:
    job_name = sys.argv[1]
    odate = sys.argv[2]
except IndexError as err:
    print('OS error: {0}'.format(err))
else:
    print('Correct arguments were provided')

如何产生错误让用户知道缺少哪个 arg？

Answer 1

异常IndexError 不包括导致错误的访问索引，您可以通过traceback.format_exc() 或sys.exc_info() 或其他方式检查堆栈回溯，以捕获访问的索引，但这太复杂了。

替代方案 1

首先检查参数的长度是否正确并采取相应措施：

ARG_COUNT = 2
assert len(sys.argv) == ARG_COUNT + 1

# No more need of try-except block
job_name = sys.argv[1]
odate = sys.argv[2]

替代方案 2

这是最好的解决方案。请改用argparse。它处理您需要的行为。您还可以进一步自定义每个参数的设置，例如指定参数的预期数据类型。

import argparse

parser = argparse.ArgumentParser(description='Process some inputs.')
parser.add_argument('job_name')
parser.add_argument('odate')
args = parser.parse_args()
print(args, args.job_name, args.odate)

输出

$ python script.py 
usage: script.py [-h] job_name odate
script.py: error: the following arguments are required: job_name, odate
$ python script.py some
usage: script.py [-h] job_name odate
script.py: error: the following arguments are required: odate
$ python script.py some thing
Namespace(job_name='some', odate='thing') some thing