2018-09-27 21:33:08 -04:00
|
|
|
# case where generator doesn't intercept the thrown/injected exception
|
|
|
|
def gen():
|
|
|
|
yield 123
|
|
|
|
yield 456
|
2021-04-20 21:04:23 -04:00
|
|
|
|
2018-09-27 21:33:08 -04:00
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
try:
|
|
|
|
g.throw(KeyError)
|
|
|
|
except KeyError:
|
|
|
|
print('got KeyError from downstream!')
|
|
|
|
|
|
|
|
# case where a thrown exception is caught and stops the generator
|
|
|
|
def gen():
|
|
|
|
try:
|
|
|
|
yield 1
|
|
|
|
yield 2
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
try:
|
|
|
|
g.throw(ValueError)
|
|
|
|
except StopIteration:
|
|
|
|
print('got StopIteration')
|
|
|
|
|
|
|
|
# generator ignores a thrown GeneratorExit (this is allowed)
|
|
|
|
def gen():
|
|
|
|
try:
|
|
|
|
yield 123
|
2021-04-22 20:55:39 -04:00
|
|
|
except GeneratorExit as e:
|
|
|
|
print('GeneratorExit', repr(e.args))
|
2018-09-27 21:33:08 -04:00
|
|
|
yield 456
|
2021-04-20 21:04:23 -04:00
|
|
|
|
2018-09-27 21:33:08 -04:00
|
|
|
# thrown a class
|
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
print(g.throw(GeneratorExit))
|
|
|
|
|
|
|
|
# thrown an instance
|
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
print(g.throw(GeneratorExit()))
|
2021-04-22 20:55:39 -04:00
|
|
|
|
|
|
|
# thrown an instance with None as second arg
|
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
print(g.throw(GeneratorExit(), None))
|
|
|
|
|
|
|
|
# thrown a class and instance
|
|
|
|
g = gen()
|
|
|
|
print(next(g))
|
|
|
|
print(g.throw(GeneratorExit, GeneratorExit(123)))
|