- 装饰器：
把一个函数当作参数，返回一个替代版的函数
本质上就是一个返回函数的函数
“在不改变原函数的基础上，给函数增加功能”
代码:
def func1():
print('have a nice day!')
def outer(fun):
def inner():
print('-----------')
fun()
return inner
func1 = outer(func1)
func1()
def say(age):
print('man is %d years old'%(age))
def outer(func):
def inner(age):
if age < 0:
age = 0
func(age)
return inner
say =outer(say)
运行
- @语法糖，相当于say=outer(say)
使用@符号将装饰器应用到函数
代码：
def outer(func):
def inner(age):
if age<0:
age =0
func(age)
return inner
@outer #语法糖 #相当于say=outer(say)
使用@符号将装饰器应用到函数
def say(age):
print('man is %d years old'%(age))
say(10)
运行：
- 装饰器的第一个用法：给一个函数加内容
代码：
def desc(fun): # 需要传递一个函数，要装饰的函数
def add_info(): # 装饰器函数里要嵌套函数
print('元旦快乐～')
fun()
#print('欢迎光临～')
return add_info # 返回值是嵌套的函数对象
@desc # 如何调用装饰器(两种方式)
def login():
print('login....')
login = desc(login) # 返回值是一个函数
login()
@desc
def logout():
print('logout...')
@desc
def savemoney():
print('存钱...')
@desc
def transfermoney():
print('转账...')
login()
logout()
savemoney()
transfermoney()
运行：
- 装饰器的应用：记时器
4.1 需求：装饰器实现记时器
import time
import string
import random
import functools #导入装饰器函数名和帮助信息文档的包
def timenum(fun):
def swapper(*args,**kwargs):#接收可变参数和关键字参数
#*args:可变参数 **kargs:关键字参数
#记录在函数执行之前的系统时间
time_start = time.time()
fun(*args,**kwargs)
#记n录函数执行完毕的系统时间
time_end = time.time()
print('运行时间为%.6f'%(time_end - time_start))
return swapper
li =[random.choice(string.ascii_letters) for i in range(100)]
print(li)
@timenum
def con_add():
s = ''
for i in li:
s +=i +'.'
print(s)
@timenum
def jion_add():
print(','.join(li))
@timenum
def fun_list(n):
print([2*i for i in range(n)])
@timenum
def fun_map(n):
print(list(map(lambda x:x*2,range(n))))
con_add()
jion_add()
fun_list(1000)
fun_map(1000)
运行
4.2 问题一：装饰的函数有返回值时
首先导入所需要的包
import time
import string
import random
import functools #导入装饰器函数名和帮助信息文档的包
def timenum(fun):
"""这是一个装饰器timenum"""
@functools.wraps(fun) #可以保留被装饰函数的函数名和帮助信息文档，不加这一句用装饰器装饰函数的时候打印信息会显示装饰器中的swaper函数的信息
def swapper(*args,**kwargs):#接收可变参数和关键字参数
"""这是一个swapper函数"""
#*args:可变参数 **kargs:关键字参数
#记录在函数执行之前的系统时间
time_start = time.time()
res =fun(*args,**kwargs)
#记n录函数执行完毕的系统时间
time_end = time.time()
print('运行时间为%.6f'%(time_end - time_start))
return res
return swapper
@timenum
def fun_list(n):
"""这是一个fun_map函数，被timenum装饰"""
return [2*i for i in range(n)]
@timenum
def fun_map(n):
"""这是一个map函数，被timenum装饰"""
return [list(map(lambda x:x*2,range(n)))]
fun_list(700)
fun_map(700)
运行：
4.3 问题二：如何保留被装饰函数的函数名和帮助信息文档
#导入第三方包 import functools
def timenum(fun):
“”“这是一个装饰器timenum”""
@functools.wraps(fun) #可以保留被装饰函数的函数名和帮助信息文档，不加这一句用装饰器装饰函数的时候打印信息会显示装饰器中的swaper函数的信息
def swapper(*args,**kwargs):#接收可变参数和关键字参数
“”“这是一个swapper函数”""
#args:可变参数 **kargs:关键字参数
#记录在函数执行之前的系统时间
time_start = time.time()
res =fun(args,**kwargs)
#记n录函数执行完毕的系统时间
time_end = time.time()
print(‘运行时间为%.6f’%(time_end - time_start))
return res
return swapper
@timenum
def fun_list(n):
“”“这是一个fun_map函数，被timenum装饰”""
return [2i for i in range(n)]
@timenum
def fun_map(n):
“”“这是一个map函数，被timenum装饰”""
return [list(map(lambda x:x2,range(n)))]
def fun():
“”“这是fun函数”""
print(‘hello’)
print(fun.doc)
print(fun.name)
print(fun_map.doc)
print(fun_map.name)
print(fun_list.doc)
print(fun_list.name)
运行：
- 超级用户有添加用户权限
import functools
import inspect #导入字典的第三方文件库
def is_admin(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
inspect_res = inspect.getcallargs(fun,*args,**kwargs)
print('inspact的返回值为：%s'%inspect_res)
if inspect_res.get('name')=='root':
temp = fun(*args,**kwargs)
return temp
else:
print('非超级用户，无权限')
return wrapper
@is_admin
def add_student(name):
print('添加学生信息')
@is_admin
def del_student(name):
print('删除学生信息～')
add_student('westos')
del_student('root')
运行：
创建装饰器，要求如下：
- 创建add_log装饰器，被装饰的函数打印日志信息；
- 日志格式为：[字符串时间] 函数名：xxx, 运行时间：xxx,运行返回结果：xxx
import functools #保留被装饰函数的函数名和信息
import time
def add_log(fun):
""" A """
@functools.wraps(fun)
def message(*args,**kwargs):
""" B """
string_time = time.ctime() #.ctime 字符串格式时间
start_time = time.time()
res = fun(*args,**kwargs)
end_time = time.time()
#打印所需要的信息
print('[%s] 函数名：%s，运行时间:%.6f 运行返回结果为:%s'%(string_time,fun.__name__,end_time-start_time,res))
return res
return message
@add_log
def add_num(x,y):
""" C """
time.sleep(0.1)#设定休眠时间
return x+y
print(add_num(1,3))
运行：
- 多个装饰器装饰一个函数
当有多个装饰器的时候，从下到上调用
真实的wapper内容是从上到下执行
def decorator_a(fun):
print('Get in decorator_a')
def inner_a(*args,**kwargs):
print('Get in inner_a')
return fun(*args,**kwargs)
return inner_a
def decorator_b(fun):
print('Get in decorator_b')
def inner_b(*args,**kwargs):
print('Get in inner_b')
return fun(*args,**kwargs)
return inner_b
@decorator_a
@decorator_b
def f(x):
print('Get in f')
return x*2
f(2)
运行:
- 多个装饰器验证多个条件
在实际的应用场景中，会采用多个装饰器先验证是否登陆成功，再验证权限是否足够。
代码：
import functools
import inspect
def is_admin(fun):
@functools.wraps(fun)
def wrapper(*args, **kwargs):
# inspect.getcallargs返回一个字典，key值：形参 value值：对应的实参
inspect_res = inspect.getcallargs(fun, *args, **kwargs)
print('inspect的返回值:%s' %(inspect_res))
if inspect_res.get('name') == 'root':
temp = fun(*args,**kwargs)
return temp
else:
print('not root user,no permisson add stduent')
return wrapper
login_session = ['root','admin','redhat']
def is_login(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
if args[0] in login_session:
temp = fun(*args,**kwargs)
return temp
else:
print('Error:%s未登陆！' %(args[0]))
return wrapper
@is_login
@is_admin
def add_student(name):
print('添加学生信息～')
add_student('admin')
运行：
- 编写装饰器required_ints, 条件如下:
1). 确保函数接收到的每一个参数都是整数;
2). 如果参数不是整形数， 打印 TypeError:参数必须为整形
代码：
import functools
def required_ints(fun):
@functools.wraps(fun)
def wrapper(*args):
for i in args:
if not isinstance(i,int):
print('TypeError:参数必须为整型')
return
else:
res =fun(*args)
return res
return wrapper
@required_ints
def instance(*args):
a = sum(args)
print(a)
instance(1,4,7,2,9)
运行：
- 装饰器函数09—函数加参数
代码：
import functools #保留被装饰函数的函数名和信息
import time
def log(kind): #添加外置参数接口
def add_log(fun):
""" A """
@functools.wraps(fun)
def message(*args,**kwargs):
""" B """
string_time = time.ctime() #.ctime 字符串格式时间
start_time = time.time()
res = fun(*args,**kwargs)
end_time = time.time()
#打印所需要的信息
print('%s [%s] 函数名：%s，运行时间:%.6f 运行返回结果为:%s'%(kind,string_time,fun.__name__,end_time-start_time,res))
return res
return message
return add_log #返回装饰器中包含外置参数
@log('hello')
def add_num(x,y):
""" C """
time.sleep(0.1)#设定休眠时间
return x+y
add_num(3,5)
运行：
- 装饰器应用
编写装饰器required_types, 条件如下:
1). 当装饰器为@required_types(int,float)确保函数接收到的每一个参数都是int或者float类型;
2). 当装饰器为@required_types(list)确保函数接收到的每一个参数都是list类型;
3). 当装饰器为@required_types(str,int)确保函数接收到的每一个参数都是str或者int类型;
4). 如果参数不满足条件， 打印 TypeError:参数必须为xxxx类型
代码：
import functools
def required_type(*kinds):
def required(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
for i in args:
if not isinstance(i,kinds):
print ('函数所有参数并非',kinds)
break
else:
res = fun(*args,**kwargs)
return res
return wrapper
return required
#@required_type(int,float)
#def add(a,b):
#return a+b
#print(add('1','2'))
@required_type(int)
def mymax(a,b,c,d):
return max(a,b,c,d)
print(mymax(1.2,2,3,4))
运行：
扫一扫
173万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
