noobs_nerd_notifier/host/setcolor.py

35 lines
755 B
Python

import usb.util
import sys
def set_color(r, g, b):
"""
Sets the color of the LED.
:param r: the 8 bit red value of the color
:param g: the 8 bit green value of the color
:param b: the 8 bit blue value of the color
:return: True if 3 bytes were transmitted, False otherwise
"""
d = [r, g, b]
result = dev.ctrl_transfer(0x40, 4, 0, 0, d)
return result == 3
if len(sys.argv) != 4:
print("must be called with 'color.py red blue green'")
sys.exit(1)
red = int(sys.argv[1])
green = int(sys.argv[2])
blue = int(sys.argv[3])
# find our device
dev = usb.core.find(idVendor=0x16c0, idProduct=0x05dc)
# was it found?
if dev is None:
raise ValueError('Device not found')
print(set_color(red, green, blue))