装饰器浅谈

装饰器

如需转发，请注明出处：小婷儿的python https://www.cnblogs.com/xxtalhr/p/10261902.html

有问题请在博客下留言或加作者微信：tinghai87605025 或 QQ ：87605025

python QQ交流群：py_data 483766429

一、概念

l　　装饰器其实就是一个以函数作为参数并返回一个替换函数的可执行函数（注：与闭包类似，参数是函数）；

l　　wrapper()函数的参数定义是(*args, **kw)，因此，wrapper()函数可以接受任意参数的调用。在 wrapper()函数内，首先打印日志，再紧接着调用原始函数；

 

二、练习

import time
import functools
def add():
    print('调用(add)函数时间:%s' % time.strftime('%Y-%m-%d %H:%M:%S'))
print('add=============')

 

结果：

　　add=============

# 装饰器函数
# 方法一：打印每一个函数调用时间
def log(func):
    def wrapper():
        # __name__可以获得函数名称
        print('调用%s函数时间:%s' % (func.__name__, time.strftime('%Y-%m-%d %H:%M:%S')))
        func()
    return wrapper
add_wrap = log(add)
add_wrap()
print('log=============')

结果：

　　调用add函数时间:2019-01-13 18:02:27

　　调用(add)函数时间:2019-01-13 18:02:27

　　log=============

# 方法一，每次写都要print('调用%s函数时间:%s'%(func.__name__, time.strftime('%Y-%m-%d %H:%M:%S')))
# 方法二：打印每一个函数调用时间
def mul():
    print('mul')
mul_wrap = log(mul)
mul_wrap()
# add不是原来的add函数了，只是名字相同
add = log(add)
add()
print('mul=============')

结果：

　　调用mul函数时间:2019-01-13 18:02:27

　　mul

　　调用add函数时间:2019-01-13 18:02:27

　　调用(add)函数时间:2019-01-13 18:02:27

　　mul=============

 

# 方法三：借助@的语法，将装饰器作用到指定的函数
@log  # sub = log(sub)
def sub():
    print('sub')
sub()
print('===============')

结果：

　　调用sub函数时间:2019-01-13 18:02:27

　　sub

　　===============

# 方法四：带参数装饰器
def divide_function(funct):
    def wrap(*kwargs, **kw):  # 可以传任一参数
        print('调用%s函数时间:%s' % (funct.__name__, time.strftime('%Y-%m-%d %H:%M:%S')))
        print('实际应用中，大多时候只改变print部分，其他都不变')
        return funct(*kwargs, **kw)  # 原始函数有返回值，所以return,若没有就返回none

    return wrap
@divide_function
def power_function(a, b):
    return a**b
f = power_function(2, 3)
print('调用函数：%s，f：%s'%(power_function.__name__,f))

结果：

　　调用power_function函数时间:2019-01-13 18:02:27

　　实际应用中，大多时候只改变print部分，其他都不变

　　调用函数：wrap，f：8

# 方法五：functools
def divide_function(funct):
    @functools.wraps(funct)#作用：wrapper.__name__ = func.__name__
    def wrap(*kwargs, **kw):  # 可以传任一参数
        print('调用%s函数时间:%s' % (funct.__name__, time.strftime('%Y-%m-%d %H:%M:%S')))
        print('实际应用中，大多时候只改变print部分，其他都不变')
        return funct(*kwargs, **kw)  # 原始函数有返回值，所以return,若没有就返回none

    return wrap
@divide_function
def power_function(a, b):
    return a**b
f = power_function(2, 3)
print('调用函数：%s，f：%s'%(power_function.__name__,f))

结果：

　　调用power_function函数时间:2019-01-13 18:02:27

　　实际应用中，大多时候只改变print部分，其他都不变

　　调用函数：power_function，f：8

三、进阶

编写一个 decorator（wrapper），能在函数调用的前后打印出'begin call'和'end call'的日志。

import functools

def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print('call %s():' % func.__name__)
        return func(*args, **kw)
    return wrapper

@log
def now():
    print('2019-01-13')

now()

def logger(text):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print('%s %s():' % (text, func.__name__))
            return func(*args, **kw)
        return wrapper
    return decorator

@logger('DEBUG')
def today():
    print('2019-01-13')

today()
print(today.__name__)

结果：

　　call now():

　　2019-01-13

　　DEBUG today():

　　2019-01-13

　　today

 

欢迎关注小婷儿的博客：

博客园：http://www.cnblogs.com/xxtalhr/

csdn：https://blog.csdn.net/u010986753

 

OCP培训说明连接：https://mp.weixin.qq.com/s/2cymJ4xiBPtTaHu16HkiuA

OCM培训说明连接：https://mp.weixin.qq.com/s/7-R6Cz8RcJKduVv6YlAxJA

 

小婷儿的python正在成长中，其中还有很多不足之处，随着学习和工作的深入，会对以往的博客内容逐步改进和完善哒。

重要的事多做几遍。。。。。。