0

0

Python中如何使用装饰器?语法糖原理与应用实例

絕刀狂花

絕刀狂花

发布时间:2025-07-09 13:02:02

|

1025人浏览过

|

来源于php中文网

原创

python中的装饰器本质上是一个接收函数并返回新函数的特殊函数,它通过@符号实现语法糖机制,使得在不修改原函数代码的前提下扩展其行为。装饰器的执行顺序遵循从下往上的原则,但调用时最外层装饰器先执行;使用functools.wraps可保留原函数元数据,确保装饰后函数信息完整;带参数的装饰器通过三层嵌套结构实现,由装饰器工厂函数接收参数并返回实际装饰器;类也可作为装饰器,通常通过__call__方法实现功能扩展或修改。这些特性使装饰器成为封装日志记录、性能分析、权限校验等通用逻辑的强大工具

Python中如何使用装饰器?语法糖原理与应用实例

Python中的装饰器,本质上是一个特殊的函数,它能接收另一个函数作为参数,并返回一个新的函数,这个新函数通常在不修改原函数源代码的情况下,扩展或修改了原函数的行为。它提供了一种优雅的方式来封装可重用的代码逻辑,比如日志记录、性能分析、权限校验等。

Python中如何使用装饰器?语法糖原理与应用实例

解决方案

使用Python装饰器,最直观的方式就是利用它的“语法糖”特性,即在定义函数时,在其上方使用 @ 符号加上装饰器函数的名称。

Python中如何使用装饰器?语法糖原理与应用实例
import time
import functools

# 一个简单的日志装饰器
def log_calls(func):
    @functools.wraps(func) # 保持原函数的元数据,非常重要!
    def wrapper(*args, **kwargs):
        print(f"正在调用函数: {func.__name__},参数: {args}, {kwargs}")
        result = func(*args, **kwargs)
        print(f"函数 {func.__name__} 调用完成,结果: {result}")
        return result
    return wrapper

# 一个简单的性能测量装饰器
def time_it(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        start_time = time.perf_counter()
        result = func(*args, **kwargs)
        end_time = time.perf_counter()
        print(f"函数 {func.__name__} 执行耗时: {end_time - start_time:.4f} 秒")
        return result
    return wrapper

@log_calls
@time_it
def calculate_sum(a, b):
    """计算两个数的和"""
    time.sleep(0.1) # 模拟耗时操作
    return a + b

@log_calls
def greet(name):
    """向某人打招呼"""
    return f"你好,{name}!"

# 使用被装饰的函数
print("--- 第一次调用 calculate_sum ---")
total = calculate_sum(10, 20)
print(f"最终结果: {total}\n")

print("--- 第二次调用 greet ---")
message = greet("张三")
print(f"最终消息: {message}")

# 验证functools.wraps的作用
print(f"\ncalculate_sum 的文档字符串: {calculate_sum.__doc__}")
print(f"calculate_sum 的名称: {calculate_sum.__name__}")

上面这个例子展示了如何定义和使用两个不同的装饰器,并且它们可以堆叠使用。当你调用 calculate_sum(10, 20) 时,它会先被 time_it 装饰器处理,再被 log_calls 装饰器处理。执行顺序是从下往上(离函数定义最近的先执行),但实际执行时,最外层的装饰器(log_calls)的 wrapper 会先被调用,然后它会调用内层的 wrappertime_it 的),最终才调用原始函数 calculate_sum

立即学习Python免费学习笔记(深入)”;

@符号背后的魔法:Python装饰器的语法糖原理

当我第一次看到 @ 符号时,它给我的感觉就像是某种魔法,能凭空给函数加上新功能。但深入了解后,才发现它不过是Python提供的一个非常巧妙的语法糖。理解这背后的原理,能帮助我们更好地掌握装饰器,甚至在调试时茅塞顿开。

Python中如何使用装饰器?语法糖原理与应用实例

简单来说,@decorator_name 放在函数定义上方,等同于在函数定义完成后,执行 函数名 = decorator_name(函数名) 这行代码。

以我们之前的 log_calls 装饰器为例:

@log_calls
def my_function():
    print("这是我的函数")

这段代码在Python解释器看来,实际上是这样的:

def my_function():
    print("这是我的函数")

my_function = log_calls(my_function) # 这一步就是语法糖的本质

这意味着 log_calls 函数接收了 my_function 这个函数对象作为参数,然后 log_calls 返回了一个新的函数对象(也就是我们 wrapper),这个新的函数对象又被重新赋值给了 my_function 这个名字。所以,当你之后调用 my_function() 时,你实际上调用的是 log_calls 返回的那个 wrapper 函数,而不是最初定义的 my_function。这个 wrapper 函数内部才负责调用原始的 my_function,并在其前后添加额外的逻辑。

北极象沉浸式AI翻译
北极象沉浸式AI翻译

免费的北极象沉浸式AI翻译 - 带您走进沉浸式AI的双语对照体验

下载

这种机制的精妙之处在于,它让我们可以“无痛”地修改或扩展函数的行为,而不需要去动原始函数的代码。对于代码的维护和复用来说,简直是福音。

装饰器的进阶用法:带参数的装饰器与类装饰器

当你对基础装饰器运用自如后,自然会遇到更复杂的需求:比如,我希望我的装饰器能接收一些配置参数,或者我需要装饰的不是函数,而是整个类。别担心,Python的装饰器机制同样提供了优雅的解决方案。

带参数的装饰器

想象一下,你希望一个限速装饰器能够自定义每分钟允许的调用次数。这时候,装饰器本身就需要接收参数。这通常通过一个“装饰器工厂”函数来实现,它接收参数,然后返回一个真正的装饰器。

def rate_limit(calls_per_minute):
    # 这是一个装饰器工厂,它接收参数,并返回一个装饰器
    def decorator(func):
        last_called = {} # 简单模拟,实际生产环境需更健壮的机制

        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            current_time = time.time()
            if func.__name__ not in last_called:
                last_called[func.__name__] = []

            # 移除一分钟前的调用记录
            last_called[func.__name__] = [
                t for t in last_called[func.__name__] if current_time - t < 60
            ]

            if len(last_called[func.__name__]) >= calls_per_minute:
                print(f"函数 {func.__name__} 调用过于频繁,请稍后再试!")
                return None # 或者抛出异常

            last_called[func.__name__].append(current_time)
            return func(*args, **kwargs)
        return wrapper
    return decorator

@rate_limit(calls_per_minute=2) # 传入参数
def send_email(to_address, subject):
    print(f"发送邮件到 {to_address},主题:{subject}")
    return True

print("\n--- 测试带参数的装饰器 ---")
send_email("test@example.com", "Hello") # 第一次,成功
send_email("test@example.com", "Hello") # 第二次,成功
send_email("test@example.com", "Hello") # 第三次,失败(被限速)
time.sleep(61) # 等待超过一分钟
send_email("test@example.com", "Hello again") # 第四次,成功

这里,rate_limit 函数首先被调用,它接收 calls_per_minute 参数,然后返回 decorator 函数。接着,这个 decorator 函数作为真正的装饰器,接收 send_email 函数,并返回 wrapper 函数。所以,这实际上是三层嵌套:rate_limit -> decorator -> wrapper

类装饰器

虽然不如函数装饰器常见,但类同样可以作为装饰器。一个类作为装饰器时,它的实例通常会被设计成可调用的(通过实现 __call__ 方法),或者类本身作为装饰器工厂。

以下是一个简单的类装饰器示例,它给被装饰的类添加一个方法:

class AddMethodDecorator:
    def __init__(self, new_method_name):
        self.new_method_name = new_method_name

    def __call__(self, cls):
        # 这是一个简单的类装饰器,给类添加一个新方法
        def added_method(self_instance):
            return f"这是 {self_instance.__class__.__name__} 类通过装饰器添加的方法 '{self.new_method_name}'。"

        setattr(cls, self.new_method_name, added_method)
        return cls

@AddMethodDecorator("say_hello")
class MyClass:
    def __init__(self, name):
        self.name = name

    def get_name(self):
        return self.name

print("\n--- 测试类装饰器 ---")
obj = MyClass("Alice")
print(obj.get_name())
print(obj.say_hello()) # 调用通过装饰器添加的方法

这个 AddMethodDecorator 类在被实例化时(例如 @AddMethodDecorator("say_hello")),其 __call__ 方法会被调用,并将 MyClass 作为参数传入。__call__ 方法随后动态地向 MyClass 添加了一个名为 say_hello 的新方法,并返回修改后的 MyClass。类装饰器通常用于为类添加功能、注册类、修改类属性或方法等场景。

掌握了这些,你对Python装饰器的理解就更上一层楼了。它们确实是Python中一个强大而灵活的特性,能够极大地提升代码的模块化和复用性。不过,在使用时也要注意,过度复杂的装饰器链可能会增加调试的难度,所以保持简洁和单一职责原则依然很重要。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

713

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

625

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

738

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

617

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1235

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

547

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

574

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

696

2023.08.11

俄罗斯搜索引擎Yandex最新官方入口网址
俄罗斯搜索引擎Yandex最新官方入口网址

Yandex官方入口网址是https://yandex.com;用户可通过网页端直连或移动端浏览器直接访问,无需登录即可使用搜索、图片、新闻、地图等全部基础功能,并支持多语种检索与静态资源精准筛选。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1

2025.12.29

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 0.6万人学习

Django 教程
Django 教程

共28课时 | 2.5万人学习

SciPy 教程
SciPy 教程

共10课时 | 0.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号