tools/roll_swarming_client.py

   1 #!/usr/bin/env python
   2 # Copyright 2014 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Rolls swarming_client.
   7
   8 While it is currently hard coded for swarming_client/, it is potentially
   9 modifiable to allow different dependencies. Works only with git checkout and git
  10 dependencies.
  11 """
  12
  13 import optparse
  14 import os
  15 import re
  16 import subprocess
  17 import sys
  18
  19 SRC_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  20
  21
  22 def is_pristine(root, merge_base='origin/master'):
  23   """Returns True is a git checkout is pristine."""
  24   cmd = ['git', 'diff', '--ignore-submodules', merge_base]
  25   return not (
  26       subprocess.check_output(cmd, cwd=root).strip() or
  27       subprocess.check_output(cmd + ['--cached'], cwd=root).strip())
  28
  29
  30 def roll(deps_dir, key, reviewer, bug):
  31   if not is_pristine(SRC_ROOT):
  32     print >> sys.stderr, 'Ensure %s is clean first.' % SRC_ROOT
  33     return 1
  34
  35   full_dir = os.path.join(SRC_ROOT, deps_dir)
  36   head = subprocess.check_output(
  37       ['git', 'rev-parse', 'HEAD'], cwd=full_dir).strip()
  38   deps = os.path.join(SRC_ROOT, 'DEPS')
  39   with open(deps, 'rb') as f:
  40     deps_content = f.read()
  41
  42   if not head in deps_content:
  43     print('Warning: %s is not checked out at the expected revision in DEPS' %
  44           deps_dir)
  45     # It happens if the user checked out a branch in the dependency by himself.
  46     # Fall back to reading the DEPS to figure out the original commit.
  47     for i in deps_content.splitlines():
  48       m = re.match(r'\s+"' + key + '": "([a-z0-9]{40})",', i)
  49       if m:
  50         head = m.group(1)
  51         break
  52     else:
  53       print >> sys.stderr, 'Expected to find commit %s for %s in DEPS' % (
  54           head, key)
  55       return 1
  56
  57   print('Found old revision %s' % head)
  58
  59   subprocess.check_call(['git', 'fetch', 'origin'], cwd=full_dir)
  60   master = subprocess.check_output(
  61       ['git', 'rev-parse', 'origin/master'], cwd=full_dir).strip()
  62   print('Found new revision %s' % master)
  63
  64   if master == head:
  65     print('No revision to roll!')
  66     return 1
  67
  68   commit_range = '%s..%s' % (head[:9], master[:9])
  69   logs = subprocess.check_output(
  70       ['git', 'log', commit_range, '--date=short', '--format=%ad %ae %s'],
  71       cwd=full_dir).strip()
  72   logs = logs.replace('@chromium.org', '')
  73   cmd = (
  74       'git log %s --date=short --format=\'%%ad %%ae %%s\' | '
  75       'sed \'s/@chromium\.org//\'') % commit_range
  76
  77   msg = (
  78       'Roll %s/ to %s.\n'
  79       '\n'
  80       '$ %s\n'
  81       '%s\n\n'
  82       'R=%s\n'
  83       'BUG=%s') % (
  84           deps_dir,
  85           master,
  86           cmd,
  87           logs,
  88           reviewer,
  89           bug)
  90
  91   print('Commit message:')
  92   print('\n'.join('    ' + i for i in msg.splitlines()))
  93   deps_content = deps_content.replace(head, master)
  94   with open(deps, 'wb') as f:
  95     f.write(deps_content)
  96   subprocess.check_call(['git', 'add', 'DEPS'], cwd=SRC_ROOT)
  97   subprocess.check_call(['git', 'commit', '-m', msg], cwd=SRC_ROOT)
  98   print('Run:')
  99   print('  git cl upl --send-mail')
 100   return 0
 101
 102
 103 def main():
 104   parser = optparse.OptionParser(description=sys.modules[__name__].__doc__)
 105   parser.add_option(
 106       '-r', '--reviewer', default='',
 107       help='To specify multiple reviewers, use comma separated list, e.g. '
 108            '-r joe,jack,john. Defaults to @chromium.org')
 109   parser.add_option('-b', '--bug', default='')
 110   options, args = parser.parse_args()
 111   if args:
 112     parser.error('Unknown argument %s' % args)
 113   if not options.reviewer:
 114     parser.error('Pass a reviewer right away with -r/--reviewer')
 115
 116   reviewers = options.reviewer.split(',')
 117   for i, r in enumerate(reviewers):
 118     if not '@' in r:
 119       reviewers[i] = r + '@chromium.org'
 120
 121   return roll(
 122       'tools/swarming_client',
 123       'swarming_revision',
 124       ','.join(reviewers),
 125       options.bug)
 126
 127
 128 if __name__ == '__main__':
 129   sys.exit(main())