testtools: Import new upstream snapshot.
[nivanova/samba-autobuild/.git] / 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'])