pymongo 生成器失败 - “返回”生成器内有参数

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 可以

Answer 2

您的问题在于必须返回 None 事件，但它被检测为语法错误，因为返回会破坏迭代循环。

打算使用 yield 在循环中传递值的生成器不能将 return 与参数值一起使用，因为这会触发 StopIteration 错误。您可能希望引发异常并在调用上下文中捕获它，而不是返回 None。

http://www.answermysearches.com/ python-fixing-syntaxerror-return-with-argument-inside-generator/354/

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

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

如果需要的话，您也可以为此制定一个特殊的例外。