python - 特定代码行的类似装饰器的语法

Question

链接主题(但不重复):Decorator to time specific lines of the code instead of whole method?

我知道装饰器通常如何用于 Python 函数。

单行代码是否有类似的概念/语法？

例子:用

def measuretime(lineofcode):
    start = time.time()
    lineofcode()
    print time.time() - start

然后

@measuretime
im = Image.open(BytesIO(base64.b64decode(data)))

会被解释为

start = time.time()
im = Image.open(BytesIO(base64.b64decode(data)))
print time.time() - start

注意事项:

• 我知道这样测量执行时间不是最优的，最好使用 timeit等等，但这只是一个随机示例，用于展示我正在寻找的内容(单行代码的装饰器)

• 我正在寻找 1 或 2 行代码解决方案(当然还有函数定义)。如果解决方案需要超过 2 行代码(即超过 @measuretime 之类的代码)，那么最好放弃并按正常方式执行:

    start = time.time()
    im = Image.open(BytesIO(base64.b64decode(data)))
    print time.time() - start
  

Answer 1

如果你想在一行代码前后做一些事情，一个context manager将是合适的:

from contextlib import contextmanager
import time

@contextmanager
def measuretime():
    start = time.time()
    try:
        yield
    finally:
        print(time.time() - start)

with measuretime():
    do_stuff()