当您使用装饰器修饰函数时,装饰器通常会返回一个新的函数对象,这个对象取代了原始函数在程序中的位置。因此,当您通过变量或直接调用装饰后的函数时,您实际上是在调用装饰器返回的新函数,这就是为什么您得到的函数名是装饰器的函数名。
要获取原始函数的名称,您可以在装饰器内部使用functools.wraps
装饰器。functools.wraps
接受一个函数作为参数,并返回一个新的装饰器函数,这个新函数具有与原始函数相同的元数据,包括函数名、文档字符串等。通过这种方式,您可以确保即使在装饰后,原始函数的名称和其他元数据也能被保留。
下面是一个使用functools.wraps
的示例:
from functools import wraps
def my_decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
# 装饰器逻辑
return func(*args, **kwargs)
return wrapper
@my_decorator
def original_function():
pass
print(original_function.__name__) # 输出 "original_function"
def QhStarEndTime(QhFunc):
@wraps(QhFunc) # 取到的将还是原始函数名
def QhWrapper(*args, **kwargs):
Star = time.time()
QhFuncName = QhFunc.__name__
QhJieGuo = QhFunc(*args, **kwargs)
End = time.time()
logger.info("【程序运行时间】程序*{}*运行耗时[{}]秒!QueHui!".format(QhFuncName,format(End-Star,'.2f')))
return QhJieGuo
return QhWrapper