Burathar
/
OpenRazer-Command
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

Include changed files../

master
Burathar 5 years ago
parent
93b7350b3d
commit
f99ed334b6
4 changed files with 95 additions and 169 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 55
      create_keyboard_map.py
  2. 59
      keyboard.py
  3. 81
      main.py
  4. 69
      test_keyboard_map.py

55
create_keyboard_map.py
Unescape View File

@ -1,34 +1,20 @@
#! /bin/python3 #! /bin/python3
from .keyboard import Keyboard
from .keymap import Keymap
from .color import Color
import openrazer.client import openrazer.client
import json
from pathlib import Path
from pynput import keyboard
def get_keyboard(): from pynput import keyboard as keyboardnput
devman = openrazer.client.DeviceManager()
for device in devman.devices:
if (device.name == "Razer BlackWidow Chroma V2"):
return device
return None
def clear_all(kbd): keyboard = Keyboard(no_keymap=True)
# Set full keyboard to green
for row in range(0,kbd_rows):
for column in range(0, kbd_columns):
kbd.fx.advanced.matrix[row,column] = [0, 0, 0]
if not keyboard.found_keyboard():
kbd = get_keyboard() print("No keyboard was found")
if (kbd == None):
print("No Razer device with devicename 'Razer Blackwidow Chroma V2' found.")
exit() exit()
kbd_rows, kbd_columns = kbd.fx.advanced.rows, kbd.fx.advanced.cols
last_key_pressed = None last_key_pressed = None
def on_press(key): def on_press(key):
global last_key_pressed global last_key_pressed
@ -36,35 +22,30 @@ def on_press(key):
return False return False
def get_key(): def get_key():
with keyboard.Listener(on_press=on_press) as listener: with keyboardnput.Listener(on_press=on_press) as listener:
listener.join() listener.join()
def set_key_columns(keymap, column_range, prefix=""): def set_key_columns(keymap, column_range, prefix=""):
for row in range(0,kbd_rows): for row in range(0, 6):
for column in column_range: for column in column_range:
clear_all(kbd) keyboard.clear()
kbd.fx.advanced.matrix[row,column] = [255, 255, 255] keyboard.set_key(row,column, Color(255, 255, 255))
kbd.fx.advanced.draw() keyboard.draw()
get_key() get_key()
print('{} was pressed'.format(last_key_pressed)) print('{} was pressed'.format(last_key_pressed))
keymap[prefix + str(last_key_pressed)] = row * kbd_columns + column keymap[prefix + str(last_key_pressed)] = row * 22 + column
def iter_keys(): def iter_keys():
keymap = dict() keymap = dict()
print("Make sure caps lock and numpad are disabled, if not, do it now and restart the script") print("Make sure caps lock and numpad are disabled, if not, do it now and restart the script")
print("This script will iterate over all keys, row by row. When a key lights up white, press it, and use the mouse to cancel any F-key effects. First, we'll skip the numpad. If no key lights up, press the bottom-right most key, excluding the numpad().") print("This script will iterate over all keys, row by row. When a key lights up white, press it, and use the mouse to cancel any F-key effects. First, we'll skip the numpad. If no key lights up, press the bottom-right most key, excluding the numpad().")
set_key_columns(keymap, range(0,18)) set_key_columns(keymap, range(0, 18))
print("Now we'll iterate over the numpad, if no key lights up, press the bottom right key(enter)") print("Now we'll iterate over the numpad, if no key lights up, press the bottom right key(enter)")
set_key_columns(keymap, range(18,kbd_columns), "num_") set_key_columns(keymap, range(18, 22), "num_")
return keymap return keymap
def write_keymap_file(dictionary):
keymap_json = json.dumps(keymap)
path = Path.home() / '.config/openrazer_scripter/keymap.json'
Path(path.parent).mkdir(parents=True, exist_ok=True)
with open(path,'w') as keymap_file:
keymap_file.write(keymap_json)
keymap = iter_keys() keymap = Keymap()
write_keymap_file(keymap) keymap.set_keys(iter_keys())
keymap.write_keymap_file()
print('Saved keymap to file (.config/openrazer_scripter/keymap.json)') print('Saved keymap to file (.config/openrazer_scripter/keymap.json)')

