python基础——8（装饰器）

一、nonlocal关键字

def outer():
    num = 0
    def inner():
        # 如果想在被嵌套的函数中修改外部函数变量（名字）的值
        nonlocal num  # 将 L 与 E(E中的名字需要提前定义) 的名字统一
        num = 10
        print(num) # 10
    inner()
    print(num)  # 10
outer()
print(num)  # name 'num' is not defined

二、装饰器

定义：

本质是函数，（装饰其他函数）就是为其他的函数添加功能。

开放封闭原则：

1、不能修改被装饰的函数的源代码

2、不能修改被装饰的函数的调用方式

理解装饰器的知识储备：

1、函数即‘变量’

2、高阶函数

3、嵌套函数

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

高阶函数：变量可以指向函数，函数的参数可以接收变量，那么一个函数就可以接收另一个函数作为参数，这就是高阶函数。

# 1、把一个函数名当作实参传给另一个函数（在不修改被装饰函数源代码的情况下为其添加功能）

# 2、返回值中包含函数名（不修改函数的调用方式）

高阶函数实例：

def add(x,y,f):
    return f(x)+f(y)
res = add(3,-6,abs)
print(res)

装饰器写法：

def wrap(fn):
    def inner(*args,**kwargs):
        print("前增功能")
        result = fn(*args,**kwargs)
        print("后增功能")
        return result
    return inner

@wrap
def fn1():
    print("fn1的原有功能")
@wrap
def fn2(a,b):
    print("fn2的原有功能")
@wrap
def fn3():
    print("fn3的原有功能")
    return True
@wrap
def fn4(a,*,x):
    print("fn4的原有功能")
    return True

fn1()
fn2(10,20)
fn3()
fn4(10,x=20)

装饰器演示：

import time
def timmer(func):
    def warpper(*args,**kwargs):
        start_time=time.time()
        func()
        stop_time=time.time()
        print('the func run time is %s'%(stop_time-start_time))
    return warpper()

@timmer   #语法糖
def test1():
    time.sleep(3)
    print('in the test1')
test1()

有参有返的函数装饰器示例：

# 增加一个账号处理功能：3位及以上英文字母或汉字
def check_usr(fn):
    def inner(usr,pwd):
        if not (len(usr) >= 3 and usr.isalpha()):
            print("账号验证失败")
            return False
        result = fn(usr,pwd)  # login
        return result
    return inner

# 增加一个密码处理功能：6位及以上英文和数字
def check_pwd(fn):
    def inner(usr,pwd):
        if not (len(pwd) >= 6 and pwd.isalnum()):
            print("密码验证失败")
            return False
        return fn(usr,pwd)
    return inner

# 登录功能
@check_usr
@check_pwd
def login(usr,pwd):
    if usr == 'abc' and pwd == '123qwe':
        print("登陆成功")
        return True
    print("登录失败")
    return False

res = login('abc','123qwe')  # inner
print(res)

#总结
1、login有参数，所以inner与fn都有相同参数
2、login有返回值，所以inner与fn都有返回值

装饰器结合可变长参数

def wrap(fn):
    def inner(*args,**kwargs):
        result = fn(*args,**kwargs)
        print("新增功能")
        return result
    return inner

@wrap
def func(a,b,c,*,x,y,z):
    print(a,b,c,x,y,z)
    print("原有功能")

func(1,2,3,x=10,y=20,z=30)

带参装饰器

def outer(input_color):
    def wrap(fn):
        if input_color == 'red':
            info = '\033[31mnew action\033[0m'
        else:
            info = '\033[32mGreen:new action\033[0m'
        def inner(*args,**kwargs):
            result = fn(*args,**kwargs)
            print(info)
            return result
        return inner
    return wrap   # outer(color) --> wrap

color = input("color:")
@outer(color)
def func():
    print('func run')

func()

登录认证案例：

​
is_login = False  # 登陆状态

def login():
    global ​is_login
    usr = input("usr:")
    if not (len(usr) >= 3 and usr.isalpha()):
        print('账号验证失败')
        return False
    pwd = input("pwd:")
    if usr == 'abc' and pwd == '123qwe':
        print('login success')
        is_login = True
    else:
        print('login fail')
        is_login = False

# 完成一个登录状态校验的装饰器
def check_login(fn):
    def inner(*args,**kwargs):
        if is_login != True:
            print("你未登录")
            login()
        result = fn(*args,**kwargs)
        return result
    return inner

#查看个人主页
@check_login
def home():
    print("个人主页")

#销售功能
@check_login
def sale():
    print("可以销售")

home()
sale()

​

装饰器案例高级：

import time
user,passwd = 'shj','abc123'
def auth(auth_type):
    print("auth func:",auth_type)
    def outer_wrapper(func):
        def wrapper(*args,**kwargs):
            print("wrapper func args:",*args,**kwargs)
            if auth_type == 'local':
                username = input("Username:").strip()
                password = input("Password:").strip()
                if user==username and passwd==password:
                    print("User has passed authentication")
                    res = func(*args,**kwargs)
                    print("-------after authentication")
                    return res
                else:
                    exit("Invalid username or password")
            elif auth_type == 'ldap':
                print("It is so difficult !")
        return wrapper
    return outer_wrapper
def index():
    print("welcome to index page")
@auth(auth_type="local")  #相当于 home = wrapper()
def home():
    print("welcome to home page")
@auth(auth_type="ldap")
def bbs():
    print("welcome to bbs page")

index()
home()
bbs()