专栏首页python3python中的装饰器decorator

python中的装饰器decorator

python中的装饰器

装饰器是为了解决以下描述的问题而产生的方法

我们在已有的函数代码的基础上,想要动态的为这个函数增加功能而又不改变原函数的代码

例如有三个函数:

def f1(x):
    return x
def f2(x):
    return x*x
def f3(x):
    return x*x*x

而我们想为这三个函数增加一个函数调用打印功能 类似print("call f1()")

如果我们直接修改的话,需要对每个函数的内部进行改写。

所以为了简化代码,我们可以使用python内置的@装饰器的方法,可以做到修饰函数的功能

Python的 decorator 本质上就是一个高阶函数,它接收一个函数作为参数,然后,返回一个新函数。

装饰器可以极大地简化代码,避免每个函数写重复性代码

不带参数的decorator

例如我们可以编写一个@log可以打印函数调用的装饰器

def log(f):
    def fn(x):
        print ('call ' + f.__name__ + '()...')
        return f(x)
    return fn

然后我们可以测试一下

@log
def f1(x):
    return x

a=f1(1)
print(a)

结果

call f1()...
1

但是,对于参数不是一个的函数,调用将报错:

@log
def add(x, y):
    return x + y
print (add(1, 2))
Traceback (most recent call last):
  File "D:/pythonwork/cvworktest/final/practice.py", line 12, in <module>
    print (add(1, 2))
TypeError: fn() takes 1 positional argument but 2 were given

这是因为add() 函数需要传入两个参数,但是 @log 写死了只含一个参数的返回函数。

要让 @log 自适应任何参数定义的函数,可以利用Python的 *args 和 **kw,保证任意个数的参数总是能正常调用:

def log(f):
    def fn(*args, **kw):
        print ('call ' + f.__name__ + '()...')
        return f(*args, **kw)
    return fn

现在,对于任意函数,@log 都能正常工作。

带参数decorator

上面的@log不带任何参数,同样我们可以编写一个带参数的decorator

例如如果有的函数非常重要,希望打印出'[INFO] call xxx()...',有的函数不太重要,希望打印出'[DEBUG] call xxx()...',这时

log函数本身就需要传入'INFO'或'DEBUG'这样的参数,类似这样:

@log('DEBUG')
def my_func():
    pass

我们把它翻译成高阶函数就是这样:

my_func = log('DEBUG')(my_func)

展开:

log_decorator = log('DEBUG')
my_func = log_decorator(my_func)

又相当于:

log_decorator = log('DEBUG')
@log_decorator
def my_func():
    pass

所以,带参数的log函数首先返回一个decorator函数,再让这个decorator函数接收my_func并返回新函数:

def log(content):
    def log_decorator(f):
        def fn(*args, **kw):
            print ('[%s] %s()...' % (content, f.__name__))
            return f(*args, **kw)
        return fn
    return log_decorator

@log('DEBUG')
def test():
    pass
print (test())

结果:

[DEBUG] test()...
None

decorator的注意事项:

经过@decorator“改造”后的函数,和原函数相比,除了功能多一点外,还有很重要的一点就是函数自身的改变

在没有decorator的情况下,打印函数名:

def f1(x):
    pass
print (f1.__name__)

结果:

f1

有decorator的情况下,再打印函数名:

def log(f):
    def wrapper(*args, **kw):
        print ('call...')
        return f(*args, **kw)
    return wrapper
@log
def f2(x):
    pass
print (f2.__name__)

结果:

wrapper

由于decorator返回的新函数函数名已经不是'f2',而是@log内部定义的'wrapper'。这对于那些依赖函数名的代码就会失效。decorator还改变了函数的__doc__等其它属性。如果要让调用者看不出一个函数经过了@decorator的“改造”,就需要把原函数的一些属性复制到新函数中:

def log(f):
    def wrapper(*args, **kw):
        print ('call...')
        return f(*args, **kw)
    wrapper.__name__ = f.__name__
    wrapper.__doc__ = f.__doc__
    return wrapper

这样写decorator很不方便,因为我们也很难把原函数的所有必要属性都一个一个复制到新函数上,所以Python内置的functools可以用来自动化完成这个“复制”的任务:

import functools
def log(f):
    @functools.wraps(f)
    def wrapper(*args, **kw):
        print ('call...')
        return f(*args, **kw)
    return wrapper

注意:对于函数的参数信息哦我们无法确定,因为装饰器与原函数的参数名不一定一样

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Python 迭代器、生成器和列表解析

    迭代器在 Python 2.2 版本中被加入, 它为类序列对象提供了一个类序列的接口。 Python 的迭代无缝地支持序列对象, 而且它还允许迭代非序列类型, ...

    用户2398817
  • python开发第六篇--递归函数和面

    面向过程的程序设计的核心是过程(流水线式思维),过程即解决问题的步骤,面向过程的设计就好比精心设计好一条流水线,考虑周全什么时候处理什么东西。

    用户2398817
  • 论文式编程

    文学编程(Literate programming)的一些概念,上个世纪 70 年代就有人提出来了。

    用户2398817
  • Python中的正则表达式

    句点(.)符号匹配除了换行符\n以外的任何字符。无论字母、数字、空格(不包括\n换行符)、可打印字符、不可打印字符,使用.都可以匹配。

    用户2398817
  • Python 的异步 IO:Async

    首先,为了便于测试,我们用 Python 内建的 http 模块,运行一个简单的 HTTP Server。

    用户2398817
  • Python基础练习100题 ( 11

    这十道题的代码在我的github上,如果大家想看一下每道题的输出结果,可以点击以下链接下载:

    用户2398817
  • Flask 扩展系列之 Flask-R

    Flask-RESTful是一个Flask的扩展,它增加了对快速构建REST APIs的支持。它是一种轻量级的抽象,可以与现有的ORM/库一起工作。Flask-...

    用户2398817
  • python setup.py 浅析

    对于所有 packages 列表里提到的纯 Python 模块做处理 需要在 setup 脚本里有一个包名到目录的映射。 默认对于 setup 脚本所在...

    用户2398817
  • 学习笔记-小甲鱼Python3学习第十八

                print('传递进来的' + name + '叫实参,因为它是具体的参数值')

    用户2398817
  • Python中的文本和字节序列

    原因在于这个latin1不会对中午编码,所以我们需要处理这个EncodeError,处理如下:

    用户2398817

扫码关注云+社区

领取腾讯云代金券