swarmbucket: add support for blamelist

appengine/cr-buildbucket/swarming/swarming.py

 # Copyright 2014 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.
 
 """This module integrates buildbucket with swarming.
 
 A bucket config may have "swarming" field that specifies how a builder
 is mapped to a recipe. If build is scheduled for a bucket/builder
 with swarming configuration, the integration overrides the default behavior.
 
@@ skipping 33 matching lines @@
 
 import config
 import errors
 import model
 import notifications
 
 PUBSUB_TOPIC = 'swarming'
 BUILDER_PARAMETER = 'builder_name'
 PARAM_PROPERTIES = 'properties'
 PARAM_SWARMING = 'swarming'
+PARAM_CHANGES = 'changes'
 DEFAULT_URL_FORMAT = 'https://{swarming_hostname}/user/task/{task_id}'
 
 
 ################################################################################
 # Creation/cancellation of tasks.
 
 
 @ndb.tasklet
 def get_task_template_async():
   """Returns a task template (dict) if it exists, otherwise None.
@@ skipping 61 matching lines @@
           'Unrecognized keys in swarming.recipe: %r' % recipe)
 
     if swarming:
       raise errors.InvalidInputError('Unrecognized keys: %r', swarming)
 
 
 @ndb.tasklet
 def create_task_def_async(swarming_cfg, builder_cfg, build):
   """Creates a swarming task definition for the |build|.
 
+  Supports build properties that are supported by Buildbot-Buildbucket
+  integration. See
+  https://chromium.googlesource.com/chromium/tools/build/+/eff4ceb/scripts/master/buildbucket/README.md#Build-parameters
+
   Raises:
     errors.InvalidInputError if build.parameters are invalid.
   """
   params = build.parameters or {}
   validate_build_parameters(builder_cfg.name, params)
   swarming_param = params.get(PARAM_SWARMING) or {}
 
   # Render task template.
   task_template = yield get_task_template_async()
   task_template_params = {
     'bucket': build.bucket,
     'builder': builder_cfg.name,
   }
 
   is_recipe = builder_cfg.HasField('recipe')
   if is_recipe:  # pragma: no branch
     recipe = builder_cfg.recipe
     revision = swarming_param.get('recipe', {}).get('revision') or ''
 
     build_properties = dict(
       p.split(':', 1) for p in builder_cfg.recipe.properties or [])
     build_properties.update(build.parameters.get(PARAM_PROPERTIES) or {})
     build_properties['buildername'] = builder_cfg.name
 
+    # Convert changes in build to blamelist property, like Buildbot-Buildbucket
+    # integration. In Buildbot the property value is a list of emails.
+    changes = params.get(PARAM_CHANGES)
+    if changes:  # pragma: no branch
+      emails = [c.get('author', {}).get('email') for c in changes]
+      build_properties['blamelist'] = filter(None, emails)
+
     task_template_params.update({
       'repository': recipe.repository,
       'revision': revision,
       'recipe': recipe.name,
       'properties_json': json.dumps(build_properties, sort_keys=True),
     })
 
   task_template_params = {
     k: v or '' for k, v in task_template_params.iteritems()}
   task = format_obj(task_template, task_template_params)
@@ skipping 416 matching lines @@
 def _extend_unique(target, items):
   for x in items:
     if x not in target:  # pragma: no branch
       target.append(x)
 
 
 class TaskToken(tokens.TokenKind):
   expiration_sec = 60 * 60 * 24  # 24 hours.
   secret_key = auth.SecretKey('swarming_task_token', scope='local')
   version = 1