import logging
import time
def log(func):
def wrapper(*args, **kwatgs):
start_time = time.time()
logging.info('start %s()' % func.__name__)
ret = func(*args, **kwatgs)
end_time = time.time()
logging.info('end {}()\ncost {:.5f} seconds'.format(func.__name__, end_time - start_time))
return ret
return wrapper
@log
def add(a, b):
return a + b
@log
def multiply(a, b):
return a * b
logger = logging.getLogger()
logger.setLevel(logging.INFO)
sum_ = add(1000000, 1000000)
product = multiply(66, 88)
'''result
INFO:root:start add()
INFO:root:end add()
cost 0.00012 seconds
INFO:root:start multiply()
INFO:root:end multiply()
cost 0.00004 seconds
'''
logging模块,记录函数运行时间
最新推荐文章于 2024-08-25 20:01:14 发布