Python之函式裝飾器
阿新 • • 發佈:2018-12-09
一、map函式
- map()傳入的第一個引數是一個函式,第二個引數是一個序列
- map 將傳入的函式依次作用到序列的每個元素,並把結果作為新的 list 返回
def f(x):
return x * x
print map(f, [1, 2, 3]) ##返回[1, 4, 9]
- 把列表中的所有數字轉為字串
print(list(map(str,[1,2,3,4])))
##呼叫map的結果為map型別,強制轉化為list型別顯示
二、匿名函式lambda
def f(x):
return x * x
print map(f, [1, 2, 3]) ##輸出:[1,4,9]
print map(lambda x: x * x, [1, 2, 3]) ##輸出:[1,4,9]
三、函式裝飾器
1、每個函式的執行時間
"""
需求:
裝飾器需求:獲取每個函式的執行時間
1.函式執行之前計算時間
2.函式執行之後計算時間
問題1:被裝飾的函式有返回值的時候怎麼辦?
問題2:如何保留被裝飾函式的函式名和幫助文件資訊
"""
import time
import random
import string
#from functools import reduce
# print string.ascii_letters
import functools
li = [random.choice(string.ascii_letters) for i in range(100)]
def timeit(fun):
"""這是一個裝飾器"""
@functools.wraps(fun)
def wrapper(*args, **kwargs): # 接收可變引數和關鍵字引數
"""這是一個wrapper函式"""
# args:元組 kwargs:字典
# 函式執行之前
start_time = time.time()
# 執行函式
# 接收被裝飾函式的返回值
res = fun(*args, **kwargs)
# 函式執行之後
end_time = time.time()
print '執行的時間為:%.6f' % (end_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)
@timeit
def fun_list(n):
"""這是fun_list函式,被timeit裝飾"""
return [2 * i for i in range(n)]
@timeit
def fun_map(n):
"""這是fun_map函式"""
return list(map(lambda x: x * 2, range(n)))
con_add()
join_add()
print fun_list(50000)
print fun_map(50000)
print fun_list.__name__
print fun_list.__doc__
2、編寫裝飾器
"""
需求:
編寫裝飾器required_ints, 條件如下:
# 1). 確保函式接收到的每一個引數都是整數;
# 2). 如果引數不是整形數, 列印 TypeError:引數必須為整形
"""
import functools
def required_ints(fun):
@functools.wraps(fun)
def wrapper(*args, **kwargs):
for i in args:
if not isinstance(i, int):
print 'TypeError:引數必須為整形'
break
else:
res = fun(*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)
print myMax(1,2,3,4)
3、使用者登陸驗證的裝飾器
"""
需求:使用者登陸驗證的裝飾器 is_login
1.如果使用者登陸成功,則執行被裝飾的函式
2.如果使用者登陸不成功,則執行登陸函式
"""
import functools
login_users = ['admin', 'root']
def is_login(fun):
@functools.wraps(fun)
def wrapper(*args, **kwargs):
# 判斷寫部落格這個使用者是否登陸成功
if kwargs.get('name') in login_users:
res = fun(*args, **kwargs)
return res
else:
res = login()
return res
return wrapper
def login():
return '登陸。。。'
@is_login
def writeBlog(name):
return '編寫部落格。。'
print writeBlog(name='adminss')
4、列印日誌資訊的裝飾函式
"""
建立裝飾器, 要求如下:
1. 建立add_log裝飾器, 被裝飾的函式列印日誌資訊;
2. 日誌格式為: [字串時間] 函式名: xxx, 執行時間:xxx, 執行返回值結果:xxx
"""
import time
import functools
#print time.ctime()
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,執行時間:%.5f,執行返回值結果:%d' %(time.ctime(),fun.__name__,end_time-start_time,res)
return res
return wrapper
@add_log
def add(x,y):
time.sleep(1)
return x+y
print add(1,2)
5、帶引數的裝飾器
import time
import functools
#print time.ctime()
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('westos')
# log('dubug) --> 返回值是add_log
# add = add_log(add)
def add(x,y):
time.sleep(1)
return x+y
# wrapper(1,2)
print add(1,2)
裝飾器的概念
- 裝飾器的實現是函式裡面巢狀函式;
- 裝飾器的本質是一個函式, 它可以讓其他函式在不需要做任何程式碼改動的前提下增加額外的功能;
- 裝飾器需要傳遞一個函式, 返回值也是一個函式物件.
"""
import functools
def required_types(*kinds):
def required_ints(fun):
@functools.wraps(fun)
def wrapper(*args, **kwargs):
for i in args:
if not isinstance(i, kinds):
print 'TypeError:引數必須為',kinds
break
else:
res = fun(*args, **kwargs)
return res
return wrapper
return required_ints
@required_types(float,float)
def add(a, b):
return a + b
print add(1.1,3)
6、多個裝飾器的順序
def decorator_a(func):
print 'Get in decorator_a'
def inner_a(*args, **kwargs):
print 'Get in inner_a'
res = func(*args, **kwargs)
return res
return inner_a
def decorator_b(func):
print 'Get in decorator_b'
def inner_b(*args, **kwargs):
print 'Get in inner_b'
res = func(*args, **kwargs)
return res
return inner_b
# 當有多個裝飾器時,從下到上呼叫裝飾器
@decorator_b
@decorator_a
def f(x):
print 'Get in f'
return x * 2
f(1)
7、多個裝飾器的應用
在我們實際的應用場景中,會採用多個裝飾器先驗證是否登陸成功再驗證許可權
import functools
import inspect
def is_root(fun):
@functools.wraps(fun)
def wrapper(*args, **kwargs):
# inspect.getcallargs返回值是字典,key值為:形參 value值:形參對應的實參
inspect_res = inspect.getcallargs(fun, *args, **kwargs)
print 'inspect_res的返回值:%s' % inspect_res
if inspect_res.get('name') == 'root':
res = fun(*args, **kwargs)
return res
else:
print 'Error:no permisson add studnet'
return wrapper
login_session = ['root', 'admin', 'redhat']
def is_login(fun):
@functools.wraps(fun)
def warrper(*args, **kwargs):
if args[0] in login_session:
res = fun(*args, **kwargs)
return res
else:
print 'Error:%s未登陸!' % args[0]
return warrper
@is_login
@is_root
def add_student(name):
print '新增學生資訊。。'
add_student('root')