Python学习笔记--装饰器

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都是不同实例,数值不会混乱,是不是特别方便?

让我们总结一下今天的学习.

             装饰器可以在不修改代码下增加功能
             装饰器可以叠加
             装饰器可以写成一个类