-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtasks.py
196 lines (168 loc) · 5.6 KB
/
tasks.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
from gpiozero import LEDBoard, PWMLED
from gpiozero.tools import random_values
from random import *
import time
from celery import Celery
app = Celery("tasks")
# set up various delay values
delay = 0.05 #used in spiral function
delay2 = 0.03 #used in spiral function
on_delay = 0.01 #used in random_led function
top_down_delay = 0.125 # used in top_down function
top_down_delay2 = 0.05 # used in top_down function
sides_delay = 0.075 # used in the sides function
# Just flicker the star led
star = PWMLED(2)
star.source = random_values()
period_min = 3 # base number of seconds for each function
period_max = 10 # maximum number of seconds for each function
@app.task
def spiral():
# function to create a spiral pattern that repeats for a while
branch1_leds = LEDBoard(27,17,4,18,15,14)
branch2_leds = LEDBoard(11,5,8,12,6,7)
branch3_leds = LEDBoard(19,16,26,13,20,21)
branch4_leds = LEDBoard(25,9,22,24,23,10)
branch1_leds.off()
branch2_leds.off()
branch3_leds.off()
branch4_leds.off()
time.sleep(delay)
spiral_time = time.time() + uniform(period_min, period_max)
while spiral_time > time.time():
# spiral down
for point in range(6):
branch1_leds[point].on()
time.sleep(delay2)
branch2_leds[point].on()
time.sleep(delay2)
branch3_leds[point].on()
time.sleep(delay2)
branch4_leds[point].on()
time.sleep(delay2)
time.sleep(delay)
# spiral up
for point in range(5, -1, -1):
branch1_leds[point].off()
time.sleep(delay2)
branch2_leds[point].off()
time.sleep(delay2)
branch3_leds[point].off()
time.sleep(delay2)
branch4_leds[point].off()
time.sleep(delay2)
time.sleep(delay)
return True
@app.task
def random_led():
# function that turns off and on random leds
tree = LEDBoard(*range(4,28))
random_time = time.time() + uniform(period_min, period_max)
while random_time > time.time():
# Turn on a random led
on_lamp = randint(0, 23)
tree[on_lamp].on()
time.sleep(on_delay)
# Turn off a random led
off_lamp = randint(0, 23)
tree[off_lamp].off()
return True
@app.task
def top_down(forward):
# function that lights the leds from top down.
# Or from bottom up if forward is not True.
branch1_leds = LEDBoard(27,17,4,18,15,14)
branch2_leds = LEDBoard(11,5,8,12,6,7)
branch3_leds = LEDBoard(19,16,26,13,20,21)
branch4_leds = LEDBoard(25,9,22,24,23,10)
top_down_time = time.time() + uniform(period_min, period_max)
while top_down_time > time.time():
branch1_leds.off()
branch2_leds.off()
branch3_leds.off()
branch4_leds.off()
time.sleep(top_down_delay2)
if forward:
for point in range(6):
branch1_leds[point].on()
branch2_leds[point].on()
branch3_leds[point].on()
branch4_leds[point].on()
time.sleep(top_down_delay)
else:
for point in range(5, -1, -1):
branch1_leds[point].on()
branch2_leds[point].on()
branch3_leds[point].on()
branch4_leds[point].on()
time.sleep(top_down_delay)
return True
@app.task
def sides(forward):
# function that lights the leds on the sides in sequence
# direction is defined by forward being True or False
side1_leds = LEDBoard(8,6,7)
side2_leds = LEDBoard(11,5,12)
side3_leds = LEDBoard(19,26,21)
side4_leds = LEDBoard(16,13,20)
side5_leds = LEDBoard(25,24,23)
side6_leds = LEDBoard(9,22,10)
side7_leds = LEDBoard(17,4,14)
side8_leds = LEDBoard(27,18,15)
sides_time = time.time() + uniform(period_min, period_max)
while sides_time > time.time():
if forward:
# spin the side leds anticlockwise
side1_leds.on()
time.sleep(sides_delay)
side1_leds.off()
side2_leds.on()
time.sleep(sides_delay)
side2_leds.off()
side3_leds.on()
time.sleep(sides_delay)
side3_leds.off()
side4_leds.on()
time.sleep(sides_delay)
side4_leds.off()
side5_leds.on()
time.sleep(sides_delay)
side5_leds.off()
side6_leds.on()
time.sleep(sides_delay)
side6_leds.off()
side7_leds.on()
time.sleep(sides_delay)
side7_leds.off()
side8_leds.on()
time.sleep(sides_delay)
side8_leds.off()
else:
# spin the side leds clockwise
side8_leds.on()
time.sleep(sides_delay)
side8_leds.off()
side7_leds.on()
time.sleep(sides_delay)
side7_leds.off()
side6_leds.on()
time.sleep(sides_delay)
side6_leds.off()
side5_leds.on()
time.sleep(sides_delay)
side5_leds.off()
side4_leds.on()
time.sleep(sides_delay)
side4_leds.off()
side3_leds.on()
time.sleep(sides_delay)
side3_leds.off()
side2_leds.on()
time.sleep(sides_delay)
side2_leds.off()
side1_leds.on()
time.sleep(sides_delay)
side1_leds.off()
return True
if __name__ == '__main__':
app.worker_main()