python：如何包装“with”参数？

Question

我有with构造的两个参数：

with A(), B():
    do_something()

A()和B()内置__enter__和__exit__个方法。但是我想在一些函数中包含A和B来得到这样的结构：

with my_wrapper():
    do_something()

我该怎么做？我试着这样做：

def my_wrapper():
    return A(), B()

但当然，我得到AttributeError