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
add(1,2)    #调用函数add

练习：编写装饰器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_types(*kinds):
    def required(fun):
        @functools.wraps(fun)
        def wrapper(*args,**kwargs):
            for i in args:
                if not isinstance(i,kinds):
                    print('TypeError:参数必须为',kinds)
                    break
                    # raise TypeError('参数必须为%s,%s' %kinds)    #利用内置功能直接报错
            else:
                res = fun(*args,**kwargs)
                return res
        return wrapper
    return required

@required_types(int,int)
def add(a,b):
    return a + b
print(add(1.2,2.0))

二、多个装饰器同时使用

def decorator_a(fun):
    def inner_a(*args,**kwargs):
        print('Get in inner_a')
        return fun(*args,**kwargs)
    return inner_a

def decorator_b(fun):
    def inner_b(*args,**kwargs):
        print('Get in inner_b')
        return fun(*args,**kwargs)
    return inner_b

@decorator_b
@decorator_a
def f(x):
    print('Get in f')
    return x*2
f(1)

import functools
import inspect
login_tables = ['root','admin','westos']
def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        #inspect.getcallargs返回一个字典,key值:形参 value值:对应的实参
        inspect_res = inspect.getcallargs(fun,*args,**kwargs)
        print('inspect的返回值: %s' %inspect_res)
        if inspect_res.get('name') == 'root':
            res = fun(*args,**kwargs)
            return res
        else:
            print('Permission denied!')
    return wrapper


def login(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        if args[0] in login_tables:
            res = fun(*args,**kwargs)
            return res
        else:
            print('Error:无此用户')
    return wrapper

@login
@is_admin
def add_student(name):
    print('添加学生信息')
def del_student(name):
    print('删除学生信息')

add_student('root')