引入装饰器,在函数执行的过程中,遇到错误的,直接截图,下面做了一个简单的demo实现
def log1(func):
@functools.wraps(func)
def wrapper(*arg,**kw):
print 'begin call %s' %time.ctime()
try:
func(*arg,**kw)
time.sleep(1)
except:
print 'end call %s' %time.ctime()
return wrapper
@log1
def now():
print 'ddd'
assert 1==2
now()
执行结果如下:
begin call Wed Feb 15 16:17:01 2017
ddd
end call Wed Feb 15 16:17:01 2017
替换:将except中的替换为截图函数,即可在出错的时候,做出处理。