python协程

coroutine.py

from functools import wraps

def coroutine(func):
  """
  装饰器，用来预激活协程，协程用之前需要用next来激活
  """
  @wraps(func)
  def primer(*args, **kwargs):
    gen = func(*args, **kwargs)
    next(gen)   # 激活
    return gen
  return primer