Add tree.on() to main entry point in tree.py, update readme and add t…

…ry/except for KeyboardInterrupt

README.md

@@ -10,6 +10,9 @@ Start by downloading the xmas tree file. Open a terminal and type:
 wget https://bit.ly/2Lr9CT3 -O tree.py
 ```
 
+Test the tree by running `python3 tree.py` (or running it from an IDE like Mu,
+Thonny or IDLE). All the lights should come on (white).
+
 When you write your own Python code, make sure you keep this file in the same
 folder.
 
@@ -20,8 +23,7 @@ using Raspbian Lite, you'll need to install gpiozero with:
 sudo apt install python3-gpiozero
 ```
 
-Open a Python shell or IDE (like Mu, Thonny or IDLE), import `RGBXmasTree` and
-initialise your tree:
+Open a Python shell or IDE, import `RGBXmasTree` and initialise your tree:
 
 ```python
 from tree import RGBXmasTree

examples/huecycle.py

@@ -5,5 +5,8 @@
 
 tree.color = Color('red')
 
-while True:
-    tree.color += Hue(deg=1)
+try:
+    while True:
+        tree.color += Hue(deg=1)
+except KeyboardInterrupt:
+    tree.close()

examples/onebyone.py

@@ -5,7 +5,10 @@
 
 colors = [Color('red'), Color('green'), Color('blue')] # add more if you like
 
-while True:
-    for color in colors:
-        for pixel in tree:
-            pixel.color = color
+try:
+    while True:
+        for color in colors:
+            for pixel in tree:
+                pixel.color = color
+except KeyboardInterrupt:
+    tree.close()

examples/randomsparkles.py

@@ -12,5 +12,8 @@ def random_color():
 def random_colors(n):
     return [random_color() for i in range(n)]
 
-while True:
-    tree.value = random_colors(25)
+try:
+    while True:
+        tree.value = random_colors(25)
+except KeyboardInterrupt:
+    tree.close()

examples/rgb.py

@@ -5,6 +5,9 @@
 
 colors = [Color('red'), Color('green'), Color('blue')] # add more if you like
 
-while True:
-    for color in colors:
-        tree.color = color
+try:
+    while True:
+        for color in colors:
+            tree.color = color
+except KeyboardInterrupt:
+    tree.close()

tree.py

@@ -104,3 +104,5 @@ def close(self):
 
 if __name__ == '__main__':
     tree = RGBXmasTree()
+
+    tree.on()