Python如何将方法结果转换为生成器

Question

我有以下继承：

class Processor(object):
    def get_listings(self):
        """
        returns a list of data
        """
        raise NotImplemented()

    def run(self):
        for listing in get_listings():
           do_stuff(listing)

class DBProcessor(Processor):
    def get_listings(self):
        """
        return a large set of paginated data
        """
        ...
        for page in pages:
            for data in db.fetch_from_query(...):
                yield data

尽管此方法有效，但在len(self.get_listings())或任何其他列表操作上均失败。

我的问题是如何重构我的代码，使DBProcessor.get_listings可以处理列表操作，但是当它的迭代器被调用时，它将返回一个生成器？

Answer 1

我想我有个主意：

class DBListings(object):
    def __iter__(self):
        for page in pages:
            for data in db.fetch_from_query(...):
                yield data

    def __len__(self):
        return db.get_total_from_query(...)
        """
        Or the following
        counter = 0
        for x in self:
           counter += 1
        return counter
        """

class DBProcessor(Processor):
    def get_listings(self):
        """
        return a large set of paginated data
        """
        return DBListings()

更新 ：刚刚测试了上面的代码，可以正常工作。

Answer 2

这取决于要支持的list操作。 其中一些仅在默认为iter时才会消耗生成器。

如果您事先知道操作的结果（例如len ），则可以通过创建GeneratorContainer绕过它：

class GeneratorContainer():
    def __init__(self, generator, length):
        self.generator = generator
        self.length = length

    def __iter__(self):
        return self.generator

    def __len__(self):
        return self.length

result = GeneratorContainer(DBProcessor().get_listings(), length) 
# But you need to know the length-value.

然后，调用len不会尝试迭代生成器。 但是，您始终可以只创建一个列表，以便不会耗尽数据：

result = list(DBProcessor().get_listings())

并将其用作没有生成器优点和缺点的列表。

Answer 3

如果您希望将get_listings生成的生成器（非Python迭代器） get_listings为列表，只需使用

listings = list(get_listings())