Optionally support setuptools/distribute

1 files changed, 4 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 20fbeca..ad9981f 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,10 @@
 import os
 import sys
 
-from distutils.core import setup
+try:
+    from setuptools import setup
+except ImportError:
+    from distutils.core import setup
 
 def publish():
     """Publish to Pypi"""