site_config/builder_name_schema.py

# Copyright (c) 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.


""" Utilities for dealing with builder names. This module obtains its attributes
dynamically from builder_name_schema.json. """


import json
import os


def _LoadSchema():
  """ Load the builder naming schema from the JSON file. """

  def _UnicodeToStr(obj):
    """ Convert all unicode strings in obj to Python strings. """
    if isinstance(obj, unicode):
      return str(obj)
    elif isinstance(obj, dict):
      return dict(map(_UnicodeToStr, obj.iteritems()))
    elif isinstance(obj, list):
      return list(map(_UnicodeToStr, obj))
    elif isinstance(obj, tuple):
      return tuple(map(_UnicodeToStr, obj))
    else:
      return obj

  builder_name_json_filename = os.path.join(
      os.path.dirname(__file__), 'builder_name_schema.json')
  builder_name_schema_json = json.load(open(builder_name_json_filename))
  for k, v in builder_name_schema_json.iteritems():
    globals()[str(k).upper()] = _UnicodeToStr(v)

  # Roles specify different types of builders.
  builder_role_template = 'BUILDER_ROLE_%s'

  #pylint: disable=E0602
  for role in BUILDER_NAME_SCHEMA.keys():
    globals()[builder_role_template % role.upper()] = role

_LoadSchema()


def MakeBuilderName(role, extra_config=None, is_trybot=False, **kwargs):
  #pylint: disable=E0602
  schema = BUILDER_NAME_SCHEMA.get(role)
  if not schema:
    raise ValueError('%s is not a recognized role.' % role)
  for k, v in kwargs.iteritems():
    if BUILDER_NAME_SEP in v:
      raise ValueError('%s not allowed in %s.' % (BUILDER_NAME_SEP, v))
    if not k in schema:
      raise ValueError('Schema does not contain "%s": %s' %(k, schema))
  if extra_config and BUILDER_NAME_SEP in extra_config:
    raise ValueError('%s not allowed in %s.' % (BUILDER_NAME_SEP,
                                                extra_config))
  name_parts = [role]
  name_parts.extend([kwargs[attribute] for attribute in schema])
  if extra_config:
    name_parts.append(extra_config)
  if is_trybot:
    name_parts.append(TRYBOT_NAME_SUFFIX)
  return BUILDER_NAME_SEP.join(name_parts)


def BuilderNameFromObject(obj, is_trybot=False):
  """Create a builder name based on properties of the given object.

  Args:
      obj: the object from which to create the builder name. The object must
          have as properties:
          - A valid builder role, as defined in the JSON file
          - All properties listed in the JSON file for that role
          - Optionally, an extra_config property
      is_trybot: bool; whether or not the builder is a trybot.
  Returns:
      string which combines the properties of the given object into a valid
          builder name.
  """
  #pylint: disable=E0602
  schema = BUILDER_NAME_SCHEMA.get(obj.role)
  if not schema:
    raise ValueError('%s is not a recognized role.' % role)
  name_parts = [obj.role]
  for attr_name in schema:
    attr_val = getattr(obj, attr_name)
    name_parts.append(attr_val)
  extra_config = getattr(obj, 'extra_config', None)
  if extra_config:
    name_parts.append(extra_config)
  if is_trybot:
    name_parts.append(TRYBOT_NAME_SUFFIX)
  return BUILDER_NAME_SEP.join(name_parts)


def IsTrybot(builder_name):
  """ Returns true if builder_name refers to a trybot (as opposed to a
  waterfall bot). """
  #pylint: disable=E0602
  return builder_name.endswith(TRYBOT_NAME_SUFFIX)


def GetWaterfallBot(builder_name):
  """Returns the name of the waterfall bot for this builder. If it is not a
  trybot, builder_name is returned unchanged. If it is a trybot the name is
  returned without the trybot suffix."""
  #pylint: disable=E0602
  if not IsTrybot(builder_name):
    return builder_name
  return _WithoutSuffix(builder_name, BUILDER_NAME_SEP +
                        TRYBOT_NAME_SUFFIX)


def _WithoutSuffix(string, suffix):
  """ Returns a copy of string 'string', but with suffix 'suffix' removed.
  Raises ValueError if string does not end with suffix. """
  if not string.endswith(suffix):
    raise ValueError('_WithoutSuffix: string %s does not end with suffix %s' % (
        string, suffix))
  return string[:-len(suffix)]


def DictForBuilderName(builder_name):
  """Makes a dictionary containing details about the builder from its name."""
  split_name = builder_name.split(BUILDER_NAME_SEP)
  result = {}
  if split_name[0] in BUILDER_NAME_SCHEMA.keys():
    key_list = BUILDER_NAME_SCHEMA[split_name[0]]
    result['role'] = split_name[0]
    for idx, key in enumerate(key_list):
      result[key] = split_name[idx+1]
    if len(split_name) > len(key_list):
      result['badParams'] = BUILDER_NAME_SEP.join(
          split_name[len(split_name):])
  else:
    raise ValueError("DictForBuilderName: Unable to determine buildbot type "
        "from builder name; no builder data emitted")
  return result