装饰器装饰函数后恢复原函数名--两种方法

装饰器装饰一个函数会改变函数的名字

def set_func(func):
    def call_func(*args, **kwargs):
        # 业务代码
        print('xx')
        return func(*args, **kwargs)

    return call_func

@set_func
def func_test():
    print("func_test")

if __name__ == '__main__':
    print(func_test.__name__)

## result:   call_func

方式一:

使用__name__重新赋值
def set_func(func):
    def call_func(*args, **kwargs):
        # 业务代码
        print('xx')
        return func(*args, **kwargs)

    call_func.__name__ = func.__name__
    return call_func

@set_func
def func_test():
    print("func_test")

if __name__ == '__main__':
    print(func_test.__name__)

## result:   func_test

方式二:

使用functools.wrap()
import functools


def set_func(func):
    @functools.wraps(func)
    def call_func(*args, **kwargs):
        # 业务代码
        print('xx')
        return func(*args, **kwargs)

    return call_func


@set_func
def func_test():
    print("func_test")

if __name__ == '__main__':
    print(func_test.__name__)

## result: func_test
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Python装饰器是一种特殊的Python函数,它可以在不修改被装饰函数源代码的情况下,为被装饰函数添加额外的功能。装饰器本质上是一个闭包,它接收一个函数作为参数,并返回一个新的函数对象。这个新的函数对象包装了原始函数,并在其前后执行额外的代码。通过使用装饰器,我们可以在不改变原函数结构的情况下,扩展函数的功能,比如添加日志记录、性能计时、权限验证等功能。 装饰器的返回值也是一个函数对象,所以它可以像普通函数一样被调用。装饰器可以用来装饰函数方法或者类,使其拥有额外的功能。装饰器的语法是在被装饰函数定义的上方加上@符号,后面紧跟装饰器函数称。 装饰器可以分为函数装饰器和类装饰器两种类型函数装饰器是最常见的一种形式,它是一个函数,接收一个函数作为参数,并返回一个新的函数对象。类装饰器是指以类的形式实现的装饰器,它必须实现`__call__`方法,并接收一个函数作为参数。 装饰器可以带有参数,这时候需要编写一个返回装饰器的高阶函数。高阶函数接收装饰器参数,并返回一个装饰器函数装饰器函数再接收被装饰函数,并返回一个新的函数对象。通过这种方式,我们可以传递不同的参数给装饰器,实现不同的功能。 在使用装饰器时,为了保留原函数的元信息,如函数、文档字符串等,可以使用`functools.wraps`修饰内部新函数,将新函数的属性设置为原始函数的属性,以便更好地还原被装饰函数的原始属性。 总结一下,Python装饰器是一种为函数或类添加额外功能的机制,它可以在不修改原函数源代码的情况下,通过闭包的方式实现功能的扩展。装饰器可以带有参数,并且可以用来装饰函数方法或类。通过使用装饰器,我们可以更加灵活地扩展和改变函数的行为。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值