码迷,mamicode.com
首页 > 编程语言 > 详细

Python 中实现装饰器时使用 @functools.wraps 的理由

时间:2017-05-09 12:32:48      阅读:181      评论:0      收藏:0      [点我收藏+]

标签:replace   lan   logs   过程   为我   函数   ace   功能   ble   

Python 中使用装饰器对在运行期对函数进行一些外部功能的扩展。但是在使用过程中,由于装饰器的加入导致解释器认为函数本身发生了改变,在某些情况下——比如测试时——会导致一些问题。Python 通过 functool.wraps 为我们解决了这个问题:在编写装饰器时,在实现前加入 @functools.wraps(func) 可以保证装饰器不会对被装饰函数造成影响。比如,在 Flask 中,我们要自己重写 login_required 装饰器,但不想影响被装饰器装饰的方法,则 login_required 装饰器本身可以写成下面的样子:

def login_required_(func):
    @wraps(func)
    def decorated_view(*args, **kwargs):
        if current_app.login_manager._login_disabled:
            return func(*args, **kwargs)
        elif not current_user.is_authenticated:
            # return current_app.login_manager.unauthorized()
            return redirect(url_for("login.loginPage", next=request.url))
        return func(*args, **kwargs)

    return decorated_view

  

Python 中实现装饰器时使用 @functools.wraps 的理由

标签:replace   lan   logs   过程   为我   函数   ace   功能   ble   

原文地址:http://www.cnblogs.com/terrycy/p/6829891.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!