mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2024-11-23 11:48:07 -06:00
120 lines
3.4 KiB
Python
120 lines
3.4 KiB
Python
import functools
|
|
import wx
|
|
import platform
|
|
from main import KeyboardHandler
|
|
|
|
__all__ = ['WXKeyboardHandler', 'WXControlKeyboardHandler']
|
|
|
|
|
|
def call_after(func):
|
|
def wrapper(*args, **kwargs):
|
|
wx.CallAfter(func, *args, **kwargs)
|
|
functools.update_wrapper(wrapper, func)
|
|
return wrapper
|
|
|
|
|
|
class BaseWXKeyboardHandler(KeyboardHandler):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(BaseWXKeyboardHandler, self).__init__(*args, **kwargs)
|
|
#Setup the replacement dictionaries.
|
|
for i in dir(wx):
|
|
if i.startswith('WXK_'):
|
|
key = i[4:].lower()
|
|
self.replacement_keys[key] = getattr(wx, i)
|
|
elif i.startswith('MOD_'):
|
|
key = i[4:].lower()
|
|
self.replacement_mods[key] = getattr(wx, i)
|
|
|
|
def parse_key (self, keystroke, separator="+"):
|
|
keystroke = [self.keycode_from_key(i) for i in keystroke.split(separator)]
|
|
mods = 0
|
|
for i in keystroke[:-1]:
|
|
mods = mods | i #or everything together
|
|
return (mods, keystroke[-1])
|
|
|
|
def keycode_from_key(self, key):
|
|
if key in self.replacement_mods:
|
|
result = self.replacement_mods[key]
|
|
elif key in self.replacement_keys:
|
|
result = self.replacement_keys[key]
|
|
if result >= 277:
|
|
result -= 277
|
|
elif len(key) == 1:
|
|
result = ord(key.upper())
|
|
print "result: ", result
|
|
return result
|
|
|
|
|
|
|
|
#try:
|
|
if platform.system() == "Windows":
|
|
from windows import WindowsKeyboardHandler as keyboard_handler
|
|
elif platform.system() == "Linux":
|
|
from linux import LinuxKeyboardHandler as keyboard_handler
|
|
elif platform.system() == "Darwin":
|
|
from osx import OSXKeyboardHandler as keyboard_handler
|
|
|
|
class WXKeyboardHandler(keyboard_handler):
|
|
def __init__ (self, parent, *args, **kwargs):
|
|
super(WXKeyboardHandler, self).__init__(*args, **kwargs)
|
|
self.parent = parent
|
|
self.key_ids = {}
|
|
|
|
@call_after
|
|
def register_key(self, key, function):
|
|
super(WXKeyboardHandler, self).register_key(key, function)
|
|
key_id = wx.NewId()
|
|
parsed = self.parse_key(key)
|
|
self.parent.RegisterHotKey(key_id, *parsed)
|
|
self.parent.Bind(wx.EVT_HOTKEY, lambda evt: self.process_key(evt, key_id), id=key_id)
|
|
self.key_ids[key] = key_id
|
|
|
|
@call_after
|
|
def unregister_key (self, key, function):
|
|
super(WXKeyboardHandler, self).unregister_key(key, function)
|
|
if key not in self.key_ids:
|
|
return #there's nothing we can do.
|
|
key_id = self.key_ids[key]
|
|
self.parent.UnregisterHotKey(key_id)
|
|
self.parent.Unbind( wx.EVT_HOTKEY, id=key_id)
|
|
self.key_ids.pop(key)
|
|
|
|
def process_key (self, evt, id):
|
|
evt.Skip()
|
|
key_ids = self.key_ids.keys()
|
|
for i in key_ids:
|
|
if self.key_ids.get(i) == id:
|
|
self.handle_key(i)
|
|
|
|
class WXControlKeyboardHandler(wx.StaticText, KeyboardHandler):
|
|
|
|
def __init__(self, parent=None, *a, **k):
|
|
wx.StaticText.__init__(self, parent=parent)
|
|
KeyboardHandler.__init__(self, *a, **k)
|
|
self.wx_replacements = {}
|
|
for i in [d for d in dir(wx) if d.startswith('WXK_')]:
|
|
self.wx_replacements[getattr(wx, i)] = i[4:].lower()
|
|
self.Bind(wx.EVT_KEY_DOWN, self.process_key, self)
|
|
self.SetFocus()
|
|
|
|
def process_key(self, evt):
|
|
keycode = evt.GetKeyCode()
|
|
keyname = self.wx_replacements.get(keycode, None)
|
|
modifiers = ""
|
|
replacements = ( (evt.ControlDown(), 'control+'),
|
|
(evt.AltDown(), 'alt+'),
|
|
(evt.ShiftDown(), 'shift+'),
|
|
(evt.MetaDown(), 'win+')
|
|
)
|
|
for mod, ch in (replacements):
|
|
if mod:
|
|
modifiers += ch
|
|
if keyname is None:
|
|
if 27 < keycode < 256:
|
|
keyname = chr(keycode).lower()
|
|
else:
|
|
keyname = "(%s)unknown" % keycode
|
|
key = modifiers + keyname
|
|
self.handle_key(key)
|