python 模拟检测函数形参类型-优快云博客

本文链接：https://blog.youkuaiyun.com/jsphpyyds/article/details/119988021

这篇博客介绍了如何利用Python3的函数注解，结合装饰器和inspect模块，实现一个简单的类型检测机制。通过定义一个`typeof`装饰器，可以在函数调用时检查传入参数和返回值的类型是否符合预期，如果类型不匹配则抛出异常。示例代码中展示了如何应用这个装饰器来确保`test`函数的参数和返回值类型正确。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python3 新增了函数注解，大致用法就是 def func(a:int,b:str) -> str ，用：来表示函数形参的类型，但是目前为止这个注解在python解释器中不会起到任何作用，实参类型不会因为形参注解类型而发生任何异常，所以我突发奇想，结合python的装饰器和inspect 模块一起做一个简单的类型检测（可能存在一点bug，但是我觉得实现过程值得被记录）

在以下的代码中，需要使用到inspect模块里面的signature、Parameter

#异常检测类，对于类型不匹配的抛出此类异常
class typenotmatch(Exception):
    def __str__(self):
        return "类型不匹配"


#检测类型装饰器
def typeof(func):
    def decorator(*args,**kwargs):
        signal=signature(func)
        bound_args=signal.bind(*args,**kwargs) #此处做一个函数签名，获取实参字典
        print(bound_args)
        #获取参数键值对字典
        params=dict(bound_args.arguments.items())
        #做一次深拷贝
        clone=copy.deepcopy(dict(signal.parameters.items()))
        #检测参数
        try:
            for item in params:
                if clone[item].kind != Parameter.POSITIONAL_OR_KEYWORD:
                    continue
                if clone[item].annotation is not type(params[item]):
                    raise typenotmatch
        except Exception as e:
            print(e)
            return False
        #检测返回值
        try:
            result=func(*args,**kwargs)
            if type(result) is not signal.return_annotation:
                raise typenotmatch
        except Exception as e:
            print(e)
            return False
        return result
    return decorator

@typeof
def test(a:int,b:str,tup:tuple) -> str:
    print("Hello 我成功执行了")
    return repr(a)+repr(b)
test(1,"2",(1,))