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

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

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
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值