AlienPyFX/AlienPyFX_c.py at master · ybotspawn/AlienPyFX · GitHub

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
#!/usr/bin/python2.7

import usb.core
import usb.util
import argparse

class AlienwareDeviceManager(object):
    vendorID = 0x187c
    productID = 0x0527
    dev = None

    def __init__(self):
        print("Init")

    def apply_theme(self, themeFilePath):
        self.__attach_usb_device__()
        self.__apply_theme__(themeFilePath)

    def lights_off(self):
        print("turning lights off")

    def lights_on(self):
        print("turning lights on")

    def __send_usb_message__(self, line):
        if not line.startswith('#'):
            message = r"\x{0}".format(line.replace(':', r'\x').replace('\n', ''))
            while True:
                if self.dev.ctrl_transfer(0x21, 9, 0, 0, message.decode("string-escape")) == len(message.decode("string-escape")):
                    break

    def __apply_theme__(self, themeFilePath):
        f = open(themeFilePath)
        for line in f.readlines():
            self.__send_usb_message__(line)

    def __attach_usb_device__(self):
        self.dev = usb.core.find(idVendor=self.vendorID, idProduct=self.productID)
        endpoint = self.dev[0][(0,0)][0]
        config = self.dev.get_active_configuration()
        intf = config[(0,0)]
        if self.dev.is_kernel_driver_active(0) is True:
            self.dev.detach_kernel_driver(0)
        elif self.dev.is_kernel_driver_active(1) is True:
            self.dev.detach_kernel_driver(1)
        usb.util.claim_interface(self.dev, 0)

def __setup_argument_parser__():
    parser = argparse.ArgumentParser("AlienPyFX console")
    parser.add_argument('--theme', dest='theme_path', metavar="FILE", help='Path to theme file to apply.  In test mode this validates the configuration of the theme')
    parser.add_argument('--lights-off', help='Toggle lights off', action='store_true')
    parser.add_argument('--lights-on',  help='Toggle lights on', action='store_true')
    parser.set_defaults(off=False)
    parser.set_defaults(on=True)
    return parser

if __name__ == '__main__':
    parser = __setup_argument_parser__()
    option = parser.parse_args()
    adm = AlienwareDeviceManager()
    if option.theme_path is not None:
        adm.apply_theme(option.theme_path)
    elif option.lights_off is True:
        adm.lights_off()
    else:
        adm.lights_on()