Fix setup.py import

1 files changed, 59 insertions, 3 deletions

diff --git a/setup.py b/setup.py
index dea37fa..fa2a0b7 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,11 @@
 #!/usr/bin/env python
 
+import os
+import sys
+from distutils import log
+from distutils.cmd import Command
+from distutils.errors import DistutilsError
+
 try:
     from setuptools import setup, find_packages
 except ImportError:
@@ -7,7 +13,57 @@ except ImportError:
     use_setuptools()
     from setuptools import setup, find_packages
 
-import distutils_ext
+
+class SimpleCommand(Command):
+
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+
+class cover_test(SimpleCommand):
+
+    description = "run unit tests with coverage"
+
+    def run(self):
+        from coverage import coverage
+
+        cov = coverage(data_file=".coverage", branch=True,
+                       source=self.distribution.packages)
+        cov.start()
+
+        # Unittest calls exit. How naughty.
+        try:
+            self.run_command("test")
+        except SystemExit:
+            pass
+
+        cov.stop()
+        cov.xml_report(outfile="coverage.xml")
+
+
+class check_style(SimpleCommand):
+
+    description = "run PEP8 style validations"
+
+    def run(self):
+        from pep8 import StyleGuide
+
+        self.run_command("egg_info")
+        files = self.get_finalized_command("egg_info")
+
+        report = StyleGuide().check_files([
+            p for p in files.filelist.files if p.endswith(".py")])
+
+        if report.total_errors:
+            msg = "Found {} PEP8 violations".format(report.total_errors)
+            raise DistutilsError(msg)
+        else:
+            log.info("No PEP8 violations found")
 
 
 setup(
@@ -20,8 +76,8 @@ setup(
     url="https://github.com/mcrute/pydora",
     test_suite="tests.discover_suite",
     cmdclass={
-        "check_style": distutils_ext.check_style,
-        "cover_test": distutils_ext.cover_test,
+        "check_style": check_style,
+        "cover_test": cover_test,
     },
     packages=find_packages(exclude=["tests", "tests.*"]),
     tests_require=[