Fixed SyntaxError when installing via Python 3

1 files changed, 10 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index b9a7677..dacbf91 100755
--- a/setup.py
+++ b/setup.py
@@ -2,8 +2,10 @@
 """Distutils installer for testtools."""
 
 from setuptools import setup
+from distutils.command.build_py import build_py
 import email
 import os
+import sys
 
 import testtools
 cmd_class = {}
@@ -11,6 +13,14 @@ if getattr(testtools, 'TestCommand', None) is not None:
     cmd_class['test'] = testtools.TestCommand
 
 
+class testtools_build_py(build_py):
+    def build_module(self, module, module_file, package):
+        if sys.version_info >= (3,) and module == '_compat2x':
+            return
+        return build_py.build_module(self, module, module_file, package)
+cmd_class['build_py'] = testtools_build_py
+
+
 def get_version_from_pkg_info():
     """Get the version from PKG-INFO file if we can."""
     pkg_info_path = os.path.join(os.path.dirname(__file__), 'PKG-INFO')