From 8c20d1dad3c1fe687dba5aceb1ba5f3e12aedda5 Mon Sep 17 00:00:00 2001
From: Barbara Miller <barbara@archive.org>
Date: Fri, 13 Sep 2024 14:09:06 -0700
Subject: [PATCH] add (back) env param

---
 brozzler/cli.py    | 14 ++++++++++++++
 brozzler/worker.py |  2 +-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/brozzler/cli.py b/brozzler/cli.py
index 0ae7993..1db25b0 100755
--- a/brozzler/cli.py
+++ b/brozzler/cli.py
@@ -246,6 +246,12 @@ def brozzle_page(argv=None):
         default=None,
         help="Prometheus scrape target registry URL",
     )
+    arg_parser.add_argument(
+        "--env",
+        dest="env",
+        default=None,
+        help="env for Prometheus target registry",
+    )
     arg_parser.add_argument(
         "--screenshot-full-page", dest="screenshot_full_page", action="store_true"
     )
@@ -293,6 +299,7 @@ def brozzle_page(argv=None):
         stealth=args.stealth,
         metrics_port=args.metrics_port,
         registry_url=args.registry_url,
+        env=args.env,
     )
 
     def on_screenshot(screenshot_jpeg):
@@ -543,6 +550,12 @@ def brozzler_worker(argv=None):
         default=None,
         help="Prometheus scrape target registry URL",
     )
+    arg_parser.add_argument(
+        "--env",
+        dest="env",
+        default=None,
+        help="env for Prometheus target registry",
+    )
     add_common_options(arg_parser, argv)
 
     args = arg_parser.parse_args(args=argv[1:])
@@ -601,6 +614,7 @@ def brozzler_worker(argv=None):
         stealth=args.stealth,
         metrics_port=args.metrics_port,
         registry_url=args.registry_url,
+        env=args.env,
     )
 
     signal.signal(signal.SIGQUIT, dump_state)
diff --git a/brozzler/worker.py b/brozzler/worker.py
index d470bd7..2a63dec 100644
--- a/brozzler/worker.py
+++ b/brozzler/worker.py
@@ -112,7 +112,7 @@ class BrozzlerWorker:
         self._shutdown = threading.Event()
 
         # Setup metrics
-        metrics.register_prom_metrics(self._metrics_port, self._registry_url)
+        metrics.register_prom_metrics(self._metrics_port, self._registry_url, self._env)
 
     def _choose_warcprox(self):
         warcproxes = self._service_registry.available_services("warcprox")