宜配屋

python装饰器常见使用方法分析

yipeiwu_com6年前 (2020-03-06)Python基础

本文实例讲述了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()

运行结果：

开始执行
in foo()
花费时间: 6.55415628267e-05

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

二、对在 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()

运行结果：

开始执行
in foo()
花费时间: 4.42208134735e-05

三、对在 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()

运行结果：