Validate that away command has option of either away or here

jsquyres · Jan 6, 2013 · 3cf5015 · 3cf5015
1 parent 510885f
commit 3cf5015
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nest.py b/nest.py
@@ -282,6 +282,9 @@ def main():
     elif (cmd == "curhumid"):
         print n.status["device"][n.serial]["current_humidity"]
     elif (cmd == "away"):
+        if len(args)<2 or args[1] not in {"away", "here"}:
+            print "please specify a state of 'away' or 'here'"
+            sys.exit(-1)
         n.set_away(args[1])
     else:
         print "misunderstood command:", cmd