diff --git a/setup.py b/setup.py index f75910ea..82522ab5 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,7 @@ -from distutils.core import setup +try: + from setuptools import setup +except ImportError: + from distutils.core import setup import os import codecs @@ -23,6 +26,12 @@ if os.path.isdir(os.path.join('html5lib', name)) and not name.startswith('.') and name != 'tests'] +install_requires = ['six'] +try: + from collections import OrderedDict +except ImportError: + install_requires.append('ordereddict') + current_dir = os.path.dirname(__file__) with codecs.open(os.path.join(current_dir, 'README.rst'), 'r', 'utf8') as readme_file: with codecs.open(os.path.join(current_dir, 'CHANGES.rst'), 'r', 'utf8') as changes_file: @@ -38,7 +47,5 @@ maintainer='James Graham', maintainer_email='james@hoppipolla.co.uk', packages=packages, - install_requires=[ - 'six', - ], + install_requires=install_requires, )