commit dea56dc50dcbf81ce73e94b0837e74c9d5feb34c
author Anuj Jamwal <anujjamwal@google.com> Tue Jun 06 19:35:30 2023 +0000
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Sat Jun 10 04:59:42 2023 +0000
tree 18f0e61269a3eecb91f31a1fdcdcb951a7e6034b
parent 94fa64405ec9e8cfab5078b469fd855ce0abb508

utils: update telemetry.initialize to accept config_file and opts

Update the initialize method to accept config file and opts as args.
This decouple the telemetry lib from chromite_config and allows the
initialization to take commandline arguments into account like enabling
console exports for --debug.

BUG=b:285374846
TEST=Run Unit Tests
TEST=./scripts/telemetry_poc
TEST=./bin/build_packages

Change-Id: I483160e81307318227fd8668bfc69676d2fd5e5e
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/4595281
Reviewed-by: Alex Klein <saklein@chromium.org>
Auto-Submit: Anuj Jamwal <anujjamwal@google.com>
Commit-Queue: Anuj Jamwal <anujjamwal@google.com>
Tested-by: Anuj Jamwal <anujjamwal@google.com>

scripts/telemetry_poc.py

diff --git a/scripts/telemetry_poc.py b/scripts/telemetry_poc.py
index f9c52a1..60c027c 100644
--- a/scripts/telemetry_poc.py
+++ b/scripts/telemetry_poc.py
@@ -47,10 +47,7 @@
     opts = parse_arguments(argv)
 
     chromite_config.initialize()
-    telemetry.initialize(chromite_config.TELEMETRY_CONFIG)
-
-    if opts.debug:
-        telemetry.export_to_console(chromite_config.TELEMETRY_CONFIG)
+    telemetry.initialize(chromite_config.TELEMETRY_CONFIG, opts.debug)
 
     with tracer.start_as_current_span("test") as span:
         time.sleep(opts.time / 2)