python——函数之装饰器

1 问题

　　实际生活中，我们很难一次性就把一个函数代码写得完美无缺。当我们需要对以前的函数添加新功能时，我们应该怎么做？

2 问题解决思路

　　 （1）可以直接修改原来的函数，在函数内直接修改。当我们对多个函数都新增这个功能时，这样修改显得代码冗余。

　　（2）采用装饰器来解决。装饰器是python里面的一种特殊函数（装饰其它函数），它可以在函数名以及函数体不改变的前提下，给一个函数附加一些额外代码。

3 简单装饰器案例（python3.x）

　　3.1 原代码功能：根据你输入的内容，输出你输入的城市名称。原代码如下。

def bj():
    print("北京！")

def sz():
    print("深圳！")

def sh():
    print("上海！")

# 功能逻辑
chioce = input("请输入你喜欢的城市名称:")

if chioce == "北京":
    bj()
elif chioce == "深圳":
    sz()
else:
    sh()

　　上面代码执行后结果为：
　　　　

　　 3.2 现在要添加新功能：在原来的输出城市名称前先输出"你好:"。

　　3.3 解决方案1：直接在原来的函数里面添加输出“你好”相应代码。修改后的代码如下：

def bj():
    print("你好:北京！")

def sz():
    print("你好:深圳！")

def sh():
    print("你好:上海！")

# 功能逻辑
chioce = input("请输入你喜欢的城市名称:")

if chioce == "北京":
    bj()
elif chioce == "深圳":
    sz()
else:
    sh()

 　　　上面代码执行后结果为：

　　　　

　　　3.3 解决方案2：把新增功能写到一个新的函数里面，这个新的函数就是装饰器。修改后代码如下：

# 装饰器函数
def hello(func):
    def innner():
        print("你好:", end='')
        func()
    return innner

@hello
def bj():
    print("北京！")

@hello
def sz():
    print("深圳！")

@hello
def sh():
    print("上海！")

# 功能逻辑
chioce = input("请输入你喜欢的城市名称:")

if chioce == "北京":
    bj()
elif chioce == "深圳":
    sz()
else:
    sh()

 　　　　上面代码执行后结果为：

　　　　　

4 复杂装饰器案例（python3.x）　　　

 　　 4.1 代码功能：根据选择进入不同区域，如图片区域或音乐区域。原代码如下,

def ftp():
    print("这里是图片区")

def fyy():
    print("这里是音乐区")
    

# 功能代码
ychoice = input("请输如的你的选择:")

if ychoice == "1":
    ftp()
elif ychoice == "2":
    fyy()
else:
    print("输入错误:请输入1或2!")

　　上面代码执行后结果为：
　　　　

4.2 需要新增的功能：根据不同情况，在输出时首先输出"---------------------------"或者"****************************"的分隔符。

4.3 使用装饰器来增加这个新功能，修改后的代码如下，

def login(char):
    def checkin(func):
        def inner():
            print(char * 50)
            func()
        return inner
    return checkin

@login("*")
def ftp():
    print("这里是图片区")

@login("-")
def fyy():
    print("这里是音乐区")

# 逻辑功能
ychoice = input('''请输如的你的选择:''')

if ychoice == "1":
    ftp()
elif ychoice == "2":
    fyy()
else:
    print("输入错误:请输入'1'或者'2'!")

　　上面代码执行后结果为：
　　　　

2018-08-12