脚本执行过程中 遇到错误截图的demo

引入装饰器，在函数执行的过程中，遇到错误的，直接截图，下面做了一个简单的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中的替换为截图函数，即可在出错的时候，做出处理。