[debian] py3k: change build command from new distutils

1 files changed, 8 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 72881eb..e485805 100644
--- a/setup.py
+++ b/setup.py
@@ -38,6 +38,12 @@ except ImportError:
     from distutils.command import install_lib
     USE_SETUPTOOLS = 0
 
+try:
+    # python3
+    from distutils.command.build_py import build_py_2to3 as build_py
+except ImportError:
+    # python2.x
+    from distutils.command.build_py import build_py
 
 sys.modules.pop('__pkginfo__', None)
 # import optional features
@@ -156,7 +162,8 @@ def install(**kwargs):
                  scripts = ensure_scripts(scripts),
                  data_files = data_files,
                  ext_modules = ext_modules,
-                 cmdclass = {'install_lib': MyInstallLib},
+                 cmdclass = {'install_lib': MyInstallLib,
+                             'build_py': build_py},
                  **kwargs
                  )