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 78 matching lines @@
   cancelation_reason = messages.EnumField(model.CancelationReason, 9)
   lease_expiration_ts = messages.IntegerField(10)
   lease_key = messages.IntegerField(11)
   url = messages.StringField(12)
   created_ts = messages.IntegerField(13)
   updated_ts = messages.IntegerField(14)
   completed_ts = messages.IntegerField(15)
   created_by = messages.StringField(16)
   status_changed_ts = messages.IntegerField(17)
   utcnow_ts = messages.IntegerField(18, required=True)
+  retry_of = messages.IntegerField(19)
 
 
 class BuildResponseMessage(messages.Message):
   build = messages.MessageField(BuildMessage, 1)
   error = messages.MessageField(ErrorMessage, 2)
 
 
 def build_to_message(build, include_lease_key=False):
   """Converts model.Build to BuildMessage."""
   assert build
@@ skipping 11 matching lines @@
     cancelation_reason=build.cancelation_reason,
     failure_reason=build.failure_reason,
     lease_key=build.lease_key if include_lease_key else None,
     url=build.url,
     created_ts=datetime_to_timestamp_safe(build.create_time),
     updated_ts=datetime_to_timestamp_safe(build.update_time),
     completed_ts=datetime_to_timestamp_safe(build.complete_time),
     created_by=build.created_by.to_bytes() if build.created_by else None,
     status_changed_ts=datetime_to_timestamp_safe(build.status_changed_time),
     utcnow_ts=datetime_to_timestamp_safe(utils.utcnow()),
+    retry_of=build.retry_of,
   )
   if build.lease_expiration_date is not None:
     msg.lease_expiration_ts = utils.datetime_to_timestamp(
       build.lease_expiration_date)
   return msg
 
 
 def build_to_response_message(build, include_lease_key=False):
   return BuildResponseMessage(build=build_to_message(build, include_lease_key))
 
@@ skipping 140 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),
     result=messages.EnumField(model.BuildResult, 5),
     cancelation_reason=messages.EnumField(model.CancelationReason, 6),
     failure_reason=messages.EnumField(model.FailureReason, 7),
     created_by=messages.StringField(8),
     max_builds=messages.IntegerField(9, variant=messages.Variant.INT32),
+    retry_of=messages.IntegerField(10),
   )
 
   class SearchResponseMessage(messages.Message):
     builds = messages.MessageField(BuildMessage, 1, repeated=True)
     next_cursor = messages.StringField(2)
     error = messages.MessageField(ErrorMessage, 3)
 
   @buildbucket_api_method(
     SEARCH_REQUEST_RESOURCE_CONTAINER, SearchResponseMessage,
     path='search', http_method='GET')
   @auth.public
   def search(self, request):
     """Searches for builds."""
     assert isinstance(request.tag, list)
     builds, next_cursor = service.search(
       buckets=request.bucket,
       tags=request.tag,
       status=request.status,
       result=request.result,
       failure_reason=request.failure_reason,
       cancelation_reason=request.cancelation_reason,
       max_builds=request.max_builds,
       created_by=request.created_by,
-      start_cursor=request.start_cursor)
+      start_cursor=request.start_cursor,
+      retry_of=request.retry_of,
+    )
     return self.SearchResponseMessage(
       builds=map(build_to_message, builds),
       next_cursor=next_cursor,
     )
 
   ###################################  PEEK  ###################################
 
   PEEK_REQUEST_RESOURCE_CONTAINER = endpoints.ResourceContainer(
     message_types.VoidMessage,
     bucket=messages.StringField(1, repeated=True),
@@ skipping 240 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()