Python——装饰器练习

练习

练习一

import time
import functools

def log_kind(kind):
    def add_log(fun):
        @functools.wraps(fun)
        def wrapper(*args,**kwargs):
            start_time = time.time()
            res = fun(*args,**kwargs)
            end_time = time.time()
            print('<%s>[%s] 函数名: %s,运行时间: %.6f,运行返回值结果: %d' %(kind,time.ctime(),fun.__name__,end_time-start_time,res))
            return res
        return wrapper
    return add_log

@log_kind('debug')
def add(x,y):
    time.sleep(1)
    return x + y
print(add(1,2))

运行结果：

练习二

编写装饰器required_types, 条件如下：
1). 当装饰器为@required_types(int,float)确保函数接收到的每一个参数都是int或者float类型;
2). 当装饰器为@required_types(list)确保函数接收到的每一个参数都是list类型;
3). 当装饰器为@required_types(str,int)确保函数接收到的每一个参数都是str或者int类型;
4). 如果参数不满足条件， 打印 TypeError:参数必须为xxxx类型
注意：

使用示例：

代码如下：

import functools
def required_type(*kind):   #设置装饰器接受任意多个参数
    def required(fun):     #定义函数，接收参数
        @functools.wraps(fun)
        def wrapper(*args,**kwargs):
            for i in args:   #遍历所有参数
                if not isinstance(i,kind):    #判断每个参数的类型是否为函数中给定的并且传到装饰器的参数
                    print('TypeError:参数必须为',kind)     #不是，报错
            else:
                res = fun(*args,**kwargs)    
                return res
        return wrapper
    return required   #给装饰器返回指定的多个参数

@required_type(int,float)
def add(a,b):
    return a + b
print(add(1,2.2))

执行结果：

import functools
def required_type(*kind):   #设置装饰器接受任意多个参数
    def required(fun):
        @functools.wraps(fun)
        def wrapper(*args,**kwargs):
            for i in args:   #遍历所有参数
                if not isinstance(i,kind):    #判断每个参数的类型是否为函数中给定的并且传到装饰器的参数
                   raise TypeError('参数必须为%s,%s' %kind)
            else:
                res = fun(*args,**kwargs)
                return res
        return wrapper
    return required   #给装饰器返回指定的多个参数

@required_type(int,float)
def add(a,b):
    return a + b
print(add([1,2,3],2.2))