다른 컨텍스트 관리자 내부의 컨텍스트 관리자 인스턴스 처리

Question

다른 컨텍스트 관리자 내부에서 작성된 컨텍스트 관리자는 어떻게 파이썬에서 처리해야합니까?

예 : 컨텍스트 관리자 역할을하는 A 클래스와 컨텍스트 관리자 역할을하는 B 클래스가 있다고 가정합니다. 그러나 클래스 B 인스턴스는 인스턴스화하고 클래스 A의 인스턴스를 사용해야합니다. 저는 PEP 343을 생각해 봤는데 이것이 제가 생각한 해결책입니다 :

class A(object): 
    def __enter__(self): 
     # Acquire some resources here 
     return self 

    def __exit__(seplf, exception_type, exception, traceback): 
     # Release the resources and clean up 
     pass 


class B(object): 
    def __init__(self): 
     self.a = A() 

    def __enter__(self): 
     # Acquire some resources, but also need to "start" our instance of A 
     self.a.__enter__() 
     return self 

    def __exit__(self, exception_type, exception, traceback): 
     # Release the resources, and make our instance of A clean up as well 
     self.a.__exit__(exception_type, exception, traceback) 

올바른 방법입니까? 아니면 내가 몇 가지 잡았을까?

Answer 1

또는, 당신은과 같이 코드를 작성할 수

with A() as a: 
    with B(a) as b: 
     # your code here 

이 될 수 시도 할 수 있습니다 또 다른 방법 : 상황의 설명을 고려한 후에

class A: 

    def __init__(self): 
     pass 

    def __enter__(self): 
     return self 

    def __exit__(self, exc_type, exc_val, exc_tb): 
     pass 

class B(A): 

    def __init__(self): 
     super().__init__() 

    def __enter__(self): 
     super().__enter__() 
     return self 

    def __exit__(self, exc_type, exc_val, exc_tb): 
     super().__exit__(exc_type, exc_val, exc_tb) 

을,이있을 수 있습니다 더 나은 해결책 :

class Resource: 

    def __init__(self, dependency=None): 
     self.dependency = dependency 
     # your code here 

    def __enter__(self): 
     if self.dependency: 
      self.dependency.__enter__() 
     # your code here 
     return self 

    def __exit__(self, exc_type, exc_val, exc_tb): 
     # your code here 
     if self.dependency: 
      self.dependency.__exit__(exc_type, exc_val, exc_tb) 

다음 구현 정확하지만, __exit__은 예외를 적절하게 처리해야합니다. 예외를 적절하게 처리하면서 호출을 순환 적으로 연결하는 방법을 상상하는 것은 다소 어렵습니다.

class Resource: 

    def __init__(self, dependency=None): 
     self.dependency = dependency 
     self.my_init() 

    def __enter__(self): 
     if self.dependency: 
      self.dependency.__enter__() 
     return self.my_enter() 

    def __exit__(self, exc_type, exc_val, exc_tb): 
     suppress = False 
     try: 
      suppress = self.my_exit(exc_type, exc_val, exc_tb): 
     except: 
      exc_type, exc_val, exc_tb = sys.exc_info() 
     if suppress: 
      exc_type = exc_val = exc_tb = None 
     if self.dependency: 
      suppress = self.dependeny.__exit__(exc_type, exc_val, exc_tb) 
      if not supress: 
       raise exc_val.with_traceback(exc_tb) from None 
     return suppress 

    def my_init(self): 
     pass 

    def my_enter(self): 
     pass 

    def my_exit(self, exc_type, exc_val, exc_tb): 
     pass 

Answer 2

당신이 당신의 인생을 훨씬 쉽게 얻는 @contextlib.contextmanager 장식 사용할 수있는 경우 :

import contextlib 

@contextlib.contextmanager 
def internal_cm(): 
    try: 
     print "Entering internal_cm" 
     yield None 
     print "Exiting cleanly from internal_cm" 
    finally: 
     print "Finally internal_cm" 


@contextlib.contextmanager 
def external_cm(): 
    with internal_cm() as c: 
     try: 
      print "In external_cm_f", c 
      yield [c] 
      print "Exiting cleanly from external_cm_f", c 
     finally: 
      print "Finally external_cm_f", c 


if "__main__" == __name__: 
    with external_cm(): 
     print "Location A" 
    print 
    with external_cm(): 
     print "Location B" 
     raise Exception("Some exception occurs!!")