python使用装饰器捕获异常

    

        可以编写一个通用的捕获异常的装饰器, 当程序发生异常时可以继续执行后续动作。 尤其适合于使用大量断言的验证性程序。

        装饰器的实现原理使用了回调技术。

        如下所示, robust 是一个装饰器。 当在普通函数 func 加上 @robust 注解时, 即是给 func 加上了 robust 装饰。 当调用 func 时, 就会实际地执行 robust , 获取装饰后的函数 add_robust , 调用 add_robust 来完成实际的动作。  即调用:     func(*arg, **keyargs)  等效于  robust(simple)(*arg, **keyargs)

 

#-------------------------------------------------------------------------------
# Name:        deco.py
# Purpose:     demo of decoration in python
#
# Author:      qin.shuq
#
# Created:     27/10/2014
#-------------------------------------------------------------------------------

import traceback

def robust(actual_do):
    def add_robust(*args, **keyargs):
        try:
            return actual_do(*args, **keyargs)
        except:
            print 'Error execute: %s' % actual_do.__name__
            #traceback.print_exc()
    return add_robust

@robust
def simple():
    return 5 / 0

@robust
def readFile(filename):
    f = open(filename, "r")
    print len(f.readlines())
    f.close()

def add(a,b):
    return int(a)+int(b)

@robust
def assertSumIsPositive(*args):
    sum = reduce(add, *args)
    assert sum >= 0

@robust
def checkLen(**keyargs):
    if len(keyargs) < 3:
        raise Exception('Number of key args should more than 3.')

if __name__ == '__main__':
    simple()
    readFile("UnexistFile.txt")
    assertSumIsPositive(1,2,-3,-4)
    checkLen(a=5,b=2)
    print 'Yet still reach here.'

 

posted @ 2014-10-29 18:44  琴水玉  阅读(6746)  评论(0编辑  收藏  举报