59
keyboard.py
Unescape View File

@ -1,26 +1,71 @@
import openrazer.client
import json
import warnings import warnings
from pathlib import Path from collections.abc import Iterable
import openrazer.client
class keyboard: from .color import Color
from .keymap import Keymap
class Keyboard:
"""Provides tooling to call openrazer.client.device""" """Provides tooling to call openrazer.client.device"""
def __init__(self): def __init__(self, no_keymap = False):
self.kbd = None self.kbd = None
self.find_keyboard() self.find_keyboard()
if self.kbd is None: if self.kbd is None:
warnings.warn('Compatible keyboard was not detected', RuntimeWarning) warnings.warn('Compatible keyboard was not detected', RuntimeWarning)
if not no_keymap:
self.keymap = Keymap()
self.keymap.load_from_file()
def find_keyboard(self): def find_keyboard(self):
if self.kbd is not None: if self.kbd is not None:
print('keyboard was already initiated') print('keyboard was already initiated')
return return True
devman = openrazer.client.DeviceManager() devman = openrazer.client.DeviceManager()
for device in devman.devices: for device in devman.devices:
if (device.name == "Razer BlackWidow Chroma V2"): if (device.name == "Razer BlackWidow Chroma V2"):
self.kbd = device self.kbd = device
return return True
return False
def found_keyboard(self):
return self.kbd is not None
def set_keys(self, color, *keys):
if not isinstance(color, Color):
print(f"Color {color} for key(s) '{keys}' is no instance of Color, skipping.")
return False
for key in keys:
if isinstance(key, Iterable) and not isinstance(key, str): # Allow for iterables (lists etc) of keys to be parsed
for single_key in key:
self.set_keys(color, single_key)
else:
location = self.keymap.get_location(key)
self.kbd.fx.advanced.matrix[location] = [color.red, color.green, color.blue]
return True
def set_key(self, row, column, color):
if not isinstance(color, Color):
print(f"Color '{color}' for key {row},{column} is no instance of Color, skipping.")
return False
self.kbd.fx.advanced.matrix[row, column] = [color.red, color.green, color.blue]
def set_static(self, color):
if not isinstance(color, Color):
print(f"Color for set_static() is no instance of Color, skipping.")
return False
for row in range(0,6):
for column in range(0, 22):
self.kbd.fx.advanced.matrix[row,column] = [color.red, color.green, color.blue]
def clear(self):
self.set_static(Color(0, 0, 0))
def draw(self):
self.kbd.fx.advanced.draw()

81
main.py
Unescape View File

