Merge pull request #3 from agershman/collector-addr-env-var

Allow set of prometheus collector addr as env var
Gandi · Dec 20, 2023 · 4d57701 · 4d57701
2 parents def3163 + 6f7014d
commit 4d57701
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/CONTRIBUTORS.txt b/CONTRIBUTORS.txt
@@ -1,2 +1,3 @@
 Guillaume Gauvrit <[email protected]>
-Michael Vieira <[email protected]>
+Michael Vieira <[email protected]>
+Andrew Gershman <[email protected]>
diff --git a/src/celery_prometheus/prometheus_bootstep.py b/src/celery_prometheus/prometheus_bootstep.py
@@ -1,5 +1,6 @@
 """Helper for celery."""
 import logging
+import os
 from argparse import ArgumentParser
 from typing import Any, Mapping, Optional, cast
 
@@ -26,7 +27,11 @@ def add_prometheus_option(app: Celery) -> None:
     if celery_version.major < 5:
 
         def add_preload_arguments(parser: ArgumentParser) -> None:
-            parser.add_argument("--prometheus-collector-addr", default=None, help=help)
+            parser.add_argument(
+                "--prometheus-collector-addr",
+                default=os.getenv("CELERY_PROMETHEUS_COLLECTOR_ADDR"),
+                help=help,
+            )
 
         app.user_options["preload"].add(add_preload_arguments)
     else: