装饰器

[TOC]

函数

函数定义

def function():
    return

函数调用

function()

高阶函数

def func(function):
  print(function)
  return

嵌套函数

def function1:
  def function2:
    print("this is second function")
  function2()
function1()

---

装饰器

装饰器=高阶函数+嵌套函数

基础装饰器

假设有一个计算器函数1：

def calculator():
    x=float(input("please input x value:"))
    y=float(input("please input y value:"))
    z=x+y
    print(z)
calculator()

以上运行结果为：

计算器函数

现在要在不修改源代码及调用方式的情况下为计算过程添加装饰性文字，故必须使用装饰器：

# 装饰器函数
def decoration(function):
    def wrapper():
        print("----计算函数开始-------")
        function()
        print("----计算函数终止-------")
    return wrapper

# 语法糖
@decoration
# 计算器函数1
def calculator():
    x=float(input("please input x value:"))
    y=float(input("please input y value:"))
    z=x+y
    print(z)
calculator()

带形参的装饰器

现在，由于新需求产生，我需要计算一些带形参的函数：

def calculator_1(a,b):
    z=a+b
    print(z)

要想使其拥有计算器函数1相同的装饰，需要更改装饰器，但同时要兼容计算器函数1!

# 装饰器函数
def decoration(function):
    def wrapper(*args,**kwargs):
        print("----计算函数开始-------")
        function(*args,**kwargs)
        print("----计算函数终止-------")
    return wrapper

# 语法糖
@decoration
# 计算器函数1
def calculator():
    x=float(input("please input x value:"))
    y=float(input("please input y value:"))
    z=x+y
    print(z)

@decoration
# 计算器函数2
def calculator_1(a,b):
    n=a+b
    print(n)
calculator()
calculator_1(5,7)

运行结果：

形参装饰器

带返回值的装饰器

若函数返回值为，运行结果为None，如果将计算器函数1加入返回值，运行结果为：

image

None ？怎么会为空呢，原因是装饰器在函数运行时无返回 。

# 装饰器函数
def decoration(function):
    def wrapper(*args,**kwargs):
        print("----计算函数开始-------")
        return function(*args,**kwargs) #返回调用函数结果
        print("----计算函数终止-------")
    return wrapper

# 语法糖
@decoration
# 计算器函数1
def calculator():
    x=float(input("please input x value:"))
    y=float(input("please input y value:"))
    z=x+y
    print(z)
    return z

@decoration
# 计算器函数2
def calculator_1(a,b):
    n=a+b
    print(n)


print("this is printer:",calculator())
calculator_1(5,7)

image

为不同函数提供不同装饰

# 装饰器函数
def choose(way):
    def decoration(function):
        def wrapper(*args,**kwargs):
            if way==1:
                print("----计算函数开始1-------")
                return function(*args, **kwargs)
                print("----计算函数终止1-------")
            else:
                print("----计算函数开始2-------")
                function(*args, **kwargs)
                print("----计算函数终止2-------")


        return wrapper
    return decoration

# 语法糖
@choose(way=1)
# 计算器函数1
def calculator():
    x=float(input("please input x value:"))
    y=float(input("please input y value:"))
    z=x+y
    print(z)
    return z

@choose(way=2)
# 计算器函数2
def calculator_1(a,b):
    n=a+b
    print(n)


print("this is printer:",calculator())
calculator_1(5,7)

运行结果：

choose