如何在Python中断言函数对象的属性（如指定关键字参数）？

当然有办法实现你想要的功能！在Python中，我们可以借助标准库的inspect模块来解析函数的参数签名，从而判断它是否接受某个特定的关键字参数。这在元类这类需要做元编程的场景里确实非常实用，尽管鸭子类型是Python的惯用风格，但特殊需求下这种检查完全合理。

实现takes_argument函数 #

下面是一个可以直接使用的实现，它会处理多种参数情况：

import inspect

def takes_argument(func, arg_name):
    # 获取函数的参数签名
    sig = inspect.signature(func)
    params = sig.parameters
    
    # 情况1：参数直接在函数定义中存在
    if arg_name in params:
        param = params[arg_name]
        # 允许的参数类型：可关键字传递的参数（包括位置/关键字兼容、仅关键字）
        return param.kind in (param.POSITIONAL_OR_KEYWORD, param.KEYWORD_ONLY)
    
    # 情况2：函数接受任意关键字参数（即带有**kwargs）
    for param in params.values():
        if param.kind == param.VAR_KEYWORD:
            return True
    
    # 两种情况都不满足，返回False
    return False

用你的示例代码测试 #

把这个函数集成到你的代码里，就能实现预期的断言功能：

def example(foo=1):
    pass

def g(f):
    assert takes_argument(f, "foo"), "Function must accept 'foo' keyword argument"

g(example)  # 正常执行，不会触发断言

处理边界场景 #

这个实现还能覆盖一些特殊情况：

• 无默认值的位置参数：比如def bar(foo): pass，takes_argument(bar, "foo")返回True，因为你可以通过bar(foo=2)的方式传递参数
• 仅接受可变位置参数的函数：比如def baz(*args): pass，takes_argument(baz, "foo")返回False，因为这类函数不接受关键字参数
• 接受任意关键字参数的函数：比如def qux(**kwargs): pass，takes_argument(qux, "foo")返回True，因为**kwargs会捕获所有传入的关键字参数

元类中的适用性 #

在元类里，你可以对类的方法、属性对应的函数做这类检查，比如在__new__方法中遍历类的成员，用takes_argument验证特定方法是否符合参数要求，完全能满足你的特殊需求。

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