Add implementation

g203-led.py

@@ -0,0 +1,138 @@
+#!env/bin/python
+
+# Logitech G203 Prodigy LED Control
+# https://github.com/smasty/g203-led
+# Author: Smasty, hello@smasty.net
+# Licensed under the MIT license.
+
+import sys
+import usb.core
+import usb.util
+import re
+
+g203_vendor_id =  0x046d
+g203_product_id = 0xc084
+
+default_rate = 10000
+default_brightness = 100
+
+
+dev = None
+intf = None
+
+
+def help():
+    print("""Logitech G203 Prodigy LED control
+
+Usage:
+\tg203-led solid {color} - Solid color mode
+\tg203-led cycle [{rate} [{brightness}]] - Cycle through all colors
+\tg203-led breathe {color} [{rate} [{brightness}]] - Single color breathing
+
+Arguments:
+\tColor: RRGGBB (RGB hex value)
+\tRate: 100-60000 (Number of milliseconds. Default: 10000ms)
+\tBrightness: 0-100 (Percentage. Default: 100%)""")
+
+
+def main():
+    if(len(sys.argv) < 2):
+        help()
+        sys.exit()
+
+    args = sys.argv + [None] * (5 - len(sys.argv))
+
+    mode = args[1]
+    if mode == 'solid':
+        set_led_solid(process_color(args[2]))
+    elif mode == 'cycle':
+        set_led_cycle(process_rate(args[2]), process_brightness(args[3]))
+    elif mode == 'breathe':
+        set_led_breathe(
+            process_color(args[2]),
+            process_rate(args[3]),
+            process_brightness(args[4])
+        )
+    else:
+        print_error('Unknown mode.')
+
+
+def print_error(msg):
+    print('Error: ' + msg)
+    sys.exit(1)
+
+
+def process_color(color):
+    if not color:
+        print_error('No color specifed.')
+    if color[0] == '#':
+        color = color[1:]
+    if not re.match('^[0-9a-fA-F]{6}$', color):
+        print_error('Invalid color specified.')
+    return color.lower()
+
+def process_rate(rate):
+    if not rate:
+        rate = default_rate
+    try:
+        return '{:04x}'.format(max(100, min(65535, int(rate))))
+    except ValueError:
+        print_error('Invalid rate specified.')
+
+def process_brightness(brightness):
+    if not brightness:
+        brightness = default_brightness
+    try:
+        return '{:02x}'.format(max(1, min(100, int(brightness))))
+    except ValueError:
+        print_error('Invalid brightness specified.')
+
+
+def set_led_solid(color):
+    return set_led('01', color + '0000000000')
+
+def set_led_breathe(color, rate, brightness):
+    return set_led('03', color + rate + '00' + brightness + '00')
+
+def set_led_cycle(rate, brightness):
+    print(rate, brightness)
+    return set_led('02', '0000000000' + rate + brightness)
+
+
+def set_led(mode, data):
+    global dev
+
+    prefix = '11ff0e3b00'
+    suffix = '000000000000'
+    data = prefix + mode + data + suffix
+    data = [ int(''.join([data[i], data[i+1]]), base=16) for i in range(0, len(data), 2)]
+
+    attach_mouse()
+    dev.ctrl_transfer(0x21, 0x09, 0x0211, 0x01, data)
+    detach_mouse()
+
+
+def attach_mouse():
+    global dev
+    global intf
+    dev = usb.core.find(idVendor=g203_vendor_id, idProduct=g203_product_id)
+    if dev is None:
+        print_error('Device {:04x}:{:04x} not found.'.format(g203_vendor_id, g203_product_id))
+    intf = 1
+    if dev.is_kernel_driver_active(intf) is True:
+        dev.detach_kernel_driver(intf)
+        usb.util.claim_interface(dev, intf)
+
+
+def detach_mouse():
+    global dev
+    global intf
+    if intf is not None:
+        usb.util.release_interface(dev, intf)
+        dev.attach_kernel_driver(intf)
+        dev = None
+        intf = None
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1 @@
+pyusb==1.0.2