1 files changed, 9 insertions, 8 deletions
diff --git a/setup.py b/setup.py
index 1aa6e190..2ee0628c 100644
--- a/setup.py
+++ b/setup.py
@@ -3,13 +3,19 @@
 
 from setuptools import setup
 
+import os, re
+
+path = os.path.join(os.path.dirname(__file__), 'djangorestframework', '__init__.py')
+init_py = open(path).read()
+VERSION = re.match("__version__ = '([^']+)'", init_py).group(1)
+
 setup(
-    name = "djangorestframework",
-    version = "0.1",
+    name = 'djangorestframework',
+    version = VERSION,
     url = 'https://bitbucket.org/tomchristie/django-rest-framework/wiki/Home',
     download_url = 'https://bitbucket.org/tomchristie/django-rest-framework/downloads',
     license = 'BSD',
-    description = "A lightweight REST framework for Django.",
+    description = 'A lightweight REST framework for Django.',
     author = 'Tom Christie',
     author_email = 'tom@tomchristie.com',
     packages = ['djangorestframework',
@@ -30,8 +36,3 @@ setup(
         'Topic :: Internet :: WWW/HTTP',
     ]
 )
-
-import os, shutil
-shutil.rmtree(os.path.join(os.path.dirname(__file__), 'djangorestframework.egg-info'), True)
-
-