Сбой генератора pymongo - возврат с аргументом внутри генератора - PullRequest
Новая
       17

Сбой генератора pymongo - возврат с аргументом внутри генератора

5 голосов
/ 29 ноября 2011

Я пытаюсь сделать следующее: 

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

что я делаю здесь неправильно?

Ответы [ 2 ]

12 голосов
/ 29 ноября 2011

Кажется, проблема в том, что Python не позволяет вам смешивать return и yield - вы используете оба в get_collection_iterator.

Разъяснение (благодаря Робу Мейоффу): return x и yield нельзя смешивать, но голая return может

3 голосов
/ 29 ноября 2011

Ваша проблема в том, что 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

Вы также можете сделать специальное исключение для этого, если это будет необходимо.

...