Fixing exception if HOME is missing.

1 files changed, 4 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 00a7a51..f075dcb 100755
--- a/setup.py
+++ b/setup.py
@@ -137,7 +137,7 @@ def main(argv=None):
         # Settings:
         nice = False
         pretend = False
-        home = os.environ["HOME"]
+        home = os.environ.get("HOME")
 
         for o, a in opts:
             if o in ("-h", "--help"):
@@ -153,6 +153,9 @@ def main(argv=None):
             elif o in ("-d", "--home"):
                 home = a
 
+        if not home:
+            raise Usage("No home provided")
+
         makeDots(os.getcwd(), home, nice, pretend)
 
     except Usage, err: