当前位置: 首页
业界动态
我被 Python 装饰器坑了三个月,直到搞懂它的"套路"

我被 Python 装饰器坑了三个月,直到搞懂它的"套路"

热心网友 时间:2026-04-20
转载

别再把装饰器当成高深莫测的高级语法来回避了

装饰器的本质,说到底就是一个函数包裹函数的技巧。想彻底掌握它,只需要搞清楚三件事:谁包裹谁、参数怎么传递、元数据怎么保留。

免费影视、动漫、音乐、游戏、小说资源长期稳定更新! 👉 点此立即查看 👈

先讲一个真实发生过的故事。

去年有个接口日志项目,需求是记录每个API的调用时间、请求参数和返回状态。团队里一位经验丰富的同事建议用装饰器,说写个@log标签就能搞定。听起来很美好,对吧?于是照着网上的文档抄了一段代码,运行——结果直接报错:TypeError: wrapper() missing 1 required positional argument

对着屏幕愣了半天。文档明明就是这样写的,问题出在哪里?

后来排查才发现,抄的那篇博客太老了,用的是Python 2的写法。到了Python 3,连个清晰的错误提示都没有,就扔给你一句“缺少参数”。这个细节差点让项目延期。也正是那次经历让人意识到,对装饰器这种工具,必须从原理到实战彻底搞清楚,不能停留在半懂不懂、照抄代码的阶段。

今天,就把那些踩过的坑、总结的经验、提炼的套路一次性讲清楚。不堆砌概念,不照本宣科,直接聚焦你在实际编码中会遇到的问题。

一、装饰器到底是什么?

很多资料会说“装饰器是一种闭包”,这话没错,但看完你可能还是不会写。

换个更直观的理解方式:装饰器就是函数的“包装膜”。

想象一下去水果店买水果。水果本身没变,但套上包装盒之后,功能就多了:可以保鲜,可以印上品牌Logo,还能加个防伪标签。装饰器干的就是这个事——在不改动原函数内部代码的前提下,为函数增添额外的功能。

最常见的应用场景有哪些?计时、登录校验、日志记录、性能监控。这些需求如果每个函数都手动写一遍,代码会变得冗长且难以维护。装饰器的价值就在于,让你只写一次核心逻辑,然后像贴标签一样应用到任何需要的地方。

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print(f"函数 {func.__name__} 耗时 {end - start:.4f} 秒")
        return result
    return wrapper

@timer
def slow_function():
    time.sleep(1)
    return "Done"

slow_function()

看,slow_function本身没有任何改动,但贴上@timer之后,就自动拥有了计时能力。这就是装饰器的核心魅力——增强函数功能,但不污染函数本身的纯净性。

二、它为什么能工作?

很多人会用装饰器,但未必清楚其背后的运作机制。看懂下面这段拆解,你就能自己创造装饰器了:

def slow_function():
    return "Done"

def timer(func):
    def wrapper():
        start = time.time()
        result = func()
        end = time.time()
        print(f"耗时 {end - start:.4f} 秒")
        return result
    return wrapper

# 这一行就是 @timer 的本质
slow_function = timer(slow_function)

装饰器的语法糖@timer,其实就是上面那行手动包装代码的简写形式。

所以,装饰器的本质可以归结为:一个高阶函数,它接收一个函数作为输入,经过包装后,返回另一个函数作为输出。

三、踩坑实录:那些让新手崩溃的瞬间

1. 坑一:*args, **kwargs写错一个符号

def my_decorator(func):
    def wrapper(*args, **kwargs):  # 星号必须带
        print("开始执行")
        return func(*args, **kwargs)  # 这里也是
    return wrapper

*args写成args,或者把**kwargs写成kwargs,Python通常会报一个比较隐晦的错误。关键在于,*是解包操作。func(*args)是把元组里的每个元素拆开传给函数;而func(args)则是把整个元组当作一个参数传进去,函数签名自然就对不上了。

自检方法很简单:用签名复杂的函数来测试你的装饰器。

@my_decorator
def complex_func(a, b=10, *args, **kwargs):
    print(f"a={a}, b={b}, args={args}, kwargs={kwargs}")

complex_func(1, 2, 3, 4, name="test")

2. 坑二:装饰器带参数,写成了套娃

def log(level="INFO"):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"[{level}] 调用 {func.__name__}")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@log("ERROR")
def run():
    pass

这里有个记忆诀窍:带参数的装饰器,本质是“工厂的工厂”。

拆解一下这三层各自负责什么:

  • 第一层log(“ERROR”)负责接收参数,并返回一个真正的装饰器函数。
  • 第二层decorator负责接收被装饰的函数。
  • 第三层wrapper才是真正包裹原函数、执行增强逻辑的地方。

所以,@log(“ERROR”)等价于run = log(“ERROR”)(run)

3. 坑三:装饰器改变了原函数的“身份”

@timer
def add(a, b):
    """两数相加,返回结果"""
    return a + b

print(add.__name__)   # 输出什么? wrapper  ← 函数名丢了
print(add.__doc__)    # 输出什么? None    ← 文档丢了

