Added check for mode.

Valid modes are 'cool', 'heat' or 'range'
jsquyres · Jan 6, 2013 · bc69abc · bc69abc
1 parent e1bacdf
commit bc69abc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nest.py b/nest.py
@@ -259,7 +259,7 @@ def main():
             sys.exit(-1)
         n.set_fan(args[1])
     elif (cmd == "mode"):
-        if len(args)<2:
+        if len(args)<2 or args[1] not in {"cool", "heat", "range"}:
             print "please specify a thermostat mode of 'cool', 'heat'  or 'range'"
             sys.exit(-1)
         n.set_mode(args[1])