Script for the Pi Hut 3D Xmas Tree

pihut_xmas_tree.py

from gpiozero import LEDBoard
from time import sleep
import random

tree = LEDBoard(*range(2, 28), pwm=True)

# first led is the top one in the star
tree[0].value = 0.6
leds = range(1, len(tree))


def startup():
    for bright in (0.25, 0.50, 0.75, 1.0):
        for led in leds:
            tree[led].value = bright
            sleep(0.1)
        for led in reversed(leds):
            tree[led].off()
            sleep(0.1)


def run():
    # each iteration, pick a few leds and give them random values
    shuffle_leds = leds[:]
    while True:
        count = random.randint(4, len(shuffle_leds))
        random.shuffle(shuffle_leds)
        leds_on = shuffle_leds[0:count]
        leds_off = shuffle_leds[count:]
        for led in leds_off:
            tree[led].off()
        for led in leds_on:
            tree[led].value = random.randint(2, 7) / 10.0
        sleep(1)


if __name__ == '__main__':
    try:
        startup()
        run()
    finally:
        tree[0].off()
        for led in leds:
            tree[led].off()