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.errors
  13     import bzrlib.workingtree
  14     try:
  15         t = bzrlib.workingtree.WorkingTree.open_containing(__file__)[0]
  16     except (bzrlib.errors.NotBranchError, bzrlib.errors.NoWorkingTree):
  17         return None
  18     else:
  19         return t.branch.revno()
  20
  21
  22 def get_version_from_pkg_info():
  23     """Get the version from PKG-INFO file if we can."""
  24     pkg_info_path = os.path.join(os.path.dirname(__file__), 'PKG-INFO')
  25     try:
  26         pkg_info_file = open(pkg_info_path, 'r')
  27     except (IOError, OSError):
  28         return None
  29     try:
  30         pkg_info = email.message_from_file(pkg_info_file)
  31     except email.MessageError:
  32         return None
  33     return pkg_info.get('Version', None)
  34
  35
  36 def get_version():
  37     """Return the version of testtools that we are building."""
  38     version = '.'.join(
  39         str(component) for component in testtools.__version__[0:3])
  40     phase = testtools.__version__[3]
  41     if phase == 'final':
  42         return version
  43     pkg_info_version = get_version_from_pkg_info()
  44     if pkg_info_version:
  45         return pkg_info_version
  46     revno = get_revno()
  47     if revno is None:
  48         return "snapshot"
  49     if phase == 'alpha':
  50         # No idea what the next version will be
  51         return 'next-r%s' % revno
  52     else:
  53         # Preserve the version number but give it a revno prefix
  54         return version + '-r%s' % revno
  55
  56
  57 def get_long_description():
  58     manual_path = os.path.join(
  59         os.path.dirname(__file__), 'doc/overview.rst')
  60     return open(manual_path).read()
  61
  62
  63 setup(name='testtools',
  64       author='Jonathan M. Lange',
  65       author_email='jml+testtools@mumak.net',
  66       url='https://launchpad.net/testtools',
  67       description=('Extensions to the Python standard library unit testing '
  68                    'framework'),
  69       long_description=get_long_description(),
  70       version=get_version(),
  71       classifiers=["License :: OSI Approved :: MIT License"],
  72       packages=['testtools', 'testtools.testresult', 'testtools.tests'],
  73       cmdclass={'test': testtools.TestCommand})