buildbot/trybot_patch_pool.py

# Copyright (c) 2011-2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Module that contains trybot patch pool code."""


class TrybotPatchPool(object):
  """Represents patches specified by the user to test."""
  def __init__(self, gerrit_patches=(), local_patches=(), remote_patches=()):
    self.gerrit_patches = tuple(gerrit_patches)
    self.local_patches = tuple(local_patches)
    self.remote_patches = tuple(remote_patches)

  def __nonzero__(self):
    """Returns True if the pool has no patches."""
    return any([self.gerrit_patches, self.local_patches, self.remote_patches])

  def Filter(self, **kwargs):
    """Returns a new pool with only patches that match constraints.

    **kwargs: constraints in the form of attr=value.  I.e.,
              project='chromiumos/chromite', tracking_branch='master'.
    """
    filtered = []
    for patch_list in [self.gerrit_patches, self.local_patches,
                       self.remote_patches]:
      sub_filtered = []
      for patch in patch_list:
        for key in kwargs:
          if getattr(patch, key, object()) != kwargs[key]:
            break
        else:
          sub_filtered.append(patch)

      filtered.append(sub_filtered)

    return TrybotPatchPool(*filtered)

  def __iter__(self):
    for source in [self.local_patches, self.remote_patches,
                   self.gerrit_patches]:
      for patch in source:
        yield patch