python: cree un bloque "con" en varios gestores de contexto

Question

Supongamos que tiene tres objetos que adquiere a través del gestor de contexto, por ejemplo, un bloqueo, una conexión db y un socket ip. Puede adquirirlos por:

with lock: 
    with db_con: 
     with socket: 
      #do stuff 

pero hay una manera de hacerlo en un bloque? algo así como

with lock,db_con,socket: 
    #do stuff 

Por otra parte, es posible que, dada una matriz de longitud desconocida de objetos que tienen gestores de contexto, es posible de alguna manera hacer:

a=[lock1, lock2, lock3, db_con1, socket, db_con2] 
with a as res: 
    #now all objects in array are acquired 

Si la respuesta es "no", ¿es porque la necesidad de tal característica implica un mal diseño, o tal vez debería sugerirlo en un pep? :-P

Answer 1

En Python 2.6 y continuación, puede utilizar contextlib.nested:

from contextlib import nested 

with nested(A(), B(), C()) as (X, Y, Z): 
    do_something() 

es equivalente a:

m1, m2, m3 = A(), B(), C() 
with m1 as X: 
    with m2 as Y: 
     with m3 as Z: 
      do_something() 

Tenga en cuenta que esto no es exactamente el mismo que normalmente utilizando anidada with, porque A(), B() y C() se llamarán inicialmente, antes de ingresar a los administradores de contexto. Esto no funcionará correctamente si una de estas funciones puede generar excepciones, pero funcionará para los ejemplos en la pregunta.

---

En Python 2.7 y 3.1, la sintaxis se ha añadido para esto, y contextlib.nested ha quedado obsoleto:

with A() as X, B() as Y, C() as Z: 
    do_something() 

---

En Python 3.3, también puede introducir un desconocido -length list of context managers using contextlib.ExitStack:

with ExitStack() as stack: 
    for mgr in ctx_managers: 
     stack.enter_context(mgr) 
    # ... 

Esto le permite crear los administradores de contexto al agregarlos al ExitStack, lo que evita el posible problema con contextlib.nested.

contextlib2 proporciona a backport of ExitStack para Python 2.6 y 2.7.

Answer 2

La primera parte de su pregunta es posible en Python 3.1.

Con más de un artículo, los gestores de contexto se procesan como si instrucciones múltiples se anida:

with A() as a, B() as b: 
    suite 

es equivalente a

with A() as a: 
    with B() as b: 
     suite 

cambiado en la versión 3.1: Apoyo para expresiones de contexto múltiples

Answer 3

La segunda parte de su pregunta se resuelve con contextlib.ExitStack en Python 3.3.

Answer 4

@ interjay's La respuesta es correcta. Sin embargo, si necesita hacer esto para administradores de contexto largos, por ejemplo administradores de contexto de mock.patch, entonces se da cuenta rápidamente de que desea dividir esto entre líneas. Resulta que no puedes envolverlos en paréntesis, por lo que debes usar barras diagonales inversas. Esto es lo que parece:

with mock.patch('aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa') as a, \ 
     mock.patch('bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb') as b, \ 
     mock.patch('cccccccccccccccccccccccccccccccccccccccccc') as c: 
    do_something()