python中的装饰器 | 易学教程

把一个函数当作参数传递给另一个函数，返回一个替代版的函数，本质上就是一个返回函数的函数，在不改变原函数的基础上给函数增加功能

语法塘

#装饰器可以被多个函数调用
def fun(z):           
    def inner():
        print('@@@@@@@')
        z()
    return inner()
    
@fun  #语法塘
def outher():
    print('hello')

@fun
def lala():
    print('haha')

#打印结果
@@@@@@@
hello
@@@@@@@
haha

**装饰器实现一个函数计时器
1.被装饰的函数如果有返回值的时候怎么办？
2.如何保留被装饰函数的函数名字和帮助文档信息？
**

import functools
import random
import string
import time
li = [random.choice(string.ascii_letters) for i in range(100)]
print(time.time())
def timeit(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs): #接收可变参数 和 关键字参数
        """这是一个装饰器timeit"""
        # 在函数执行之前
        start_time = time.time()
        # 执行函数
        res = fun(*args,**kwargs)
        # 在函数执行之后
        end_time = time.time()
        print('运行时间为:%.6f' %(end_time - start_time))
        return res
    return wrapper

@timeit
def con_add():
    s = ''
    for i in li:
        s += (i+'')
    print(s)
@timeit
def join_add():
    print(','.join(li))
    
con_add()
join_add()

@timeit
def fun_list(n):
    """这是一个fun_list函数"""
    return [2 * i for i in range(n)]
@timeit
def fun_map(n):
    return list(map(lambda x:x*2,range(n)))

fun_list(1000000)
fun_map(1000000)

print(fun_list.__name__)
print(fun_list.__doc__)


#打印结果
1576490387.797476
MQLyrUYLMoufBkaQNsCVBKznvvXotXPkbutpceqMWjAQivODbmUEbatDVWUemGImuCBAVWEOpppJIiRlqZBlmLNRAUxEPZqDHzwB
运行时间为:0.000027
M,Q,L,y,r,U,Y,L,M,o,u,f,B,k,a,Q,N,s,C,V,B,K,z,n,v,v,X,o,t,X,P,k,b,u,t,p,c,e,q,M,W,j,A,Q,i,v,O,D,b,m,U,E,b,a,t,D,V,W,U,e,m,G,I,m,u,C,B,A,V,W,E,O,p,p,p,J,I,i,R,l,q,Z,B,l,m,L,N,R,A,U,x,E,P,Z,q,D,H,z,w,B
运行时间为:0.000010

多个装饰器

ef decorator_a(fun):

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

def decorator_b(fun):

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


# 多个装饰器装饰函数 从上到下去执行的
@decorator_a
@decorator_b
def f(x):
    print('Gat in f')
f(1)

#执行结果
Get in inner_a
Get in inner_b
Gat in f

多个装饰器的应用

"""
['root','admin','redhat']
id+vip
多个装饰器的应用场景
会采用多个装饰器先验证是否登陆成功 再验证权限是否足够
"""
import inspect
login_session = ['root','admin','redhat']
import functools
def is_login(fun):
    @functools.wraps(fun)
    def warapper(*args,**kwargs):#('root',)
        if args[0] in login_session:
            temp = fun(*args,**kwargs)
            return temp
        else:
            print('Error:%s 没有登陆成功' %(args[0]))

    return warapper


def is_admin(fun):
    @functools.wraps(fun)
    #inspect_res 会返回一个字典
    # key:形参 value:对应的实参
    def wrapper(*args,**kwargs):
        inspect_res = inspect.getcallargs(fun,*args,**kwargs)
        print('inspect的返回值是:%s' %(inspect_res))
        if inspect_res.get('name') == 'root':
            temp = fun(*args,**kwargs)
            return temp

        else:
            print('not root user,no permisson add user')
    return wrapper

@is_login
@is_admin
def adduser(nam):
    print('add_user')

adduser('admin')

#打印结果
inspect的返回值是:{'nam': 'admin'}
not root user,no permisson add user

编写装饰器required_ints, 条件如下:
#1). 确保函数接收到的每一个参数都是整数;
2). 如果参数不是整形数，打印 TypeError:参数必须为整形

import functools
def required_ints(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):  # args=(1,2,..)
        for i in args:
            # if isinstance(i,int):
            #     pass
            # else:
            #     print('函数所有的参数并非都是int型')
            #     break
            if not isinstance(i, int):
                print('函数所有的参数并非都是int型')
                break
        else:
            res = func(*args, **kwargs)
            return res

    return wrapper
@required_ints
def add(a, b):
    return a + b
@required_ints
def mymax(a,b,c,d):
    return max(a,b,c,d)

print(add(1,9.9))

#打印结果
函数所有的参数并非都是int型
None

来源：CSDN

作者：房渊

链接：https://blog.csdn.net/fangyuan1997/article/details/103483594

标签

python装饰器

python函数

fun函数

python