Add option to print version number

1 files changed, 11 insertions, 3 deletions

diff --git a/setup.py b/setup.py
index a14a6c6..f41e668 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,16 @@
-from distutils.core import setup
-import dotfiles
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+try:
+    from setuptools import setup
+except ImportError:
+    from distutils.core import setup
+
+from dotfiles.core import __version__
+
 
 setup(name='dotfiles',
-      version=dotfiles.__version__,
+      version=__version__,
       description='Easily manage your dotfiles',
       long_description=open('README.rst').read(),
       author='Jon Bernard',