35 lines
755 B
Python
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))
|
||
|
|
|