在安卓审计项目里，我们使用python脚本来对APK进行扫描检测，为了统计每一步操作的耗时情况编写了一个简单的类，并通过下面的方式使用：
#coding:utf-8
import time
class TimeRecorder:
def __init__(self, name):
print(name + u"开始")
self.name = name
self.startTime = time.time()
def __del__(self):
print(u"{0}结束，耗时：{1}".format(self.name, time.time() - self.startTime))
def scan1():
t = TimeRecorder(scan1.func_name)
time.sleep(1)
return 1
print scan1()
运行后打印输出：
scan1开始
scan1结束，耗时：1.0
1
主要利用了类的构造函数和异构函数，实际上在scan1函数里创建的TimeRecorder对象并没有真实使用，需要统计耗时的函数开始都要声明一个TimeRecorder对象，虽然也比较简单，但是使用者有时会莫名其妙。 在学习了python的装饰器后，便有了更好的设计，让装饰器来统一调用即可，实现方式如下：
# 函数装饰器，让函数打印耗时
def deco(func):
def _deco():
t = TimeRecorder(func.func_name)
return func()
return _deco
@deco
def scan2():
time.sleep(1)
return 2
print scan2()
TimeRecorder对象不变，使用时调用者函数代码不变，只需要在前面那个@装饰器即可。
运行输出：
scan2开始
scan2结束，耗时：1.0
2
在上面的例子中，如果打印scan2的函数名：
print scan2.func_name
会发现输出的是_deco 这显然会影响开发者的使用习惯和判断，怎么办呢？ 这个时候就需要functools.wraps
from functools import wraps
def decorator2(func):
@wraps(func)
def wrapper(*args, **kwargs):
t = TimeRecorder(func.func_name)
return func(*args, **kwargs)
return wrapper
@decorator2
def scan3():
u"""扫描APK的权限。。。。"""
time.sleep(1)
return 3
print scan3.func_name, scan3.func_doc
print scan3()
运行输出：
scan3 扫描APK的权限。。。。
scan3开始
scan3结束，耗时：0.0
3
这个时候函数名和文档描述就是正确的了。
文档信息
- 本文作者：zhupite
- 本文链接：https://zhupite.com/python/Python%E6%89%93%E5%8D%B0%E5%87%BD%E6%95%B0%E7%94%A8%E6%97%B6%E7%9A%84%E4%B8%A4%E7%A7%8D%E6%96%B9%E6%B3%95-%E7%B1%BB%E7%9A%84%E6%9E%84%E9%80%A0%E6%9E%90%E6%9E%84%E5%87%BD%E6%95%B0%E4%B8%8E%E5%87%BD%E6%95%B0%E8%A3%85%E9%A5%B0%E5%99%A8.html
- 版权声明：自由转载-非商用-非衍生-保持署名（创意共享3.0许可证）