1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
#!/usr/bin/python
import codecs, os, re, sys, unittest, doctest
try:
from setuptools import setup
except ImportError:
from distutils.core import setup
setuptools_options = {}
else:
setuptools_options = dict(
test_suite='tests.test_suite',
)
try:
unichr
except NameError:
# Python 3.x support
unichr = chr
def read(filename):
f = codecs.open(filename, 'r', 'utf-8')
try:
return f.read()
finally:
f.close()
def unsphinx(text):
# remove Sphinx extensions used in CHANGES.rst from reStructuredText
# so that it can be handled by plain docutils
return text.replace(':func:', '').replace('.. currentmodule:: objgraph', '')
def get_version():
r = re.compile('^__version__ = "(.+)"$')
for line in read('objgraph.py').splitlines():
m = r.match(line)
if m:
# our read() returns unicode; coerce it back into str, or
# python2.7 setup.py sdist will try to mix a unicode filename with
# the byte stream of the .tar file
return str(m.group(1))
raise AssertionError('Could not determine version number from objgraph.py')
def get_description():
readme = read('README.rst')
changelog = read('CHANGES.rst')
description = unsphinx(readme + '\n\n\n' + changelog)
if '--unicode-description' in sys.argv:
sys.argv.remove('--unicode-description')
else:
# can't use u'' literals, this is supposed to work on both Py2 and Py3
description = description.replace('Kristj%sn' % unichr(0xe1),
'Kristjan')
description = description.encode('ascii', 'replace').decode('ascii')
return description
def build_images(doctests=()):
import tests
if not doctests:
doctests = tests.find_doctests()
suite = doctest.DocFileSuite(optionflags=doctest.ELLIPSIS,
checker=tests.IgnoreNodeCountChecker(),
*doctests)
os.chdir('docs')
result = unittest.TextTestRunner().run(suite)
if not result.wasSuccessful():
sys.exit(1)
if len(sys.argv) > 1 and sys.argv[1] == '--build-images':
build_images(sys.argv[2:])
sys.exit(0)
setup(name='objgraph',
version=get_version(),
author='Marius Gedminas',
author_email='marius@gedmin.as',
url='http://mg.pov.lt/objgraph/',
license='MIT',
description='Draws Python object reference graphs with graphviz',
long_description=get_description(),
classifiers=[
'Intended Audience :: Developers',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent',
'Programming Language :: Python',
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 2.4',
'Programming Language :: Python :: 2.5',
'Programming Language :: Python :: 2.6',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.1',
'Programming Language :: Python :: 3.2',
'Programming Language :: Python :: 3.3',
'Programming Language :: Python :: 3.4',
],
py_modules=['objgraph'],
**setuptools_options)
|