Merge pull request #177 from msabramo/require_ordereddict_for_py26

jgraham · jgraham · commit 3b3c1031b3da · 2014-12-12T14:26:06.000Z
setup.py: Require ordereddict for Python 2.6
diff --git 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,
       )
