4

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

@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

有任何想法吗？

4

3 回答 3

6

我将为您提供有关完成此操作必须做什么的基本概述。下面是一个接受两个参数并执行函数的装饰器。缺少的功能显示为注释，将它们添加到：

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 上有一篇关于装饰器的好文章。

于 2016-01-24T21:29:46.133 回答
3

快速组合起来，但在一些功能上使用 @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 带有一个非常好的日志模块；为什么要重新发明轮子？

于 2016-01-24T21:29:34.783 回答
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
于 2016-01-24T23:23:19.813 回答