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
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
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
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))
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
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')