Python装饰器

装饰器

什么是装饰器：

装饰器本质上是一个Python函数，它可以让其他函数在不需要做任何代码变动的前提下增加额外功能，装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

概括的讲，装饰器的作用就是为已经存在的函数或对象添加额外的功能。

语法：@

1.装饰器的原型

def printInfo():
   print("我是一个普通的函数")

printInfo()

如何在不允许修改printInfo函数的前提下，在执行printInfo函数前进行权限校验；执行后进行日志记录。

def check(func):

   def checkAndLog():
      print("函数执行前：权限校验")
      func()
      print("函数执行后：日志记录")
   return checkAndLog       #返回内部函数

def printInfo():
   print("我是一个普通的函数")

# printInfo()

newFunc = check(printInfo)
newFunc()        #执行内部函数

2.装饰器的定义和调用

#装饰器函数定义
def check(func):

   def checkAndLog():
      print("函数执行前：权限校验")
      func()
      print("函数执行后：日志记录")
   return checkAndLog       #返回内部函数（包装过的函数）
@check    #定义装饰器修饰的位置，用来修饰谁，相当于check(printInfo)
def printInfo():
   print("我是一个普通的函数")

printInfo()#执行函数前，会先执行装饰器

3.装饰器的嵌套

def zs1(func):
   def f():
      print("我是装饰器1，我来啦")
      func()
      print("我是装饰器1，我走啦")
   return f
def zs2(func):
   def f():
      print("我是装饰器2，我来啦")
      func()
      print("我是装饰器2，我走啦")
   return f

@zs2
@zs1      #自下而上逐步修饰，zs2会把zs1返回的函数进行修饰，一层套一层
def printInfo():
   print("我是一个普通的函数")

printInfo()

4.带参数的装饰器

如果原函数带有参数，那么返回的包装后的新函数也要带有参数，且参数需要一一对应

def tool(func):
   def log(name,pwd):       #如果log不传入参数，func将接收不到参数
      print(f"日志记录【{name}:{pwd}】")
      func(name,pwd)
   return log       #返回包装过的函数
@tool
def login(name,pwd):
   print(f"用户名:{name},密码:{pwd}")

login("admin",123456)

5.带返回值的装饰器

返回值和原函数定义时相同

def tool(func):
   def log(name,pwd):       #如果log不传入参数，func将接收不到参数
      print(f"日志记录【{name}:{pwd}】")
      state = func(name,pwd)
      return state
   return log       #返回包装过的函数
@tool
def login(name,pwd):
   print(f"用户名:{name},密码:{pwd}")
   if name == "admin" and pwd == 123456:
      return "登陆成功"
   return "登录失败"


res = login("admin",123456)
print(res)

6.定义通用的装饰器

增加装饰器的适用性

def tool(func):
   def log(*args,**kwargs):			#通过使用可变参数来增加装饰器的适用性
      print("日志记录前。。。")
      state = func(*args,**kwargs)
      print("日志记录后。。。")
      return state
   return log       #返回包装过的函数

@tool
def test():
   print("我是一个测试函数")

@tool
def login(name,pwd):
   print(f"用户名:{name},密码:{pwd}")
   if name == "admin" and pwd == 123456:
      return "登陆成功"
   return "登录失败"


res = login("admin",123456)
print(res)
test()