Multiple Databases / Connections using Mogo With Statement

mogo_with_statement_example.py

import mogo

DBNAME1 = "__testing__"
DBNAME2 = "__testing2__"

# Basic model
class Person(mogo.Model):
    name = mogo.Field(unicode)

# Automatically opens and closes DB connection (or should)
with mogo.session(DBNAME1) as session1:
    person1 = Person.use(session1)(name=u"Josh")
    person1.save(safe=True)
    # Using another db connection to another database
    with mogo.session(DBNAME2) as session2:
        person2 = Person.use(session2)(name=u"Troy")
        person2.save(safe=True)
        # checking that we only have one person in this db connection
        assert(Person.use(session2).find().count() == 1)
        assert(Person.use(session2).find().first() == person2)
    # checking that we only have one person in the outer db connection
    assert(Person.use(session1).find().count() == 1)
    assert(Person.use(session1).find().first() == person1)

# Just clearing the database
connection = mogo.connect(DBNAME1)
connection.drop_database(DBNAME1)
connection.drop_database(DBNAME2)
connection.disconnect()