From 3c5abee77cfc74341cd9cf6094ff45a5b4892ede Mon Sep 17 00:00:00 2001 From: Iain Learmonth Date: Sat, 8 Oct 2022 19:20:15 +0100 Subject: [PATCH] metrics: do not initialise metrics when running db migrations --- app/__init__.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) 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('/')