Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-02-23 16:19:49 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

add (back) env param

Browse Source
This commit is contained in:
Barbara Miller 2024-09-13 14:09:06 -07:00
parent 9d7b7c4167
commit 8c20d1dad3
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
brozzler/cli.py
View File

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

2
brozzler/worker.py
View File

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