def get_db(settings): mongo_replicaset = settings.get('mongo_replicaset', None) if mongo_replicaset is not None: mongodb = MongoDB(db_uri=settings['mongo_uri'], replicaSet=mongo_replicaset) else: mongodb = MongoDB(db_uri=settings['mongo_uri']) return mongodb.get_database()
def setup_celeryd(sender, conf, **kwargs): settings = read_configuration() conf.update(settings) db_conn = MongoDB(settings.get('MONGO_URI', DEFAULT_MONGODB_URI)) db = db_conn.get_database() db.attributes.ensure_index('mail', name='mail-index', unique=True) db.attributes.ensure_index('eduPersonPrincipalName', name='eppn-index', unique=True) db.attributes.ensure_index([('norEduPersonNIN.norEduPersonNIN', 1), ('norEduPersonNIN.verified', 1), ('norEduPersonNIN.status', 1)], name='norEduPersonNIN-index') db.attributes.ensure_index([('mobile.mobile', 1), ('mobile.verified', 1)], name='mobile-index') db.attributes.ensure_index([('mailAliases.email', 1), ('mailAliases.verified', 1)], name='mailAliases-index')
def conn(self): """ Get the MongoDB connection object. :return: connection object """ if self._conn is None: self._conn = MongoDB( self.app.conf.get('MONGO_URI', DEFAULT_MONGODB_URI)) return self._conn