迫于年纪大健忘，写了个简陋的装饰器，转换输入类型

可作为无参数装饰器
varnames 默认 None，就对传入的参数轮询一遍，符合 type hint 的，或者 funs / despatcher 没有的类型跳过。如果非 None，格式是个字符串列表（ list/tuple/set 等），列出需要检查的变量名。没做递归，所以 args，kwargs 传入的就没用了（这两个也没 type hint 吧？）
despatcher 默认 None，就是预置的 funs，是个 “类型：函数名” 的键值对 dict，可以外部传入。那些 to 什么的函数我就不提供了，放出来会害死人的，例如我 toStr 直接把列表类型用逗号串起来了，因为几乎只用在 sql 相关的函数，字段名是字串，但总是忘记上一个函数输出的是列表还是字串
checktypes 及 is_instance 是人家写的，这里就不贴了，代码在 https://stackoverflow.com/questions/55503673/how-do-i-check-if-a-value-matches-a-type-in-python 很长看上去也很严谨，我就没详细研究了，伸手了事
type hint 就不用说版本了吧，这个仅在 3.7.5 测试，手头也没更多 py 版本
有错误请指出，也帮我以后避免引入新问题
这段代码无版权，各取所需（谁用谁负责），至于前面第 4 点引用的代码，版权自理，我也不清楚

import inspect
from functools import singledispatch, wraps, partial
from checktypes import is_instance # 说明 4

def toStr(source)->str:
    return str(source)

def toInt(source)->int:
    return int(source)

def typeshint(func=None, *, varnames=None, despatcher=None):
    if func is None:
        return partial(typeshint, varnames=varnames, despatcher=despatcher)

    funs = { # 说明 3
        int: toInt,
        str: toStr,
        # list: toList,
        # tuple: toTuple,
        # set: toSet,
        # dict: toDict,
    }
    despatcher = despatcher if despatcher else funs
    sig = inspect.signature(func)
    parameters = sig.parameters
    varnames = varnames if varnames else parameters.keys()

    @wraps(func)
    def wrapper(*args, **kwargs):
        varkw = inspect.getfullargspec(func).varkw
        binds = sig.bind(*args, **kwargs)
        for n in varnames:
            annotation = parameters[n].annotation
            var = binds.arguments[n] if n in binds.arguments else kwargs[n]
            if is_instance(var, annotation) or (annotation not in despatcher): continue
            ff = despatcher.get(annotation)
            if n in binds.arguments:
                binds.arguments[n] = ff(var)
            else:
                binds.arguments[varkw][n] = ff(var)
        return func(*binds.args, **binds.kwargs)
    return wrapper

if __name__ == "__main__":
    @typeshint
    def myfunc(a:str, b:int=4, c=0):
        print(type(a), a, b, c)

    myfunc(20, '123', '0') # <class 'str'> 20 123 0

两三事：

python 真是什么都是对象，才发现类型也能做 keys

这个 varnames 报错查了几个小时，就因为最初 sig 是写在 wrapper 里面，在 wrapper 里面用 if varnames is None 就报错，要一起移到 wrapper 外面。唉，水平低，会抄会改不会写……

另外有两个类似的，一个自动适应格式的，例如 json/yaml 转字典、csv 转二维，诸如此类，一个根据 return type hint 转换输出类型的，因为魔改得太厉害（只有自己能用），害死人的代码就不放出来了，反正目的和思路差不多——先跑通再说

varnames

despatcher

函数

none

3 条回复 • 2020-11-24 22:13:43 +08:00