Add publish and test routines to setup.py

1 files changed, 13 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index f41e668..df99c5a 100644..100755
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
+import os
+import sys
+
 try:
     from setuptools import setup
 except ImportError:
@@ -9,6 +12,16 @@ except ImportError:
 from dotfiles.core import __version__
 
 
+if sys.argv[-1] == "publish":
+    os.system("python setup.py sdist upload")
+    sys.exit()
+
+
+if sys.argv[-1] == "test":
+    os.system("python test_dotfiles.py")
+    sys.exit()
+
+
 setup(name='dotfiles',
       version=__version__,
       description='Easily manage your dotfiles',