如何基于argparse参数字典为Python类程序化添加类型注解？

你可以通过动态创建类的方式，从argparse配置字典中提取类型信息，自动生成带类型注解的Args类，彻底避免重复定义类型的问题。以下是两种实用的实现方式：

方法1：用type()手动生成带注解的类 #

利用Python内置的type()函数动态创建类，同时将从配置中提取的类型信息写入类的__annotations__字典，编辑器和类型检查工具能正常识别这些注解。

argparse_args = [
    {
        "name": "some_arg",
        "config": {
            "type": int,
            "help": "This is some arg"
        }
    },
    {
        "name": "another_arg",
        "config": {
            "type": str,
            "help": "This is another arg"
        }
    }
]

# 从配置中提取字段名和对应类型，生成注解字典
annotations = {arg["name"]: arg["config"]["type"] for arg in argparse_args}

# 动态创建Args类，继承自object
def args_init(self, **kwargs):
    for key, value in kwargs.items():
        setattr(self, key, value)

Args = type(
    'Args',
    (object,),
    {
        '__annotations__': annotations,
        '__init__': args_init  # 添加初始化方法，方便将解析后的参数转为实例
    }
)

# 使用示例
def someFunc(args: Args):
    # 编辑器会识别some_arg是int类型，提供类型提示和检查
    print(args.some_arg + 1)

配合argparse使用时，只需将解析后的参数转为Args实例：

import argparse

parser = argparse.ArgumentParser()
for arg in argparse_args:
    parser.add_argument(f'--{arg["name"]}', **arg["config"])

# 解析命令行参数并转为Args实例
parsed_args = parser.parse_args()
args_instance = Args(**vars(parsed_args))

someFunc(args_instance)

方法2：用dataclasses.make_dataclass生成数据类 #

如果需要更简洁的实现，Python的dataclasses模块提供了make_dataclass函数，可以一键生成带类型注解的数据类，自带__init__、__repr__等实用方法，无需手动编写初始化逻辑。

from dataclasses import make_dataclass
import argparse

argparse_args = [
    {
        "name": "some_arg",
        "config": {
            "type": int,
            "help": "This is some arg"
        }
    },
    {
        "name": "another_arg",
        "config": {
            "type": str,
            "help": "This is another arg"
        }
    }
]

# 生成字段列表，格式为 (字段名, 类型)
fields = [(arg["name"], arg["config"]["type"]) for arg in argparse_args]

# 动态创建数据类
Args = make_dataclass('Args', fields)

# 解析参数并转为Args实例
parser = argparse.ArgumentParser()
for arg in argparse_args:
    parser.add_argument(f'--{arg["name"]}', **arg["config"])

parsed_args = parser.parse_args()
args_instance = Args(**vars(parsed_args))

def someFunc(args: Args):
    print(args.some_arg + 1)  # 同样支持类型提示

someFunc(args_instance)

如果需要给字段添加默认值，只需从配置的default字段提取，修改字段定义为(arg["name"], arg["config"]["type"], arg["config"]["default"])即可。

内容的提问来源于stack exchange，提问作者Alex Long