Python 使用带生成器函数的Flask缓存

Python 使用带生成器函数的Flask缓存,python,flask,Python,Flask,我正在尝试使用Flask缓存缓存生成器函数。烧瓶缓存需要酸洗,由于发电机无法酸洗,这是一个问题。使用带烧瓶缓存的生成器有什么解决方法吗?如果要缓存结果,只需在生成器上调用列表,并将其存储即可。如果结果太大,无法安全地存储在内存中(即使是临时存储),则将结果包装在类实例中,该类实例可以: # Warning: This is terribly tightly bound together # There are better ways to do this - this is just by wa

我正在尝试使用Flask缓存缓存生成器函数。烧瓶缓存需要酸洗,由于发电机无法酸洗,这是一个问题。使用带烧瓶缓存的生成器有什么解决方法吗?

如果要缓存结果,只需在生成器上调用
列表
,并将其存储即可。如果结果太大,无法安全地存储在内存中(即使是临时存储),则将结果包装在类实例中,该类实例可以:

# Warning: This is terribly tightly bound together
# There are better ways to do this - this is just by way of example

def some_generator(identifier):
    """This could be anything
        * A database wrapper
        * A permutation generator
        * Something that streams (and transforms) data from disk
        * The possibilities are endless"""
    if identifier == "result_set_1":
        results = [(x, y) for x in range(10) for y in range(x)]
    elif identifier == "result_set_2":
        results = [(x, 0) for x in range(20)]
    else:
        results = [(1, 2), (3, 4)]

    for row in results:
        yield row


class GeneratorWrapper(object):
    """
    Use::

        @cache
        some_function(identifier):
            return GeneratorWrapper(identifier)
    """

    def __init__(self, identifier=None):
        self.identifier = identifier
        if identifier is not None:
            self.data = some_generator(identifier)

    def __getstate__(self):
        """When pickled return the key to rebuilding the generator"""
        return self.identifier

    def __setstate__(self, state):
        """When unpickled, restore the generator using the pickled key"""
        self.identifier = state
        if state is not None:
            self.data = some_generator(state)

    def __iter__(self):
        return self.data