-
Notifications
You must be signed in to change notification settings - Fork 0
/
controller.py
64 lines (56 loc) · 1.89 KB
/
controller.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
from neopixel import Neopixel
class Controller:
def __init__(self, count, pin, type):
self.strip = None
self.count = count
self.pin = pin
self.type = type
self.isOn = True
self.color = (0, 80, 0)
self.brightness = 50
self.initialize()
def initialize(self):
self.strip = Neopixel(self.count, 0, self.pin, self.type)
self.updateLeds()
def changeColor(self, color):
try:
rgb = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
if rgb[0] < 0 or rgb[0]>255 or rgb[1] < 0 or rgb[1]>255 or rgb[2] < 0 or rgb[2]>255:
raise Exception("Color not valid")
self.color = rgb
if (self.type[0] == 'G'):
self.color = (rgb[1], rgb[0], rgb[2])
self.updateLeds()
print(f'Changed color to {rgb}')
return 'OK'
except Exception as e:
print(e)
return 'Color changing failed'
def changeBrightness(self, brightness):
try:
brightness = int(brightness)
if brightness < 0 or brightness > 255:
raise Exception("Brightness range not valid")
self.brightness = brightness
self.updateLeds()
print(f'Changed brightness to {brightness}')
return 'OK'
except Exception as e:
print(e)
return e
def toggleLeds(self, isOn):
try:
self.isOn = isOn
self.updateLeds()
print(f'Changed powering to {isOn}')
return 'OK'
except Exception as e:
print(e)
return e
def updateLeds(self):
if self.isOn:
self.strip.fill(self.color)
else:
self.strip.fill((0,0,0))
self.strip.brightness(self.brightness)
self.strip.show()