如何实现装饰发电机

Question

我有一些发电机：

def my_gen():
    while True:
        #some code
        yield data_chunk

我有一些功能，用数据格式进行一些操作

def my_formatting_func(data_chunk):
    #some code
    return formated_data_chunk

是什么创造发电机至极的最短途径产生data_chunks通过格式化my_formatting_func没有对矫正my_gen ？

Answer 1

假设Python 3.x并且生成器不接受任何参数（后者添加起来很简单）：

def wrapper(generator):
    def _generator():
        return map(my_formatting_func, generator())
    return _generator

@wrapper
def my_gen():
    # do stuff

对于2.x，请使用itertools.imap而不是map 。