有没有办法直接“装饰”一段 Python 代码？

Question

我有一堆代码：

statement1(args)
statement2(args)
statement3(args)
statement4(args)
statement5(args)

我想将语句分成块并在每个块之后写入日志。日志记录有点复杂：我想记录每个块的运行时间和块执行后特定数据结构的状态。所以我创建了一个名为log_block 的装饰器来处理所有这些细节。现在我的代码如下所示：

@log_block()
def block1():
    statement1(args)
    statement2(args)

@log_block()
def block2()
    statement3(args)

@log_block()
def block3():
    statement4(args)
    statement5(args)

block1()
block2()
block3()

这很好用，但有点笨拙。我不得不分别调用三个块函数，这很烦人，如果我想在块之间共享一个变量，那么我要么必须给块函数参数和返回语句，要么使用全局变量，这两者都不是特别可口。我真正想要的是看起来像这样的语法：

@log_block()
    statement1(args)
    statement2(args)

@log_block()
    statement3(args)

@log_block()
    statement4(args)
    statement5(args)

所以我直接装饰语句而不是将它们包含在辅助块函数中。有没有办法实现这样的目标？

Answer 1

Context managers 正是您正在寻找的。您将它们与 with 语句一起使用，它们定义了在进入和退出 with 块时要运行的代码。