Python装饰器

Life is short, you need Python

最近在开发公司内部的一个性能网站，网站是diango+mongodb架设的，部分功能已经被其他同事实现，但某些地方的代码冗余太多，一点都不符合”生命苦短，快用Python”的格言。比如类似下面的代码

#!/usr/bin/python

def testfunc1(arg1,arg2):
    if arg1 == None or arg2 == None: return #1判断参数是否为None
    print "testfunc1 args===",arg1

def testfunc2(arg1,arg2):
    if arg1 == None or arg2 == None: return #2判断参数是否为None
    print "testfunc2 args===",arg1

if __name__ == '__main__':
    testfunc1('a','c')
    testfunc2(1,'')

以上代码在每一个方法执行时，都判断传入参数是否有None值，有的话就直接返回，方法有N多个，竟然每个方法都将#1,#2的判断语句重复写，代码冗余太严重了。那么有什么机制能将这种判断方法参数是否为空的同类操作抽离出来呢？自然的就想到了Python中的装饰器。

在优化代码时，需要考虑方法参数的个数是不定的，而*args的形式能解决这个问题。来看看优化后的代码

#!/usr/bin/python

def check_args(func):
    def wrapper(*args):
        if None in args: return
        return func(*args)
    return wrapper

@check_args
def testfunc1(arg1,arg2):
    print "testfunc1 args===",arg1

@check_args
def testfunc2(arg1,arg2):
    print "testfunc2 args===",arg1

@check_args
def testfunc3(arg1,arg2,arg3):
    print "testfunc3 args===",args

if __name__ == '__main__':
    testfunc1('a','c')
    testfunc2(1,'')
    testfunc3(2,None)

上述修改原有的方法基本不用更改，代码也简洁了许多.