这个问题很关键。很多Web框架会根据函数名和文档字符串来生成路由或进行参数校验。元数据一旦丢失,轻则路由名变成莫名其妙的wrapper,重则导致整个应用行为异常。

解决方法只有一行代码:

import functools

def timer(func):
    @functools.wraps(func)  # 把原函数的元数据复制过来
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print(f"耗时 {end - start:.4f} 秒")
        return result
    return wrapper

加上@functools.wraps(func)这一行,原函数的__name____doc____annotations__等元数据就全部保留下来了。这行代码没有理由不写,务必养成习惯。

4. 坑四:装饰器返回了None

def bad_decorator(func):
    def wrapper(*args, **kwargs):
        print("做一些操作")
        func(*args, **kwargs)  # 没有return!
    return wrapper

@bad_decorator
def get_data():
    return [1, 2, 3]

result = get_data()
print(result)  # 输出:None ← 返回值丢了

记住一个铁律:装饰器内部的wrapper函数,永远要return原函数的执行结果。否则,被装饰的函数就失去了返回值。

四、实战:写一个带缓存的装饰器

import functools, time

def memoize(ttl=300):
    """
    带过期时间的内存缓存装饰器
    ttl: 缓存有效期(秒),默认5分钟
    """
    cache = {}

    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            key = str(args) + str(sorted(kwargs.items()))
            now = time.time()

            if key in cache:
                val, t, expire = cache[key]
                if now - t < expire:
                    print(f"命中缓存 (剩余 {expire - (now - t):.0f}s)")
                    return val

            result = func(*args, **kwargs)
            cache[key] = (result, now, ttl)
            print("缓存未命中,执行函数")
            return result
        return wrapper
    return decorator

@memoize(ttl=10)
def fetch_api(user_id, include_history=True):
    time.sleep(2)  # 假设每次请求耗时2秒
    return {"user_id": user_id, "data": [1, 2, 3]}

data1 = fetch_api(1001)  # 耗时2秒
data2 = fetch_api(1001)  # 0秒返回,命中缓存
data3 = fetch_api(1002)  # 耗时2秒,不同参数

同样的参数在10秒内重复调用,2秒的请求瞬间变成0秒返回。这个模式稍加扩展,配合Redis,就能做成一个分布式的缓存装饰器。

五、类装饰器:不是函数也可以被装饰

1. 单例模式装饰器

def singleton(cls):
    instances = {}

    @functools.wraps(cls)
    def get_instance(*args, **kwargs):
        if cls not in instances:
            instances[cls] = cls(*args, **kwargs)
        return instances[cls]
    return get_instance

@singleton
class DatabaseConnection:
    def __init__(self):
        print("创建数据库连接...")
        self.connected = True

db1 = DatabaseConnection()
db2 = DatabaseConnection()
db3 = DatabaseConnection()
print(db1 is db2 is db3)  # True,只初始化一次

虽然调用了三次DatabaseConnection(),但__init__只执行了一次。这种模式非常适合数据库连接池、全局配置管理器、线程池或日志实例等场景。

2. 路由注册装饰器

class Router:
    routes = {}

    @classmethod
    def route(cls, path):
        def decorator(func):
            cls.routes[path] = func
            print(f"注册路由: {path} -> {func.__name__}")
            return func
        return decorator

class MyApp(Router):
    @Router.route("/home")
    def home(self): return "Welcome home"

    @Router.route("/about")
    def about(self): return "About us"

print(MyApp.routes)

3. 多个装饰器的叠加顺序

@log_decorator
@timer_decorator
def process():
    time.sleep(0.1)
    return "Done"

装饰器是从下往上依次应用的。可以想象成从里到外穿衣服:先穿内衣@timer_decorator,再套外套@log_decorator

从调用角度看:process()先经过外套log_decorator,再进入内衣timer_decorator,最后才到达process函数本体。这也解释了为什么内层装饰器必须加@functools.wraps——它确保了外层装饰器看到的是真实的函数名,而不是一个wrapper

六、真实项目的两个实用场景

场景一:重试机制

import functools, time, logging
logger = logging.getLogger(__name__)

def retry(max_attempts=3, delay=1):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            for attempt in range(max_attempts):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    if attempt == max_attempts - 1:
                        raise
                    logger.warning(f"第{attempt+1}次失败,{delay}s后重试: {e}")
                    time.sleep(delay)
        return wrapper
    return decorator

@retry(max_attempts=3, delay=2)
def call_api(url):
    import random
    if random.random() < 0.7:
        raise ConnectionError("网络超时")
    return "成功"

场景二:权限校验

def requires_auth(permission):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            user = get_current_user()
            if permission not in user.permissions:
                raise PermissionError(f"需要权限: {permission}")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@requires_auth("admin")
def delete_user(user_id): pass

@requires_auth("finance")
def export_report(): pass

这样一来,权限校验逻辑和业务逻辑完全分离,修改一处不会影响另一处,代码的清晰度和可维护性大大提升。

七、行动框架

最后,给出一套可行的学习路径:

第一步,理解闭包的原理。装饰器背后站着的就是闭包——即函数可以记住并访问其定义时所在作用域的变量。理解了闭包,装饰器就是顺水推舟。

