Support multiple try masters when sending tries to rietveld.

rietveld.py

 # coding: utf-8
 # Copyright (c) 2012 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.
 """Defines class Rietveld to easily access a rietveld instance.
 
 Security implications:
 
 The following hypothesis are made:
 - Rietveld enforces:
@@ skipping 313 matching lines @@
         # It's an error message. Return as no result.
         break
       data = json.loads(output) or {}
       if not data.get('results'):
         break
       for i in data['results']:
         yield i
       cursor = '&cursor=%s' % data['cursor']
 
   def trigger_try_jobs(
-      self, issue, patchset, reason, clobber, revision, builders_and_tests):
+      self, issue, patchset, reason, clobber, revision, builders_and_tests,
+      master=None):
     """Requests new try jobs.
 
     |builders_and_tests| is a map of builders: [tests] to run.
+    |master| is the name of the try master the builders belong to.
 
     Returns the keys of the new TryJobResult entites.
     """
     params = [
       ('reason', reason),
       ('clobber', 'True' if clobber else 'False'),
       ('builders', json.dumps(builders_and_tests)),
       ('xsrf_token', self.xsrf_token()),
     ]
     if revision:
       params.append(('revision', revision))
+    if master:
+      # Temporarily allow empty master names for old configurations. The try
+      # job will not be associated with a master name on rietveld. This is
+      # going to be deprecated.
+      params.append(('master', master))
     return self.post('/%d/try/%d' % (issue, patchset), params)
 
+  def trigger_distributed_try_jobs(
+      self, issue, patchset, reason, clobber, revision, masters):
+    """Requests new try jobs.
+
+    |masters| is a map of masters: map of builders: [tests] to run.
+    """
+    for (master, builders_and_tests) in masters.iteritems():
+      self.trigger_try_jobs(
+          issue, patchset, reason, clobber, revision, builders_and_tests,
+          master)
+
   def get_pending_try_jobs(self, cursor=None, limit=100):
     """Retrieves the try job requests in pending state.
 
     Returns a tuple of the list of try jobs and the cursor for the next request.
     """
     url = '/get_pending_try_patchsets?limit=%d' % limit
     extra = ('&cursor=' + cursor) if cursor else ''
     data = json.loads(self.get(url + extra))
     return data['jobs'], data['cursor']
 
@@ skipping 171 matching lines @@
                   add_as_reviewer=False):
     logging.info('ReadOnlyRietveld: posting comment "%s" to issue %d' %
         (message, issue))
 
   def set_flag(self, issue, patchset, flag, value):  # pylint:disable=R0201
     logging.info('ReadOnlyRietveld: setting flag "%s" to "%s" for issue %d' %
         (flag, value, issue))
     ReadOnlyRietveld._local_changes.setdefault(issue, {})[flag] = value
 
   def trigger_try_jobs(  # pylint:disable=R0201
-      self, issue, patchset, reason, clobber, revision, builders_and_tests):
+      self, issue, patchset, reason, clobber, revision, builders_and_tests,
+      master=None):
     logging.info('ReadOnlyRietveld: triggering try jobs %r for issue %d' %
         (builders_and_tests, issue))
+
+  def trigger_distributed_try_jobs(  # pylint:disable=R0201
+      self, issue, patchset, reason, clobber, revision, masters):
+    logging.info('ReadOnlyRietveld: triggering try jobs %r for issue %d' %
+        (masters, issue))