装饰器（Decorator）概念
把一个函数当作参数传递给另一个函数，返回一个替代版的函数
在代码运行期间动态增加功能的方式，称之为装饰器，本质上就是一个以函数为返回值的函数，目的是在不改变原函数的基础上，给函数增加功能。
装饰器1
1.函数也是一个对象，而且函数对象可以被赋值给变量，所以，通过变量也能调用该函数。
函数say（）的定义
def fun():
print('hello python')
假设要增强say()函数的功能，比如在调用函数时要对实参进行正负的判断，但又不希望修改say（）的定义，这时采用装饰器，即能在代码运行期间增加功能。
可以定义如下:
def outer(f):
def inner():
print('@@@@@@@')
f()
print('!!!!')
return inner
观察上面的outer（）函数，因为它是一个decorator，所以接受一个函数作为参数，并返回一个函数。借助python的语法糖，把decorator置于函数的定义处
@outer
def fun():
print('hello python')
把语法糖@outer放到fun（）函数的定义处，相当于执行了语句：
由于outer（）只是一个decorator，返回一个函数，故原来的fun（）函数仍然存在，只是现在同名的fun（）变量指向了新的函数，于是调用你fun（)将执行新函数，即在outer（）函数中返回的inner（）函数
fun = outer（fun）
调用fun（）函数，不仅会运行fun（）本身，还会打印出新的内容：
例子：
def outer(f):
def inner(age):
if age <=0:
age = 0
f(age)
return inner
@outer #语法糖
def say(age):
print('%d years old' %(age))
执行say（-12）结果为
装饰器2
装饰器实现一个函数计时器
1.被装饰的函数如果有 返回值的时候 怎么办？
2.如何保留被装饰函数的函数名字和帮助文档信息？ 装饰器实现一个函数计时器
1.被装饰的函数如果有 返回值的时候 怎么办？
2.如何保留被装饰函数的函数名字和帮助文档信息？
import random
import string
import time
li = [random.choice(string.ascii_letters) for i in range(100)]
def timeit(fun):
#@functools.wraps(fun)
def wrapper(*args,**kwargs):# 接收可变参数和关键字参数
"""这是一个装饰器timeit"""
# 在函数执行之前
start_time = time.time()
# 执行函数
res = fun(*args,**kwargs)
# 在函数执行之后
stop_time = time.time()
print('运行事件为:%.6f' %(stop_time-start_time))
return res
return wrapper
# @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函数"""
return [2 * i for i in range(n)]
@timeit
def fun_map(n):
"""这是fun_map函数 被装饰器timeis装饰"""
return list(map(lambda x:x*2,range(n)))
# fun_list(10000)
# fun_map(10000)
print(fun_list.__name__)
print(fun_list.__doc__)
print(fun_map.__name__)
print(fun_map.__doc__)
装饰器3
创建装饰器， 要求如下：
- 创建add_log装饰器， 被装饰的函数打印日志信息；
- 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx,
运行返回值结果:xxx
# import time
# print(time.ctime())
import time
import functools
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)
装饰器4
- 基础版(无参数的装饰器)
编写装饰器required_ints, 条件如下:
1). 确保函数接收到的每一个参数都是整数;
2). 如果参数不是整形数， 打印 TypeError:参数必须为整形
import functools
def required_ints(func):
@functools.wraps(func)
def wrapper(*args, **kwargs): # args=(1,2,..)
for i in args:
# if isinstance(i,int):
# pass
# else:
# print('函数所有的参数并非都是int型')
# break
if not isinstance(i, int):
print('函数所有的参数并非都是int型')
break
else:
res = func(*args, **kwargs)
return res
return wrapper
@required_ints
def add(a, b):
return a + b
@required_ints
def mymax(a,b,c,d):
return max(a,b,c,d)
print(add(1,2.0))
创建装饰器， 要求如下：
- 创建add_log装饰器， 被装饰的函数打印日志信息；
- 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx, 运行返回值结果:xxx
import time
import functools
def log(kind):
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] 函数名:%s,运行时间:%.6f,运行返回值的结果'
':%d' %(kind,time.ctime(),func.__name__,
end_time-start_time,res))
return res
return wrapper
return add_log
@log('local2')
def add(x,y):
time.sleep(1)
return x+y
add(1,10)
装饰器5
[‘root’,‘admin’,‘redhat’]
id id+vip
多个装饰器的应用场景：
会采用多个装饰器先验证是否登陆成功 再验证权限是否足够
import functools
import inspect
def is_login(fun):
@functools.wraps(fun)
def warapper(*args, **kwargs): # ('admin',)
if args[0] in login_session:
temp = fun(*args, **kwargs)
return temp
else:
print('Error:%s 没有登陆成功' % (args[0]))
return warapper
def is_admin(fun):
@functools.wraps(fun)
# nspect.getcallargs会返回一个字典
# key:形参 value:对应的实参
def wrapper(*args, **kwargs):
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 user')
return wrapper
login_session = ['root', 'admin', 'redhat']
@is_login
@is_admin
def add_user(name):
print('add_user~~')
add_user('root')
装饰器6
- 基础版(无参数的装饰器)
编写装饰器required_ints, 条件如下:
1). 确保函数接收到的每一个参数都是整数;
2). 如果参数不是整形数， 打印 TypeError:参数必须为整形
import functools
def required_ints(func):
@functools.wraps(func)
def wrapper(*args, **kwargs): # args=(1,2,..)
for i in args:
# if isinstance(i,int):
# pass
# else:
# print('函数所有的参数并非都是int型')
# break
if not isinstance(i, int):
print('函数所有的参数并非都是int型')
break
else:
res = func(*args, **kwargs)
return res
return wrapper
@required_ints
def add(a, b):
return a + b
@required_ints
def mymax(a,b,c,d):
return max(a,b,c,d)
print(add(1,2.0))
装饰器7
创建装饰器， 要求如下：
- 创建add_log装饰器， 被装饰的函数打印日志信息；
- 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx, 运行返回值结果:xxx
import time
import functools
def log(kind):
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] 函数名:%s,运行时间:%.6f,运行返回值的结果'
':%d' %(kind,time.ctime(),func.__name__,
end_time-start_time,res))
return res
return wrapper
return add_log
@log('local2')
def add(x,y):
time.sleep(1)
return x+y
add(1,10)
装饰器8
- 升级版(有参数的装饰器)
编写装饰器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(*kind):
def required(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
for i in args:
if not isinstance(i,kind):
print('函数所有的参数并非',kind)
break
else:
res = fun(*args,**kwargs)
return res
return wrapper
return required
@required_type(int)
def add(a, b):
return a + b
print(add(1,1.2))
扫一扫
1413
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
