apppublic/appPublic/timecost.py

import time
import datetime
from .Singleton import SingletonDecorator

timerecord = {}

class TimeCost:
	def __init__(self,name):
		self.name = name
	
	def __enter__(self):
		self.begin_time = time.time()
	
	def __exit__(self,*args):
		self.end_time = time.time()
		d = timerecord.get(self.name,[])
		d.append(self.end_time - self.begin_time)
		timerecord[self.name] = d

	def clear(self):
		timerecord = {}

	def getTimeCost(self,name):
		x = timerecord.get(name,[])
		if len(x) == 0:
			return 0,0,0
		return len(x), sum(x), sum(x)/len(x)
	
	def show(self):
		print('TimeCost ....')
		for name in timerecord.keys():
			print(name, * self.getTimeCost(name))