summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xsetup.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index 726c07d..71f4fd2 100755
--- a/setup.py
+++ b/setup.py
@@ -141,7 +141,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"):
@@ -157,6 +157,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: