from main import KeyboardHandler
import threading
import thread
import pyatspi
def parse(s):
 """parse a string like control+f into (modifier, key).
Unknown modifiers will return ValueError."""
 m = 0
 lst = s.split('+')
 if not len(lst): return (0, s)
#Are these right?
 d = {
 "shift": 1<<pyatspi.MODIFIER_SHIFT,
 "control": 1<<pyatspi.MODIFIER_CONTROL,
 "alt": 1<<pyatspi.MODIFIER_ALT,
"win":1<<pyatspi.MODIFIER_META3,
 }
 for item in lst:
  if item in d:
   m|=d[item]
   lst.remove(item)
#end if
 if len(lst) > 1: #more than one key, parse error
  raise ValueError, 'unknown modifier %s' % lst[0]
 return (m, lst[0].lower())
class AtspiThread(threading.Thread):
 def run(self):
  pyatspi.Registry.registerKeystrokeListener(handler, kind=(pyatspi.KEY_PRESSED_EVENT,),
  mask=pyatspi.allModifiers())
  pyatspi.Registry.start()
#the keys we registered
keys = {}
def handler(e):
 m,k = e.modifiers,e.event_string.lower()
#not sure why we can't catch control+f. Try to fix it.
 if (not e.is_text) and e.id >= 97 <= 126:
  k = chr(e.id)
 if (m,k) not in keys: return False
 thread.start_new(keys[(m,k)], ())
 return True #don't pass it on
class LinuxKeyboardHandler(KeyboardHandler):
 def __init__(self, *args, **kwargs):
  KeyboardHandler.__init__(self, *args, **kwargs)
  t = AtspiThread()
  t.start()
 def register_key(self, key, function):
  """key will be a string, such as control+shift+f.
We need to convert that, using parse_key,
into modifier and key to put into our dictionary."""
#register key so we know if we have it on event receive.
  t = parse(key)
  keys[t] = function
#if we got this far, the key is valid.
  KeyboardHandler.register_key(self, key, function)

 def unregister_key (self, key, function):
  KeyboardHandler.unregister_key(self, key, function)
  del keys[parse(key)]