如何在对象列表中使用“with”

Question

假设我有一个类，它将生成一个线程并实现.__enter__和.__exit__，因此我可以这样使用它：

with SomeAsyncTask(params) as t:
    # do stuff with `t`
    t.thread.start()
    t.thread.join()

.__exit__可能出于清理目的执行某些操作（即删除临时文件等）

在我有一个SomeAsyncTask列表之前，这一切都很好。你知道吗

list_of_async_task_params = [params1, params2, ...]

我应该如何在列表中使用with？我希望这样：

with [SomeAsyncTask(params) for params in list_of_async_task_params] as tasks:
    # do stuff with `tasks`
    for task in tasks:
        task.thread.start()
    for task in tasks:
        task.thread.join()

Answer 1

@martineau答案应该有用。这里有一个更通用的方法，应该适用于其他情况。请注意，__exit__()中不处理异常。如果一个__exit__()函数失败，其余的函数将不会被调用。通用解决方案可能会抛出聚合异常并允许您处理它。另一种情况是当您的第二个管理器的__enter__()方法抛出异常时。在这种情况下，不会调用第一个管理器的__exit__()。你知道吗

class list_context_manager:
  def __init__(self, managers):
    this.managers = managers

  def __enter__(self):
    for m in self.managers:
      m.__enter__()
    return self.managers

  def __exit__(self):
    for m in self.managers:
      m.__exit__()

然后可以像在你的问题中那样使用：

with list_context_manager([SomeAsyncTask(params) for params in list_of_async_task_params]) as tasks:
    # do stuff with `tasks`
    for task in tasks:
        task.thread.start()
    for task in tasks:
        task.thread.join()