如果調用者對其調用close（），生成器是否可以注意到？

Question

給出一個簡單的發電機：

def myGenerator(max): 
    for i in range(max): 
     yield i 

它可以用於像：

>>> gen = myGenerator(10) 
>>> next(gen) 
0 
>>> next(gen) 
1 

當我在StopIteration例外，對發電機進行close()，所有後續調用next結果。

>>> gen.close() 
>>> next(gen) 
StopIteration exception 

發電機可以注意嗎？ yield不會引發異常。我在尋找類似：

def myGenerator(max): 
    for i in range(max): 
     try: 
      yield i 
     except CloseIteration as ex: 
      print("I got closed") 

Answer 1

由於喬恩在評論中提到，calling close() raises a GeneratorExit exception，您可以觀看：

In [1]: def mygenerator(max): 
    ...:  try: 
    ...:   for i in range(max): 
    ...:    yield i 
    ...:  except GeneratorExit: 
    ...:   print('I got closed') 
    ...:   raise 
    ...:  

In [2]: gen = mygenerator(10) 

In [3]: next(gen) 
Out[3]: 0 

In [4]: next(gen) 
Out[4]: 1 

In [5]: gen.close() 
I got closed