快捷導(dǎo)航

如何在Python函數(shù)執(zhí)行前后增加額外的行為

更新時間：2016年10月20日 16:39:38 作者：瓜園耕讀

有的時候會需要在函數(shù)前后添點(diǎn)額外的功能（比如過濾、計時等）時，以前總是首先想到裝飾器。最近學(xué)習(xí)了Python的上下文管理器，所以本文就給大家介紹了如何在Python函數(shù)執(zhí)行前后增加額外的行為，有需要的朋友們可以參考借鑒，下面來一起看看吧。

首先來看一個小程序，這個是計量所花費(fèi)時間的程序，以下是以往的解決示例

from functools import wraps, partial
from time import time

def timing(func=None, frequencies=1):
 if func is None:
  # print("+None")
  return partial(timing, frequencies=frequencies)
 # else:
  # print("-None")

 @wraps(func)
 def _wrapper(*args, **kwargs):
  start_time = time()
  for t in range(frequencies):
   result = func(*args, **kwargs)
  end_time = time()
  print('運(yùn)行花費(fèi)時間：{:.6f}s。'.format(end_time-start_time))
  return result

 return _wrapper


@timing
def run():
 l = []
 for i in range(5000000):
  l.extend([i])
 return len(l)

運(yùn)行如下：

In [4]: run()
運(yùn)行花費(fèi)時間：2.383398s。
Out[4]: 5000000

（喜歡刨根問底的可以去掉注釋，并思考預(yù)計會有什么樣的輸出）。

今天無意間看到了Python的上下文管理器（Context Manager），發(fā)現(xiàn)也非常不錯，其實這跟with語句是息息相關(guān)的，竟然以前一直未在意。

from time import time

def run2():
 l = []
 for i in range(5000000):
  l.extend([i])
 return len(l)

class ElapsedTime():
 def __enter__(self):
  self.start_time = time()
  return self

 def __exit__(self, exception_type, exception_value, traceback):
  self.end_time = time()
  print('運(yùn)行花費(fèi)時間：{:.6f}s。'.format(self.end_time - self.start_time))

with ElapsedTime():
 run2()

總結(jié)

初略看了一點(diǎn)官方文檔，上下文管理還是有點(diǎn)多內(nèi)容的。Python發(fā)展到現(xiàn)在，其實不簡單了。說簡單，只是你自己不夠與時俱進(jìn)，掌握的都是老式三板斧而已。所以，知識需要不斷更新，才能彌補(bǔ)自己的盲點(diǎn)，以上就是本文的全部內(nèi)容，希望能大家的學(xué)習(xí)或者工作帶來一定的幫助。

您可能感興趣的文章:

相關(guān)文章

matplotlib基礎(chǔ)繪圖命令之bar的使用方法
這篇文章主要介紹了matplotlib基礎(chǔ)繪圖命令之bar的使用方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-08-08
Python生成指定大小的文件兩種解決方案
這篇文章主要介紹了Python生成指定大小的文件,這里提供兩種解決方案幫助python完成我們生成任意大小的文件，需要的朋友可以參考下
2023-06-06
python os模塊在系統(tǒng)管理中的應(yīng)用
這篇文章主要介紹了python os模塊在系統(tǒng)管理中的應(yīng)用，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2020-06-06
基于pytorch 預(yù)訓(xùn)練的詞向量用法詳解
今天小編就為大家分享一篇基于pytorch 預(yù)訓(xùn)練的詞向量用法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-01-01
詳解python 模擬豆瓣登錄（豆瓣6.0）
這篇文章主要介紹了python模擬豆瓣登錄，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-04-04
Python模塊WSGI使用詳解
這篇文章主要為大家詳細(xì)介紹了Python模塊WSGI使用的相關(guān)資料，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-02-02
Python添加進(jìn)度條tqdm進(jìn)階使用實例
這篇文章主要為大家介紹了Python添加進(jìn)度條tqdm進(jìn)階使用實例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
2023-06-06
Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離問題詳解
這篇文章主要介紹了Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離問題,簡單描述了Dijkstra算法的原理并結(jié)合具體實例形式分析了Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離的相關(guān)步驟與操作技巧,需要的朋友可以參考下
2018-05-05
Python基礎(chǔ)之輸入,輸出與高階賦值詳解
這篇文章主要為大家介紹了Python基礎(chǔ)之輸入,輸出與高階賦值，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助
2021-11-11
python eventlet綠化和patch原理
這篇文章主要介紹了python eventlet綠化和patch原理，幫助大家更好的理解和學(xué)習(xí)python eventlet工具的使用，感興趣的朋友可以了解下
2020-11-11