From 430ef0f86aba327822b0785bac2898d2535a6aa4 Mon Sep 17 00:00:00 2001 From: Iain Learmonth Date: Wed, 2 Nov 2022 14:24:18 +0000 Subject: [PATCH] metrics: don't fail if no argument is given to command --- app/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/app/__init__.py b/app/__init__.py index df77d00..9cc211d 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -27,7 +27,11 @@ app.register_blueprint(portal, url_prefix="/portal") app.register_blueprint(tfstate, url_prefix="/tfstate") -if 'nose' not in sys.modules.keys() and sys.argv[1] != "db": +def not_migrating() -> bool: + return len(sys.argv) < 2 or sys.argv[1] != "db" + + +if 'nose' not in sys.modules.keys() and not_migrating(): from app.metrics import DefinedProxiesCollector, BlockedProxiesCollector, AutomationCollector REGISTRY.register(DefinedProxiesCollector(app))