python高手之路 笔记2装饰器专题 简明易懂

677 查看

python装饰器

创建装饰器

  • 装饰器本质上就是一个函数,这个函数接收其他函数作为参数,并将其以一个新的修改后的函数进行替换。
  • 最简单的装饰器就是本体函数(identify function)
    def identify(f):
      #A#
      return f
    然后就可以像下面这样使用这个装饰器:
    @identify
    def foo():
      return 'bar'
    它和下面的过程类似:
    def foo():
      return 'bar'
    foo = identify(foo)
    #A#处,可以把一批函数都需要进行的操作都放在这,再用装饰器的形式来装饰那批函数,来简化代码。

注册装饰器

  • _functions = {}
    def register(f):
      global _functions
      _functions[f.__name__] = foo
      return foo
    ##使用##
    @register
    def foo():
      return 'bar'
  • 在这个例子中,函数被注册并存储在一个字典里,以便后续可以根据函数名字提取函数

装饰器升级

  • import functools
    import inspect
    def check_is_admin(f):
      @functools.wraps(f)
      #使用functools使返回的wrapper函数的属性与原函数一致
      def wrapper(*args, **kwargs):
      #args得到函数传参
          func_args = inspect.getcallargs(f, *args, **kwargs)
          #返回一个将参数名字和值作为键值对的字典,不论参数的传递方式
          if func_args.get('username') != 'admin':
              raise Exception("This user is not allowed to get food")
          return f(*args, **kwargs)
          #return时执行函数,返回函数返回值,有点混,注意
      return wrapper
    ##使用##
    @check_is_admin
    def get_food(username, type='chocolate'):
      return type + " nom nom nom!"

python方法类装饰器

  • 在python3中不仅可以向方法传入该类的任意实例,还可以传入任何对象,只要它包含方法期望的属性
  • 静态方法 @staticmethod
  • 类方法@classmethod 第一个参数cls
  • 抽象方法 import abc; @abc.abstractmethod