python inspect模块

作用

用于获取函数签名对象

基本用法

import inspect

# 待获取签名的函数

def func(x:int, y:str, z:list) -> None:

    pass

# 获取签名对象，包括函数名，参数等

sig = inspect.signature(func) 

print(sig)

 # 获取签名中的参数,返回有序字典

params = sig.parameters 

print(params)

# 通过循环并解构

for k, v in enumerate(params.items()):

    name, param = v

    print(param.annotation, param.kind, param.name)

# 获取签名中的返回值类型

ret = sig.return_annotation

print(ret)

例: 判断输入的实参是否为标注的类型

import datetime, time

import inspect

def check_str(func):

   def wrapper(*args, **kwargs):

       obj = inspect.signature(func)

       params = obj.parameters

       for k, v in enumerate(params.items()):

           name, param = v

           for i in args:

               if not isinstance(i, param.annotation):

                   print("{}: {} is not string".format(name, i))

                   raise TypeError

       ret = func(*args, **kwargs)

       return ret

   return wrapper

@check_str

def str_add(str1:str, str2:str):

   t1 = datetime.datetime.now()

   new_str = str1 + str2

   time.sleep(3)

   delta_time = (datetime.datetime.now() - t1).total_seconds()

   print(new_str, round(delta_time), sep="\n")

str_add(3, "4")