python裝飾器計算函數執行時間

import time
import logging
from functools import wraps
import datetime

log = logging.getLogger(__name__)

def count_time(func):
    @wraps(func)
    def _wrapper(*args, **kwargs):
        start_time = time.time()  # 程序開始時間
        log.info("task begin at: %s" %(datetime.datetime.fromtimestamp(start_time)))
        func()
        end_time = time.time()   # 程序結束時間
        log.info("task end at: %s" %(datetime.datetime.fromtimestamp(end_time)))
        total_time = end_time - start_time
        log.info('程序共計%s秒' % total_time)
　　return _wrapper