Enforce wheel check before allowing 'setup.pu publish'

1 files changed, 3 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 2c56cd75..e64346d4 100755
--- a/setup.py
+++ b/setup.py
@@ -59,6 +59,9 @@ version = get_version('rest_framework')
 
 
 if sys.argv[-1] == 'publish':
+    if os.system("pip freeze | grep wheel"):
+        print "wheel not installed.\nUse `pip install wheel`.\nExiting."
+        sys.exit()
     os.system("python setup.py sdist upload")
     os.system("python setup.py bdist_wheel upload")
     print("You probably want to also tag the version now:")