buildbucket: add retry API.

appengine/cr-buildbucket/api.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 functools
 import json
 import logging
 
 from google.appengine.ext import ndb
@@ skipping 269 matching lines @@
         one_res.build = build_to_message(build, include_lease_key=True)
       except errors.Error as ex:
         one_res.error = exception_to_error_message(ex)
       return one_res
 
     res.results = [
       to_msg(req, build)
       for req, build in zip(request.builds, build_futures)]
     return res
 
+  ##################################  RETRY   ##################################
+
+  class RetryRequestMessage(messages.Message):
+    client_operation_id = messages.StringField(1)
+    lease_expiration_ts = messages.IntegerField(2)
+    pubsub_callback = messages.MessageField(PubSubCallbackMessage, 3)
+
+  @buildbucket_api_method(
+    id_resource_container(RetryRequestMessage),
+    BuildResponseMessage,
+    path='builds/{id}/retry', http_method='PUT')
+  @auth.public
+  def retry(self, request):
+    """Retries an existing build."""
+    build = service.retry(
+      request.id,
+      lease_expiration_date=parse_datetime(request.lease_expiration_ts),
+      client_operation_id=request.client_operation_id,
+      pubsub_callback=pubsub_callback_from_message(request.pubsub_callback),
+    )
+    return build_to_response_message(build, include_lease_key=True)
+
+
   ##################################  SEARCH   #################################
 
 
   SEARCH_REQUEST_RESOURCE_CONTAINER = endpoints.ResourceContainer(
     message_types.VoidMessage,
     start_cursor=messages.StringField(1),
     bucket=messages.StringField(2, repeated=True),
     # All specified tags must be present in a build.
     tag=messages.StringField(3, repeated=True),
     status=messages.EnumField(model.BuildStatus, 4),
@@ skipping 279 matching lines @@
     ),
     DeleteManyBuildsResponse,
     path='bucket/{bucket}/delete', http_method='POST')
   @auth.public
   def delete_many_builds(self, request):
     """Deletes scheduled or started builds in a bucket."""
     service.delete_many_builds(
       request.bucket, request.status,
       tags=request.tag[:], created_by=request.created_by)
     return self.DeleteManyBuildsResponse()