示例#1
0
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()
示例#2
0
文件: celery.py 项目: Ratler/eduid-am
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')
示例#3
0
文件: tasks.py 项目: Ratler/eduid-am
    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
示例#4
0
文件: celery.py 项目: Ratler/eduid-am
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')