PythonTip >> 博文 >> python

python中的函数修饰符

zihua 2014-01-26 15:01:41 点击: 1008 | 收藏


首先,什么是函数修饰符?函数修饰符就是对原有函数做一层包装。比如有以下两个函数:

 

def func1():
    print 'I am function func1'
def func2():
    print 'I am function func2'

 

现在我们想为这两个函数增加一段共同的操作,这时就可以使用函数修饰符。

首先我们定义一个修饰函数,先不做任何处理,仅仅打印一条横线:

 

def de(f):
    def _call_():
        print '-------------------------------'
        return f()
    return _call_

 

使用这个修饰符:

 

@de
def func1():
    print 'I am function func1'
@de
def func2():
    print 'I am function func2'
    
if __name__ == '__main__':
    func1()
    func2()

 

运行结果为:

 

E:\开发文档2\python\demo>修饰符.py

-------------------------------

I am function func1

-------------------------------

I am function func2

 

可以看到,修饰符起作用了。 

 

注意: 

1.修饰符等价于包装调用:

 

@de

def func1:

----- 等价于 ------

func1 = de(func1)

 

2.修饰函数必须返回一个“可调用对象”:

 

def de( f ):

    def _call_():

        return f()

    return _call_ # 返回的是一个函数体,而非调用_call_()

 

 

下一个问题:如果各个被修饰函数有不同的参数,怎么处理呢?

例如:

 

def func1(lst1,lst2):

    # 合并两个list,并打印各项

    for item in lst1+lst2:

        print item

def func2(dic):

    # 循环dict,打印各key-value

    for k, v in dic.items():

        print k, v

 

这两个函数参数不同,又都没有异常处理,现在我们想增加上,使用修饰符这样处理:

 

import sys
import traceback
def de(f):
    def _call_( *args, **kwargs ):
        try:
            return f( *args, **kwargs )
        except:
            print 'param type error'
    return _call_
@de
def func1(lst1,lst2 ):
    for item in lst1+lst2:
        print item
@de
def func2( dic ):
    for k, v in dic.items():
        print k, v
if __name__ == '__main__':
    func1( [1,2], [3,4] )
    #func2( {1:'a',2:'b'})
    func2( [1,2])

 

这里使用了可变参数*args和**kwargs,这样会把所有接收来的参数,原样不动的再转给原函数,是惯用法。 

以上程序运行结果: 

 

E:\开发文档2\python\demo>修饰符带参数.py

1

2

3

4

param type error

原文链接:http://lizhao6210-126-com.iteye.com/blog/2009092

作者:zihua | 分类: python | 标签: python 修饰符 | 阅读: 1008 | 发布于: 2014-01-26 15时 |