Python的基础知识之装饰器

一、基本概念

装饰器：给函数增加一些额外的功能，但是又不改变函数原来的调用，我们叫做装饰器Decorator，Decorator就是一个返回函数的高阶函数

tmpdir__17_9_7_15_04_07.jpg

二、使用场景

基础场景：按钮index = 0的时候进行发图片，index = 1的时候进行发发文字,基本功能实现

#功能函数
def ftp():
    print("发图片")
def fwz():
    print("发文字")
 # 业务逻辑
buttonIndex = 1
if buttonIndex == 0:
    ftp()
else:
    fwz()

现在有个额外的功能需求，就是在发送之前判断用户是否登录？
要解决这个需求，我们考虑有三种方式解决：
1、直接在代码里面增加一个校验的函数，然后在业务逻辑的时候调用，缺点：就是改变原来的业务逻辑

def checkIsLogin(): 
    print("登录验证......")
def ftp():
    print("发图片")
def fwz():
    print("发文字")
 # 业务逻辑
buttonIndex = 1
if buttonIndex == 0:
   checkIsLogin()
    ftp()
else:
   checkIsLogin()
    fwz()

2、增加函数闭包调用，缺点：代码太多，想是否有简便的方法

 # 功能函数
def checkIsLogin(func):  # 闭包的使用
    def inner():
           print("登录验证......")
           func()
     return inner
def ftp():
    print("发图片")

def fwz():
    print("发文字")

ftp = checkIsLogin(ftp)
fwz = checkIsLogin(fwz)

#业务逻辑
buttonIndex = 1
if buttonIndex == 0:
    ftp()
else:
    fwz()

3、采用装饰器的语法糖实现

 # 功能函数
def checkIsLogin(func):  # 闭包的使用
     def inner():
          print("登录验证......")
          func()
    return inner
@checkIsLogin   # ftp = checkIsLogin(ftp)就相当于@checkIsLogin 
def ftp():
     print("发图片")

@checkIsLogin  # fwz = checkIsLogin(fwz) 就相当于@checkIsLogin 
def fwz():
     print("发文字")

#业务逻辑
buttonIndex = 1
if buttonIndex == 0:
   ftp()
else:
    fwz()

三、装饰的常用操作

注意：写出装饰器语法糖的时候就已经执行装饰函数，并返回内部函数啦

1、装饰器装饰有参数的函数

def zsq(func):
    def inner(*args,**kwargs):
    print("*" * 30)
    func(*args,**kwargs)
return inner
@zsq
def pnum(num):
   print('打印参数---',num)
@zsq
def pnum2(num1,num2,num3):
   print('打印参数---',num1,num2,num3)
pnum2(1122,33,num3=4)

函数执行过程：
@zsq装饰pnum函数之后是指向inner 函数的地址，当执行inner函数的时候，传参数就是innner接受参数 执行func 就是执行pnum函数

2、装饰器自身有参数
因为装饰函数默认都会传入一个函数参数，如果想要让装饰器自身有参数，需要在外层增加一个函数，主要保证回来的是一个装饰器就可以

def getzsq(c):
    def zsq(func):
        def innner(*args, **kwargs):
            print("装饰器")
            print(c * 30)
            res = func(*args, **kwargs)
        return res
      return innner
    return zsq

 @getzsq("===")  #主要保证回来的是一个装饰器就可以
 def pnum(num):
       print(num)
  pnum(3)

3、装饰器有返回值

def computeSum(func):
    def inner(*args,**kwargs):
        print("检测参数")
        print(args,kwargs)
        res = func(*args,**kwargs)
        return res
     return inner

@computeSum
def pnum(num1,num2,num3):
     return num1 + num2 + num3

@computeSum
def pnum2(num):
    print("没有返回值",num)


resut = pnum(1,2,num3=3)
print(resut)
pnum2(2)    

函数执行过程：@computeSum函数装饰的时候返回inner函数的地址pnum = computeSum（pnum） ，执行pnum函数 就是执行就是inner函数，执行inner里面的func 才是真正执行pnum（）函数，返回值在返回给外界

4、装饰器的叠加
从函数的头部开始 向上装饰 执行 从代码的行数 从上到下

def print_line(func):
    print("------print_line-----")
    def inner():
        print("-----------")
        func()
        print("-----------")
    return inner
def print_star(func):
    print("*****print_star******")
    def inner():
       print("***********")
       func()
       print("***********")
    return inner

@print_line  #print_content =  print_line（print_content）
@print_star  # print_content =  print_star（print_content）
def print_content():
    print("刘子彬笔记")
print_content()

#执行结果
#*****print_star******
# ------print_line-----
# -----------
# ***********
# 刘子彬笔记
# ***********
# -----------

如果需要了解更多Python知识，请查看
Python的基础知识之生成器
Python的基础知识之装饰器
Python的基础知识之迭代器

最后赠言

学无止境，学习Python的伙伴可以多多交流