我应该如何从 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- manager

创建一个类，用于保存成功和错误计数，并实现 __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

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

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

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

Answer 3

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

使函数更具体地解决问题/更少可重用”

错误。

这使得上下文管理器保留状态。

如果您不实现除此之外的任何内容，它将是可重用的。

但是，您实际上无法使用元组，因为它是不可变的。 你需要一些可变的集合。 我想到了字典和类定义。

因此，推荐的实现是

“将特定属性作为上下文管理器参数的实例”。

您只需要一个具有两个属性的简单类定义。 但是，您的事务状态是有状态的，您需要在某处保留状态。

class Counters(dict):
    SUCCEED= 0
    FAIL= 1
    def __init__( self ):
        self[ self.SUCCEED ]= 0
        self[ self.FAIL ]= 0 
    def increment( self, status ):
        self[status] += 1

class Transaction(object):
    def __init__( self, worker, counters ):
        self.worker= worker
        self.counters= counters
    def __enter__( self ):
        self.counters.status= None
    def process( self, *args, **kw ):
        status= self.worker.execute( *args, **kw )
        self.counters.increment( status )
    def __exit__( self ):
        pass

counts= Counters()
for q in queryList:
    with Transaction(execQuery,counts) as t:
        t.process( q )
print counts