python - 是否有自动检测数据类型的 argparse 模块

Question

是否有任何其他 python 包或 argparse 是否具有我期待的以下功能

arg_parser.add_argument("-v", "--some_data", required=True,
                        help="data id(int) or vehicle name (str) ")

input -v 123 output type(args.some_data) --> int
input -v abc output type(args.some_data) --> str
input -v a12 output type(args.some_data) --> str

Answer 1

您可以定义自定义 type尝试解析一个 int，如果失败，则返回字符串。

(为简单起见，我将您的选项作为此示例中的参数。)

import argparse

def int_or_str(s):
    try:
        return int(s)
    except ValueError:
        return s

parser = argparse.ArgumentParser()
parser.add_argument("x", type=int_or_str, help="int or str")
print(parser.parse_args(['123']))  # -> Namespace(x=123)
print(parser.parse_args(['hello']))  # -> Namespace(x='hello')

但是，请注意 int 会忽略前导和尾随空格，因此例如 int(' 1 ') 会成功。

print(parser.parse_args([' 1 ']))  # -> Namespace(x=1)