Python学习笔记--装饰器

当你要提升自己颜值的时候该如何做? 可以整容,但是能不能在不修改本身器官下来提升颜值?
答案就是装饰器

def add(x,y):
        print("I am working now")
        return x+y
  • 你编写了一段程序,但是你觉得功能不够,例如你希望在运行add的时候,增加一段解释性的输出.就可以这样子.
def deco(func):
      print("The value is ")
      return func
      print(deco(add(3,9)))
  • 输出是
    I am working now
    The value is
    12
  • 这样就做到了在没有改写代码的情况下增加了功能.可能你觉得这个例子有点简单,我们在看一个复杂点的.

  • 需求1:在执行add()方法的时候增加类型检查.
def requires_int(decorated):
      def inner(*args,  **kwargs):
          kwargs_value=[i for i in kwargs.values()]
          for arg in list(args)+kwargs_value:
              print(arg)
              if not isinstance(arg,int):
                  raise TypeError('%s only accepts integers as arguments'%decorated.__name__)
          return decorated(*args,**kwargs)
      return inner



      def add(x,y):
        """ Return the sum of x and y"""
        return x+y

      add=requires_int(add)
      add('3',5)

      TypeError: add only accepts integers as arguments
  • 但是有人疑问,写的太麻烦了,不如直接在代码上加一个类型检查. 但是如果再增加一个需求呢?

需求二:计算程序运行事件.

def addTime(decorated):
              def inner(x,y):
                  start=time.clock()
                  decorated(x,y)
                  end=time.clock()
                  addTime=(end-start)*1000
                  print("Time used",addTime)
                  return decorated(x,y)
              return inner
              add=addTime(add)
          Time used 0.03699999999999884
  • 如果再有新需求,再写一个装饰器就可以了.是不是很方便?
并且,两个装饰器还可以一起用.像这样
add=addTime(add)
            add=requires_int(add)
            add(4,8)
但是这样写感觉好麻烦,每次都要赋值一次,没关系,python为你专门准备了语法糖.其实可以这样写
@addTime
          @requires_int
          def add(x,y):
              """ Return the sum of x and y"""
              time.sleep(0.7)
              return x+y
感觉有点眼熟,Django的login_required()不就是这样写的吗.不仅是这样,Flask的路由也是装饰器写的@route.
需要注意的是,如果多个装饰器,运行顺序是从下到上.
瞬间觉得装饰器好有用,不好好学习不行了.

让我们再来学习一个实例. 需求3如果我们要写一个注册表装饰器.#每次运行一个函数,都会有记录这个函数#.

registers=[]
            def register(decorated):
                registers.append(decorated)
                return decorated

            @register
            def foo():
                return 3

            @register
            def bar():
                return 5

            for i in registers:
                print(i())
最后输出3,5. 但是有点问题是,我觉得应该不同的函数的register应该分开.这应该怎么办呢
答案是写一个装饰器类
class Register(object):
              def __init__(self):
                  self._functions=[]#定义一个列表来装被装饰的函数

              def register(self,decorated):#装饰函数
                  self._functions.append(decorated)
                  return decorated

              def run_all_function(self,  *args,  **kwargs):
                  return_values=[]
                  for func in self._functions:
                          return_values.append(func(*args,  **kwargs))
                  return return_values



                  a=Register()
                  b=Register()

                  @a.register
                  def foo(x=5):
                      return x

                  @b.register
                  def bar(x=4):
                      return x


                  @a.register
                  @b.register
                  def baz(x=7):
                      return x

                  print(a.run_all_function())
                  print(b.run_all_function())

                  [5, 7]
                  [4, 7]
这样就可以做到互相不影响,每一个register都是不同实例,数值不会混乱,是不是特别方便?
让我们总结一下今天的学习.
装饰器可以在不修改代码下增加功能
             装饰器可以叠加
             装饰器可以写成一个类