#309 add check for devel environment

idealista · Sep 29, 2022 · 0c9ae64 · 0c9ae64
1 parent 3b24412
commit 0c9ae64
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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')