python - pymongo 生成器失败 - 'return' 生成器内有参数

Question

我正在尝试执行以下操作:

def get_collection_iterator(collection_name, find={}, criteria=None):
    collection = db[collection_name]
    # prepare the list of values of collection
    if collection is None:
        logging.error('Mongo could not return the collecton - ' + collection_name)
        return None

    collection = collection.find(find, criteria)
    for doc in collection:
        yield doc 

并像这样调用:

def get_collection():
    criteria = {'unique_key': 0, '_id': 0}
    for document in Mongo.get_collection_iterator('contract', {}, criteria):
        print document 

我看到错误说:

File "/Users/Dev/Documents/work/dw/src/utilities/Mongo.py", line 96
    yield doc
SyntaxError: 'return' with argument inside generator 

我在这里做错了什么？

Answer 1

看来问题是 Python 不允许您混合使用 return 和 yield —— 您在 get_collection_iterator 中同时使用这两者。

澄清(感谢 rob mayoff):return x 和 yield 不能混用，但是一个简单的 return 可以