“”"
装饰器:
概念:把一个函数当作参数,返回一个替代版的函数
本质上就是一个返回函数的函数
“在不改变原函数的基础上,给函数增加功能”
“”"
def func():
print('have a nice day!')
def func2():
print('hello world')
def outer():
print('**********')
func()
#func()
outer()
“”"
其实简单的理解装饰器,就是一个化妆师,给函数化妆
“”"
def outer(func):
def inner():
print('****************')
func()
return inner #inner其实就是我们装饰后的新函数
@outer #语法糖
def func1():
print('have a nice day!')
def func2():
print('welcome!')
#func1= outer(func1)
func1()
#outer(func2)
“”"
outer()就相当于一个‘化妆师’
将传进来的‘参数’装饰,然后返回‘一个装饰好的函数’
“”"
def outer(func):
def inner(age):
if age<0:
age = 0
func(age)
return inner
@outer
def say(age):
print('name is %d years old' %(age))
say(-10)
def outer(func):
# 不定长参数
def inner(*args):
#添加功能
print('#################')
func(*args)
return inner
@outer
def say(name,age):
print('my name is %s,I am %d years old' %(name,age))
say('tom',18)
def desc(aa): #需要传递的函数(要装饰的函数)
def add_info():
print('儿童节快乐~~~')
aa()
print('欢迎来西部开源学习~')
return add_info
@desc
def login():
print('login...')
@desc
def logout():
print('logout...')
@desc
def savemoney():
print('save money...')
def transfermoney():
print('transfermoney...')
login()
logout()
savemoney()
“”"
装饰器实现一个函数计时器
“”"
import random
import string
import time
import functools
li = [random.choice(string.ascii_letters)for i in range(100)]
#print(time.time())
def timeit(fun):
# 问题1:被装饰的函数有返回值的时候怎么办?
# 问题2:如何保留被装饰函数的函数名和帮助信息文档
@functools.wraps(fun)
def wapper(*args,**kwargs):
"""这是一个wapper函数"""
# args:元组 kwargs:字典
# 在函数执行之前
start_time = time.time()
# 执行函数
res = fun(*args,*kwargs)
# 在函数执行之后
end_time = time.time()
print('运行时间为:%.6f' %(end_time - start_time))
return res
return wapper
@timeit
def con_add():
s = ''
for i in li:
s += (i + ',')
print(s)
@timeit
def join_add():
print(','.join(li))
con_add()
join_add()
@timeit
def fun_list(n):
"""这是fun_list函数,被timeit装饰"""
return [2 * i for i in range(n)]
@timeit
def fun_map(n):
"""这是fun_map函数,被timeit装饰"""
return list(map(lambda x:x*2,range(n)))
#print(fun_list(500000))
#print(fun_map(500000))
#print(fun_list.__name__)
#print(fun_list.__doc__)
#print(fun_map.__name__)
#print(fun_map.__doc__)
“”"
#创建装饰器, 要求如下:
#1. 创建add_log装饰器, 被装饰的函数打印日志信息;
#2. 日志格式为: [字符串时间] 函数名: xxx, 运行时间:xxx, 运行返回值结果:xxx
“”"
import functools
import time
#print(time.ctime())
def add_log(func):
@functools.wraps(func)
def wrapper(*args,**kwargs):
start_time = time.time()
res = func(*args,**kwargs)
end_time = time.time()
print('[%s] 函数名:%s,运行时间:%.6f,运行返回值结果:%d' %(
time.ctime(),func.__name__,end_time-start_time,res
))
return res
return wrapper
@add_log
def add(x,y):
time.sleep(1)
return x+y
add(1,10)
“”"
[‘root’,‘admin’,‘redhat’]
多个装饰器的应用场景:
会采用多个装饰器先验证是否登陆成功,再验证权限是否足够
“”"
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 student')
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('westos')
“”
带参数的装饰器
“”
import functools
import time
def log(kind):
def add_log(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
start_time = time.time()
res = fun(*args,**kwargs)
end_time = time.time()
print('<%s> [%s] 函数名:%s,运行时间:%.5f,运行返回值结果:%d' %
(kind,time.ctime(),fun.__name__,end_time-start_time,res))
return res
return wrapper
return add_log
@log('aaa')
def add(x,y):
time.sleep(1)
return x+y
print(add(1,2))