Ignore harmless KeyError while modifying selectors

kovidgoyal · Feb 9, 2018 · e119ed0 · e119ed0
1 parent 8c18486
commit e119ed0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/kittens/tui/loop.py b/kittens/tui/loop.py
@@ -290,9 +290,15 @@ def wakeup(self):
 
     def _modify_output_selector(self, waiting_for_write):
         if waiting_for_write:
-            self.sel.register(self.output_fd, selectors.EVENT_WRITE, self._write_ready)
+            try:
+                self.sel.register(self.output_fd, selectors.EVENT_WRITE, self._write_ready)
+            except KeyError:
+                pass
         else:
-            self.sel.unregister(self.output_fd)
+            try:
+                self.sel.unregister(self.output_fd)
+            except KeyError:
+                pass
 
     def loop(self, handler):
         select = self.sel.select