第一種：
import time
def time_me(fn): #fn 是要修飾/修改 的函數
def _wrapper(*args, **kwargs): #這個 _wrapper(*args, **kwargs) 則代指fn, *args 代表一般變量參數， **kwargs代表 字典，哈希等參數
start = time.perf_counter()
fn(*args, **kwargs)
print ("%s cost %s second"%(fn.__name__, time.perf_counter() - start))
return _wrapper
#這個裝飾器可以在方便地統計函數運行的耗時。
#用來分析腳本的性能是最好不過了。
#這樣用：
@time_me() 這里 沒有指定
def test(x, y):
time.sleep(0.1)
@time_me()
def test2(x):
time.sleep(0.2)
test(1, 2)
test2(2)
第二種：
import time
import functools
def time_me(info="used"):
def _time_me(fn):
@functools.wraps(fn) #轉化為輸入函數
def _wrapper(*args, **kwargs): #*args, **kwargs 代指所有 輸入函數的參數
start = time.perf_counter()
fn(*args, **kwargs)
print("%s %s %s"%(fn.__name__, info, time.perf_counter() - start), "second")
return _wrapper
return _time_me
#應用
@time_me() #添加修飾器，就是函數嵌套了函數，中間加了一些功能和執行內容，而不需要 修改 test() 函數的內容
def test(x, y):
time.sleep(0.1)
@time_me("cost") #添加修飾器
def test2(x):
time.sleep(0.2)
test(1, 2)
test2(2)
image.png
第三種改進版：
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import division
from __future__ import unicode_literals
import sys,time,datetime,functools
#info
__author__ = ''
__email__ = 'xxx@xxx.com'
__version__ = '0.0.1'
__status__ = 'Dev'
def time_me(info="used"):
def _time_me(fn):
@functools.wraps(fn) #轉化為輸入函數
def _wrapper(*args, **kwargs): #*args, **kwargs 代指所有 輸入函數的參數
if sys.version[0]=="3":
start = time.perf_counter()
else:
start = time.clock()
fn(*args, **kwargs)
if sys.version[0]=="3":
end = time.perf_counter()
else:
end = time.clock()
print("%s %s %s"%(fn.__name__, info, str(datetime.timedelta(seconds = end - start))))
return _wrapper
return _time_me
#應用
@time_me() #添加修飾器，就是函數嵌套了函數，中間加了一些功能和執行內容，而不需要 修改 test() 函數的內容
def test(x, y):
time.sleep(0.3)
@time_me("cost") #添加修飾器，給個參數cost
def test2(x):
time.sleep(0.4)
test(1, 2)
test2(2)
輸出為：
image.png
第四種 在實際應用時 發現會報錯
TypeError: cannot unpack non-iterable NoneType object
自己用在某個函數上,這個函數輸入和返回都不止 一個參數
@time_me()
def bam2fa(bam,lst,fz=1000,rz=100,pb=0.66,if="ins",if2="hr",sf=False):
pass
return list, dict, list
查詢后發現 原因是：
TypeError: cannot unpack non-iterable NoneType object
報錯的原因是函數返回值得數量不一致，查看函數返回值數量和調用函數時接收返回值的數量是不是一致，修改一致即可
發覺是是自己對裝飾器理解不到位，逐
參考了這里，發現 之前的例子里少了 return func(*args, **kwargs) 返回這個原始函數func
Python 函數裝飾器
image.png
看完了這篇才明白
如何理解Python裝飾器？
如下圖中
fn(*args, **kwargs) 放在了 獲取時間的中間，但是我的函數是有返回值的
image.png
完善版本（支持具有返回值的函數）
8881910.png
'''
import sys,time,datetime,functools,logging
#logging 用于輸出日志
def main():
#logging 設置
LOG_FORMAT = "%(asctime)s %(name)s %(levelname)s %(pathname)s %(message)s "#配置輸出日志格式
DATE_FORMAT = '%Y-%m-%d %H:%M:%S' #配置輸出時間的格式，注意月份和天數不要搞亂了
logging.basicConfig(level=logging.DEBUG,format=LOG_FORMAT,datefmt = DATE_FORMAT,filename=r"logging.log" )
'''
def time_me(info="used"):
def _time_me(func):
if sys.version[0]=="3": #獲得python版本號,截取第一個字符，2.7就是2 ，3.8就是3
start = time.perf_counter() #python3 中使用perf_counter()
else:
start = time.clock() #python2 中使用clock()
@functools.wraps(func) #效果是保持當前裝飾器去裝飾的函數的 __name__ 的值不變；
#函數的函數名即 __name__默認被裝飾器改變為_wrapper，但是當存在多個修飾器時，重名會發生錯誤，
def _wrapper(*args, **kwargs): #*args, **kwargs 代指所有 輸入函數的參數
return func(*args, **kwargs) #這里等于執行了一次這個函數
if sys.version[0]=="3": #獲得python版本號,截取第一個字符，2.7就是2 ，3.8就是3
end = time.perf_counter()
else:
end = time.clock()
#logging.info("%s %s %s\n"%(func.__name__, info, str(datetime.timedelta(seconds = end - start)))) #這里可以替換 logging 日志，func.__name__ 是該函數的名字，也可以自己在函數中設置 __name__="xxx"
print("%s %s %s"%(fn.__name__, info, str(datetime.timedelta(seconds = end - start))))
return _wrapper
return _time_me
args表示任何多個無名參數，它是一個tuple；kwargs表示關鍵字參數，它是一個dict。并且同時使用args和kwargs時，必須args參數列要在*kwargs前，像foo(a=1, b='2', c=3, a', 1, None, )這樣調用的話，會提示語法錯誤“SyntaxError: non-keyword arg after keyword arg”。
獲得的啟示：編寫函數時，參數中字典類型的放在后面 func() 的后面 ，如 func( arg,tuple,list,dict)
這么慢？)
