獲得通過__enter__在__exit__

Question

例如回來後，我有這樣的代碼：

with MyClass() as x: 
    print 'I have only {0}'.format(x) 
    with MyClass() as y: 
     print 'I have {0} and {1}'.format(x, y) 
    print 'Again only {0}'.format(x) 

x和y都應該相應with塊退出後去初始化。 x和y也不是MyClass的實例。

__exit__只有三個參數，每個參數都是無（如​​果沒有提供異常）。

我如何確定__exit__哪個塊剛剛退出，以及__enter__返回的值是什麼？

（N.B.代碼應該是線程安全的）。

---

例子：

class MyClass(object): 
    def __enter__(self): 
     if moon_phase > 0: 
      return 123 
     else: 
      return 456 
    def __exit__(self): 
     number = what_was_returned_by_enter() 
     print 'End of block with', number 


with MyClass() as x: 
    print x # 123 
    with MyClass() as y: 
     print x, 'and', y # 123 and 456 
    # printed "End of block with 456" 
    print x # 123 
# printed "End of block with 123" 

Answer 1

既然你有一個自定義類處理您的背景下，self將上下文管理器實例。

您需要檢查其狀態（存儲在__init__()創建時間或調用__enter__()時）以確定您剛剛退出的哪一個。

下面的示例存儲的__enter__實例上的返回值，以便可以再次找回它，當__exit__被稱爲：

class MyClass(object): 
    def __enter__(self): 
     if moon_phase > 0: 
      self.returnval = 123 
     else: 
      self.returnval = 456 
     return self.returnval 

    def __exit__(self, *args): 
     number = self.returnval 
     print 'End of block with', number