python: tạo khối "có" trên một số trình quản lý ngữ cảnh

Question

Giả sử bạn có ba đối tượng bạn có được thông qua trình quản lý ngữ cảnh, ví dụ: Khóa, kết nối db và ổ cắm ip. Bạn có thể mua chúng bằng cách:

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

Nhưng có một cách để làm điều đó trong một khối? cái gì đó như

with lock,db_con,socket: 
    #do stuff 

Bên cạnh đó, là nó có thể, đưa ra một mảng không rõ chiều dài của đối tượng mà có người quản lý bối cảnh, là nó có thể bằng cách nào đó làm:

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

Nếu câu trả lời là "không", có phải vì sự cần thiết cho một tính năng như vậy có nghĩa là thiết kế xấu, hoặc có lẽ tôi nên đề nghị nó trong một pep? :-P

Answer 1

Trong Python 2.6 và dưới đây, bạn có thể sử dụng contextlib.nested:

from contextlib import nested 

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

tương đương với:

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

Lưu ý rằng đây không phải là chính xác giống như bình thường sử dụng lồng nhau with, bởi vì A(), B() và C() tất cả sẽ được gọi ban đầu, trước khi vào trình quản lý ngữ cảnh. Điều này sẽ không hoạt động chính xác nếu một trong các hàm này có thể tăng ngoại lệ, nhưng sẽ làm việc cho các ví dụ trong câu hỏi.

---

Trong Python 2.7 và 3.1, cú pháp đã được thêm vào cho điều này, và contextlib.nested đã được chấp nhận:

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

---

Trong Python 3.3, bạn cũng có thể nhập một không rõ danh sách độ dài của người quản lý ngữ cảnh bằng cách sử dụng contextlib.ExitStack:

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

Điều này cho phép bạn tạo các trình quản lý ngữ cảnh khi bạn thêm chúng vào ExitStack, ngăn sự cố có thể xảy ra với contextlib.nested.

contextlib2 cung cấp a backport of ExitStack cho Python 2.6 và 2,7.

Answer 2

Phần đầu tiên của câu hỏi là có thể trong Python 3.1.

Với nhiều hơn một mục, các nhà quản lý bối cảnh được chế biến như nhiều với báo cáo được lồng nhau:

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

tương đương với

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

Thay đổi trong phiên bản 3.1: Hỗ trợ cho nhiều biểu thức ngữ cảnh

Answer 3

Phần thứ hai của câu hỏi của bạn được giải quyết với contextlib.ExitStack trong Python 3.3.

Answer 4

Câu trả lời của @ interjay là chính xác. Tuy nhiên, nếu bạn cần làm điều này cho các nhà quản lý ngữ cảnh dài, ví dụ như các trình quản lý ngữ cảnh mock.patch, thì bạn nhanh chóng nhận ra rằng bạn muốn phá vỡ điều này trên các dòng. Hóa ra bạn không thể quấn chúng trong parens, vì vậy bạn phải sử dụng dấu gạch chéo ngược. Dưới đây là hình thức:

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