python装饰器

最新推荐文章于 2022-10-06 19:53:25 发布

liuzhang2012

最新推荐文章于 2022-10-06 19:53:25 发布

阅读量174

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python decorator装饰器

本文链接：https://blog.youkuaiyun.com/liuzhang2012/article/details/79506742

python 专栏收录该内容

1 篇文章

订阅专栏

转载自廖雪峰的官方网站 decorator装饰器

python中函数也是一个对象，而且函数对象可以被赋值给变量，所以变量也可以调用该函数

>>> def now():
...     print('2015-3-25')
...
>>> f = now
>>> f()
2015-3-25

函数对象有一个__name__属性，可以拿到函数的名称：

>>> now.__name__
'now'
>>> f.__name__
'now'

现在，假设我们要增强now()函数的功能，比如在函数调用前后自动打印日志，但又不希望修改now()函数的定义，这种在代码运行期间动态增加功能的方式，称为装饰器（Decorator）

本质上，decorator是一个返回函数的高阶函数，所以定义一个能打印日志的decorator:

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

一个decorator接受一个函数作为参数，

@log
def now():
    print('2015-3-25')

decorator分两种，一种传参，一种不传参。不传参可以想上面定义的那样。如果decorator需要传参，需要编写一个高阶函数。

def log(text):
    def decorator(func):
        def wrapper(*args, **kw):
            print('%s %s():' % (text, func.__name__))
            return func(*args, **kw)
        return wrapper
    return decorator

@log('execute')
def now():
    print('2015-3-25')

为了防止decorator装饰的函数的__name__改变，需要在decorator定义时加上@functools.wrags(func).一个完整decorator定义如下

import functools

def log(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

注意的地方https://weibo.com/ttarticle/p/show?id=2309404216708578595735