使用装饰器为Python实现运行时类型检查

来源：segmentfault

背景

闲来无事想熟悉一下Python的装饰器写法，就写了一个简单的类型检查装饰器。

思路

利用带参数的装饰器直接传入变量的预期类型，然后使用isinstance检查类型。

代码

def typeLimits(*types, **typesdict):
    def desc(func):
        def f(*args, **kwargs):
            for i, v in enumerate(types):
                if not isinstance(args[i], v):
                    raise Exception("Need a {} where {} is not.".format(v.__name__, args[i]))
            for parm in typesdict:
                if kwargs.get(parm, None) is None:
                    raise Exception("Need parameter {}.".format(parm))
                if not isinstance(kwargs[parm],typesdict[parm]):
                    raise Exception('Need a {} for {} where {} is not.'.format(typesdict[parm].__name__, parm, kwargs[parm]))
            return func(*args, **kwargs)
        return f
    return desc

@typeLimits(int, str)
def printOneIntAndOneString(n, s):
    print(n ** 2, s.upper())

@typeLimits(int, name = str)
def printOneIntAndName(n, name):
    print(n + n, name.upper())

printOneIntAndOneString(1,'abc') # 正确执行

# printOneIntAndOneString(1 + 2,3 + 4) 报错

printOneIntAndName(1, name = "aaa") 正确执行

# printOneIntAndName(1) 报错

# printOneIntAndName(1, name = 2.0) 报错

感想

这只是一个游戏之作，检查类型会耗费一些性能，真正部署运行的项目是不需要类型检查的，但可以在测试的时候使用类型检查来避免一些无谓的错误。