diff --git a/app/__init__.py b/app/__init__.py index ba97cb5..82f29ee 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -3,6 +3,7 @@ from flask.typing import ResponseReturnValue from werkzeug.middleware.dispatcher import DispatcherMiddleware from prometheus_client import make_wsgi_app, REGISTRY import yaml +import sys from app.extensions import db from app.extensions import migrate @@ -26,11 +27,13 @@ bootstrap.init_app(app) app.register_blueprint(portal, url_prefix="/portal") app.register_blueprint(tfstate, url_prefix="/tfstate") -from app.metrics import DefinedProxiesCollector, BlockedProxiesCollector, AutomationCollector -REGISTRY.register(DefinedProxiesCollector()) -REGISTRY.register(BlockedProxiesCollector()) -REGISTRY.register(AutomationCollector()) +if sys.argv[1] != "db": + from app.metrics import DefinedProxiesCollector, BlockedProxiesCollector, AutomationCollector + + REGISTRY.register(DefinedProxiesCollector()) + REGISTRY.register(BlockedProxiesCollector()) + REGISTRY.register(AutomationCollector()) @app.route('/')