@ -1,80 +1,15 @@
import openrazer.client from .keyboard import Keyboard
from .color import Color
import json import json
from pathlib import Path from pathlib import Path
import math import math
def get_keyboard(): keyboard = Keyboard()
devman = openrazer.client.DeviceManager()
for device in devman.devices:
if (device.name == 'Razer BlackWidow Chroma V2'):
return device
return None
def read_keymap_file(path = ''): if not keyboard.found_keyboard():
if path == '': print("No keyboard was found")
path = Path.home() / '.config/openrazer_scripter/keymap.json'
keymap = None
with open(path,'r') as keymap_file:
keymap = json.load(keymap_file)
return keymap
kbd = get_keyboard()
if (kbd == None):
print("No Razer device with devicename 'Razer Blackwidow Chroma V2' found.")
exit() exit()
keymap = read_keymap_file()
def set_static(kbd, color):
color = get_rgb(color)
if color is None:
print(f"Color set_static() was incorrect, skipping.")
return
for row in range(0,6):
for column in range(0, 22):
kbd.fx.advanced.matrix[row,column] = [color[0], color[1], color[2]]
def check_color_value(value):
if not isinstance(value, int):
print(f"Colorvalue '{value}' is not a number")
return False
if value < 0 or value > 255:
print(f"Color value '{value}' is out of bounds (0-255)")
return False
return True
def get_rgb(color):
r, g, b = None, None, None
if isinstance(color, str):
if len(color) == 6:
color = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
if len(color) == 3:
r = color[0]
g = color[1]
b = color[2]
else:
print(f"Could not interpert color '{color}'")
return None
if check_color_value(r) and check_color_value(g) and check_color_value(b):
return r, g, b
else:
print(f"Color '{color}' is out of range")
return None
def get_row_column(index):
row = math.floor(index / 22)
column = index % 22
return row, column
def set_keys(kbd, color, *keys):
color = get_rgb(color)
if color is None:
print(f"Color for key(s) '{keys}' was incorrect, skipping.")
return
for key in keys:
if key in keymap:
position = get_row_column(keymap[key])
kbd.fx.advanced.matrix[position[0], position[1]] = [color[0], color[1], color[2]]
set_static(kbd, (255,255,255)) keyboard.set_static(Color(255,255,255))
set_keys(kbd, 'ff0000', 'enter') keyboard.set_keys('ff0000', 'enter')
kbd.fx.advanced.draw() keyboard.draw()

69
test_keyboard_map.py
Unescape View File

@ -1,65 +1,30 @@
#! /bin/python3 #! /bin/python3
import openrazer.client from .keyboard import Keyboard
import json from .color import Color
from pathlib import Path from .keymap import Keymap
import math
def get_keyboard(): keyboard = Keyboard(no_keymap=False)
devman = openrazer.client.DeviceManager() if not keyboard.found_keyboard():
print("No keyboard was found")
for device in devman.devices:
if (device.name == "Razer BlackWidow Chroma V2"):
return device
return None
def clear_all(kbd):
# Set full keyboard to green
for row in range(0, 6):
for column in range(0, 22):
kbd.fx.advanced.matrix[row,column] = [0, 0, 0]
kbd = get_keyboard()
if (kbd == None):
print("No Razer device with devicename 'Razer Blackwidow Chroma V2' found.")
exit() exit()
def read_keymap_file():
path = Path.home() / '.config/openrazer_scripter/keymap.json'
keymap = None
with open(path,'r') as keymap_file:
keymap = json.load(keymap_file)
return keymap
def get_row_column(index):
row = math.floor(index / 22)
column = index % 22
return row, column
def set_all_white(kbd, keymap):
clear_all(kbd)
for key, value in keymap.items():
position = get_row_column(value)
kbd.fx.advanced.matrix[position[0], position[1]] = [255,255,255]
kbd.fx.advanced.draw()
def sort_dict(dictionary): def sort_dict(dictionary):
return {k: v for k, v in sorted(dictionary.items(), key=lambda item: item[1])} return {k: v for k, v in sorted(dictionary.items(), key=lambda item: item[1])}
def iter_keys(kbd, keymap): def iter_keys(keymap):
keymap = sort_dict(keymap) sorted_keys = sort_dict(keymap.keys)
for key, value in keymap.items(): for key in sorted_keys.keys():
clear_all(kbd) keyboard.clear()
position = get_row_column(value)
print(f'key: {key}') print(f'key: {key}')
kbd.fx.advanced.matrix[position[0], position[1]] = [255,255,255] keyboard.set_keys(Color(255, 255, 255), key)
kbd.fx.advanced.draw() keyboard.draw()
input() input()
keymap = read_keymap_file() keymap = keyboard.keymap
while True: while True:
set_all_white(kbd, keymap) keyboard.set_keys(Color(255, 255, 255), list(keymap.keys.keys()))
keyboard.draw()
input("Press Enter to continue...") input("Press Enter to continue...")
iter_keys(kbd, keymap) iter_keys(keymap)

Write Preview
Loading…
Cancel
Save