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.'