testing/mozbase/versioninfo.py

   1 #!/usr/bin/env python
   2
   3 # This Source Code Form is subject to the terms of the Mozilla Public
   4 # License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5 # You can obtain one at http://mozilla.org/MPL/2.0/.
   6
   7 """
   8 List mozbase package dependencies or generate changelogs
   9 from commit messages.
  10 """
  11
  12 from collections import Iterable
  13 from distutils.version import StrictVersion
  14 import argparse
  15 import os
  16 import subprocess
  17 import sys
  18
  19 import setup_development
  20
  21 here = os.path.abspath(os.path.dirname(__file__))
  22
  23 def run_hg(command):
  24     command = command[:]
  25     if not isinstance(command, Iterable):
  26         command = command.split()
  27     command.insert(0, 'hg')
  28     try:
  29         output = subprocess.check_output(command, cwd=here)
  30     except subprocess.CalledProcessError:
  31         sys.exit(1)
  32     return output
  33
  34
  35 def changelog(args):
  36     setup = os.path.join(args.module, 'setup.py')
  37
  38     def get_version_rev(v=None):
  39         revisions = run_hg(['log', setup, '--template={rev},']).split(',')[:-1]
  40         for rev in revisions:
  41             diff = run_hg(['diff', '-c', rev, setup, '-U0'])
  42             minus_version = None
  43             plus_version = None
  44             for line in diff.splitlines():
  45                 if line.startswith('-PACKAGE_VERSION'):
  46                     minus_version = StrictVersion(line.split()[-1].strip('"\''))
  47                 elif line.startswith('+PACKAGE_VERSION'):
  48                     plus_version = StrictVersion(line.split()[-1].strip('"\''))
  49
  50                     # make sure the change isn't a backout
  51                     if not minus_version or plus_version > minus_version:
  52                         if not v:
  53                             return rev
  54
  55                         if StrictVersion(v) == plus_version:
  56                             return rev
  57         print("Could not find %s revision for version %s." % (args.module, v or 'latest'))
  58         sys.exit(1)
  59
  60     from_ref = args.from_ref or get_version_rev()
  61     to_ref = args.to_ref or 'tip'
  62
  63     if '.' in from_ref:
  64         from_ref = get_version_rev(from_ref)
  65     if '.' in to_ref:
  66         to_ref = get_version_rev(to_ref)
  67
  68     delim = '\x12\x59\x52\x99\x05'
  69     changelog = run_hg(['log', '-r', '%s:children(%s)' % (to_ref, from_ref), '--template={desc}%s' % delim, '-M', args.module]).split(delim)[:-1]
  70
  71     def prettify(desc):
  72         lines = desc.splitlines()
  73         lines = [('* %s' if i == 0 else '  %s') % l for i, l in enumerate(lines)]
  74         return '\n'.join(lines)
  75
  76     changelog = map(prettify, changelog)
  77     print '\n'.join(changelog)
  78
  79
  80 def dependencies(args):
  81     # get package information
  82     info = {}
  83     dependencies = {}
  84     for package in setup_development.mozbase_packages:
  85         directory = os.path.join(setup_development.here, package)
  86         info[directory] = setup_development.info(directory)
  87         name, _dependencies = setup_development.get_dependencies(directory)
  88         assert name == info[directory]['Name']
  89         dependencies[name] = _dependencies
  90
  91     # print package version information
  92     for value in info.values():
  93         print '%s %s : %s' % (value['Name'], value['Version'],
  94                               ', '.join(dependencies[value['Name']]))
  95
  96
  97 def main(args=sys.argv[1:]):
  98     parser = argparse.ArgumentParser()
  99     subcommands = parser.add_subparsers(help="Sub-commands")
 100
 101     p_deps = subcommands.add_parser('dependencies', help="Print dependencies.")
 102     p_deps.set_defaults(func=dependencies)
 103
 104     p_changelog = subcommands.add_parser('changelog', help="Print a changelog.")
 105     p_changelog.add_argument('module', help="Module to get changelog from.")
 106     p_changelog.add_argument('--from', dest='from_ref', default=None,
 107                              help="Starting version or revision to list changes from. [defaults to latest version]")
 108     p_changelog.add_argument('--to', dest='to_ref', default=None,
 109                              help="Ending version or revision to list changes to. [defaults to tip]")
 110     p_changelog.set_defaults(func=changelog)
 111
 112     # default to showing dependencies
 113     if args == []:
 114         args.append('dependencies')
 115     args = parser.parse_args(args)
 116     args.func(args)
 117
 118
 119 if __name__ == '__main__':
 120     main()