【Python】使用裝飾器爲Python實現運行時類型檢查

背景

閒來無事想熟悉一下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) 報錯

感想

這只是一個遊戲之做，檢查類型會耗費一些性能，真正部署運行的項目是不須要類型檢查的，但能夠在測試的時候使用類型檢查來避免一些無謂的錯誤。