我正在编写一个基于 MPI 的应用程序(但 MPI 在我的问题中并不重要，我提到它只是为了揭示基本原理)并且在某些情况下，当工作项少于流程时，我需要创建一个新的通信器，排除了无关的进程。最后，新的通信器必须由有工作要做的进程释放(并且只能由它们完成)。

一个简洁的方法是写:

with filter_comm(comm, nworkitems) as newcomm:
    ... do work with communicator newcomm...

正文仅由有工作要做的进程执行。

在上下文管理器中有没有办法避免执行正文？ 我知道上下文管理器的设计是为了避免隐藏控制流，但我想知道是否有可能规避这一点，因为在我的例子中，为了清楚起见，我认为这是合理的。

最佳答案

有条件地跳过上下文管理器主体的能力已被提议但被拒绝，如 PEP 377 中所述.

我做了一些关于替代品的研究。这是我的发现。

首先让我解释一下我的代码示例的背景。您有一堆要使用的设备。对于每个设备，您都必须获取该设备的驱动程序；然后使用驱动程序与设备一起工作；最后释放驱动程序，以便其他人可以获得驱动程序并使用该设备。

这里没有什么不寻常的。代码大致如下所示:

driver = getdriver(devicename)
try:
  dowork(driver)
finally:
  releasedriver(driver)

但是每当满月时，当行星未正确对齐时，获得的设备驱动程序就会出现问题，并且无法使用该设备完成任何工作。这没什么大不了的。只需跳过这一轮的设备，然后在下一轮重试。通常司机是好的。但即使是坏的驱动程序也需要发布，否则无法获取新的驱动程序。

(固件是专有的，供应商不愿意修复甚至承认这个错误)

代码现在看起来像这样:

driver = getdriver(devicename)
try:
  if isgooddriver(driver):
    dowork(driver)
  else:
    handledrivererror(geterrordetails(driver))
finally:
  release(driver)

每次需要使用设备完成工作时，都需要重复大量样板代码。 python context manager 的主要候选者也称为 with statement .它可能看起来像这样:

# note: this code example does not work
@contextlib.contextmanager
def contextgetdriver(devicename):
  driver = getdriver(devicename)
  try:
    if isgooddriver(driver):
      yield driver
    else:
      handledrivererror(geterrordetails(driver))
  finally:
    release(driver)

然后使用设备时的代码又短又甜:

# note: this code example does not work
with contextgetdriver(devicename) as driver:
  dowork(driver)

但这行不通。因为上下文管理器必须屈服。它可能不会屈服。不屈服将导致 contextmanager 引发的 RuntimeException。

所以我们必须从上下文管理器中取出检查

@contextlib.contextmanager
def contextgetdriver(devicename):
  driver = getdriver(devicename)
  try:
    yield driver
  finally:
    release(driver)

并将其放在with语句的主体中

with contextgetdriver(devicename) as driver:
  if isgooddriver(driver):
    dowork(driver)
  else:
    handledrivererror(geterrordetails(driver))

这很丑陋，因为现在我们再次有了一些样板，每次我们想要使用设备时都需要重复这些样板。

所以我们想要一个可以有条件地执行主体的上下文管理器。但我们没有，因为PEP 377 (恰好暗示了这个功能)被拒绝了。

我们可以自己引发一个异常，而不是不屈服:

@contextlib.contextmanager
def contextexceptgetdriver(devicename):
  driver = getdriver(devicename)
  try:
    if isgooddriver(driver):
      yield driver
    else:
      raise NoGoodDriverException(geterrordetails(driver))
  finally:
    release(driver)

但是现在你需要处理异常:

try:
  with contextexceptgetdriver(devicename) as driver:
    dowork(driver)
except NoGoodDriverException as e:
  handledrivererror(e.errordetails)

这与上面对良好驱动程序的显式测试具有几乎相同的代码复杂性成本。

区别:除了异常，我们可以决定不在这里处理它，而是让它在调用堆栈中冒泡并在别处处理。

另外的区别:当我们处理异常时，驱动程序已经被释放。虽然通过显式检查驱动程序尚未发布。 (except在with语句外，else在with语句内)

我发现滥用 generator作为可以跳过正文的上下文管理器的替代品，效果很好

def generatorgetdriver(devicename):
  driver = getdriver(devicename)
  try:
    if isgooddriver(driver):
      yield driver
    else:
      handledrivererror(geterrordetails(driver))
  finally:
    release(driver)

但是调用代码看起来很像一个循环

for driver in generatorgetdriver(devicename):
  dowork(driver)

如果您可以接受(请不要)，那么您就有了一个可以有条件地执行正文的上下文管理器。

似乎防止样板代码的唯一方法是使用回调

def workwithdevice(devicename, callback):
  driver = getdriver(devicename)
  try:
    if isgooddriver(driver):
      callback(driver)
    else:
      handledrivererror(geterrordetails(driver))
  finally:
    release(driver)

调用代码

workwithdevice(devicename, dowork)

关于Python 上下文管理器 : conditionally executing body?，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/10447818/