submodule/skia/bin/gerrit-number

   1 #!/usr/bin/env python2
   2 # Copyright 2017 Google Inc.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import json
   7 import re
   8 import subprocess
   9 import sys
  10 import urllib
  11
  12 # TODO(halcanary): document functions and script usage.
  13
  14 def retrieve_changeid(commit_or_branch):
  15   try:
  16     cmd = ['git', 'log', '-1', '--format=%B', commit_or_branch, '--']
  17     body = subprocess.check_output(cmd)
  18   except OSError:
  19     raise Exception('git not found')
  20   except subprocess.CalledProcessError:
  21     raise Exception('`%s` failed' % ' '.join(cmd))
  22   match = re.search(r'^Change-Id: *(.*) *$', body, re.MULTILINE)
  23   if match is None:
  24     raise Exception('Change-Id field missing from commit %s' % commit_or_branch)
  25   return match.group(1)
  26
  27
  28 def gerrit_change_id_to_number(site, cid):
  29   url = 'https://%s/changes/?q=change:%s' % (site, cid)
  30   try:
  31     content = urllib.urlopen(url).read()
  32   except IOError:
  33     raise Exception('error reading "%s"' % url)
  34   try:
  35     parsed = json.loads(content[content.find('['):])
  36   except ValueError:
  37     raise Exception('unable to parse content\n"""\n%s\n"""' % content)
  38   try:
  39     return parsed[0]['_number']
  40   except (IndexError, KeyError):
  41     raise Exception('Content missing\n"""\n%s\n"""' %
  42                     json.dumps(parsed, indent=2))
  43
  44
  45 def args_to_changeid(argv):
  46   if len(argv) == 2 and len(argv[1]) == 41 and argv[1][0] == 'I':
  47     return argv[1]
  48   else:
  49     return retrieve_changeid(argv[1] if len(argv) == 2 else 'HEAD')
  50
  51
  52 if __name__ == '__main__':
  53   try:
  54     sys.stdout.write('%d\n' %
  55         gerrit_change_id_to_number('skia-review.googlesource.com',
  56                                    args_to_changeid(sys.argv)))
  57   except Exception as e:
  58     sys.stderr.write('%s\n' % e)
  59     sys.exit(1)
  60
  61