import enum
import logging
import time
from functools import wraps
logger = logging.getLogger('sls')
# time装饰器
def timer(func):
@wraps(func)
def wrap(*args, **kwargs):
begin_time = time.perf_counter()
result = func(*args, **kwargs)
start_time = time.perf_counter()
logger.info('func:%r took: %2.4f sec' % (func.__name__, start_time - begin_time))
return result
return wrap
使用
# 导入装饰器
from conf import timer
@ timer
def func(a, b):
return a+b
func(1, 5)