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中内置数据类型list,tuple,dict,set的区别和用法

    Python中内置数据类型list,tuple,dict,set的区别和用法

    这篇文章主要给大家介绍了Python中内置数据类型list,tuple,dict,set的区别和用法,都是非常基础的知识,十分的细致全面,有需要的小伙伴可以参考下。
    2015-12-12
  • Selenium+Python自动化脚本环境搭建的全过程

    Selenium+Python自动化脚本环境搭建的全过程

    说到自动化测试,就不得不提大名鼎鼎的Selenium,Selenium 是如今最常用的自动化测试工具之一,支持快速开发自动化测试框架,且支持在多种浏览器上执行测试,下面这篇文章主要给大家介绍了关于Selenium+Python自动化脚本环境搭建的相关资料,需要的朋友可以参考下
    2021-09-09
  • python中使用 xlwt 操作excel的常见方法与问题

    python中使用 xlwt 操作excel的常见方法与问题

    这篇文章主要给大家介绍了关于python中使用 xlwt 操作excel的常见方法与问题的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-01-01
  • Python pyecharts Boxplot箱线图的实现

    Python pyecharts Boxplot箱线图的实现

    本文主要介绍了Python pyecharts Boxplot箱线图的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-05-05
  • matplotlib绘制直方图的基本配置(万能模板案例)

    matplotlib绘制直方图的基本配置(万能模板案例)

    本文主要介绍了matplotlib绘制直方图的基本配置(万能模板案例),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-04-04
  • python添加模块搜索路径方法

    python添加模块搜索路径方法

    下面小编就为大家带来一篇python添加模块搜索路径方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-09-09
  • Python数据结构dict常用操作代码实例

    Python数据结构dict常用操作代码实例

    这篇文章主要介绍了Python数据结构dict常用操作代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-03-03
  • python顺序的读取文件夹下名称有序的文件方法

    python顺序的读取文件夹下名称有序的文件方法

    今天小编就为大家分享一篇python顺序的读取文件夹下名称有序的文件方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07
  • python使用PythonMagick将jpg图片转换成ico图片的方法

    python使用PythonMagick将jpg图片转换成ico图片的方法

    这篇文章主要介绍了python使用PythonMagick将jpg图片转换成ico图片的方法,涉及PythonMagick模块操作图片的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-03-03
  • 用django设置session过期时间的方法解析

    用django设置session过期时间的方法解析

    这篇文章主要介绍了用django设置session过期时间的方法解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08

最新评论