buildbucket: add retry API.

appengine/cr-buildbucket/service.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.
 
 import datetime
 import logging
 import urlparse
 
 from google.appengine.api import taskqueue
 from google.appengine.api import modules
@@ skipping 164 matching lines @@
   if client_operation_id is not None:
     yield ctx.memcache_set(client_operation_cache_key, build.key.id(), 60)
   raise ndb.Return(build)
 
 
 def add(*args, **kwargs):
   """Sync version of add_async."""
   return add_async(*args, **kwargs).get_result()
 
 
+def retry(
+    build_id, lease_expiration_date=None, client_operation_id=None,
+    pubsub_callback=None):
+  """Adds a build with same bucket, parameters and tags as the given one."""
+  build = model.Build.get_by_id(build_id)
+  if not build:
+    raise errors.BuildNotFoundError('Build %s not found' % build_id)
+  return add(

[Vadim Sh., 2016/04/11 23:30:10]

does it make sense to add some sort of a "backlink" to original build_id? E.g
"build #1234123213 is a retry of build #234234234"?

[nodir, 2016/04/11 23:57:36]

Done

+    build.bucket,
+    tags=build.tags,
+    parameters=build.parameters,
+    lease_expiration_date=lease_expiration_date,
+    client_operation_id=client_operation_id,
+    pubsub_callback=pubsub_callback,
+  )
+
+
 def get(build_id):
   """Gets a build by |build_id|.
 
   Requires the current user to have permissions to view the build.
   """
   build = model.Build.get_by_id(build_id)
   if not build:
     return None
   if not acl.can_view_build(build):
     raise current_identity_cannot('view build %s', build.key.id())
@@ skipping 582 matching lines @@
   if isinstance(identity, basestring):
     if not identity:  # pragma: no cover
       return None
     if ':' not in identity:  # pragma: no branch
       identity = 'user:%s' % identity
     try:
       identity = auth.Identity.from_bytes(identity)
     except ValueError as ex:
       raise errors.InvalidInputError('Invalid identity identity: %s' % ex)
   return identity