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