Add a flag to pick up timestamp of commits and print information if found.

bin/cbuildbot.py

 #!/usr/bin/python
 
 # Copyright (c) 2010 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.
 
 import errno
 import optparse
 import os
 import shutil
 import subprocess
 import sys
 
 from cbuildbot_config import config
 
 _DEFAULT_RETRIES=3
 
 # Utility functions
 
-def RunCommand(cmd, error_ok=False, error_message=None, exit_code=False,
-               redirect_stdout=False, redirect_stderr=False, cwd=None,
-               input=None):
+def RunCommand(cmd, print_cmd=True, error_ok=False, error_message=None,
+               exit_code=False, redirect_stdout=False, redirect_stderr=False,
+               cwd=None, input=None):
   # Print out the command before running.
-  print >>sys.stderr, "CBUILDBOT -- RunCommand:", ' '.join(cmd)
+  if print_cmd:
+    print >> sys.stderr, "CBUILDBOT -- RunCommand:", ' '.join(cmd)
   if redirect_stdout:
     stdout = subprocess.PIPE
   else:
     stdout = None
   if redirect_stderr:
     stderr = subprocess.PIPE
   else:
     stderr = None
   if input:
     stdin = subprocess.PIPE
   else:
       stdin = None
   proc = subprocess.Popen(cmd, cwd=cwd, stdin=stdin,
                           stdout=stdout, stderr=stderr)
   (output, error) = proc.communicate(input)
   if exit_code:
     return proc.returncode
   if not error_ok and proc.returncode != 0:
     raise Exception('Command "%s" failed.\n' % (' '.join(cmd)) +
                     (error_message or error or output or ''))
   return output
 
 def MakeDir(path, parents=False):
   try:
     os.makedirs(path)
-  except OSError,e:
+  except OSError, e:
     if e.errno == errno.EEXIST and parents:
       pass
     else:
       raise
 
 def RepoSync(buildroot, retries=_DEFAULT_RETRIES):
   while retries > 0:
     try:
       RunCommand(['repo', 'sync'], cwd=buildroot)
       retries = 0
@@ skipping 32 matching lines @@
 
 def _SetupBoard(buildroot, board='x86-generic'):
   cwd = os.path.join(buildroot, 'src', 'scripts')
   RunCommand(['./setup_board', '--fast', '--default', '--board=%s' % board],
              cwd=cwd)
 
 def _Build(buildroot):
   cwd = os.path.join(buildroot, 'src', 'scripts')
   RunCommand(['./build_packages'], cwd=cwd)
 
-def _UprevPackages(buildroot):
+def _UprevAllPackages(buildroot):
   cwd = os.path.join(buildroot, 'src', 'scripts')
   RunCommand(['./enter_chroot.sh', '--', './cros_mark_all_as_stable',
               '--tracking_branch="cros/master"'],
              cwd=cwd)
 
+def _UprevPackages(buildroot, revisionfile):
+  revisions = None
+  if (revision_file):
+    rev_file = revisionfile.open(revisionfile)
+    revisions = rev_file.read()
+    rev_file.close()
+
+  # Note:  Revisions == "None" indicates a Force Build.
+  if revisions and revisions != 'None':
+    print 'CBUILDBOT - Revision list found %s' % revisions
+    print 'Revision list not yet propagating to build, marking all instead'
+
+  _UprevAllPackages(buildroot)
+
 def _UprevCleanup(buildroot):
   cwd = os.path.join(buildroot, 'src', 'scripts')
   RunCommand(['./cros_mark_as_stable', '--srcroot=..',
               '--tracking_branch="cros/master"', 'clean'],
              cwd=cwd)
 
 def _UprevPush(buildroot):
   cwd = os.path.join(buildroot, 'src', 'scripts')
   RunCommand(['./cros_mark_as_stable', '--srcroot=..',
               '--tracking_branch="cros/master"',
@@ skipping 11 matching lines @@
   return buildconfig
 
 def main():
   # Parse options
   usage = "usage: %prog [options] cbuildbot_config"
   parser = optparse.OptionParser(usage=usage)
   parser.add_option('-r', '--buildroot',
                     help='root directory where build occurs', default=".")
   parser.add_option('-n', '--buildnumber',
                     help='build number', type='int', default=0)
+  parser.add_option('-f', '--revisionfile',
+                    help='file where new revisions are stored')
+  parser.add_option('--noclobber', action='store_false', dest='clobber',
+                    default=True,
+                    help='Disables clobbering the buildroot on failure')
   (options, args) = parser.parse_args()
 
   buildroot = options.buildroot
+  revisionfile = options.revisionfile
+  clobber = options.clobber
+
   if len(args) == 1:
     buildconfig = _GetConfig(args[0])
   else:
-    print >>sys.stderr, "Missing configuration description"
+    print >> sys.stderr, "Missing configuration description"
     parser.print_usage()
     sys.exit(1)
   try:
     if not os.path.isdir(buildroot):
       _FullCheckout(buildroot)
     else:
       _IncrementalCheckout(buildroot)
     chroot_path = os.path.join(buildroot, 'chroot')
     if not os.path.isdir(chroot_path):
       _MakeChroot(buildroot)
     boardpath = os.path.join(chroot_path, 'build', buildconfig['board'])
     if not os.path.isdir(boardpath):
       _SetupBoard(buildroot, board=buildconfig['board'])
     if buildconfig['uprev']:
-      _UprevPackages(buildroot)
+      _UprevPackages(buildroot, revisionfile)
     _Build(buildroot)
     if buildconfig['uprev']:
       _UprevPush(buildroot)
       _UprevCleanup(buildroot)
   except:
     # something went wrong, cleanup (being paranoid) for next build
-    RunCommand(['sudo', 'rm', '-rf', buildroot])
+    if clobber:
+      RunCommand(['sudo', 'rm', '-rf', buildroot], print_cmd=False)
     raise
 
 if __name__ == '__main__':
     main()