我应该如何从with语句中返回有趣的值？

Question

有没有比使用全局变量从上下文管理器获取有趣值更好的方法？

@contextmanager
def transaction():
    global successCount
    global errorCount
    try:
        yield
    except:
        storage.store.rollback()
        errorCount += 1
    else:
        storage.store.commit()
        successCount += 1

其他可能性:

• 单身

  有点全局...

• 元组作为上下文管理器的参数

  使功能更加特定于问题/更少可重用

• 将特定属性作为参数保存到上下文管理器的实例

  与元组相同的问题,但更清晰

• 在持有值的上下文管理器的末尾引发异常.

  真的很糟糕

Answer 1

请参阅http://docs.python.org/reference/datamodel.html#context-managers

创建一个包含成功和错误计数的类,并实现__enter__和__exit__方法.

Answer 2

我仍然认为您应该创建一个类来保存错误/成功计数，正如我在上一个问题中所说的那样。我猜你有自己的类，所以只需添加如下内容：

class transaction:
    def __init__(self):
        self.errorCount = 0
        self.successCount = 0  

    def __enter__(*args):
        pass  

    def __exit__(self, type, value, traceback):
        if type:
            storage.store.rollback()
            self.errorCount += 1
        else:
            storage.store.commit()
            self.successCount += 1

（type如果调用后没有异常，则为 None contextmanager）

然后您可能已经在某处使用它，它将调用contextmanager并运行您的__exit__()代码。 编辑：正如 Eli 所评论的，只有在您想要重置计数器时才创建一个新的事务实例。

t = transaction()
for q in queries:
    with t:
        t.execute(q)