"""
預激協程的裝飾器

"""


from inspect import getgeneratorstate
from functools import wraps


def coroutine(func):
? ? """裝飾器：向前執行到第一個`yield`表達式，預激`func`"""

? ? # 把被裝飾的生成器函數替換成這里的 primer 函數；
? ? # 調用 primer 函數時，返回預激后的 生成器。
? ? @wraps(func)
? ? def primer(*args, **kwargs):
? ? ? ? # 調用被裝飾的函數，獲取生成器對象。
? ? ? ? gen = func(*args, **kwargs)
? ? ? ? # 預激生成器。
? ? ? ? next(gen)
? ? ? ? # 返回生成器。
? ? ? ? return gen

? ? return primer


@coroutine
def averager():
? ? total = 0.0
? ? count = 0
? ? average = None
? ? while True:
? ? ? ? term = yield average
? ? ? ? total += term
? ? ? ? count += 1
? ? ? ? average = total / count


if __name__ == '__main__':
? ? coro_avg = averager()
? ? # print(getgeneratorstate(coro_avg))
? ? print(coro_avg.send(10))
? ? print(coro_avg.send(30))
? ? # 發送的值不是數字，導致協程內部有異常拋出。
? ? print(coro_avg.send('spam'))
? ? # 由于在協程內沒有處理異常，協程會終止。
? ? # 如果試圖重新激活協程，會拋出 StopIteration 異常。
? ? print(coro_avg.send(60))

generator.throw(exc_type[, exc_value[, traceback]])

generator.close()

"""
學習在協程中處理異常的測試代碼
"""
from inspect import getgeneratorstate

class DemoException(Exception):
? ? """為這次演示定義的異常類型。"""

def demo_exc_handling():
? ? print('-> coroutine started')
? ? try:
? ? ? ? while True:
? ? ? ? ? ? try:
? ? ? ? ? ? ? ? x = yield
? ? ? ? ? ? # ?特別處理 DemoException 異常
? ? ? ? ? ? except DemoException:
? ? ? ? ? ? ? ? print('*** DemoException handled. Continuing...')
? ? ? ? ? ? # 如果沒有異常，那么顯示接收到的值。
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print('-> coroutine received: {!r}'.format(x))
? ? finally:
? ? ? ? # 如果不管協程如何結束都想做些清理工作，
? ? ? ? # 要把協程定義體中相關的代碼放入try/ finally 塊中
? ? ? ? print('-> coroutine ending')

if __name__ == '__main__':
? ? exc_coro = demo_exc_handling()
? ? next(exc_coro)
? ? exc_coro.send(11)
? ? exc_coro.send(22)
? ? # 激活和關閉 demo_exc_handling，沒有異常
? ? # exc_coro.close()

? ? # 如果把 DemoException 異常傳入 demo_exc_handling 協程，
? ? # 它會處理，然后繼續運行
? ? # exc_coro.throw(DemoException)
? ? # exc_coro.send(33)

? ? # 如果無法處理傳入的異常，協程會終止
? ? exc_coro.throw(ZeroDivisionError)

? ? print(getgeneratorstate(exc_coro))