Beispiel #1
0
def get_models(**config):
    
    settings, storage = utils.get_db_config(config.get('db_settings', {}))
    
    if storage == "mongo":
        from mongrey.storage.mongo.utils import create_mongo_connection
        from mongrey.storage.mongo import models
        create_mongo_connection(**settings)
        _models = {
            'domain': models.Domain,
            'mynetwork': models.Mynetwork,
            'mailbox': models.Mailbox,
            'whitelist': models.WhiteList,
            'blacklist': models.BlackList,
        }        
        return _models

    elif storage == "sql":
        from mongrey.storage.sql.models import configure_peewee
        from mongrey.storage.sql import models
        configure_peewee(**settings)
        _models = {
            'domain': models.Domain,
            'mynetwork': models.Mynetwork,
            'mailbox': models.Mailbox,
            'whitelist': models.WhiteList,
            'blacklist': models.BlackList,
        }        
        return _models
Beispiel #2
0
 def _create_app(self):
     settings, storage = get_db_config(**self.db_settings)        
     models.configure_peewee(drop_before=True, **settings)
     
     from mongrey.web import create_app
     app = create_app(config=self.CONFIG)
     return app
Beispiel #3
0
def get_models(**config):

    settings, storage = utils.get_db_config(**config.get('db_settings', {}))

    if storage == "mongo":
        from mongrey.storage.mongo.utils import create_mongo_connection
        from mongrey.storage.mongo import models
        create_mongo_connection(settings)
        _models = {
            'domain': models.Domain,
            'mynetwork': models.Mynetwork,
            'mailbox': models.Mailbox,
            'whitelist': models.WhiteList,
            'blacklist': models.BlackList,
        }
        return _models

    elif storage == "sql":
        from mongrey.storage.sql.models import configure_peewee
        from mongrey.storage.sql import models
        configure_peewee(**settings)
        _models = {
            'domain': models.Domain,
            'mynetwork': models.Mynetwork,
            'mailbox': models.Mailbox,
            'whitelist': models.WhiteList,
            'blacklist': models.BlackList,
        }
        return _models
Beispiel #4
0
    def _create_app(self):
        settings, storage = get_db_config(**self.db_settings)
        models.configure_peewee(drop_before=True, **settings)

        from mongrey.web import create_app
        app = create_app(config=self.CONFIG)
        return app
Beispiel #5
0
def _configure_storage_sql(app):
    from mongrey.storage.sql import models
    from mongrey.storage.sql.admin import init_admin
    settings, storage = utils.get_db_config(**app.config.get('DB_SETTINGS'))
    models.configure_peewee(**settings)
    init_admin(app=app, url='/')

    def load_user(userid):
        try:
            return models.User.get(id=userid)
        except Exception, err:
            app.logger.warning(str(err))            
Beispiel #6
0
def _configure_storage_sql(app):
    from mongrey.storage.sql import models
    from mongrey.storage.sql.admin import init_admin
    settings, storage = utils.get_db_config(**app.config.get('DB_SETTINGS'))
    models.configure_peewee(**settings)
    init_admin(app=app, url='/')

    def load_user(userid):
        try:
            return models.User.get(id=userid)
        except Exception, err:
            app.logger.warning(str(err))
Beispiel #7
0
def get_store(**config):

    settings, storage = utils.get_db_config(**config.get('db_settings', {}))    
    
    if storage == "mongo":
        from mongrey.storage.mongo.utils import create_mongo_connection
        from mongrey.storage.mongo.policy import MongoPolicy
        from mongrey.storage.mongo import models
        create_mongo_connection(settings)
        policy_klass = MongoPolicy
        return policy_klass, models

    elif storage == "sql":
        from mongrey.storage.sql.models import configure_peewee
        from mongrey.storage.sql.policy import SqlPolicy
        from mongrey.storage.sql import models
        configure_peewee(**settings)
        policy_klass = SqlPolicy
        return policy_klass, models
    
    return None, None
Beispiel #8
0
def get_store(**config):

    settings, storage = utils.get_db_config(**config.get('db_settings', {}))

    if storage == "mongo":
        from mongrey.storage.mongo.utils import create_mongo_connection
        from mongrey.storage.mongo.policy import MongoPolicy
        from mongrey.storage.mongo import models
        create_mongo_connection(settings)
        policy_klass = MongoPolicy
        return policy_klass, models

    elif storage == "sql":
        from mongrey.storage.sql.models import configure_peewee
        from mongrey.storage.sql.policy import SqlPolicy
        from mongrey.storage.sql import models
        configure_peewee(**settings)
        policy_klass = SqlPolicy
        return policy_klass, models

    return None, None
Beispiel #9
0
 def setUp(self):
     BaseTestCase.setUp(self)
     settings, storage = get_db_config(**self.db_settings)
     models.configure_peewee(drop_before=True, **settings)
     self._cache = cache.configure_cache(cache_url='simple')
Beispiel #10
0
 def setUp(self):
     BaseTestCase.setUp(self)
     settings, storage = get_db_config(**self.db_settings)        
     models.configure_peewee(drop_before=True, **settings)
     self._cache = cache.configure_cache(cache_url='simple')