使用装饰器记录执行时间

Question

在我尝试了一段时间不成功后，我正在向这个神奇的网站寻求帮助。现在我的问题：我想创建一个装饰器，将函数的执行时间（在函数执行期间）写入日志文件，如：

@log_time("log.txt", 35)
def some_function(...):
    ...
    return result

和

from functools import wraps

def log_time(path_to_logfile, interval):
    ...

所以这log.txt看起来像

Time elapsed: 0h 0m 35s
Time elapsed: 0h 1m 10s
Time elapsed: 0h 1m 45s

有任何想法吗？

Answer 1

快速组装，但使用 @timeit 对一些功能进行了测试。

import logging
logging.basicConfig(
    level=logging.DEBUG, 
    filename='myProgramLog.txt', 
    format=' %(asctime)s - %(levelname)s - %(message)s')

import time                                                

def timeit(method):

    def timed(*args, **kw):
        ts = time.time()
        result = method(*args, **kw)
        te = time.time()

        logging.debug('%r (%r, %r) %2.2f sec' % \
              (method.__name__, args, kw, te-ts))
        return result

    return timed

来源：https://www.andreas-jung.com/contents/a-python-decorator-for-measuring-the-execution-time-of-methods，https : //automatetheboringstuff.com/chapter10/

编辑：我发现Python带有一个非常好的日志模块；为什么要重新发明轮子？

Answer 2

我将向您提供有关您必须做什么才能完成此任务的基本概述。下面是一个装饰器，它接受两个参数，并执行函数。缺少的功能以注释的形式呈现，将它们添加到：

def log_time(path_to_logfile, interval):
    def log(func):
        # 'wrap' this puppy up if needed 
        def wrapped(*args, **kwargs):
            # start timing
            func(*args, **kwargs)
            # stop timing
            with open(path_to_logfile, 'a') as f:
                pass # functionality
        return wrapped
    return log

您现在可以装饰函数，并且输出将以path_to_logfile. 因此，例如，foo在这里装饰：

@log_time('foo.txt', 40)
def foo(i, j):
    print(i, j)

foo(1, 2)

将获取 foo 并执行它。您需要time适当地将内容写入您的文件。你应该更多地尝试装饰器并阅读它们，Python Wiki上有一篇关于装饰器的好文章。

Answer 3

好吧，我最终通过线程找到了一些东西。感谢所有的建议！

import codecs, threading, time
from functools import wraps

def log_time(logpath="log.txt", interval=5):

    def log_time_decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            t = threading.Thread(target=func, args=args, kwargs=kwargs)
            log_entries = 0
            with codecs.open(logpath, "wb", "utf-8") as logfile:
               start_time = time.time()
               t.start()
               while t.is_alive():
                   elapsed_time = (time.time() - start_time)
                   if elapsed_time > interval * log_entries:
                       m, s = divmod(elapsed_time, 60)
                       h, m = divmod(m, 60)
                       logfile.write("Elapsed time: %2dh %2dm %2ds\n" %(h, m, s))
                       log_entries += 1
        return wrapper
    return log_time_decorator

一个缺点可能是您无法轻松检索函数的返回值（至少我还没有弄清楚）。

EDIT1：删除了不必要的变量并添加了一个很好的日志写入格式（请参阅此）

EDIT2：即使其他用户拒绝了他的编辑，我想包含Piotr Dabkowski的版本，因为它适用于返回值：

def log_time(logpath="log.txt", interval=5):

    def log_time_decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            RESULT = [None]
            def temp():
                RESULT[0] = func(*args, **kwargs)
            t = threading.Thread(target=temp)
            log_entries = 0
            with codecs.open(logpath, "wb", "utf-8") as logfile:
               start_time = time.time()
               t.start()
               while t.is_alive():
                   elapsed_time = (time.time() - start_time)
                   if elapsed_time > interval * log_entries:
                       m, s = divmod(elapsed_time, 60)
                       h, m = divmod(m, 60)
                       logfile.write("Elapsed time: %2dh %2dm %2ds\n" %(h, m, s))
                       log_entries += 1
            return RESULT[0]
        return wrapper
    return log_time_decorator