Improve device detection
This commit is contained in:
parent
bab993b94d
commit
ef7cfe38c5
12
README.md
12
README.md
|
@ -1,16 +1,22 @@
|
|||
# BlackWidow control
|
||||
Python 3 script to enable macro keys of BlackWidow keyboard under GNU/Linux
|
||||
|
||||
It just enables the macro keys which can then be configured as hotkeys in your desktop environment. However it does not enable the ability to record macros and to switch between different configurations like the Windows driver does.
|
||||
It just enables the macro keys which can then be configured as hotkeys in your desktop environment.
|
||||
However it does not enable the ability to record macros and to switch between different configurations
|
||||
like the Windows driver does.
|
||||
|
||||
## Supported devices
|
||||
The script is know to work with the following BlackWidow editions:
|
||||
- regular edition (the ceapest one which has no background light)
|
||||
- the regular 2013 edition
|
||||
|
||||
The script likely works with the following BlackWidow editions:
|
||||
- The script likely works with the 2014 version, too.
|
||||
- The script likely works with the ultimate editions, too.
|
||||
|
||||
If you can confirm that those or other devices work or don't work, let me now by editing this file.
|
||||
|
||||
### Notes
|
||||
- The script likely works with the ultimate editions, too. However, this has not
|
||||
been tested yet.
|
||||
- The script does not work with the 2016 editions (yet).
|
||||
- The script has been tested using the firmware update from Razer Synapse 2.0.
|
||||
Hence the script might not work when an older firmware version is used.
|
||||
|
|
|
@ -4,10 +4,6 @@ from optparse import OptionParser
|
|||
import usb
|
||||
import sys
|
||||
|
||||
VENDOR_ID = 0x1532 # Razer
|
||||
PRODUCT_ID_BLACK_WIDOW = 0x010e # BlackWidow / BlackWidow Ultimate
|
||||
PRODUCT_ID_BLACK_WIDOW_2013 = 0x011b # BlackWidow 2013
|
||||
|
||||
USB_REQUEST_TYPE = 0x21 # Host To Device | Class | Interface
|
||||
USB_REQUEST = 0x09 # SET_REPORT
|
||||
|
||||
|
@ -15,27 +11,44 @@ USB_VALUE = 0x0300
|
|||
USB_INDEX = 0x2
|
||||
USB_INTERFACE = 2
|
||||
|
||||
VENDOR_ID = 0x1532 # Razer
|
||||
PRODUCT_ID_BLACK_WIDOW = 0x010e # BlackWidow
|
||||
PRODUCT_ID_BLACK_WIDOW_ULTIMATE = 0x011a # BlackWidow Ultimate
|
||||
PRODUCT_ID_BLACK_WIDOW_2013 = 0x011b # BlackWidow 2013/2014
|
||||
|
||||
PRODUCTS = [("Black Widow", PRODUCT_ID_BLACK_WIDOW),
|
||||
("Black Widow Ultimate", PRODUCT_ID_BLACK_WIDOW_ULTIMATE),
|
||||
("Black Widow 2013/2014", PRODUCT_ID_BLACK_WIDOW_2013)]
|
||||
|
||||
LOG=sys.stderr.write
|
||||
|
||||
class blackwidow(object):
|
||||
kernel_driver_detached = False
|
||||
detected_keyboard = None
|
||||
|
||||
class BlackWidow(object):
|
||||
def __init__(self):
|
||||
self.device = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID_BLACK_WIDOW)
|
||||
self.kernel_driver_detached = False
|
||||
self.interface_claimed = False
|
||||
self.detected_keyboard = None
|
||||
|
||||
if self.device is None:
|
||||
self.device = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID_BLACK_WIDOW_2013)
|
||||
if self.device is None:
|
||||
LOG("No device found\n")
|
||||
return
|
||||
else:
|
||||
LOG("Found device: Black Widow 2013\n")
|
||||
detected_keyboard = PRODUCT_ID_BLACK_WIDOW_2013
|
||||
else:
|
||||
LOG("Found device: Black Widow\n")
|
||||
detected_keyboard = PRODUCT_ID_BLACK_WIDOW
|
||||
self.find_device()
|
||||
if self.device_found():
|
||||
self.claim_interface()
|
||||
|
||||
def __del__(self):
|
||||
if self.interface_claimed:
|
||||
self.release_interface()
|
||||
|
||||
def find_device(self):
|
||||
for product_name, product_id in PRODUCTS:
|
||||
self.device = usb.core.find(idVendor=VENDOR_ID, idProduct=product_id)
|
||||
if self.device:
|
||||
detected_keyboard = product_id
|
||||
LOG("Found device: %s\n" % product_name)
|
||||
break
|
||||
|
||||
def device_found(self):
|
||||
return self.device is not None
|
||||
|
||||
def claim_interface(self):
|
||||
try:
|
||||
if self.device.is_kernel_driver_active(USB_INTERFACE):
|
||||
LOG("Kernel driver active; detaching it\n")
|
||||
|
||||
|
@ -44,32 +57,32 @@ class blackwidow(object):
|
|||
|
||||
LOG("Claiming interface\n")
|
||||
usb.util.claim_interface(self.device, USB_INTERFACE)
|
||||
self.interface_claimed = True
|
||||
except:
|
||||
LOG("Unable to claim interface. Ensure the script is running as root.\n")
|
||||
raise
|
||||
|
||||
def __del__(self):
|
||||
if self.device_found():
|
||||
def release_interface(self):
|
||||
if self.interface_claimed:
|
||||
LOG("Releasing claimed interface\n")
|
||||
usb.util.release_interface(self.device, USB_INTERFACE)
|
||||
|
||||
if self.kernel_driver_detached:
|
||||
LOG("Reattaching the kernel driver\n")
|
||||
self.device.attach_kernel_driver(USB_INTERFACE)
|
||||
LOG("Done.\n")
|
||||
|
||||
def bwcmd(self, c):
|
||||
def format_command(self, command):
|
||||
from functools import reduce
|
||||
c1 = bytes.fromhex(c)
|
||||
c1 = bytes.fromhex(command)
|
||||
c2 = [ reduce(int.__xor__, c1) ]
|
||||
b = [0] * 90
|
||||
b[5:5+len(c1)] = c1
|
||||
b[-2:-1] = c2
|
||||
return bytes(b)
|
||||
|
||||
def device_found(self):
|
||||
return self.device is not None
|
||||
|
||||
def send(self, c):
|
||||
def _send(msg):
|
||||
USB_BUFFER = self.bwcmd(msg)
|
||||
def send(self, command):
|
||||
def internal_send(msg):
|
||||
USB_BUFFER = self.format_command(command)
|
||||
result = 0
|
||||
try:
|
||||
result = self.device.ctrl_transfer(USB_REQUEST_TYPE, USB_REQUEST, wValue=USB_VALUE, wIndex=USB_INDEX, data_or_wLength=USB_BUFFER)
|
||||
|
@ -79,14 +92,15 @@ class blackwidow(object):
|
|||
LOG("Data sent successfully\n")
|
||||
return result
|
||||
|
||||
if isinstance(c, list):
|
||||
for i in c:
|
||||
if isinstance(command, list):
|
||||
for i in command:
|
||||
print(' >> {}\n'.format(i))
|
||||
_send(i)
|
||||
elif isinstance(c, str):
|
||||
_send(c)
|
||||
internal_send(i)
|
||||
elif isinstance(command, str):
|
||||
internal_send(command)
|
||||
|
||||
def main():
|
||||
#
|
||||
init_new = '0200 0403'
|
||||
init_old = '0200 0402'
|
||||
no_pulsate = '0303 0201 0400'
|
||||
|
@ -103,7 +117,7 @@ def main():
|
|||
parser.add_option("-g", "--set-game-mode", type="string", dest="gamemode", default="unmodified", help="sets whether the game mode is enabled (on/off)")
|
||||
(options, args) = parser.parse_args()
|
||||
|
||||
bw = blackwidow()
|
||||
bw = BlackWidow()
|
||||
if bw.device_found():
|
||||
# enable macro keys
|
||||
if options.init == True:
|
||||
|
@ -183,5 +197,8 @@ def main():
|
|||
else:
|
||||
LOG("Specified value for game mode is unknown and will be ignored.\n")
|
||||
|
||||
else:
|
||||
LOG("No device found\n")
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# udev rule to initiate BlackWidow keyboard when connected
|
||||
# udev rule to initiate BlackWidow keyboards when connected
|
||||
SUBSYSTEM=="usb", ACTION=="add", ATTR{idVendor}=="1532", ATTR{idProduct}=="010e", RUN+="/usr/bin/blackwidowcontrol -i"
|
||||
SUBSYSTEM=="usb", ACTION=="add", ATTR{idVendor}=="1532", ATTR{idProduct}=="011a", RUN+="/usr/bin/blackwidowcontrol -i"
|
||||
SUBSYSTEM=="usb", ACTION=="add", ATTR{idVendor}=="1532", ATTR{idProduct}=="011b", RUN+="/usr/bin/blackwidowcontrol -i"
|
||||
|
|
Loading…
Reference in New Issue