python基础篇:什么是装饰器?装饰器有什么用?

本文涉及的产品
服务治理 MSE Sentinel/OpenSergo,Agent数量 不受限
可观测可视化 Grafana 版,10个用户账号 1个月
简介: python基础篇:什么是装饰器?装饰器有什么用?

上一篇介绍了python的函数,本文将介绍Python的装饰器,装饰器应用非常广泛,一定要好好掌握啊

什么是装饰器

装饰器是一种Python语言的特性,它允许在不修改已有函数的情况下,向函数添加额外的功能。装饰器本质上是一个函数,它接受一个函数作为参数,并返回一个新的函数。

装饰器应用场景

  • 记录函数的执行时间
  • 缓存函数的结果,以避免重复计算
  • 检查函数的参数是否合法
  • 为函数添加日志记录
  • 为函数添加事务处理
  • 为函数添加权限检查

简单的装饰器

以下是一个简单的装饰器的示例,它向函数添加了计时功能:

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print("Time elapsed: ", end_time - start_time)
        return result
    return wrapper

@timer
def my_function():
    time.sleep(2)

my_function()

在上面的示例中,timer是装饰器函数的名称。它接受一个函数作为参数,并返回一个新的函数wrapperwrapper函数计算函数执行的时间,并打印出来。@timer语法将my_function函数传递给timer装饰器,这意味着my_function函数将被timer装饰器包装。

my_function函数被调用时,它实际上是wrapper函数被调用。wrapper函数计算函数执行的时间,并打印出来。最后,wrapper函数返回my_function函数的结果。

这将打印Time elapsed: 2.000000238418579到控制台上。

带参数的装饰器

装饰器可以接受参数,以便在运行时自定义装饰器的行为。要创建带参数的装饰器,需要编写一个函数,该函数接受装饰器参数,并返回一个装饰器函数。

以下是一个带参数的装饰器的示例,它允许指定函数的重试次数:

import time

def retry(max_retries):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for i in range(max_retries):
                try:
                    result = func(*args, **kwargs)
                    return result
                except Exception as e:
                    print("Error:", e)
                    time.sleep(1)
            raise Exception("Max retries exceeded")
        return wrapper
    return decorator

@retry(max_retries=3)
def my_function():
    print("Trying...")
    raise Exception("Something went wrong")

my_function()

在上面的示例中,retry是带参数的装饰器函数的名称。它接受一个参数max_retries,并返回一个装饰器函数decoratordecorator函数接受一个函数作为参数,并返回一个新的函数wrapperwrapper函数尝试调用原始函数,如果发生异常,则等待1秒钟并重试,最多重试max_retries次。

@retry(max_retries=3)语法将my_function函数传递给retry装饰器,并指定max_retries参数为3。

my_function函数被调用时,它实际上是wrapper函数被调用。wrapper函数尝试调用原始函数,如果发生异常,则等待1秒钟并重试,最多重试3次。

这将打印以下内容到控制台上:

Trying...
Error: Something

@wraps()语法糖

@wraps()是一个装饰器,它用于将被装饰函数的元数据复制到装饰器函数中。这包括函数名称、文档字符串、参数列表等。使用@wraps()装饰器可以确保装饰器函数的元数据与原始函数的元数据相同,这对于调试和文档编写非常有用。

这个装饰器丢失了原来函数对象的一些属性,比如:__name____doc__等属性。使用wraps语法糖可以保留这些属性。

以下是一个使用@wraps()装饰器的示例:

from functools import wraps

def my_decorator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print("Before function")
        result = func(*args, **kwargs)
        print("After function")
        return result
    return wrapper

@my_decorator
def my_function():
    print("Function called")

print(my_function.__name__)

总结

装饰器可以接受参数,以便在运行时自定义装饰器的行为。要创建带参数的装饰器,需要编写一个函数,该函数接受装饰器参数,并返回一个装饰器函数。

目录
相关文章
|
6天前
|
程序员 测试技术 Python
深入理解Python中的装饰器
深入理解Python中的装饰器
18 1
|
13天前
|
Python
Python进阶---闭包和装饰器
Python进阶---闭包和装饰器
26 2
|
1天前
|
Python 缓存 开发者
Python中的装饰器:优雅而强大的代码增强工具
在Python编程中,装饰器是一种强大的工具,它允许开发者在不改变原有代码结构的情况下,增加额外的功能。本文将深入探讨Python中装饰器的概念、用法和实际应用,展示其在代码优化和增强方面的威力。
|
1天前
|
Python 设计模式 缓存
Python中的装饰器:提升代码可读性与可维护性
在Python编程中,装饰器是一种强大的工具,它可以提高代码的可读性和可维护性。本文将深入探讨装饰器的概念、用法以及实际应用场景,帮助读者更好地理解和利用Python中的装饰器。
|
1天前
|
Python 缓存
Python 中的装饰器:提升代码可读性与灵活性
Python 中的装饰器是一种强大的工具,它可以提高代码的可读性和灵活性。本文将深入探讨装饰器的概念、用法以及如何在实际项目中应用装饰器来简化代码、增加功能,以及解决常见的编程问题。
|
5天前
|
程序员 测试技术 Python
Python中的装饰器(Decorators) :深入解析与实战应用
Python中的装饰器(Decorators) :深入解析与实战应用
10 0
|
6天前
|
存储 缓存 Python
深入理解Python中的装饰器:原理与实战
深入理解Python中的装饰器:原理与实战
14 0
|
8天前
|
监控 Python
深入解析Python中的装饰器(Decorators)及其实用场景
装饰器是Python中一个强大且灵活的功能,它允许在不修改原有函数或类代码的情况下,动态地为其添加新的功能。本文不仅介绍了装饰器的基本概念、语法和工作原理,还通过具体实例展示了装饰器在日志记录、权限校验、性能分析等实用场景中的应用,旨在帮助读者深入理解装饰器的强大之处,并激发其在实际项目中的创新应用。
|
8天前
|
缓存 监控 数据安全/隐私保护
探索Python中的装饰器:一种强大的元编程工具
在Python编程中,装饰器是一个强大而优雅的元编程工具,它允许我们在不修改原有函数或类代码的情况下,为其添加新的功能或修改其行为。本文将带您深入了解Python装饰器的概念、工作原理及其在实际编程中的应用,通过示例演示如何创建和使用装饰器,并探讨其在代码复用、性能监控和日志记录等方面的强大作用。
|
9天前
|
监控 程序员 Python
Python装饰器:深入解析与应用
Python装饰器:深入解析与应用