一、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)
装饰器写法:
from functools import wraps
def wrap(fn):
@wraps(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)
装饰器演示:
from functools import wraps
import time
def timmer(func):
@wraps(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()
有参有返的函数装饰器示例:
from functools import wraps
# 增加一个账号处理功能:3位及以上英文字母或汉字
def check_usr(fn):
@wraps(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):
@wraps(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都有返回值
装饰器结合可变长参数
from functools import wraps
def wrap(fn):
@wraps(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)
带参装饰器
from functools import wraps
def outer(input_color):
@wraps(fn)
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()