Python装饰器中的@wraps的作用和使用

最新推荐文章于 2025-04-21 11:48:03 发布

Refrain__WG

最新推荐文章于 2025-04-21 11:48:03 发布

阅读量1.4w

点赞数 3

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/Refrain__WG/article/details/81013374

Python 专栏收录该内容

43 篇文章

订阅专栏

本文介绍Python装饰器@wraps的用法及其如何保持被装饰函数的元信息不变，避免因装饰器使用不当导致的错误。通过具体示例展示如何创建一个带有用户登录验证的装饰器。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python 装饰器中的@wraps的作用：

装饰器的作用: 在不改变原有功能代码的基础上,添加额外的功能,如用户验证等
@wraps(view_func)的作用: 不改变使用装饰器原有函数的结构(如__name__, __doc__)
不使用wraps可能出现的ERROR: view_func...endpoint...map...

使用方法：

# 创建验证用户登陆功能装饰器
from functools import wraps

def user_login_confirm(view_func):
    @wraps(view_func)
    def wrapper(*args, **kwargs):

        # 验证用户是否登陆 --从session中获取user_id
        user_id = session.get('user_id')

        user = None
        if user_id:
            try:
                from info.models import User
                user = User.query.get(user_id)
            except Exception as e:
                current_app.logger.error(e)
                return jsonify(errno=RET.DBERR, errmsg='查询用户失败')

        # 将user对象保存到给g变量(g对象的值在本次的请求中有效)
        g.user = user

        return view_func(*args, **kwargs)

    return wrapper

参考文章：https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584

-----END-----