telemetry_utils.py

# Copyright 2026 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys

from opentelemetry import trace

# Appends third_party/depot_tools/infra_lib so that we can import telemetry.
sys.path.append(
    os.path.join(os.path.dirname(__file__), os.path.pardir, os.path.pardir,
                 'third_party', 'depot_tools', 'infra_lib'))

import telemetry

tracer = telemetry.get_tracer(__name__)


def Initialize():
  """Initializes telemetry."""
  telemetry.initialize('chromium.tools.code_coverage')


def RecordMainAttributes(targets, out_dir):
  """Records main attributes to the current span."""
  span = trace.get_current_span()
  if not span.is_recording():
    return

  if targets:
    span.set_attribute('main.targets', str(targets))
  if out_dir:
    span.set_attribute('build.out_dir', out_dir)