根据条件使用不同的上下文管理器

Question

是否可以根据某些条件使用不同的上下文管理器执行单个块？

例：

if some_condition:
    with ContextManager(**args) as contex:
        ... # some block
else:
    with OtherContextManager(**other_args) as contex:
        ... # the same block

一种方法是将...包装成一个函数，但在我的情况下这可能不太方便。 还有其他可能吗？

Answer 1

您可以将构造的对象存储在变量中，例如：

if some_condition:
    cm = ContextManager(**args)
else:
    cm = OtherContextManager(**other_args)

with cm as contex:
        ... # some block

以上内容可以很容易地扩展到三个可能的上下文管理器等。您还可以决定在“进入”上下文之前首先“修补”上下文管理器。

虽然通常会看到类似于with foo() as bar:的模式with foo() as bar:实际上Python只是计算foo() ，获取该元素，并调用.__enter__()在对象上.__enter__() 。 该方法的结果存储在bar 。

所以关于foo() 调用没有任何“特殊”，你可以在左侧使用任何类型的对象。 因此，您可以将if - else逻辑封装在单独的函数中，并返回上下文管理器，然后使用该变量，或者将上下文管理器作为参数传递。 只要你在with语句中使用它，Python就会调用.__enter__(..)在窗帘后面.__enter__(..)和.__exit__(..) 。

Answer 2

怎么样...

with ContextManager(**args) if some_condition else OtherContextManager(**other_args) as contex:
    ... # some block

...？

Answer 3

我们可以疯狂地利用__enter__和__exit__都只是方法的事实，并且它们在原始对象中被调用（而不是__enter__返回的__enter__ ）：

class WrapperContext:

 def __init__(self, condition):
     if condition:
         self.real_context = ContextA()
     else:
         self.real_context = ContextB()

 def __enter__(self):
     return self.real_context.__enter__()

 def __exit__(self):
     return self.real_context.__exit__()

并像这样使用它：

 with WrapperContext(condition) as obj: