From 36fbdda492695e8f7c09d88aa2cde91f7dbaffc6 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Sun, 22 May 2011 19:43:29 -0400
Subject: [PATCH] setup.py improvements closes #5

---
 setup.py | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)
 mode change 100644 => 100755 setup.py

diff --git a/setup.py b/setup.py
old mode 100644
new mode 100755
index c4f17ac..bd9379e
--- a/setup.py
+++ b/setup.py
@@ -7,12 +7,29 @@ import sys
 from distutils.core import setup
 
 
-def publish():
-    """Publish to PyPi"""
-    os.system("python setup.py sdist upload")
 
-if sys.argv[-1] == "publish":
-    publish()
+if sys.argv[-1] == 'publish':
+    os.system("python setup.py sdist upload")
+    sys.exit()
+
+if sys.argv[-1] == 'speedups':
+    try:
+        import pip
+    except ImportError:
+        print('Pip required.')
+        sys.exit(1)
+
+    os.system('pip install simplejson pyyaml')
+    sys.exit()
+
+if sys.argv[-1] == 'test':
+    try:
+        import py
+    except ImportError:
+        print('py.test required.')
+        sys.exit(1)
+
+    os.system('pytest test_tablib.py')
     sys.exit()
 
 required = []