Python中装饰器使用方法整理

 更新时间:2023年08月23日 10:12:22   作者:小小草帽  
这篇文章主要介绍了Python中装饰器使用方法整理,装饰器是给现有的模块增添新的小功能,可以对原函数进行功能扩展,而且还不需要修改原函数的内容,也不需要修改原函数的调用,需要的朋友可以参考下

1 装饰器初阶

装饰器是一种设计模式,起到装饰的作用。就是在不修改函数代码的情况下,给函数增加一些功能。

1.1 没有使用装饰器的例子

现有一个返回字符串的函数,在不修改该函数的情况下,将返回的字符串转换为大写字符串。

def uppercase_decorator(func):  # 定义一个函数,其参数func是函数类型参数
    def inner():   # 嵌套函数inner
        s = func()  # 调用func()函数并将返回值赋值给s变量
        make_uppercase = s.upper()   # 将字符串转换为大写并赋值给make_uppercase
        return make_uppercase  # 结束嵌套函数func()调用,返回转换之后的字符串
    return inner  # 结束函数uppercase_decorator()调用, 返回嵌套函数inner
def say_hello():
    return 'hello world.'
if __name__=="__main__":
    say_hello2 = uppercase_decorator(say_hello)  # 调用uppercase_decorator()函数,实参是say_hello()函数,返回值say_hello2变量也是一个函数
    print(say_hello2())  # 调用say_hello2函数

1.2 使用装饰器

Python提供了装饰器注释功能,装饰器本质上是一个函数。 上述代码使用装饰器修改如下:

def uppercase_decorator(func):
    def inner():
        s = func()
        make_uppercase = s.upper()
        return make_uppercase
    return inner
@uppercase_decorator   # 使用@uppercase_decorator装饰器声明say_hello1()函数
def say_hello1():
    return "zhang cheng."
if __name__=="__main__":
    print(say_hello1())  # 使用装饰器不需要显式调用uppercase_decorator()函数,直接调用say_hello1()函数即可。

1.3 同时使用多个装饰器

一个函数可以有多个装饰器声明。

def uppercase_decorator(func):  # 将字符串转换成大写
    def inner():
        s = func()
        make_uppercase = s.upper()
        return make_uppercase
    return inner
def backet_decorator(func): # 给字符串添加括号
    def inner():
        s = func()
        make_bracket = '(' + s + ')'
        return make_bracket
    return inner
@backet_decorator
@uppercase_decorator
def say_hello():
    return 'hello world.'
if __name__=="__main__":
    print(say_hello())

1.4 给装饰器传递参数

装饰器本质上是函数,因此可以给装饰器传递参数。

def calc(func):  # 定义装饰器函数calc(func),其参数还是一个函数
    def wrapper(arg1):  # 定义嵌套函数wrapper(arg1),这个函数参数列表与装饰器要注释的函数参数列表一致
        return func(arg1)
    return wrapper
@calc
def square(n):
    return n * n
@calc
def abs(n):
    return n if n > 0 else -n
if __name__=="__main__":
    print(f"3's square is {square(3)}")
    print(f"-30's absulate is {abs(-30)}")

2 装饰器高阶

2.1 wraps(func)使用

需要使用包:

from functools import wraps

作用不改变使用装饰器的原有函数func的结构。(比如入参格式等,__name__, __doc__ );

因为一个原函数使用了装饰器之后,那么它就自动默认指向了装饰器函数的内存地址。 没有使用wraps装饰器,代码如下:

def add_decorator(func):
    def inner():
        """
        This is a decorator.
        :return:
        """
        func()
        pass
    return inner
@add_decorator
def add():
    """
    This is a add.
    :return:
    """
    pass
if __name__=="__main__":
    print(add.__name__)
    print(add.__doc__)

运行结果:

inner
        This is a decorator.
        :return:

结果分析:未加上@wraps()装饰器,结果得知add函数指向的是引用的装饰器函数inner.

使用wraps装饰器,代码如下:

from functools import wraps
def add_decorator(func):
    @wraps(func)
    def inner():
        """
        This is a decorator.
        :return:
        """
        func()
        pass
    return inner
@add_decorator
def add():
    """
    This is a add.
    :return:
    """
    pass
if __name__=="__main__":
    print(add.__name__)
    print(add.__doc__)

运行结果如下:

add
    This is a add.
    :return:

结果分析:加上@wraps()装饰器,结果得知add函数的结构保持不变.

提示:__name__输出函数名; __doc__输出对应函数内的注释信息。

2.2 多层嵌套使用

from functools import wraps
def add_deco(z):
    def add_decorator(func):
        @wraps(func)
        def inner(x, y):
            return z + func(x, y)
        return inner
    return add_decorator
@add_deco(z=10)
def add(x, y):
    return x + y
if __name__=="__main__":
    print(add(5, 10))

运行结果如下:

25

到此这篇关于Python中装饰器使用方法整理的文章就介绍到这了,更多相关Python中的装饰器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • 浅谈Python黑帽子取代netcat

    浅谈Python黑帽子取代netcat

    本篇文章通过一个小小实例给大家分析了Python黑帽子取代netcat的过程以及相关知识点,对此有兴趣的朋友可以学习下。
    2018-02-02
  • python解析基于xml格式的日志文件

    python解析基于xml格式的日志文件

    这篇文章主要为大家详细介绍了python如何解析基于xml格式的日志文件,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2017-02-02
  • Python中win32com模块的使用

    Python中win32com模块的使用

    本文主要介绍了Python中win32com模块的使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-01-01
  • Python正则表达式re.compile()和re.findall()详解

    Python正则表达式re.compile()和re.findall()详解

    re 模块提供了不少有用的函数,用以匹配字符串,下面这篇文章主要给大家介绍了关于Python正则表达式re.compile()和re.findall()的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
    2022-07-07
  • Python Selenium异常处理的实例分析

    Python Selenium异常处理的实例分析

    在本篇内容里小编给大家分享了关于Python Selenium异常处理的实例分析内容,对此有兴趣的朋友们可以学习参考下。
    2021-02-02
  • python 监听salt job状态,并任务数据推送到redis中的方法

    python 监听salt job状态,并任务数据推送到redis中的方法

    今天小编就为大家分享一篇python 监听salt job状态,并任务数据推送到redis中的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-01-01
  • python实现合并两个数组的方法

    python实现合并两个数组的方法

    这篇文章主要介绍了python实现合并两个数组的方法,实例分析了两种常用的合并数组的技巧,非常简单实用,需要的朋友可以参考下
    2015-05-05
  • python绘制横向水平柱状条形图

    python绘制横向水平柱状条形图

    这篇文章主要为大家详细介绍了python绘制横向水平柱状条形图,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-04-04
  • Python Serial串口基本操作(收发数据)

    Python Serial串口基本操作(收发数据)

    这篇文章主要介绍了Python Serial串口基本操作(收发数据),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11
  • PyCharm远程调试代码配置以及运行参数设置方式

    PyCharm远程调试代码配置以及运行参数设置方式

    这篇文章主要介绍了PyCharm远程调试代码配置以及运行参数设置方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-01-01

最新评论