mongomock getCollection example

gistfile1.txt

globalconnection = None


def get_mongodb_connect_url():
    if IS_PRODUCTION:
        return ('mongodb://fuck:yourmother'
                'mongodb-1.fuck.com:27017,'
                'mongodb-2.fuck.com:27017,'
                'mongodb-3.fuck.com:27017,'
                'mongodb-4.fuck.com:27017')
    else:
        return ('mongodb://love:yourmother@'
                'db-love.fuck.com:27017')


def getConnection():
    global globalconnection

    while globalconnection is None:
        try:
            if not IS_TESTING:
                globalconnection = MongoClient(get_mongodb_connect_url(),
                                               w=1,
                                               max_pool_size=500)
            else:
                import mongomock
                globalconnection = mongomock.MongoClient()

        except Exception:
            globalconnection = None

    return globalconnection


def closeConnection():
    global globalconnection
    if globalconnection:
        try:
            globalconnection.close()
        except:
            pass

        globalconnection = None


def getCollection(dbname, collname):
    conn = getConnection()
    return conn[dbname][collname]