Как написать функцию для подключения к mongodb с помощью pymongo

Question

Я пытаюсь написать функцию для подключения к mongodb и возврата объекта коллекции. У меня есть следующие:

def getCollection(dbname,collection): 
    client = MongoClient() 
    data_base = client.dbname 
    collObject = data_base.collection 
    return collObject 

Когда я бегу:

collection = getCollection(client, "hkpr_restore", "agents") 
print collection 

я получаю:

Collection(Database(MongoClient('localhost', 27017), u'dbname'), u'collection') 

Что я делаю неправильно?

Answer 1

При использовании client.dbname атрибут имя_бд называется, то есть вы извлекаете базу данных с именем имя_бд.

То же самое касается data_base.collection.

Решение:

def getCollection(dbname, collection): 
    client = MongoClient() 
    data_base = getattr(client, dbname) 
    collObject = getattr(data_base, collection) 
    return collObject 

Альтернатива: Вы можете использовать словарь доступ стиля:

def getCollection(dbname, collection): 
    client = MongoClient() 
    data_base = client[dbname] 
    collObject = data_base[collection] 
    return collObject