setup.py: run 2to3 if running under Python 3

brentp · Apr 30, 2013 · 761eb50 · 761eb50
1 parent 4ac0bed
commit 761eb50
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -3,6 +3,15 @@
 
 version = '0.4.5'
 
+# Run 2to3 builder if we're on Python 3.x, from
+#   http://wiki.python.org/moin/PortingPythonToPy3k
+try:
+    from distutils.command.build_py import build_py_2to3 as build_py
+except ImportError:
+    # 2.x
+    from distutils.command.build_py import build_py
+command_classes = {'build_py': build_py}
+
 setup(name='pyfasta',
       version=version,
       description=\
@@ -27,4 +36,5 @@
       entry_points={
       'console_scripts': ['pyfasta = pyfasta:main']
       },
+      cmdclass=command_classes,
   )