第二步,动手写。从最简单的@timer开始,重点跑通@functools.wraps这一步。然后尝试带参数的装饰器,再进阶到类装饰器。

第三步,读优质源码。Flask的@app.route、Django的@login_required、FastAPI的路径操作装饰器——这些都是工业级的高质量参考,能让你看到装饰器在真实项目中的优雅用法。

第四步,警惕过度封装。只有当同一段增强逻辑需要在多个地方重复使用时,才值得抽取成装饰器。如果只是某个函数里需要加一行日志,手动写一下就好。

说到底,别再把装饰器当成什么高深莫测的高级语法来回避了。它的本质就是一个函数包裹函数的技巧,搞清楚三件事就够了:谁包裹谁、参数怎么传递、元数据怎么保留。

下次在代码里再遇到@开头的符号,别下意识地跳过。花上五分钟,读一下它的实现,看看它给函数包裹了什么功能。你会发现,看懂别人的装饰器,比你想象的要简单得多。

来源:https://www.51cto.com/article/841096.html

游乐网为非赢利性网站,所展示的游戏/软件/文章内容均来自于互联网或第三方用户上传分享,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系youleyoucom@outlook.com。

同类文章
更多
笑不活了 AI李想在L9车顶再现一字马大劈叉

笑不活了 AI李想在L9车顶再现一字马大劈叉

理想汽车创始人李想发布AI创意视频,为北京车展预热 4月20日,理想汽车创始人李想在其社交平台发布了一段由人工智能生成的创意视频,为即将于4月24日正式开幕的北京国际车展进行预热造势。视频的核心主题聚焦于理想汽车旗下的旗舰车型——理想L9 Livis。 该视频一经发布便迅速引发网络热议,其中一个致敬

时间:2026-04-20 14:54
格局大变!2025年中国大陆电视出口超1亿台 北美退居第三

格局大变!2025年中国大陆电视出口超1亿台 北美退居第三

2025年中国大陆电视出口格局生变:中东非跃居首位,北美市场退居第三 2025年中国大陆电视出口的“成绩单”已经公布。数据显示,全年出口总量为1 0714亿台,同比下降3 1%;出口总额为1083亿元,同比下降4 2%。总量与总额双双微降,但数字背后,一场深刻的区域格局调整正在上演。 出口市场“大洗

时间:2026-04-20 14:43
雷军:小米冠名了ChinaGT、CTCC、CEC中国三大顶级汽车场地赛,几乎周周有赛事

雷军:小米冠名了ChinaGT、CTCC、CEC中国三大顶级汽车场地赛,几乎周周有赛事

小米冠名中国三大顶级汽车场地赛,车主巡游“排面夯爆” 今天上午,小米创办人、董事长兼CEO雷军在微博上分享了一则动态:小米中国超级跑车锦标赛ChinaGT的首站比赛,已经火爆开幕。作为这场赛事的冠名合作伙伴,现场还上演了特别一幕——30位小米车主驾驶着自己的爱车,在数万名观众的注视下,进行了列队巡游

时间:2026-04-20 14:42
2026 年上海靠谱网站建设公司精选榜单,专业建站服务商实力推荐

2026 年上海靠谱网站建设公司精选榜单,专业建站服务商实力推荐

2026 年年度十大上海网站建设公司推荐 对于正在筹备搭建高端企业官网、手握10万至20万预算的决策者们来说,筛选一家靠谱的服务商,无疑是项目成功的关键一步。这份榜单,正是为你们——企业负责人、部门经理、项目核心执行人——所准备的。它基于客观的行业信息与专业的甄选维度整理而成,旨在提供一个真实、可靠

时间:2026-04-20 14:36
MLGO 微算法科技的新型分布式量子算法模拟平台实现高效验证

MLGO 微算法科技的新型分布式量子算法模拟平台实现高效验证

在量子计算技术不断加速发展的背景下,如何突破单一量子处理器规模受限的问题,成为量子计算迈向实用化的重要方向之一。 当前,量子处理器的发展似乎陷入了一个“甜蜜的烦恼”:一方面,量子比特数量在稳步增长;另一方面,噪声、电路深度限制等问题依然如影随形。随着量子算法的规模日益膨胀,单台设备在资源和稳定性上逐

时间:2026-04-20 14:35
热门专题
更多
刀塔传奇破解版无限钻石下载大全 刀塔传奇破解版无限钻石下载大全
洛克王国正式正版手游下载安装大全 洛克王国正式正版手游下载安装大全
思美人手游下载专区 思美人手游下载专区
好玩的阿拉德之怒游戏下载合集 好玩的阿拉德之怒游戏下载合集
不思议迷宫手游下载合集 不思议迷宫手游下载合集
百宝袋汉化组游戏最新合集 百宝袋汉化组游戏最新合集
jsk游戏合集30款游戏大全 jsk游戏合集30款游戏大全
宾果消消消原版下载大全 宾果消消消原版下载大全
  • 日榜
  • 周榜
  • 月榜
热门教程
更多
  • 游戏攻略
  • 安卓教程
  • 苹果教程
  • 电脑教程