python - 提取函数中命令行参数的名称

Question

我正在使用 ArgParse 在 Python 中提供命令行参数。

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--quality", help="enter some quality limit")
args = parser.parse_args()
print "You gave quality = %s" % str(args.quality)

def is_number(s):
   try:
      val = int(s)
   except ValueError:
      print "That is not an int!"

is_number(args.quality)

我将其保存为 a.py 然后运行:

$python a.py --quality 10
You gave quality = 10

如果我们输入一些字符而不是数字，我想在函数“quality is not an int”内的打印语句中打印出参数名称“quality”。如何提取参数名称和在那里使用它。我还有一些命令行参数，所以我想要一个明确的错误，说明哪个参数不是 int。

Answer 1

一种方法是重写您的 is_number 以将参数名称作为字符串。

def is_number(args, argname):
    try:
        int(getattr(args, argname))
    except ValueError:
        print argname, "is not an integer"

is_number(args, "quality")

但是 Ben 建议使用 argparse 的 type 参数更好。