testing/generate_location_tags.py

#!/usr/bin/env python
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Generates the directory->tags mapping used by ResultDB."""

# pylint: disable=line-too-long
#
# For more on the tags, see
# https://source.chromium.org/chromium/infra/infra/+/main:go/src/go.chromium.org/luci/resultdb/sink/proto/v1/location_tag.proto
#
# pylint: enable=line-too-long

import argparse
import logging
import os
import subprocess
import sys

THIS_DIR = os.path.dirname(__file__)
SRC_DIR = os.path.abspath(os.path.dirname(THIS_DIR))

# //build imports.
BUILD_DIR = os.path.join(SRC_DIR, 'build')
sys.path.insert(0, BUILD_DIR)
import find_depot_tools


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('-o',
                      '--out',
                      required=True,
                      help='path to write location tag metadata to')
  args = parser.parse_args()

  logging.basicConfig(level=logging.WARNING)

  # ".git" is a directory in full checkouts, but a file in work trees.
  if not os.path.exists(os.path.join(SRC_DIR, '.git')):
    logging.warning('Skip generating location tags because the script is not '
                    'running in a git repository')
    return 0

  exe = os.path.join(find_depot_tools.DEPOT_TOOLS_PATH, 'dirmd')
  if sys.platform == 'win32':
    exe = exe + '.bat'

  return subprocess.call([
      exe,
      'location-tags',
      '-out',
      args.out,
      '-root',
      SRC_DIR,
      '-repo',
      'https://chromium.googlesource.com/chromium/src',
  ])


if __name__ == '__main__':
  sys.exit(main())