Python——装饰器

1、装饰器：

把一个函数当作参数，返回一个替代版的函数
本质就是一个返回函数的函数

“在不改变原函数的基础上，给函数增加功能”

#装饰器
def desc(fun):
    def add_info():
        print('hello hahaha～')
        fun()
    return add_info
 
 
@desc   #给函数添加装饰器
def login():
    print('login...')
@desc
def logout():
    print('logout...')
 
login()
logout()   #调用函数

2.装饰器特点：

对修改是封闭的，对扩展是开放的

import time   #导入时间模块
 
def f1():
    print('This is a function...')
 
def f2():
    print('This is a function...')
 
def print_current_time(func):
    print(time.time())
    func()
 
print_current_time(f1)
print_current_time(f2)

import time



#装饰器
def decorator(func):  
    def wrapper():
        print(time.time())
        func()
    return wrapper
 
@decorator  #调用装饰器
def f1():
    print('This is a function...')
 
def f2():   #没有装饰器
    print('This is a function...')
 
#f = decorator(f1)
f1()

import time
 
def decorator(func):   ##wrapper里不写东西不能接收参数
    def wrapper():
        print(time.time())  ##打印系统运行的时间
        func()      ##调用传进来的函数，也就是f1()——>打印：This is a function
    return wrapper  ##如果不返回，则不会执行这个wrapper函数
 
@decorator
def f1():
    print('This is a function')
 
# f1 = decorator(f1)
f1()
 
 
(2)
def desc(fun):
    def add_info():
        print('hello world')
        fun()
    return add_info()
 
@desc
def login():
    print('login...')
 
#@desc
def logout():
    print('logout...')
 
结果
1554478622.5663533
This is a function
hello world
login...

“”
装饰器实现一个函数计时器
“”"
“”"
1.问题1:被装饰的函数有返回值
2.问题2:如何保留被装饰函数的函数名和帮助信息文档
“”"

import time
import random
import string
import functools
 
li = [random.choice(string.ascii_letters) for i in range(100)]
print(li)
 
def Timer(fun):
    """这是一个装饰器Timer"""
    # @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        """这是一个wrapper函数"""
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('运行时间为: %.5f' %(end_time - start_time))
        return res
    return wrapper
 
@Timer
def con_add():
    s = ''
    for i in li:
        s += (i + ',')
    print(s)
 
@Timer
def join_add():
    print(','.join(li))
 
@Timer
def fun_list(n):
    """这是fun_list函数"""
    return [i * 2 for i in range(n)]
 
@Timer
def fun_map(n):
    return list(map(lambda x:x*2,range(n)))
 
# con_add()
# join_add()
 
# print(fun_list(100))
# print(fun_map(100))
 
print(fun_list.__doc__)
print(fun_list.__name__)
 
print(time.ctime())

实例：函数运行时间

import time
import functools
 
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, 运行时间: %6f,运行返回值结果: %d' %(time.ctime(),fun.__name__,end_time - start_time,res))
        return res
    return wrapper
 
@add_log
def add(x,y):
    time.sleep(1)
    return x + y
 
add(1,2)

import functools
import inspect
 
def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        #inspect.getcallargs返回一个字典，key值是形参，value值
        #是对应的实参{'name':'root'}
        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('not root user!')
    return wrapper
 
@is_admin
def add_user(name):
    print('添加用户信息...')
 
 
def del_user(name):
    print('删除用户信息...')
 
add_user('root')

实例：

“”"

  编写装饰器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_int(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_int
     
.
@required_types(float, float)
def add(a, b):
    return a + b
 
print(add(1.1, 2.0))

.

@required_types(float, float)
def add(a, b):
    return a + b
 
print(add(1.1, 2.0))

.

@required_types(list)
def add(a, b):
    return a + b
 
print(add(1.1, 2.0))

3.多个装饰器

"""
# _*_ coding:utf-8 _*_
Name:09_多个装饰器.py
Date:19-3-31
Author:westos-wsp
Connect:wsp439@sina.com
Desc:
"""
 
 
def decorator_a(fun):
    print('Get in decorator_a')
 
    def inner_a(*args, **kwargs):
        print('Get in inner_a')
        res = fun(*args, **kwargs)
        return res
 
    return inner_a
 
 
def decorator_b(fun):
    print('Get in decorator_b')
 
    def inner_b(*args, **kwargs):
        print('Get in inner_b')
        res = fun(*args, **kwargs)
        return res
 
    return inner_b
 
 
@decorator_b
@decorator_a
def f(x):
    print('Get in f')
    return x * 2
 
f(2)

在这个函数还未执行的时候，解释器会从上到下的读取函数，读到@decorator_b @decorator_a在这个时候，会从上到下的执行decorator_b和decorator_a函数，这个顺序是按照代码从上到下的顺序执行的，执行结果为下

然后在进一步执行@decorator_b @decorator_a装饰器内部函数，这时候的执行顺序是按照装饰器被调用的顺序，也就是先执行@decorator_b然后执行@decorator_a，所以结果为

练习：

import functools
import inspect
 
def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        #inspect.getcallargs返回一个字典，key值是形参，value值
        #是对应的实参{'name':'root'}
        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('not root user!')
    return wrapper
 
login_session = ['root', 'redhat', 'westos']
 
def is_login(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        if args[0] in login_session:
            res = fun(*args,**kwargs)
            return res
        else:
            print('Error:%s未登录' %args[0])
    return wrapper
 
 
 
@is_login
@is_admin
def add_student(name):
    print('添加学生信息...')
 
add_student('linux')