diff --git a/bin/prom2teams b/bin/prom2teams index f11f991..493451f 100644 --- a/bin/prom2teams +++ b/bin/prom2teams @@ -10,6 +10,7 @@ except ImportError: from prom2teams.app.api import app as application if __name__ == "__main__": + application.config['ENV'] = "werkzeug" host = application.config['HOST'] port = int(application.config['PORT']) run_simple(hostname=host, port=port, application=application, use_reloader=True, diff --git a/prom2teams/app/configuration.py b/prom2teams/app/configuration.py index 58711a5..eda7b36 100644 --- a/prom2teams/app/configuration.py +++ b/prom2teams/app/configuration.py @@ -130,9 +130,13 @@ def config_app(application): application.config['GROUP_ALERTS_BY'] = command_line_args.groupalertsby if command_line_args.enablemetrics or os.environ.get('PROM2TEAMS_PROMETHEUS_METRICS', False): os.environ["DEBUG_METRICS"] = "True" - from prometheus_flask_exporter.multiprocess import UWsgiPrometheusMetrics - metrics = UWsgiPrometheusMetrics(application) - metrics.start_http_server(int(os.getenv('PROMETHEUS_MULTIPROC_PORT'))) + if application.config['ENV'] == "werkzeug": + from prometheus_flask_exporter import PrometheusMetrics + metrics = PrometheusMetrics(application) + else: + from prometheus_flask_exporter.multiprocess import UWsgiPrometheusMetrics + metrics = UWsgiPrometheusMetrics(application) + metrics.start_http_server(int(os.getenv('PROMETHEUS_MULTIPROC_PORT'))) if 'MICROSOFT_TEAMS' not in application.config: raise MissingConnectorConfigKeyException('missing connector key in config')