Obsługa wyjątków przy użyciu Pythona subprocess.Popen

Question

Gdy mamy do czynienia z otwartych plików, Python posiada składnię with sprawia, że ​​się, że plik zostanie zamknięty po opuszczeniu bloku - bez wyjątków itp

with open('foo.txt') as f: 
    foo = f.read() 

Ponieważ procesy są zbyt zasoby , Zastanawiałem się: czy jest coś podobnego możliwego lub zalecanego przy użyciu Popen? Na przykład, czy powinno się uruchomić Popen.kill(); Popen.communicate() w klauzuli finally - zakładając, że nie mam nic przeciwko blokowaniu, dopóki proces się nie zakończy?

Answer 1

Począwszy od Python 3.2 Popen jest menedżerem kontekstu.

z docs:

POPEN obiekty są obsługiwane za pośrednictwem menedżerów kontekstowych ze stwierdzeniem: na wyjściu standardowe deskryptory plików są zamknięte, a proces czekał.

Powinno to zrobić dokładnie to, co chcesz.

Jest odpowiednia część z subprocess.py ze standardowego lib w Pythonie 3.4:

def __enter__(self): 
    return self 

def __exit__(self, type, value, traceback): 
    if self.stdout: 
     self.stdout.close() 
    if self.stderr: 
     self.stderr.close() 
    if self.stdin: 
     self.stdin.close() 
    # Wait for the process to terminate, to avoid zombies. 
    self.wait() 

Teraz można zrobić w Pythonie 2.7

from subprocess import Popen 

class MyPopen(Popen): 

    def __enter__(self): 
     return self 

    def __exit__(self, type, value, traceback): 
     if self.stdout: 
      self.stdout.close() 
     if self.stderr: 
      self.stderr.close() 
     if self.stdin: 
      self.stdin.close() 
     # Wait for the process to terminate, to avoid zombies. 
     self.wait() 

if __name__ == '__main__': 
    with MyPopen(['ls']) as p: 
     print(p) 

Answer 2

Można dodać tylko dwie metody niestandardowe dowolnego klasa implementująca kompilację z instrukcją with.

class CustomObject(object): 
    def __enter__(self): 
     """ This method execudes when entering block. """ 
     return thing_you_want_to_use 

    def __exit__(self, type, value, traceback): 
     """ This method execudes on block exit. """ 
     # Tear things down. 

Answer 3

2,7 można również użyć @contextlib.contextmanager:

import contextlib 

@contextlib.contextmanager 
def manage_process(process): 
    try: 
     yield process 
    finally: 
     for stream in [process.stdout, process.stdin, process.stderr]: 
      if stream: 
       stream.close() 
     process.wait() 

np:

with manage_process(Popen(['ls'])) as p: 
    print(p)