Python基础（6）——装饰器

装饰器：

 def w1(func):
     def inner():
         # 验证1
         # 验证2
         # 验证3
         return func()
     return inner

 @w1
 def f1():
     print 'f1'

如上例@w1，@函数名 是python的一种语法糖。作用是：

• 执行w1函数，并将 @w1 下面的 函数 作为w1函数的参数，即：@w1 等价于 w1(f1)
  所以，内部就会去执行：
      def inner:
          #验证
          return f1()   # func是参数，此时 func 等于 f1
      return inner     # 返回的 inner，inner代表的是函数，非执行函数
  其实就是将原来的 f1 函数塞进另外一个函数中
• 将执行完的 w1 函数返回值赋值给@w1下面的函数的函数名
  w1函数的返回值是：
     def inner:
          #验证
          return 原来f1()  # 此处的 f1 表示原来的f1函数
  然后，将此返回值再重新赋值给 f1，即：
  新f1 = def inner:
              #验证
              return 原来f1()

总而言之：@w1 等价于f1=w1(f1)

　　有参装饰器

 def auth(driver='file'):
     def auth2(func):
         def wrapper(*args,**kwargs):
             name=input("user: ")
             pwd=input("pwd: ")

             if driver == 'file':
                 if name == 'egon' and pwd == '':
                     print('login successful')
                     res=func(*args,**kwargs)
                     return res
             elif driver == 'ldap':
                 print('ldap')
         return wrapper
     return auth2

 @auth(driver='file')
 def foo(name):
     print(name)

 foo('egon')

 #就是在普通装饰器前面再加一层

　　装饰器补充：wraps

functools.wraps 则可以将原函数对象的指定属性复制给包装函数对象, 默认有 __module__、__name__、__doc__,或者通过参数选择。代码如下

 from functools import wraps

 def deco(func):
     @wraps(func) #加在最内层函数正上方
     def wrapper(*args,**kwargs):
         return func(*args,**kwargs)
     return wrapper

 @deco
 def index():
     '''哈哈哈哈'''
     print('from index')

 print(index.__doc__)