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]()