文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

如何使用Python装饰器修改函数行为

2023-06-29 17:52

关注

这篇文章主要介绍了如何使用Python装饰器修改函数行为的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇如何使用Python装饰器修改函数行为文章都会有所收获,下面我们一起来看看吧。

Python 中的装饰器是什么

装饰器在 Python中是一个非常强大和有用的工具,因为它允许程序员修改函数或类的行为。装饰器允许我们包装另一个函数,以扩展包装函数的行为,而无需修改基础函数定义。这也被称为元编程,因为程序本身在程序运行时会尝试修改自身的另一部分。

装饰器是语法糖: 在代码中利用更简洁流畅的语法实现更为复杂的功能。

我们知道,Python 一切皆对象。这意味着 Python 中的函数可以用作参数或作为参数传递。一等函数的属性:

让我们看一个这样的例子。

def hello():    print('Welcome to Python Decorator!')another_hello = hello()another_hello# Welcome to Python Decorator!

定义了一个 hello() 函数,然后将 hello 函数分配给 another_hello 变量,然后调用这个变量,得到的结果是 hello 函数被执行。

既然 Python 中的函数是对象,那么除了可以简单的调用之外,就可以把函数作为对象传递给另一个函数。

def print_welcome():    print('Welcome to Python Decorator!')def print_hello(func):    def inner():        print('Hello!')        func()    return innerdecorated = print_hello(print_welcome)decorated()# Hello!# Welcome to Python Decorator!

语法糖

但是,上面的代码使用了内部函数我们可以通过简单地用装饰器函数 print_hello() 来装饰 print_welcome() 函数。

装饰器可以简化我们的操作。功能完全一样,但它的代码更简洁。即通过 @ 符号简化装饰器的使用,如下所示:

def print_hello(func):    def inner():        print('Hello!')        func()    return inner@print_hellodef print_welcome():    print('Welcome to Python Decorator!')print_welcome()# Hello!# Welcome to Python Decorator!

通过这样做,我们能够消除将一个函数显式传递到另一个函数中的使用。Python 装饰器隐式处理这一点。

使用 Python 装饰器修改函数行为

使用 Python 装饰器对函数进行计时

为了演示它们的实用性,让我们构建一个函数,该函数采用另一个函数并对其执行进行计时。在这里,使用装饰器的好处是它允许我们遵循 DRY 编程原则。

装饰器可用于测量函数执行所需的时间。 如果你定义一个简单的睡眠函数,以计算该函数的运行时。

import timedef timeit(func):    def timed():        start = time.time()        result = func()        end = time.time()        print(f'Program took {(end - start) * 1000}s to run')        return result    return timed@timeitdef print_welcome():    print('Welcome to Python Decorator!')print_welcome()# Welcome to Python Decorator!# Program took 0.0s to run

分析一下上面的代码:

当我们将此装饰器函数应用于我们的函数 print_welcome() 时,首先会返回欢迎问候语,然后显示执行时间。

使用 Python 装饰器将有用信息记录到终端

与上面的例子类似,我们可以在程序运行时使用装饰器将有用的信息打印到终端。例如,我们可能想知道正在运行哪个函数以及当前时间。也可以使用装饰器传递到日志文件:

from datetime import datetimedef log_info(func):    def inner():        print(f'Starting run at {datetime.now()}')        print(f'Running {func.__name__}')        func()    return inner@log_infodef print_welcome():    print('Welcome to Python Decorator!')print_welcome()# Starting run at 2022-03-27 23:26:38.473310# Running print_welcome# Welcome to Python Decorator!

在上面的示例中,在运行函数之前,我们的装饰器打印当前日期和时间以及将要运行的函数的名称。如果您正在运行较长的脚本,并且只是想知道程序的位置,这可能很有用。

Web app 中使用的装饰器

让我们以 Web 应用程序的用例为例。当您在 Flask 中构建 Web 应用程序时,您总是会编写 url 路由。 每条路线都是 Web 应用程序中的特定页面。 打开页面 /about 可能会调用 about_page() 方法。

@app.route("/about")def about_page():  return "Website about nachos"

将参数传递给 Python 装饰器

到目前为止,您已经学习了如何创建一些有用的 Python 装饰器。然而,这些装饰器都没有传入参数。在本节中,您将学习如何创建接受参数的 Python 装饰器。

为此,我们将允许在 Python 语法魔术解压缩。使用 func_name(*args,**kwargs),它将解压缩所有参数和所有关键字参数。通过在装饰器中使用它,可以确保装饰器将接受任意数量的参数或关键字参数。这使得它们在重复使用时更加实用。

def print_function_name(func):    def inner(*args, **kwargs):        print(f'Running {func.__name__}...')        return func(*args, **kwargs)    return inner@print_function_namedef add_nums(a, b):    print(a + b)add_nums(1, 2)# Running add_nums...# 3

上述方法的美妙之处在于它同时接受位置和关键字参数。因此,即使我们以以下任何格式执行该函数,该函数也将运行:

使用多个 Python 装饰器

关于 Python 装饰器的一个有趣的方式是:可以同时使用多个装饰器。这意味着您可以将多个装饰器应用于单个函数。为了理解这一点,来看一个例子:

def one(func):    def inner(*args, **kwargs):        print('1')        return func(*args, **kwargs)    return innerdef two(func):    def inner(*args, **kwargs):        print('2')        return func(*args, **kwargs)    return inner@one@twodef speak(text):    print(text)speak('Hello')# 1# 2# Hello

我们的装饰器函数所做的唯一事情就是打印出数字 1 和数字 2。通过将装饰器 @one 放在 @two 之前,您可以将 two()包装的函数包装为 one()。为了说明这一点,您可以切换顺序以查看如何修改行为:

# Changing decorator order@two@onedef speak(text):    print(text)speak('Hello')# 2# 1# Hello

通过首先放置 @two 装饰器,该函数成为最外层的函数。

关于“如何使用Python装饰器修改函数行为”这篇文章的内容就介绍到这里,感谢各位的阅读!相信大家对“如何使用Python装饰器修改函数行为”知识都有一定的了解,大家如果还想学习更多知识,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