python装饰器的各种使用方法

python 的装饰器,可以用来实现,类似spring AOP 类似的功能。一样可以用来记录某个方法执行前做什么,执行后做什么,或者用来记录日志,运行的时间等,更有甚者,用这个来做权限拦截,也未尝不可。从两个方面来描述python 的装饰模式:
1. 对普通方法的装饰
2. 对在 class 类中的方法的装饰,不需要给参数的情况
3. 对在 class 类中的方法的装饰,需要给参数的情况

一,对普通方法的装饰。比如,要计算一个一个方法执行的时间.
程序代码 程序代码

#coding:utf-8
import time
def timeit(func):
    def wrapper(*args, **kv):
        start = time.clock()
        print '开始执行'
        func(*args, **kv)
        end =time.clock()
        print '花费时间:', end - start
    return wrapper

@timeit
def foo():
    print 'in foo()'
    
if __name__=='__main__':
    foo()

可以看到,计算出了时间差。而不是像普通方法一样,写在一个函数里面实现。

二、对在 class 类中的方法的装饰,不需要给参数的情况
程序代码 程序代码

#coding:utf-8
import time
def timeit(func):
    def wrapper(*args, **kv):
        start = time.clock()
        print '开始执行'
        func(*args, **kv)
        end =time.clock()
        print '花费时间:', end - start
    return wrapper
    
class MySpendTime(object):
    def __init__(self):
        pass
    @timeit
    def foo(self):
        print 'in foo()'
spendtime=MySpendTime()
spendtime.foo()

三、对在 class 类中的方法的装饰,需要给参数的情况
程序代码 程序代码

#coding:utf-8
'''
Created on 2012-11-1
@author: yihaomen.com
'''

def UpdateUI(msg, step):
    print u"内容:", msg
    print u"步骤:到第%s步了" % step
    def dec(func):
        def wapper(self, *args, **kwargs):
            func(self,*args, **kwargs)
        return wapper
    return dec

class Command(object):  
    def Excute(self):
        self.Work1st()
        self.Work2nd()
        self.Work3rd()
          
    @UpdateUI("开始第一步","1")
    def Work1st(self):
        print "Work1st"    
    @UpdateUI("开始第二步", 2)
    def Work2nd(self):
        print "Work2nd"    
    @UpdateUI("开始第三步", 3)
    def Work3rd(self):
        print "Work3rd"
        
if __name__=="__main__":
    command = Command()
    command.Excute()


除非申明,文章均为一号门原创,转载请注明本文地址,谢谢!
[本日志由 轻舞肥羊 于 2012-11-01 05:23 PM 编辑]
文章来自: 本站原创
引用通告: 查看所有引用 | 我要引用此文章
Tags: python
相关日志:
评论: 0 | 引用: 0 | 查看次数: -
发表评论
昵 称:
密 码: 游客发言不需要密码.
内 容:
验证码: 验证码
选 项:
虽然发表评论不用注册,但是为了保护您的发言权,建议您注册帐号.