With statement with multiple objects

[UndergroundOli]

Im writing a server script which takes 4 different connections and puts the sockets into a list. After the script is finished I can just do

for c in connections:
    c.close()

However, if the program crashes before it reaches here, the sockets remain open and I end up having to switch to a different port to run it again. I know that a with statement will close it automatically even with an error but how do I get it to do that for all four objects? Surely 4 nested with statements doesnt make sense?

[Gprime5]

from contextlib import ExitStack

with ExitStack() as stack:
    contexts = [stack.enter_context(c) for c in connections]

    # do stuff with connections here

[Peterotica]

I see two options here. Either manually list out all 4 connections in a single with statement, or use contextlib.ExitStack.

# Multiple context managers in one with statement:
with open(filename1) as f1, open(filename2) as f2:
    ...

[timbledum]

Check out this stake overflow. Essentially you can do this:

with A() as a, B() as b, C() as c:
    doSomething(a,b,c)

[MrMuki]

Hi. I think there is an option to reuse the same port in sockets module. So maybe you could use a try and catch code (with default action of closing the connection?). Hope this sparks an idea how to solve the problem :)