如何在Python函數執行前後增加額外的行為-Python教學-PHP中文網

首頁

後端開發

Python教學

如何在Python函數執行前後增加額外的行為

WBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWB

Dec 05, 2016 pm 01:27 PM

python 情境

首先來看一個小程序，這個是計量所花費時間的程序，以下是以往的解決範例

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('运行花费时间：{:.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)

登入後複製

運作如下：

In [4]: run()
运行花费时间：2.383398s。
Out[4]: 5000000

登入後複製

（喜歡刨根問底的可以去掉註釋，並思考預計會有什麼樣的輸出）。

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

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('运行花费时间：{:.6f}s。'.format(self.end_time - self.start_time))
 
with ElapsedTime():
 run2()