Utilize unregister_key_event_handlers

zrzka · Oct 1, 2017 · 7bad869 · 7bad869
1 parent f1db820
commit 7bad869
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/blackmamba/script/drag_and_drop.py b/blackmamba/script/drag_and_drop.py
@@ -24,7 +24,7 @@
 import ctypes
 import zipfile
 from blackmamba.uikit.keyboard import (
-    register_key_event_handler, unregister_key_event_handler,
+    register_key_event_handler, unregister_key_event_handlers,
     UIEventKeyCode, UIKeyModifier
 )
 import blackmamba.ide.tab as tab
@@ -621,14 +621,15 @@ def __init__(self):
         def handle_escape():
             self.close()
 
-        self._handlers = []
-        self._handlers.append(register_key_event_handler(UIEventKeyCode.escape, handle_escape))
-        self._handlers.append(register_key_event_handler(UIEventKeyCode.dot, handle_escape,
-                                                         modifier=UIKeyModifier.command))
+        self._handlers = [
+            register_key_event_handler(UIEventKeyCode.escape, handle_escape),
+            register_key_event_handler(UIEventKeyCode.dot, handle_escape,
+                                       modifier=UIKeyModifier.command)
+        ]
 
     def will_close(self):
-        for handler in self._handlers:
-            unregister_key_event_handler(handler)
+        if self._handlers:
+            unregister_key_event_handlers(self._handlers)
 
 
 def main():