lib/testtools/setup.py

   1 #!/usr/bin/env python
   2 """Distutils installer for testtools."""
   3
   4 from distutils.core import setup
   5 import email
   6 import os
   7
   8 import testtools
   9
  10
  11 def get_revno():
  12     import bzrlib.workingtree
  13     t = bzrlib.workingtree.WorkingTree.open_containing(__file__)[0]
  14     return t.branch.revno()
  15
  16
  17 def get_version_from_pkg_info():
  18     """Get the version from PKG-INFO file if we can."""
  19     pkg_info_path = os.path.join(os.path.dirname(__file__), 'PKG-INFO')
  20     try:
  21         pkg_info_file = open(pkg_info_path, 'r')
  22     except (IOError, OSError):
  23         return None
  24     try:
  25         pkg_info = email.message_from_file(pkg_info_file)
  26     except email.MessageError:
  27         return None
  28     return pkg_info.get('Version', None)
  29
  30
  31 def get_version():
  32     """Return the version of testtools that we are building."""
  33     version = '.'.join(
  34         str(component) for component in testtools.__version__[0:3])
  35     phase = testtools.__version__[3]
  36     if phase == 'final':
  37         return version
  38     pkg_info_version = get_version_from_pkg_info()
  39     if pkg_info_version:
  40         return pkg_info_version
  41     revno = get_revno()
  42     if phase == 'alpha':
  43         # No idea what the next version will be
  44         return 'next-r%s' % revno
  45     else:
  46         # Preserve the version number but give it a revno prefix
  47         return version + '-r%s' % revno
  48
  49
  50 def get_long_description():
  51     manual_path = os.path.join(os.path.dirname(__file__), 'MANUAL')
  52     return open(manual_path).read()
  53
  54
  55 setup(name='testtools',
  56       author='Jonathan M. Lange',
  57       author_email='jml+testtools@mumak.net',
  58       url='https://launchpad.net/testtools',
  59       description=('Extensions to the Python standard library unit testing '
  60                    'framework'),
  61       long_description=get_long_description(),
  62       version=get_version(),
  63       classifiers=["License :: OSI Approved :: MIT License"],
  64       packages=['testtools', 'testtools.testresult', 'testtools.tests'])