def setUp(self):
self.app = Flask(__name__)
self.app.testing = True
self.client = self.app.test_client()
if sys.version_info.major < 3:
self.assertRegex = self.assertRegexpMatches
self.assertNotRegex = self.assertNotRegexpMatches
registry = CollectorRegistry(auto_describe=True)
self.metrics = PrometheusMetrics.for_app_factory(registry=registry)
# DATABASE
def make_db(app):
# SQL
db = SQLAlchemy()
db.init_app(app)
# Mongo
mg = PyMongo()
mg.init_app(app)
# CORS
CORS(app)
return db, mg, app
# CELERY AND RABBITMQ
def make_celery(app):
celery = Celery(config_source=app.config)
class ContextTask(celery.Task):
def __call__(self, *args, **kwargs):
with app.app_context():
return self.run(*args, **kwargs)
celery.Task = ContextTask
return celery
# PROMETHEUS
metrics = PrometheusMetrics.for_app_factory()
aa = APIAlchemy()
registry.register('bigquery', __package__ + '.database.bigquery.dialect',
'BigQueryDialect')
registry.register('bigquery.pybigquery',
__package__ + '.database.bigquery.dialect',
'BigQueryDialect')
registry.register('db2', __package__ + '.database.db2.dialect', 'DB2Dialect')
registry.register('db2.ibm_db', __package__ + '.database.db2.dialect',
'DB2Dialect')
aps = APScheduler()
app_registry_name = 'application'
metrics = PrometheusMetrics.for_app_factory(
registry=get_registry(app_registry_name))
registered_collectors = {}
job_collector_metrics = {}
def create_app(config_name):
from .. import blueprint
app = Flask(__name__)
app.wsgi_app = ProxyFix(app.wsgi_app)
try:
config_obj = config_by_name[config_name]()
