如何在Python函数中定义类宏代码复用参数验证逻辑？

Python本身没有像C语言那样的宏定义机制，但完全可以通过装饰器实现你想要的参数验证逻辑复用效果，而且这是Python社区最推荐的做法，比模拟宏更优雅易维护。

具体实现步骤： #

1. 先封装通用的参数验证逻辑到一个装饰器里
2. 给需要验证参数的业务函数加上这个装饰器，自动复用验证代码

示例代码：

首先定义验证装饰器：

def validate_args(func):
    def wrapper(arg1, arg2, *args, **kwargs):
        # 这里写你的通用参数验证&修改逻辑
        # 示例：修正arg1为正整数，校验arg2非空
        if not isinstance(arg1, int) or arg1 <= 0:
            arg1 = abs(arg1)  # 也可以直接抛出异常终止执行
        if not isinstance(arg2, str) or len(arg2.strip()) == 0:
            raise ValueError("arg2不能为空字符串")
        # 将处理后的参数传递给原业务函数
        return func(arg1, arg2, *args, **kwargs)
    return wrapper

然后给业务函数加装饰器即可：

@validate_args
def f(arg1, arg2):
    # 这里直接写业务代码，不用重复写验证
    print(f"处理后的参数：arg1={arg1}, arg2={arg2}")
    return arg1 + len(arg2)

@validate_args
def g(arg1, arg2, extra_param):
    # 另一个业务函数，同样自动复用验证逻辑
    print(f"额外参数：{extra_param}")
    return arg1 * len(arg2)

关于模拟“宏展开”的补充： #

如果非要追求类似宏的代码嵌入效果，也可以通过动态生成代码的方式实现，但这种方式可读性差、调试困难，非常不推荐。比如：

def add_validation_macro(func_name, business_code):
    validation_code = """
    # 参数验证逻辑
    if not isinstance(arg1, int) or arg1 <= 0:
        arg1 = abs(arg1)
    if not isinstance(arg2, str) or len(arg2.strip()) == 0:
        raise ValueError("arg2不能为空字符串")
    """
    # 拼接完整函数代码并执行
    full_func_code = f"def {func_name}(arg1, arg2):\n{validation_code}\n{business_code}"
    exec(full_func_code, globals())

# 生成带验证的函数f
add_validation_macro("f", """
    # 业务代码
    print(f"处理后的参数：arg1={arg1}, arg2={arg2}")
    return arg1 + len(arg2)
""")

总之，装饰器方案是Python中解决这类复用问题的标准做法，建议优先使用。

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